Mercurial > farol > farol

annotate farol/main.py @ 125:67c4b299736e

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
Update look-and-feel
author Benoît Allard <benoit.allard@greenbone.net>
date Thu, 23 Oct 2014 16:18:30 +0200
parents 65341d05a8f8
children 75c053bf2c3a
rev   line source
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
1 # -*- encoding: utf-8 -*-
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
2 # Description:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
3 # Farol Web Application
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
4 #
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
5 # Authors:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
6 # Benoît Allard <benoit.allard@greenbone.net>
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
7 #
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
8 # Copyright:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
9 # Copyright (C) 2014 Greenbone Networks GmbH
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
10 #
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
11 # This program is free software; you can redistribute it and/or
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
12 # modify it under the terms of the GNU General Public License
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
13 # as published by the Free Software Foundation; either version 2
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
14 # of the License, or (at your option) any later version.
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
15 #
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
16 # This program is distributed in the hope that it will be useful,
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
17 # but WITHOUT ANY WARRANTY; without even the implied warranty of
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
18 # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
19 # GNU General Public License for more details.
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
20 #
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
21 # You should have received a copy of the GNU General Public License
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
22 # along with this program; if not, write to the Free Software
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
23 # Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
24
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
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
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
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
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
35 from farolluz.py2 import urlopen
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
36 from farolluz.renderer import render as render_cvrf
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
37 from farolluz.utils import utcnow
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
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
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
42 from werkzeug import secure_filename
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
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
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
47 from .vulnerability import vulnerability
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
48 from .producttree import producttree
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
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
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
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
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
56 app.register_blueprint(vulnerability, url_prefix='/vulnerability')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
57 app.register_blueprint(producttree, url_prefix='/producttree')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
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
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
63 @app.context_processor
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
64 def cache_content():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
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
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
68
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
69 @app.context_processor
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
70 def doc_properties():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
71 if not has_current():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
72 return {'has_current': False}
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
73 cvrf = get_current()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
74 vulns = [(v.getTitle(), v._ordinal) for v in cvrf._vulnerabilities]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
75 prods = []
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
76 if cvrf._producttree:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
77 prods = [(p._name, p._productid) for p in cvrf._producttree._products]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
78 try:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
79 cvrf.validate()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
80 error = None
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
81 except ValidationError as ve:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
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
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
85
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
86 @app.template_filter('secure_filename')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
87 def makeId(string):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
88 return secure_filename(string)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
89
7
8f41bb7f4681 Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents: 3
diff changeset
90 @app.route('/')
8f41bb7f4681 Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents: 3
diff changeset
91 def welcome():
125
67c4b299736e Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents: 102
diff changeset
92 return render_template('welcome.j2',
67c4b299736e Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents: 102
diff changeset
93 version=__version__,
67c4b299736e Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents: 102
diff changeset
94 imports=[('New', 100), ('CVRF', 100)],
67c4b299736e Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents: 102
diff changeset
95 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
96 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
97 ('Edit a security advisory in CVRF format', 100)]
67c4b299736e Update look-and-feel
Benoît Allard <benoit.allard@greenbone.net>
parents: 102
diff changeset
98 )
7
8f41bb7f4681 Move the Document management routines to a document Blueprint
Benoît Allard <benoit.allard@greenbone.net>
parents: 3
diff changeset
99
17
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
100 def set_url(url):
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
101 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
102 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
103 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
104 return
18
d547b6a0038e Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 17
diff changeset
105 set_text(content)
17
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
106
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
107 def set_RHSA(id_):
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
108 # validate input
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
109 if ':' not in id_:
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
110 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
111 return
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
112 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
113 try:
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
114 int(year)
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
115 int(index)
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
116 except ValueError:
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 # Process it
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
120 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
121
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
122 def set_oracle(id_):
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
123 try: int(id_)
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
124 except ValueError:
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
125 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
126 return
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
127 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
128
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
129 def set_cisco(id_):
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
130 if id_.count('-') < 2:
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
131 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
132 return
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
133 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
134 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
135 except KeyError:
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
136 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
137 return
deced0345829 Add more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 15
diff changeset
138 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
139
18
d547b6a0038e Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 17
diff changeset
140 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
141 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
142 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
143 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
144 return
d547b6a0038e Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 17
diff changeset
145 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
146
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
147 @app.route('/new', methods=['GET', 'POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
148 def new():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
149 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
150 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
151
8
2ce3676c9b2e Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents: 7
diff changeset
152 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
153 set_RHSA(request.form['id'])
8
2ce3676c9b2e Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents: 7
diff changeset
154 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
155 set_oracle(request.form['id'])
8
2ce3676c9b2e Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents: 7
diff changeset
156 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
157 set_cisco(request.form['id'])
8
2ce3676c9b2e Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents: 7
diff changeset
158 elif 'nasl' in request.form:
2ce3676c9b2e Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents: 7
diff changeset
159 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
160 return redirect(url_for('new'))
2ce3676c9b2e Ease the import of new documents
Benoît Allard <benoit.allard@greenbone.net>
parents: 7
diff changeset
161 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
162 set_url(request.form['url'])
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
163 elif 'local' in request.files:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
164 upload = request.files['local']
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
165 if not upload.filename.endswith('.xml'):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
166 flash('Uploaded files should end in .xml', 'danger')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
167 return redirect(url_for('new'))
18
d547b6a0038e Add more and more error handling in the /new page.
Benoît Allard <benoit.allard@greenbone.net>
parents: 17
diff changeset
168 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
169 secure_filename(upload.filename))
21
e66fd84439bd Create the tmp dir if not present.
Benoît Allard <benoit.allard@greenbone.net>
parents: 19
diff changeset
170 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
171 os.makedirs(os.path.dirname(fpath))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
172 upload.save(fpath)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
173 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
174 set_text(f.read())
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
175 os.remove(fpath)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
176 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
177 set_text(request.form['text'].encode('utf-8'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
178 else:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
179 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
180 return redirect(url_for('document.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
181
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
182 @app.route('/render/<format_>')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
183 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
184 def render(format_):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
185 cvrf = get_current()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
186 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
187 if 'raw' not in request.args:
891ee029a899 Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents: 27
diff changeset
188 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
189 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
190 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
191 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
192 response.headers["content-type"] = 'text/plain'
891ee029a899 Offer to download the rendered document.
Benoît Allard <benoit.allard@greenbone.net>
parents: 27
diff changeset
193 return response
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
194
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
195 @app.route('/about')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
196 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
197 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
198 ('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
199 ('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
200 return render_template('about.j2', instance_dir=app.instance_path, versions=versions)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
201
http://farol.wald.intevation.org