annotate farol/producttree.py @ 84:cc836f9e869e

Allow deletion of Products
author Benoît Allard <benoit.allard@greenbone.net>
date Thu, 09 Oct 2014 14:26:59 +0200
parents 2ecf13200e87
children b6a48669edb8
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 # Web stuff about the ProductTree
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 """\
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
26 Web-stuff about the producttree
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
27 """
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
28
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
29 from functools import wraps
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
30
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
31 from flask import Blueprint, render_template, redirect, url_for, abort, request, flash
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
32
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
33 from farolluz.cvrf import (CVRFProductBranch, CVRFFullProductName,
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
34 CVRFRelationship, CVRFGroup)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
35 from .session import document_required, get_current
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
36
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
37 producttree = Blueprint('producttree', __name__)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
38
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
39 def producttree_required(f):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
40 @wraps(f)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
41 def decorated_function(*args, **kwargs):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
42 if get_current()._producttree is None:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
43 flash('Operation invalid without producttree', 'warning')
22
dbfe89e3c6fe Fix producttree urls
Benoît Allard <benoit.allard@greenbone.net>
parents: 0
diff changeset
44 return redirect(url_for('document.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
45 return f(*args, **kwargs)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
46 return decorated_function
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
47
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
48 @producttree.route('/create', methods=['POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
49 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
50 def create():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
51 get_current().createProductTree()
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
52 return redirect(url_for('.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
53
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
54 @producttree.route('/')
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
55 @document_required
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
56 @producttree_required
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
57 def view():
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
58 return render_template('producttree/view.j2',
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
59 producttree=get_current()._producttree)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
60
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
61 @producttree.route('/branch/<path:path>/edit', methods=['GET', 'POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
62 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
63 @producttree_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
64 def edit_branch(path):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
65 path = [int(p) for p in path.split('/')]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
66 cvrf = get_current()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
67 ptree = cvrf._producttree
78
264d4579f6bf More input validation
Benoît Allard <benoit.allard@greenbone.net>
parents: 62
diff changeset
68 try: branch = cvrf._producttree.getBranch(path)
264d4579f6bf More input validation
Benoît Allard <benoit.allard@greenbone.net>
parents: 62
diff changeset
69 except (ValueError, IndexError): abort(404)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
70 if request.method != 'POST':
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
71 branches = [('', '')] + [(b.getName(), b.getPath(True)) for b in ptree.getNotTerminalBranches(branch)]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
72 return render_template('producttree/edit_branch.j2', branch=branch, branches=branches, types=branch.TYPES)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
73
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
74 pbranch = ptree
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
75 if request.form['parent']:
78
264d4579f6bf More input validation
Benoît Allard <benoit.allard@greenbone.net>
parents: 62
diff changeset
76 try: pbranch = ptree.getBranch([int(p) for p in request.form['parent'].split('/')])
264d4579f6bf More input validation
Benoît Allard <benoit.allard@greenbone.net>
parents: 62
diff changeset
77 except (ValueError, IndexError): abort(404)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
78
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
79 if pbranch is not branch.getParent():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
80 # We have to 're-link' the element ...
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
81 # 1. unlink
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
82 branch.unlink()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
83 # 2. set the new parent
82
d6fa288e92a7 Update to latest farolluz (unlink/link)
Benoît Allard <benoit.allard@greenbone.net>
parents: 80
diff changeset
84 branch.link(pbranch)
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 branch._type = request.form['type']
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
87 branch._name = request.form['name']
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
88 return redirect(url_for('.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
89
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
90 @producttree.route('/branch/add', methods=['GET', 'POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
91 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
92 @producttree_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
93 def add_branch():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
94 cvrf = get_current()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
95 ptree = cvrf._producttree
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
96 if request.method != 'POST':
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
97 branches = [('', '')] + [(b.getName(), b.getPath(True)) for b in ptree.getNotTerminalBranches()]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
98 return render_template('producttree/edit_branch.j2', branch=CVRFProductBranch('', '', ptree), action='Add', branches=branches, types=CVRFProductBranch.TYPES)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
99
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
100 pbranch = ptree
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
101 if request.form['parent']:
78
264d4579f6bf More input validation
Benoît Allard <benoit.allard@greenbone.net>
parents: 62
diff changeset
102 try: pbranch = ptree.getBranch([int(p) for p in request.form['parent'].split('/')])
264d4579f6bf More input validation
Benoît Allard <benoit.allard@greenbone.net>
parents: 62
diff changeset
103 except (ValueError, IndexError): abort(404)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
104 branch = CVRFProductBranch(request.form['type'], request.form['name'], pbranch)
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
105 return redirect(url_for('.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
106
80
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
107 @producttree.route('/branch/<path:path>/del', methods=['POST'])
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
108 @document_required
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
109 @producttree_required
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
110 def del_branch(path):
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
111 path = [int(p) for p in path.split('/')]
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
112 cvrf = get_current()
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
113 ptree = cvrf._producttree
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
114 try: branch = cvrf._producttree.getBranch(path)
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
115 except (ValueError, IndexError):
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
116 flash('Cannot find Branch', 'danger')
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
117 abort(404)
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
118 if branch._childs:
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
119 flash('Cannot delete a branch with childs', 'danger')
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
120 abort(403)
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
121 branch.unlink()
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
122 del branch
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
123 return redirect(url_for('.view'))
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
124
c00f20bd90ba Allow deletion of (orphaned) Branches
Benoît Allard <benoit.allard@greenbone.net>
parents: 78
diff changeset
125
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
126 @producttree.route('/product/<productid>')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
127 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
128 @producttree_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
129 def view_product(productid):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
130 cvrf = get_current()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
131 try:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
132 product = cvrf.getProductForID(productid)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
133 except IndexError:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
134 abort(404)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
135 return render_template('producttree/view_product.j2', product=product, cvrf=cvrf)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
136
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
137 @producttree.route('/product/<productid>/edit', methods=['GET', 'POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
138 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
139 @producttree_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
140 def edit_product(productid):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
141 cvrf = get_current()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
142 ptree = cvrf._producttree
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
143 try:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
144 product = cvrf.getProductForID(productid)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
145 except KeyError:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
146 abort(404)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
147
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
148 if request.method != 'POST':
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
149 current_rel = ''
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
150 if product.isRelationship():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
151 current_rel = str(ptree._relationships.index(product.getCurrentRelationship()))
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
152 leaves = [('', '')] + [(b.getName(), b.getPath(True)) for b in ptree.getOrphanedBranches(product)]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
153 rels = [('', '')] + [(ptree.getNameOfRelationship(r), str(i)) for i, r in ptree.getOrphanedRelationships(product)]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
154 return render_template('producttree/edit_product.j2', product=product, action='Update', orphaned_leaves=leaves, orphaned_relationships=rels, current_rel=current_rel)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
155
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
156 if request.form['parent_branch'] and request.form['parent_relationship']:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
157 flash('Cannot set a parent branch and parent relationship', 'danger')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
158 return redirect(url_for('.edit_product', productid=productid))
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
159
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
160 oldp = product._parent
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
161 if request.form['parent_branch']:
78
264d4579f6bf More input validation
Benoît Allard <benoit.allard@greenbone.net>
parents: 62
diff changeset
162 try: pbranch = ptree.getBranch([int(p) for p in request.form['parent_branch'].split('/')])
264d4579f6bf More input validation
Benoît Allard <benoit.allard@greenbone.net>
parents: 62
diff changeset
163 except (ValueError, IndexError): abort(404)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
164 if pbranch is not oldp:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
165 # Gonna be funny, needs re-link
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
166 product.unlink()
82
d6fa288e92a7 Update to latest farolluz (unlink/link)
Benoît Allard <benoit.allard@greenbone.net>
parents: 80
diff changeset
167 # And Link again
d6fa288e92a7 Update to latest farolluz (unlink/link)
Benoît Allard <benoit.allard@greenbone.net>
parents: 80
diff changeset
168 product.link(pbranch)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
169 elif request.form['parent_relationship']:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
170 prel = ptree._relationships[int(request.form['parent_relationship'])]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
171 if prel is not oldp:
82
d6fa288e92a7 Update to latest farolluz (unlink/link)
Benoît Allard <benoit.allard@greenbone.net>
parents: 80
diff changeset
172 # Unlink
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
173 product.unlink()
82
d6fa288e92a7 Update to latest farolluz (unlink/link)
Benoît Allard <benoit.allard@greenbone.net>
parents: 80
diff changeset
174 # Link again
d6fa288e92a7 Update to latest farolluz (unlink/link)
Benoît Allard <benoit.allard@greenbone.net>
parents: 80
diff changeset
175 product.link(prel)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
176 else:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
177 if ptree is not oldp:
82
d6fa288e92a7 Update to latest farolluz (unlink/link)
Benoît Allard <benoit.allard@greenbone.net>
parents: 80
diff changeset
178 # Unlink
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
179 product.unlink()
82
d6fa288e92a7 Update to latest farolluz (unlink/link)
Benoît Allard <benoit.allard@greenbone.net>
parents: 80
diff changeset
180 # Link again
d6fa288e92a7 Update to latest farolluz (unlink/link)
Benoît Allard <benoit.allard@greenbone.net>
parents: 80
diff changeset
181 product.link(ptree)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
182
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
183 product._productid = request.form['productid']
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
184 product._name = request.form['name']
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
185 product._cpe = request.form['cpe'] or None
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
186 return redirect(url_for('.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
187
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
188 @producttree.route('/product/add', methods=['GET', 'POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
189 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
190 @producttree_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
191 def add_product():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
192 cvrf = get_current()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
193 ptree = cvrf._producttree
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 if request.method != 'POST':
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
196 product=CVRFFullProductName('', '', ptree)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
197 leaves = [('', '')] + [(b.getName(), b.getPath(True)) for b in ptree.getOrphanedBranches()]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
198 rels = [('', '')] + [(ptree.getNameOfRelationship(r), str(i)) for i, r in ptree.getOrphanedRelationships()]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
199 return render_template('producttree/edit_product.j2', product=product, action='Add', orphaned_leaves=leaves, orphaned_relationships=rels, current_rel='')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
200
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
201 if request.form['parent_branch'] and request.form['parent_relationship']:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
202 flash('Cannot set a parent branch and parent relationship', 'danger')
62
ce49bd1512dd Make pyflafes a happier
Benoît Allard <benoit.allard@greenbone.net>
parents: 45
diff changeset
203 return redirect(url_for('.add_product'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
204
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
205 parent = ptree
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
206 if request.form['parent_branch']:
83
2ecf13200e87 Fix issue from a previous commit
Benoît Allard <benoit.allard@greenbone.net>
parents: 82
diff changeset
207 try: parent = ptree.getBranch([int(p) for p in request.form['parent_branch'].split('/')])
78
264d4579f6bf More input validation
Benoît Allard <benoit.allard@greenbone.net>
parents: 62
diff changeset
208 except (ValueError, IndexError): abort(404)
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
209 elif request.form['parent_relationship']:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
210 parent = ptree._relationships[int(request.form['parent_relationship'])]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
211
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
212 product = CVRFFullProductName(request.form['productid'], request.form['name'], parent, request.form['cpe'] or None)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
213 ptree.addProduct(product)
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
214 return redirect(url_for('.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
215
84
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
216 @producttree.route('/product/<productid>/del', methods=['POST'])
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
217 @document_required
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
218 @producttree_required
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
219 def del_product(productid):
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
220 cvrf = get_current()
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
221 ptree = cvrf._producttree
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
222 try:
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
223 product = cvrf.getProductForID(productid)
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
224 except KeyError:
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
225 abort(404)
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
226 product.unlink()
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
227 ptree._products.remove(product)
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
228 del product
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
229 return redirect(url_for('.view'))
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
230
cc836f9e869e Allow deletion of Products
Benoît Allard <benoit.allard@greenbone.net>
parents: 83
diff changeset
231
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
232 @producttree.route('/relationship/<int:index>/edit', methods=['GET', 'POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
233 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
234 @producttree_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
235 def edit_relationship(index):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
236 cvrf = get_current()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
237 ptree = cvrf._producttree
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
238 try:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
239 rel = ptree._relationships[index]
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
240 except IndexError:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
241 abort(404)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
242
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
243 if request.method != 'POST':
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
244 return render_template('producttree/edit_relationship.j2', productreference=rel._productreference, relationtype=rel._relationtype, relatestoproductreference=rel._relatestoproductreference, types=rel.TYPES)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
245
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
246 rel._productreference = request.form['productreference']
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
247 rel._relationtype = request.form['relationtype']
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
248 rel._relatestoproductreference = request.form['relatestoproductreference']
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
249 return redirect(url_for('.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
250
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
251 @producttree.route('/relationship/add', methods=['GET', 'POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
252 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
253 @producttree_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
254 def add_relationship():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
255 cvrf = get_current()
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
256 ptree = cvrf._producttree
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
257 if len(ptree._products) < 2:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
258 flash('You need to have at least two products to create a relationship', 'warning')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
259 return redirect(url_for('.add_product'))
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
260
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
261 if request.method != 'POST':
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
262 return render_template('producttree/edit_relationship.j2', action='Add', types=CVRFRelationship.TYPES)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
263
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
264 prodid1 = request.form['productreference']
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
265 prodid2 = request.form['relatestoproductreference']
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
266
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
267 if prodid1 == prodid2:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
268 flash('You cannot create a relationship with the same product twice', 'danger')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
269 return redirect(url_for('.add_relationship'))
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
270
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
271 rel = CVRFRelationship(prodid1, request.form['relationtype'], prodid2)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
272 ptree.addRelationship(rel)
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
273 return redirect(url_for('.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
274
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
275 @producttree.route('/group/<groupid>/edit', methods=['GET', 'POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
276 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
277 @producttree_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
278 def edit_group(groupid):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
279 try:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
280 group = get_current().getGroupForID(groupid)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
281 except KeyError:
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
282 abort(404)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
283 if request.method != 'POST':
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
284 return render_template('producttree/edit_group.j2', groupid=group._groupid, description=group._description, productids=group._productids)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
285
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
286 group._groupid = request.form['groupid']
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
287 group.setDescription(request.form['description'] or None)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
288 group._productids = []
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
289 for productid in request.form.getlist('products'):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
290 group.addProductID(productid)
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
291 return redirect(url_for('.view'))
0
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
292
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
293 @producttree.route('/group/add', methods=['GET', 'POST'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
294 @document_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
295 @producttree_required
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
296 def add_group():
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
297 if request.method != 'POST':
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
298 return render_template('producttree/edit_group.j2', action='Add')
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
299
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
300 group = CVRFGroup(request.form['groupid'])
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
301 group.setDescription(request.form['description'] or None)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
302 for productid in request.form.getlist('products'):
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
303 group.addProductID(productid)
4a9f23230eba Initial Release
Benoît Allard <benoit.allard@greenbone.net>
parents:
diff changeset
304 get_current()._producttree.addGroup(group)
45
7c4d001839e7 Move the Display of the ProductTree to its own page
Benoît Allard <benoit.allard@greenbone.net>
parents: 22
diff changeset
305 return redirect(url_for('.view'))

http://farol.wald.intevation.org