# HG changeset patch # User BenoƮt Allard # Date 1412761124 -7200 # Node ID 64403f1b424eeaa5136b116640926b84cd9da786 # Parent fbc413b8a46e0ec0b3c935f90b44bbb2227efa87 Make use of the new getNote methods on CVRFVulnerability diff -r fbc413b8a46e -r 64403f1b424e farol/vulnerability.py --- a/farol/vulnerability.py Wed Oct 08 11:34:11 2014 +0200 +++ b/farol/vulnerability.py Wed Oct 08 11:38:44 2014 +0200 @@ -106,20 +106,15 @@ @vulnerability.route('//note/') @document_required def view_note(ordinal, note_ordinal): - for note in get_vuln(ordinal)._notes: - if note._ordinal != note_ordinal: - continue - return render_template('vulnerability/view_note.j2', note=note, ordinal=ordinal) - abort(404) + note = get_vuln(ordinal).getNote(note_ordinal) + if note is None: + abort(404) + return render_template('vulnerability/view_note.j2', note=note, ordinal=ordinal) @vulnerability.route('//note//edit', methods=['GET', 'POST']) @document_required def edit_note(ordinal, note_ordinal): - note = None - for n in get_vuln(ordinal)._notes: - if n._ordinal == note_ordinal: - note = n - break + note = get_vuln(ordinal).getNote(note_ordinal) if note is None: abort(404) if request.method != 'POST':