Mercurial > farol > farol
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 |