farol/farol: farol/producttree.py annotate

author	Benoît Allard <benoit.allard@greenbone.net>
date	Wed, 24 Sep 2014 10:07:49 +0200
parents
children	dbfe89e3c6fe

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')
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	44 return redirect(url_for('document'))
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()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	52 return redirect(url_for('document'))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	53
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	54
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	55 @producttree.route('/branch/<path:path>/edit', methods=['GET', 'POST'])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	56 @document_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	57 @producttree_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	58 def edit_branch(path):
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	59 path = [int(p) for p in path.split('/')]
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	60 cvrf = get_current()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	61 ptree = cvrf._producttree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	62 try:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	63 branch = cvrf._producttree.getBranch(path)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	64 except ValueError:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	65 abort(404)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	66 if request.method != 'POST':
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	67 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	68 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	69
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	70 pbranch = ptree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	71 if request.form['parent']:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	72 pbranch = ptree.getBranch([int(p) for p in request.form['parent'].split('/')])
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 if pbranch is not branch.getParent():
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	75 # We have to 're-link' the element ...
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	76 # 1. unlink
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	77 branch.unlink()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	78 # 2. set the new parent
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	79 branch._parentbranch = pbranch
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	80 # 3. add to the Tree (again)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	81 ptree.addBranch(branch)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	82
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	83 branch._type = request.form['type']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	84 branch._name = request.form['name']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	85 return redirect(url_for('document'))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	86
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	87 @producttree.route('/branch/add', methods=['GET', 'POST'])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	88 @document_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	89 @producttree_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	90 def add_branch():
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	91 cvrf = get_current()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	92 ptree = cvrf._producttree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	93 if request.method != 'POST':
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	94 branches = [('', '')] + [(b.getName(), b.getPath(True)) for b in ptree.getNotTerminalBranches()]
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	95 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	96
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	97 pbranch = ptree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	98 if request.form['parent']:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	99 pbranch = ptree.getBranch([int(p) for p in request.form['parent'].split('/')])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	100 branch = CVRFProductBranch(request.form['type'], request.form['name'], pbranch)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	101 ptree.addBranch(branch)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	102 return redirect(url_for('document'))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	103
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	104 @producttree.route('/product/<productid>')
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	105 @document_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	106 @producttree_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	107 def view_product(productid):
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	108 cvrf = get_current()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	109 try:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	110 product = cvrf.getProductForID(productid)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	111 except IndexError:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	112 abort(404)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	113 return render_template('producttree/view_product.j2', product=product, cvrf=cvrf)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	114
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	115 @producttree.route('/product/<productid>/edit', methods=['GET', 'POST'])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	116 @document_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	117 @producttree_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	118 def edit_product(productid):
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	119 cvrf = get_current()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	120 ptree = cvrf._producttree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	121 try:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	122 product = cvrf.getProductForID(productid)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	123 except KeyError:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	124 abort(404)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	125
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	126 if request.method != 'POST':
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	127 current_rel = ''
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	128 if product.isRelationship():
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	129 current_rel = str(ptree._relationships.index(product.getCurrentRelationship()))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	130 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	131 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	132 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	133
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	134 if request.form['parent_branch'] and request.form['parent_relationship']:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	135 flash('Cannot set a parent branch and parent relationship', 'danger')
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	136 return redirect(url_for('.edit_product', productid=productid))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	137
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	138 oldp = product._parent
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	139 if request.form['parent_branch']:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	140 pbranch = ptree.getBranch([int(p) for p in request.form['parent_branch'].split('/')])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	141 if pbranch is not oldp:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	142 # Gonna be funny, needs re-link
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	143 product.unlink()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	144 product._parent = pbranch
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	145 ptree.addProduct(product)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	146 elif request.form['parent_relationship']:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	147 prel = ptree._relationships[int(request.form['parent_relationship'])]
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	148 if prel is not oldp:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	149 product.unlink()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	150 product._parent = prel
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	151 ptree.addProduct(product)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	152 else:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	153 if ptree is not oldp:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	154 product.unlink()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	155 product._parent = ptree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	156 ptree.addProduct(product)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	157
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	158 product._productid = request.form['productid']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	159 product._name = request.form['name']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	160 product._cpe = request.form['cpe'] or None
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	161 return redirect(url_for('document'))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	162
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	163 @producttree.route('/product/add', methods=['GET', 'POST'])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	164 @document_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	165 @producttree_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	166 def add_product():
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	167 cvrf = get_current()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	168 ptree = cvrf._producttree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	169
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	170 if request.method != 'POST':
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	171 product=CVRFFullProductName('', '', ptree)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	172 leaves = [('', '')] + [(b.getName(), b.getPath(True)) for b in ptree.getOrphanedBranches()]
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	173 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	174 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	175
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	176 if request.form['parent_branch'] and request.form['parent_relationship']:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	177 flash('Cannot set a parent branch and parent relationship', 'danger')
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	178 return redirect(url_for('.edit_product', productid=productid))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	179
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	180 parent = ptree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	181 if request.form['parent_branch']:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	182 parent = ptree.getBranch([int(p) for p in request.form['parent_branch'].split('/')])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	183 elif request.form['parent_relationship']:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	184 parent = ptree._relationships[int(request.form['parent_relationship'])]
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	185
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	186 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	187 ptree.addProduct(product)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	188 return redirect(url_for('document'))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	189
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	190 @producttree.route('/relationship/<int:index>/edit', methods=['GET', 'POST'])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	191 @document_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	192 @producttree_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	193 def edit_relationship(index):
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	194 cvrf = get_current()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	195 ptree = cvrf._producttree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	196 try:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	197 rel = ptree._relationships[index]
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	198 except IndexError:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	199 abort(404)
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.method != 'POST':
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	202 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	203
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	204 rel._productreference = request.form['productreference']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	205 rel._relationtype = request.form['relationtype']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	206 rel._relatestoproductreference = request.form['relatestoproductreference']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	207 return redirect(url_for('document'))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	208
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	209 @producttree.route('/relationship/add', methods=['GET', 'POST'])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	210 @document_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	211 @producttree_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	212 def add_relationship():
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	213 cvrf = get_current()
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	214 ptree = cvrf._producttree
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	215 if len(ptree._products) < 2:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	216 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	217 return redirect(url_for('.add_product'))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	218
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	219 if request.method != 'POST':
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	220 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	221
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	222 prodid1 = request.form['productreference']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	223 prodid2 = request.form['relatestoproductreference']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	224
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	225 if prodid1 == prodid2:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	226 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	227 return redirect(url_for('.add_relationship'))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	228
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	229 rel = CVRFRelationship(prodid1, request.form['relationtype'], prodid2)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	230 ptree.addRelationship(rel)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	231 return redirect(url_for('document'))
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	232
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	233 @producttree.route('/group/<groupid>/edit', methods=['GET', 'POST'])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	234 @document_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	235 @producttree_required
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	236 def edit_group(groupid):
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	237 try:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	238 group = get_current().getGroupForID(groupid)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	239 except KeyError:
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	240 abort(404)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	241 if request.method != 'POST':
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	242 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	243
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	244 group._groupid = request.form['groupid']
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	245 group.setDescription(request.form['description'] or None)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	246 group._productids = []
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	247 for productid in request.form.getlist('products'):
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	248 group.addProductID(productid)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	249 return redirect(url_for('document'))
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('/group/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_group():
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	255 if request.method != 'POST':
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	256 return render_template('producttree/edit_group.j2', action='Add')
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	257
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	258 group = CVRFGroup(request.form['groupid'])
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	259 group.setDescription(request.form['description'] or None)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	260 for productid in request.form.getlist('products'):
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	261 group.addProductID(productid)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	262 get_current()._producttree.addGroup(group)
4a9f23230eba Initial Release Benoît Allard <benoit.allard@greenbone.net> parents: diff changeset	263 return redirect(url_for('document'))

0

4a9f23230eba Initial Release