Mercurial > farol > farol
annotate farol/document.py @ 132:812e613cbbb4
Add possibility to delete the current document.
| author | Benoît Allard <benoit.allard@greenbone.net> |
|---|---|
| date | Fri, 24 Oct 2014 12:59:35 +0200 |
| parents | d2588d88d47a |
| children |
| rev | line source |
|---|---|
| 0 | 1 # -*- encoding: utf-8 -*- |
| 2 # Description: | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
3 # The Document Blueprint |
| 0 | 4 # |
| 5 # Authors: | |
| 6 # Benoît Allard <benoit.allard@greenbone.net> | |
| 7 # | |
| 8 # Copyright: | |
| 9 # Copyright (C) 2014 Greenbone Networks GmbH | |
| 10 # | |
| 11 # This program is free software; you can redistribute it and/or | |
| 12 # modify it under the terms of the GNU General Public License | |
| 13 # as published by the Free Software Foundation; either version 2 | |
| 14 # of the License, or (at your option) any later version. | |
| 15 # | |
| 16 # This program is distributed in the hope that it will be useful, | |
| 17 # but WITHOUT ANY WARRANTY; without even the implied warranty of | |
| 18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | |
| 19 # GNU General Public License for more details. | |
| 20 # | |
| 21 # You should have received a copy of the GNU General Public License | |
| 22 # along with this program; if not, write to the Free Software | |
| 23 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA. | |
| 24 | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
25 from flask import (Blueprint, render_template, abort, redirect, request, |
|
66
f1705e90695b
Don't allow to add a revision without tracking information
Benoît Allard <benoit.allard@greenbone.net>
parents:
64
diff
changeset
|
26 url_for, flash) |
| 0 | 27 |
|
62
ce49bd1512dd
Make pyflafes a happier
Benoît Allard <benoit.allard@greenbone.net>
parents:
61
diff
changeset
|
28 from farolluz.cvrf import (CVRFNote, CVRFReference, CVRFPublisher, |
|
ce49bd1512dd
Make pyflafes a happier
Benoît Allard <benoit.allard@greenbone.net>
parents:
61
diff
changeset
|
29 CVRFTracking, CVRFTrackingID, CVRFGenerator, CVRFRevision, |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
30 CVRFAggregateSeverity) |
|
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
31 from farolluz.renderer import utcnow |
| 0 | 32 |
|
60
c6e7175ff28c
Unify acknowledgments update/creation from forms
Benoît Allard <benoit.allard@greenbone.net>
parents:
58
diff
changeset
|
33 from .controller import (update_note_from_request, create_note_from_request, |
|
61
55b72057b066
Unify reference update/creation from forms
Benoît Allard <benoit.allard@greenbone.net>
parents:
60
diff
changeset
|
34 update_reference_from_request, create_reference_from_request, |
|
63
b4fb652484b4
Define a split_fields function to split fields, and use it
Benoît Allard <benoit.allard@greenbone.net>
parents:
62
diff
changeset
|
35 update_acknowledgment_from_request, create_acknowledgment_from_request, |
|
127
d49c1ee6bc07
Harden server-side version parsing
Benoît Allard <benoit.allard@greenbone.net>
parents:
100
diff
changeset
|
36 split_fields, parseDate, parseVersion) |
|
132
812e613cbbb4
Add possibility to delete the current document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
129
diff
changeset
|
37 from .session import document_required, get_current, del_current |
| 0 | 38 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
39 |
|
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
40 document = Blueprint('document', __name__) |
|
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
41 |
|
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
42 @document.route('/') |
| 0 | 43 @document_required |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
44 def view(): |
| 0 | 45 cvrf = get_current() |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
46 return render_template('document/view.j2', cvrf=cvrf) |
| 0 | 47 |
|
132
812e613cbbb4
Add possibility to delete the current document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
129
diff
changeset
|
48 @document.route('/delete', methods=['POST']) |
|
812e613cbbb4
Add possibility to delete the current document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
129
diff
changeset
|
49 def delete(): |
|
812e613cbbb4
Add possibility to delete the current document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
129
diff
changeset
|
50 del_current() |
|
812e613cbbb4
Add possibility to delete the current document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
129
diff
changeset
|
51 return redirect(url_for('welcome')) |
|
812e613cbbb4
Add possibility to delete the current document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
129
diff
changeset
|
52 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
53 @document.route('/title/edit', methods=['GET', 'POST']) |
| 0 | 54 @document_required |
| 55 def edit_title(): | |
| 56 if request.method != 'POST': | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
57 return render_template('document/edit_title.j2', title = get_current()._title, _type = get_current()._type) |
| 0 | 58 |
| 59 | |
| 60 get_current()._title = request.form['title'] | |
| 61 get_current()._type = request.form['type'] | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
62 return redirect(url_for('.view')) |
| 0 | 63 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
64 @document.route('/publisher/edit', methods=['GET', 'POST']) |
| 0 | 65 @document_required |
| 66 def edit_publisher(): | |
| 67 if request.method != 'POST': | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
68 return render_template('document/edit_publisher.j2', publisher = get_current()._publisher or CVRFPublisher(''), types=CVRFPublisher.TYPES) |
| 0 | 69 |
| 70 publisher = CVRFPublisher(request.form['type'], request.form['vendorid'] or None) | |
| 71 publisher.setContact(request.form['contact'] or None) | |
| 72 publisher.setAuthority(request.form['authority'] or None) | |
| 73 get_current().setPublisher(publisher) | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
74 return redirect(url_for('.view')) |
| 0 | 75 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
76 @document.route('/tracking/edit', methods=['GET', 'POST']) |
| 0 | 77 @document_required |
| 78 def edit_tracking(): | |
| 79 wasNone = False | |
| 80 tracking = get_current()._tracking | |
| 81 if tracking is None: | |
| 82 wasNone = True | |
|
66
f1705e90695b
Don't allow to add a revision without tracking information
Benoît Allard <benoit.allard@greenbone.net>
parents:
64
diff
changeset
|
83 tracking = CVRFTracking(CVRFTrackingID(''), 'Draft', (0,0), utcnow(), utcnow()) |
| 0 | 84 generator = tracking._generator |
| 85 if not tracking._generator: | |
| 86 generator = CVRFGenerator() | |
| 87 if request.method != 'POST': | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
88 return render_template('document/edit_tracking.j2', tracking=tracking, version='.'.join('%s'%v for v in tracking._version), generator=generator, now=utcnow(), statuses=tracking.STATUSES) |
| 0 | 89 |
| 90 tracking._identification._id = request.form['id'] | |
|
63
b4fb652484b4
Define a split_fields function to split fields, and use it
Benoît Allard <benoit.allard@greenbone.net>
parents:
62
diff
changeset
|
91 aliases = split_fields(request.form['id_aliases']) |
| 0 | 92 tracking._identification._aliases = aliases |
| 93 tracking._status = request.form['status'] | |
|
127
d49c1ee6bc07
Harden server-side version parsing
Benoît Allard <benoit.allard@greenbone.net>
parents:
100
diff
changeset
|
94 version = parseVersion(request.form['version']) |
|
129
d2588d88d47a
Move parseVersion flashing one level higher
Benoît Allard <benoit.allard@greenbone.net>
parents:
127
diff
changeset
|
95 if version is not None: |
|
127
d49c1ee6bc07
Harden server-side version parsing
Benoît Allard <benoit.allard@greenbone.net>
parents:
100
diff
changeset
|
96 tracking._version = version |
| 0 | 97 tracking._initialDate = parseDate(request.form['initial']) |
| 98 tracking._currentDate = parseDate(request.form['current']) | |
| 99 if wasNone: | |
| 100 get_current().setTracking(tracking) | |
| 101 if (not request.form['gen_engine']) and (not request.form['gen_date']): | |
| 102 generator = None | |
| 103 else: | |
| 104 generator._engine = request.form['gen_engine'] or None | |
| 105 if request.form['gen_date']: | |
| 106 generator._date = parseDate(request.form['gen_date']) | |
| 107 else: | |
| 108 generator._date = None | |
| 109 tracking.setGenerator(generator) | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
110 return redirect(url_for('.view')) |
| 0 | 111 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
112 @document.route('/revision/<int:index>/edit', methods=['GET', 'POST']) |
| 0 | 113 @document_required |
| 114 def edit_revision(index): | |
| 115 cvrf = get_current() | |
| 116 if cvrf._tracking is None: | |
| 117 abort(404) | |
| 118 try: | |
| 119 revision = cvrf._tracking._history[index] | |
| 120 except IndexError: | |
| 121 abort(404) | |
| 122 if request.method != 'POST': | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
123 return render_template('document/edit_revision.j2', number='.'.join('%s'%v for v in revision._number), date=revision._date, description=revision._description, action='Update') |
| 0 | 124 |
|
127
d49c1ee6bc07
Harden server-side version parsing
Benoît Allard <benoit.allard@greenbone.net>
parents:
100
diff
changeset
|
125 version = parseVersion(request.form['number']) |
|
129
d2588d88d47a
Move parseVersion flashing one level higher
Benoît Allard <benoit.allard@greenbone.net>
parents:
127
diff
changeset
|
126 if version is not None: |
|
127
d49c1ee6bc07
Harden server-side version parsing
Benoît Allard <benoit.allard@greenbone.net>
parents:
100
diff
changeset
|
127 revision._number = version |
| 0 | 128 revision._date = parseDate(request.form['date']) |
| 129 revision._description = request.form['description'] | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
130 return redirect(url_for('.view')) |
| 0 | 131 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
132 @document.route('/revision/add', methods=['GET', 'POST']) |
| 0 | 133 @document_required |
| 134 def add_revision(): | |
| 135 tracking = get_current()._tracking | |
| 136 if request.method != 'POST': | |
|
66
f1705e90695b
Don't allow to add a revision without tracking information
Benoît Allard <benoit.allard@greenbone.net>
parents:
64
diff
changeset
|
137 if tracking is None: |
|
f1705e90695b
Don't allow to add a revision without tracking information
Benoît Allard <benoit.allard@greenbone.net>
parents:
64
diff
changeset
|
138 flash('The tracking information should be set first to be able to add a revision.', 'danger') |
|
f1705e90695b
Don't allow to add a revision without tracking information
Benoît Allard <benoit.allard@greenbone.net>
parents:
64
diff
changeset
|
139 return redirect(url_for('.edit_tracking')) |
| 0 | 140 version = tracking._version |
| 141 version = version[:-1] + (version[-1] + 1,) | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
142 return render_template('document/edit_revision.j2', number='.'.join("%d"%v for v in version), date=utcnow(), action='Add') |
| 0 | 143 |
|
129
d2588d88d47a
Move parseVersion flashing one level higher
Benoît Allard <benoit.allard@greenbone.net>
parents:
127
diff
changeset
|
144 version = parseVersion(request.form['number']) or (0,0) |
| 0 | 145 date = parseDate(request.form['date']) |
| 146 revision = CVRFRevision(version, date, request.form['description']) | |
| 147 tracking.addRevision(revision) | |
| 148 if 'update_tracking' in request.form: | |
| 149 tracking._version = version | |
| 150 tracking._currentDate = date | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
151 return redirect(url_for('.view')) |
| 0 | 152 |
| 73 | 153 @document.route('/revision/<int:index>/del', methods=['POST']) |
|
67
023080ffd995
Allow deletion of revisions
Benoît Allard <benoit.allard@greenbone.net>
parents:
66
diff
changeset
|
154 @document_required |
| 73 | 155 def del_revision(index): |
|
68
33cdb9faed64
Improve deletion of revisions (input validation)
Benoît Allard <benoit.allard@greenbone.net>
parents:
67
diff
changeset
|
156 history = get_current()._tracking._history |
|
33cdb9faed64
Improve deletion of revisions (input validation)
Benoît Allard <benoit.allard@greenbone.net>
parents:
67
diff
changeset
|
157 if not (0 <= index < len(history)): |
|
33cdb9faed64
Improve deletion of revisions (input validation)
Benoît Allard <benoit.allard@greenbone.net>
parents:
67
diff
changeset
|
158 flash('Revision not found', 'danger') |
|
33cdb9faed64
Improve deletion of revisions (input validation)
Benoît Allard <benoit.allard@greenbone.net>
parents:
67
diff
changeset
|
159 abort(404) |
|
33cdb9faed64
Improve deletion of revisions (input validation)
Benoît Allard <benoit.allard@greenbone.net>
parents:
67
diff
changeset
|
160 del history[index] |
|
67
023080ffd995
Allow deletion of revisions
Benoît Allard <benoit.allard@greenbone.net>
parents:
66
diff
changeset
|
161 return redirect(url_for('.view')) |
|
023080ffd995
Allow deletion of revisions
Benoît Allard <benoit.allard@greenbone.net>
parents:
66
diff
changeset
|
162 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
163 @document.route('/distribution/edit', methods=['GET', 'POST']) |
| 0 | 164 @document_required |
| 165 def edit_distribution(): | |
| 166 if request.method != 'POST': | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
167 return render_template('document/edit_distribution.j2', distribution=get_current()._distribution) |
| 0 | 168 |
| 169 get_current().setDistribution(request.form['distribution']) | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
170 return redirect(url_for('.view')) |
| 0 | 171 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
172 @document.route('/severity/edit', methods=['GET', 'POST']) |
| 0 | 173 @document_required |
| 174 def edit_severity(): | |
| 175 severity = get_current()._aggregateseverity | |
| 176 if severity is None: | |
| 177 severity = CVRFAggregateSeverity('') | |
| 178 if request.method != 'POST': | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
179 return render_template('document/edit_severity.j2', severity=severity) |
| 0 | 180 if not request.form['severity']: |
| 181 severity = None | |
| 182 else: | |
| 183 severity._severity = request.form['severity'] | |
| 184 severity.setNamespace(request.form['namespace'] or None) | |
| 185 get_current().setAggregateSeverity(severity) | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
186 return redirect(url_for('.view')) |
| 0 | 187 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
188 @document.route('/note/<int:ordinal>') |
| 0 | 189 @document_required |
| 190 def view_note(ordinal): | |
| 191 note = get_current().getNote(ordinal) | |
| 192 if note is None: | |
| 193 abort(404) | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
194 return render_template('document/view_note.j2', note=note) |
| 0 | 195 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
196 @document.route('/note/<int:ordinal>/edit', methods=['GET', 'POST']) |
| 0 | 197 @document_required |
| 198 def edit_note(ordinal): | |
| 199 note = get_current().getNote(ordinal) | |
| 200 if note is None: | |
| 201 abort(404) | |
| 202 if request.method != 'POST': | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
203 return render_template('document/edit_note.j2', note=note, types = note.TYPES) |
| 0 | 204 |
|
58
fbc413b8a46e
Unify note update/creation from forms
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
205 update_note_from_request(note) |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
206 return redirect(url_for('.view_note', ordinal=note._ordinal )) |
| 0 | 207 |
| 208 | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
209 @document.route('/note/add', methods=['GET', 'POST']) |
| 0 | 210 @document_required |
| 211 def add_note(): | |
| 212 if request.method != 'POST': | |
| 213 next_ordinal = 1 | |
| 214 notes = get_current()._notes | |
| 215 if notes: | |
| 216 next_ordinal = notes[-1]._ordinal + 1 | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
217 return render_template('document/edit_note.j2', ordinal=next_ordinal, types=CVRFNote.TYPES, action='Add') |
| 0 | 218 |
|
58
fbc413b8a46e
Unify note update/creation from forms
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
219 note = create_note_from_request() |
| 0 | 220 get_current().addNote(note) |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
221 return redirect(url_for('.view')) |
| 0 | 222 |
| 73 | 223 @document.route('/note/<int:ordinal>/del', methods=['POST']) |
|
70
727c01e8e3e9
Allow deletion of Notes
Benoît Allard <benoit.allard@greenbone.net>
parents:
68
diff
changeset
|
224 @document_required |
| 73 | 225 def del_note(ordinal): |
|
70
727c01e8e3e9
Allow deletion of Notes
Benoît Allard <benoit.allard@greenbone.net>
parents:
68
diff
changeset
|
226 cvrf = get_current() |
|
727c01e8e3e9
Allow deletion of Notes
Benoît Allard <benoit.allard@greenbone.net>
parents:
68
diff
changeset
|
227 note = cvrf.getNote(ordinal) |
|
727c01e8e3e9
Allow deletion of Notes
Benoît Allard <benoit.allard@greenbone.net>
parents:
68
diff
changeset
|
228 if note is None: |
|
727c01e8e3e9
Allow deletion of Notes
Benoît Allard <benoit.allard@greenbone.net>
parents:
68
diff
changeset
|
229 flash('Note not found', 'danger') |
|
727c01e8e3e9
Allow deletion of Notes
Benoît Allard <benoit.allard@greenbone.net>
parents:
68
diff
changeset
|
230 abort(404) |
| 73 | 231 cvrf._notes.remove(note) |
|
70
727c01e8e3e9
Allow deletion of Notes
Benoît Allard <benoit.allard@greenbone.net>
parents:
68
diff
changeset
|
232 return redirect(url_for('.view')) |
|
727c01e8e3e9
Allow deletion of Notes
Benoît Allard <benoit.allard@greenbone.net>
parents:
68
diff
changeset
|
233 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
234 @document.route('/reference/<int:index>/edit', methods=['GET', 'POST']) |
| 0 | 235 @document_required |
| 236 def edit_reference(index): | |
| 237 try: | |
| 238 ref = get_current()._references[index] | |
| 239 except IndexError: | |
| 240 abort(404) | |
| 241 if request.method != 'POST': | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
242 return render_template('document/edit_reference.j2', _type=ref._type, url=ref._url, description=ref._description, types=('',) + ref.TYPES) |
| 0 | 243 |
|
61
55b72057b066
Unify reference update/creation from forms
Benoît Allard <benoit.allard@greenbone.net>
parents:
60
diff
changeset
|
244 update_reference_from_request(ref) |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
245 return redirect(url_for('.view')) |
| 0 | 246 |
| 247 | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
248 @document.route('/reference/add', methods=['GET', 'POST']) |
| 0 | 249 @document_required |
| 250 def add_reference(): | |
| 251 if request.method != 'POST': | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
252 return render_template('document/edit_reference.j2', action='Add', types=('',) + CVRFReference.TYPES) |
| 0 | 253 |
|
61
55b72057b066
Unify reference update/creation from forms
Benoît Allard <benoit.allard@greenbone.net>
parents:
60
diff
changeset
|
254 ref = create_reference_from_request() |
| 0 | 255 get_current().addReference(ref) |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
256 return redirect(url_for('.view')) |
| 0 | 257 |
| 73 | 258 @document.route('/reference/<int:index>/del', methods=['POST']) |
|
72
f9aa7b1e697e
Add possibility to delete References
Benoît Allard <benoit.allard@greenbone.net>
parents:
70
diff
changeset
|
259 @document_required |
| 73 | 260 def del_reference(index): |
|
72
f9aa7b1e697e
Add possibility to delete References
Benoît Allard <benoit.allard@greenbone.net>
parents:
70
diff
changeset
|
261 refs = get_current()._references |
|
f9aa7b1e697e
Add possibility to delete References
Benoît Allard <benoit.allard@greenbone.net>
parents:
70
diff
changeset
|
262 if not (0 <= index < len(refs)): |
|
f9aa7b1e697e
Add possibility to delete References
Benoît Allard <benoit.allard@greenbone.net>
parents:
70
diff
changeset
|
263 flash('Reference not found', 'danger') |
|
f9aa7b1e697e
Add possibility to delete References
Benoît Allard <benoit.allard@greenbone.net>
parents:
70
diff
changeset
|
264 abort(404) |
|
f9aa7b1e697e
Add possibility to delete References
Benoît Allard <benoit.allard@greenbone.net>
parents:
70
diff
changeset
|
265 del refs[index] |
|
f9aa7b1e697e
Add possibility to delete References
Benoît Allard <benoit.allard@greenbone.net>
parents:
70
diff
changeset
|
266 return redirect(url_for('.view')) |
|
f9aa7b1e697e
Add possibility to delete References
Benoît Allard <benoit.allard@greenbone.net>
parents:
70
diff
changeset
|
267 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
268 @document.route('/acknowledgment/<int:index>') |
| 0 | 269 @document_required |
| 270 def view_acknowledgment(index): | |
| 271 try: | |
| 272 ack = get_current()._acknowledgments[index] | |
| 273 except IndexError: | |
| 274 abort(404) | |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
275 return render_template('document/view_acknowledgment.j2', acknowledgment=ack, index=index, action='Update') |
| 0 | 276 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
277 @document.route('/acknowledgment/<int:index>/edit', methods=['GET', 'POST']) |
| 0 | 278 @document_required |
| 279 def edit_acknowledgment(index): | |
| 280 try: | |
| 281 ack = get_current()._acknowledgments[index] | |
| 282 except IndexError: | |
| 283 abort(404) | |
| 284 if request.method != 'POST': | |
|
64
aad7db3f93b6
Add support for Acknowledgments with multiple names and organizations
Benoît Allard <benoit.allard@greenbone.net>
parents:
63
diff
changeset
|
285 return render_template('document/edit_acknowledgment.j2', |
|
aad7db3f93b6
Add support for Acknowledgments with multiple names and organizations
Benoît Allard <benoit.allard@greenbone.net>
parents:
63
diff
changeset
|
286 names=ack._names, organizations=ack._organizations, |
|
aad7db3f93b6
Add support for Acknowledgments with multiple names and organizations
Benoît Allard <benoit.allard@greenbone.net>
parents:
63
diff
changeset
|
287 description=ack._description, url=ack._url, |
|
aad7db3f93b6
Add support for Acknowledgments with multiple names and organizations
Benoît Allard <benoit.allard@greenbone.net>
parents:
63
diff
changeset
|
288 action='Update') |
| 0 | 289 |
|
60
c6e7175ff28c
Unify acknowledgments update/creation from forms
Benoît Allard <benoit.allard@greenbone.net>
parents:
58
diff
changeset
|
290 update_acknowledgment_from_request(ack) |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
291 return redirect(url_for('.view')) |
| 0 | 292 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
293 @document.route('/acknowledgment/add', methods=['GET', 'POST']) |
| 0 | 294 @document_required |
| 295 def add_acknowledgment(): | |
| 296 if request.method != 'POST': | |
|
64
aad7db3f93b6
Add support for Acknowledgments with multiple names and organizations
Benoît Allard <benoit.allard@greenbone.net>
parents:
63
diff
changeset
|
297 return render_template('document/edit_acknowledgment.j2', |
|
aad7db3f93b6
Add support for Acknowledgments with multiple names and organizations
Benoît Allard <benoit.allard@greenbone.net>
parents:
63
diff
changeset
|
298 action='Add') |
| 0 | 299 |
|
60
c6e7175ff28c
Unify acknowledgments update/creation from forms
Benoît Allard <benoit.allard@greenbone.net>
parents:
58
diff
changeset
|
300 ack = create_acknowledgment_from_request() |
| 0 | 301 get_current().addAcknowledgment(ack) |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
302 return redirect(url_for('.view')) |
|
74
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
303 |
|
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
304 @document.route('/acknowledgment/<int:index>/del', methods=['POST']) |
|
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
305 @document_required |
|
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
306 def del_acknowledgment(index): |
|
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
307 acks = get_current()._acknowledgments |
|
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
308 if not( 0 <= index < len(acks)): |
|
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
309 flash('Acknowledgment not found', 'danger') |
|
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
310 abort(404) |
|
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
311 del acks[index] |
|
80cd8f65e72b
Add deletion of Acknowledgments
Benoît Allard <benoit.allard@greenbone.net>
parents:
73
diff
changeset
|
312 return redirect(url_for('.view')) |