Mercurial > farol > farol
annotate farol/main.py @ 140:3a56d5501ffa
Add HTTP error handling
| author | Benoît Allard <benoit.allard@greenbone.net> |
|---|---|
| date | Mon, 27 Oct 2014 16:46:18 +0100 |
| parents | 75c053bf2c3a |
| children | ce39a5267998 |
| rev | line source |
|---|---|
| 0 | 1 # -*- encoding: utf-8 -*- |
| 2 # Description: | |
| 3 # Farol Web Application | |
| 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 | |
| 25 import os | |
|
13
d5265a0da13a
Always log exceptions to file.
Benoît Allard <benoit.allard@greenbone.net>
parents:
12
diff
changeset
|
26 import logging |
|
d5265a0da13a
Always log exceptions to file.
Benoît Allard <benoit.allard@greenbone.net>
parents:
12
diff
changeset
|
27 from logging import FileHandler |
|
19
56cab60172ad
Fix issue in text import, Improve display of running versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
18
diff
changeset
|
28 import platform |
|
17
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
29 import urllib2 |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
30 from xml.etree import ElementTree as ET |
| 0 | 31 |
|
15
f8d51aaac8bc
Display sessionID in about page as well as various versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
13
diff
changeset
|
32 import farolluz |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
33 from farolluz.cvrf import CVRF, ValidationError |
|
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
34 from farolluz.parsers.cvrf import parse |
| 0 | 35 from farolluz.py2 import urlopen |
| 36 from farolluz.renderer import render as render_cvrf | |
| 37 from farolluz.utils import utcnow | |
| 38 | |
|
15
f8d51aaac8bc
Display sessionID in about page as well as various versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
13
diff
changeset
|
39 import flask |
|
29
891ee029a899
Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
27
diff
changeset
|
40 from flask import (Flask, request, render_template, redirect, url_for, flash, |
|
891ee029a899
Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
27
diff
changeset
|
41 make_response) |
| 0 | 42 from werkzeug import secure_filename |
| 43 | |
|
15
f8d51aaac8bc
Display sessionID in about page as well as various versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
13
diff
changeset
|
44 from . import __version__, cache |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
45 from .document import document |
|
12
4219d6fb4c38
Implement three kind of caches
Benoît Allard <benoit.allard@greenbone.net>
parents:
11
diff
changeset
|
46 from .session import get_current, set_current, has_current, document_required |
| 0 | 47 from .vulnerability import vulnerability |
| 48 from .producttree import producttree | |
| 49 | |
|
1
63b00c10ada8
Add Configuration support, warning when in DEBUG mode, and Deployment instruction
Benoît Allard <benoit.allard@greenbone.net>
parents:
0
diff
changeset
|
50 app = Flask(__name__, instance_relative_config=True) |
|
63b00c10ada8
Add Configuration support, warning when in DEBUG mode, and Deployment instruction
Benoît Allard <benoit.allard@greenbone.net>
parents:
0
diff
changeset
|
51 app.config.from_object('farol.config.Config') |
|
63b00c10ada8
Add Configuration support, warning when in DEBUG mode, and Deployment instruction
Benoît Allard <benoit.allard@greenbone.net>
parents:
0
diff
changeset
|
52 app.config.from_pyfile('farol.cfg', silent=True) |
| 0 | 53 |
|
12
4219d6fb4c38
Implement three kind of caches
Benoît Allard <benoit.allard@greenbone.net>
parents:
11
diff
changeset
|
54 app.register_blueprint(cache.mod, url_prefix='/cache') |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
55 app.register_blueprint(document, url_prefix='/document') |
| 0 | 56 app.register_blueprint(vulnerability, url_prefix='/vulnerability') |
| 57 app.register_blueprint(producttree, url_prefix='/producttree') | |
| 58 | |
|
13
d5265a0da13a
Always log exceptions to file.
Benoît Allard <benoit.allard@greenbone.net>
parents:
12
diff
changeset
|
59 file_handler = FileHandler(os.path.join(app.instance_path, 'farol.log')) |
|
d5265a0da13a
Always log exceptions to file.
Benoît Allard <benoit.allard@greenbone.net>
parents:
12
diff
changeset
|
60 file_handler.setLevel(logging.WARNING) |
|
d5265a0da13a
Always log exceptions to file.
Benoît Allard <benoit.allard@greenbone.net>
parents:
12
diff
changeset
|
61 app.logger.addHandler(file_handler) |
|
3
3478e20885fd
Add a ReverseProxy and fix the cache path everywhere
Benoît Allard <benoit.allard@greenbone.net>
parents:
1
diff
changeset
|
62 |
| 0 | 63 @app.context_processor |
| 64 def cache_content(): | |
| 65 """ List the documents in cache """ | |
|
12
4219d6fb4c38
Implement three kind of caches
Benoît Allard <benoit.allard@greenbone.net>
parents:
11
diff
changeset
|
66 return dict(caching=cache.caching_type(), |
|
4219d6fb4c38
Implement three kind of caches
Benoît Allard <benoit.allard@greenbone.net>
parents:
11
diff
changeset
|
67 cache=cache.cache_content()) |
| 0 | 68 |
| 69 @app.context_processor | |
| 70 def doc_properties(): | |
| 71 if not has_current(): | |
| 72 return {'has_current': False} | |
| 73 cvrf = get_current() | |
| 74 vulns = [(v.getTitle(), v._ordinal) for v in cvrf._vulnerabilities] | |
| 75 prods = [] | |
| 76 if cvrf._producttree: | |
| 77 prods = [(p._name, p._productid) for p in cvrf._producttree._products] | |
| 78 try: | |
| 79 cvrf.validate() | |
| 80 error = None | |
| 81 except ValidationError as ve: | |
| 82 error = ve | |
|
102
65341d05a8f8
Spawn a Modal when loading a doc, and another one is already loaded
Benoît Allard <benoit.allard@greenbone.net>
parents:
62
diff
changeset
|
83 return dict(has_current=True, vulnerabilities=vulns, products=prods, |
|
65341d05a8f8
Spawn a Modal when loading a doc, and another one is already loaded
Benoît Allard <benoit.allard@greenbone.net>
parents:
62
diff
changeset
|
84 error=error, current_id=cvrf.getDocId()) |
| 0 | 85 |
| 86 @app.template_filter('secure_filename') | |
| 87 def makeId(string): | |
| 88 return secure_filename(string) | |
| 89 | |
|
140
3a56d5501ffa
Add HTTP error handling
Benoît Allard <benoit.allard@greenbone.net>
parents:
131
diff
changeset
|
90 @app.errorhandler(400) |
|
3a56d5501ffa
Add HTTP error handling
Benoît Allard <benoit.allard@greenbone.net>
parents:
131
diff
changeset
|
91 @app.errorhandler(404) |
|
3a56d5501ffa
Add HTTP error handling
Benoît Allard <benoit.allard@greenbone.net>
parents:
131
diff
changeset
|
92 @app.errorhandler(405) |
|
3a56d5501ffa
Add HTTP error handling
Benoît Allard <benoit.allard@greenbone.net>
parents:
131
diff
changeset
|
93 @app.errorhandler(500) |
|
3a56d5501ffa
Add HTTP error handling
Benoît Allard <benoit.allard@greenbone.net>
parents:
131
diff
changeset
|
94 def error_page(error): |
|
3a56d5501ffa
Add HTTP error handling
Benoît Allard <benoit.allard@greenbone.net>
parents:
131
diff
changeset
|
95 return render_template('error.j2', e=error), getattr(error, 'code', 500) |
|
3a56d5501ffa
Add HTTP error handling
Benoît Allard <benoit.allard@greenbone.net>
parents:
131
diff
changeset
|
96 |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
97 @app.route('/') |
|
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
98 def welcome(): |
|
125
67c4b299736e
Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents:
102
diff
changeset
|
99 return render_template('welcome.j2', |
|
67c4b299736e
Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents:
102
diff
changeset
|
100 version=__version__, |
|
67c4b299736e
Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents:
102
diff
changeset
|
101 imports=[('New', 100), ('CVRF', 100)], |
|
67c4b299736e
Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents:
102
diff
changeset
|
102 exports=[('CVRF', 100), ('OpenVAS NASL from RHSA', 85), ('OVAL', 5) ], |
|
67c4b299736e
Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents:
102
diff
changeset
|
103 use_cases=[('Create a security advisory and publish as CVRF', 100), |
|
67c4b299736e
Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents:
102
diff
changeset
|
104 ('Edit a security advisory in CVRF format', 100)] |
|
67c4b299736e
Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents:
102
diff
changeset
|
105 ) |
|
7
8f41bb7f4681
Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents:
3
diff
changeset
|
106 |
|
17
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
107 def set_url(url): |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
108 try: content = urlopen(url).read() |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
109 except urllib2.HTTPError as e: |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
110 flash('Unable to retrieve %s: %s' % (url, e)) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
111 return |
|
18
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
112 set_text(content) |
|
17
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
113 |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
114 def set_RHSA(id_): |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
115 # validate input |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
116 if ':' not in id_: |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
117 flash('Wrong RHSA id: %s' % id_) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
118 return |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
119 year, index = id_.split(':', 1) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
120 try: |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
121 int(year) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
122 int(index) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
123 except ValueError: |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
124 flash('Wrong RHSA id: %s' % id_) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
125 return |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
126 # Process it |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
127 set_url("https://www.redhat.com/security/data/cvrf/%(year)s/cvrf-rhsa-%(year)s-%(index)s.xml" % {'year': year, 'index': index}) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
128 |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
129 def set_oracle(id_): |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
130 try: int(id_) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
131 except ValueError: |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
132 flash('Wrong Oracle id: %s' % id_) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
133 return |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
134 set_url("http://www.oracle.com/ocom/groups/public/@otn/documents/webcontent/%s.xml" % id_) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
135 |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
136 def set_cisco(id_): |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
137 if id_.count('-') < 2: |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
138 flash('Wrong cisco id: %s' % id_) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
139 return |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
140 kind, date, name = id_.split('-', 2) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
141 try: kind = {'sa': 'Advisory', 'sr': 'Response'}[kind] |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
142 except KeyError: |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
143 flash('Wrong cisco id: %s' % id_) |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
144 return |
|
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
145 set_url("http://tools.cisco.com/security/center/contentxml/CiscoSecurity%(kind)s/cisco-%(id)s/cvrf/cisco-%(id)s_cvrf.xml" % {'kind': kind, 'id': id_}) |
|
8
2ce3676c9b2e
Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
146 |
|
18
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
147 def set_text(text): |
|
19
56cab60172ad
Fix issue in text import, Improve display of running versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
18
diff
changeset
|
148 try: doc = parse(text) |
|
18
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
149 except ET.ParseError as e: |
|
19
56cab60172ad
Fix issue in text import, Improve display of running versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
18
diff
changeset
|
150 flash('Unable to parse Document: %s' % e) |
|
18
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
151 return |
|
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
152 set_current(doc) |
|
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
153 |
| 0 | 154 @app.route('/new', methods=['GET', 'POST']) |
| 155 def new(): | |
| 156 if request.method != 'POST': | |
|
102
65341d05a8f8
Spawn a Modal when loading a doc, and another one is already loaded
Benoît Allard <benoit.allard@greenbone.net>
parents:
62
diff
changeset
|
157 return render_template('new.j2', has_document=has_current(), now=utcnow()) |
|
12
4219d6fb4c38
Implement three kind of caches
Benoît Allard <benoit.allard@greenbone.net>
parents:
11
diff
changeset
|
158 |
|
8
2ce3676c9b2e
Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
159 if 'rhsa' in request.form: |
|
17
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
160 set_RHSA(request.form['id']) |
|
8
2ce3676c9b2e
Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
161 elif 'oracle' in request.form: |
|
17
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
162 set_oracle(request.form['id']) |
|
8
2ce3676c9b2e
Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
163 elif 'cisco' in request.form: |
|
17
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
164 set_cisco(request.form['id']) |
|
8
2ce3676c9b2e
Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
165 elif 'nasl' in request.form: |
|
2ce3676c9b2e
Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
166 flash("I'm not able to parse NASL scripts yet", 'danger') |
|
2ce3676c9b2e
Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
167 return redirect(url_for('new')) |
|
2ce3676c9b2e
Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents:
7
diff
changeset
|
168 elif 'url' in request.form: |
|
17
deced0345829
Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
15
diff
changeset
|
169 set_url(request.form['url']) |
| 0 | 170 elif 'local' in request.files: |
| 171 upload = request.files['local'] | |
|
18
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
172 fpath = os.path.join(app.instance_path, 'tmp', |
|
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
173 secure_filename(upload.filename)) |
|
21
e66fd84439bd
Create the tmp dir if not present.
Benoît Allard <benoit.allard@greenbone.net>
parents:
19
diff
changeset
|
174 if not os.path.exists(os.path.dirname(fpath)): |
|
e66fd84439bd
Create the tmp dir if not present.
Benoît Allard <benoit.allard@greenbone.net>
parents:
19
diff
changeset
|
175 os.makedirs(os.path.dirname(fpath)) |
| 0 | 176 upload.save(fpath) |
| 177 with open(fpath, 'rt') as f: | |
|
18
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
178 set_text(f.read()) |
| 0 | 179 os.remove(fpath) |
| 180 elif 'text' in request.form: | |
|
18
d547b6a0038e
Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents:
17
diff
changeset
|
181 set_text(request.form['text'].encode('utf-8')) |
| 0 | 182 else: |
| 183 set_current(CVRF(request.form['title'], 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
|
184 return redirect(url_for('document.view')) |
| 0 | 185 |
| 186 @app.route('/render/<format_>') | |
| 187 @document_required | |
| 188 def render(format_): | |
| 189 cvrf = get_current() | |
| 190 doc = render_cvrf(cvrf, format_ + '.j2') | |
|
29
891ee029a899
Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
27
diff
changeset
|
191 if 'raw' not in request.args: |
|
891ee029a899
Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
27
diff
changeset
|
192 return render_template('render.j2', format_=format_, title=cvrf._title, type_=cvrf._type, doc=doc) |
|
891ee029a899
Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
27
diff
changeset
|
193 response = make_response(doc) |
|
33
752b8bfe0baf
Use FarolLuz method to get a document ID (increase dependency version)
Benoît Allard <benoit.allard@greenbone.net>
parents:
29
diff
changeset
|
194 filename = secure_filename(cvrf.getDocId()) + "." + format_ |
|
29
891ee029a899
Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
27
diff
changeset
|
195 response.headers["content-disposition"] = 'attachement; filename=' + filename |
|
891ee029a899
Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
27
diff
changeset
|
196 response.headers["content-type"] = 'text/plain' |
|
891ee029a899
Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents:
27
diff
changeset
|
197 return response |
| 0 | 198 |
| 199 @app.route('/about') | |
| 200 def about(): | |
|
19
56cab60172ad
Fix issue in text import, Improve display of running versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
18
diff
changeset
|
201 versions = ((platform.python_implementation(), platform.python_version()), |
|
56cab60172ad
Fix issue in text import, Improve display of running versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
18
diff
changeset
|
202 ('Farol', __version__), ('FarolLuz', farolluz.__version__), |
|
56cab60172ad
Fix issue in text import, Improve display of running versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
18
diff
changeset
|
203 ('Flask', flask.__version__)) |
|
15
f8d51aaac8bc
Display sessionID in about page as well as various versions
Benoît Allard <benoit.allard@greenbone.net>
parents:
13
diff
changeset
|
204 return render_template('about.j2', instance_dir=app.instance_path, versions=versions) |
| 0 | 205 |