comparison farol/vulnerability.py @ 59:64403f1b424e

Make use of the new getNote methods on CVRFVulnerability
author Benoît Allard <benoit.allard@greenbone.net>
date Wed, 08 Oct 2014 11:38:44 +0200
parents fbc413b8a46e
children c6e7175ff28c
comparison
equal deleted inserted replaced
58:fbc413b8a46e 59:64403f1b424e
104 return redirect(url_for('.view', ordinal=vuln._ordinal)) 104 return redirect(url_for('.view', ordinal=vuln._ordinal))
105 105
106 @vulnerability.route('/<int:ordinal>/note/<int:note_ordinal>') 106 @vulnerability.route('/<int:ordinal>/note/<int:note_ordinal>')
107 @document_required 107 @document_required
108 def view_note(ordinal, note_ordinal): 108 def view_note(ordinal, note_ordinal):
109 for note in get_vuln(ordinal)._notes: 109 note = get_vuln(ordinal).getNote(note_ordinal)
110 if note._ordinal != note_ordinal: 110 if note is None:
111 continue 111 abort(404)
112 return render_template('vulnerability/view_note.j2', note=note, ordinal=ordinal) 112 return render_template('vulnerability/view_note.j2', note=note, ordinal=ordinal)
113 abort(404)
114 113
115 @vulnerability.route('/<int:ordinal>/note/<int:note_ordinal>/edit', methods=['GET', 'POST']) 114 @vulnerability.route('/<int:ordinal>/note/<int:note_ordinal>/edit', methods=['GET', 'POST'])
116 @document_required 115 @document_required
117 def edit_note(ordinal, note_ordinal): 116 def edit_note(ordinal, note_ordinal):
118 note = None 117 note = get_vuln(ordinal).getNote(note_ordinal)
119 for n in get_vuln(ordinal)._notes:
120 if n._ordinal == note_ordinal:
121 note = n
122 break
123 if note is None: 118 if note is None:
124 abort(404) 119 abort(404)
125 if request.method != 'POST': 120 if request.method != 'POST':
126 return render_template('vulnerability/edit_note.j2', note=note, ordinal=ordinal, types=note.TYPES) 121 return render_template('vulnerability/edit_note.j2', note=note, ordinal=ordinal, types=note.TYPES)
127 122

http://farol.wald.intevation.org