Page MenuHomec4science
Files F92249685

admin.py
No OneTemporary
Actions

File Metadata

Created
Mon, Nov 18, 18:27

admin.py
View Options

# -*- coding: utf-8 -*-
##
## This file is part of Invenio.
## Copyright (C) 2012, 2014 CERN.
##
## Invenio is free software; you can redistribute it and/or
## modify it under the terms of the GNU General Public License as
## published by the Free Software Foundation; either version 2 of the
## License, or (at your option) any later version.
##
## Invenio is distributed in the hope that it will be useful, but
## WITHOUT ANY WARRANTY; without even the implied warranty of
## MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
## General Public License for more details.
##
## You should have received a copy of the GNU General Public License
## along with Invenio; if not, write to the Free Software Foundation, Inc.,
## 59 Temple Place, Suite 330, Boston, MA 02111-1307, USA.
"""WebSearch Admin Flask Blueprint."""
from __future__ import print_function
from flask import Blueprint, g, render_template, request, flash, redirect, \
url_for, abort
from flask.ext.breadcrumbs import register_breadcrumb
from invenio.ext.sqlalchemy import db
from ..models import Collection, CollectionCollection, \
Collectionname, CollectionPortalbox, Portalbox
from invenio.base.i18n import _
from invenio.base.decorators import templated
from flask.ext.login import current_user, login_required
from invenio.ext.principal import permission_required
from invenio.base.i18n import language_list_long
# imports the necessary forms
from ..admin_forms import CollectionForm, TranslationsForm
not_guest = lambda: not current_user.is_guest
blueprint = Blueprint('websearch_admin', __name__,
url_prefix="/admin/websearch",
template_folder='../templates'
)
#breadcrumbs=[(_('Configure WebSearch'), 'websearch_admin.index')])
@blueprint.route('/', methods=['GET', 'POST'])
@blueprint.route('/index', methods=['GET', 'POST'])
@login_required
@permission_required('cfgwebsearch')
@templated('search/admin_index.html')
@register_breadcrumb(blueprint, 'admin.websearch_admin', _('WebSearch'))
def index():
"""WebSearch admin interface with editable collection tree."""
collection = Collection.query.get_or_404(1)
orphans = Collection.query.filter(
db.not_(db.or_(
Collection.id.in_(db.session.query(
CollectionCollection.id_son).subquery()),
Collection.id.in_(db.session.query(
CollectionCollection.id_dad).subquery())
))).all()
return dict(collection=collection, orphans=orphans)
@blueprint.route('/modifycollectiontree', methods=['POST'])
@login_required
@permission_required('cfgwebsearch')
def modifycollectiontree():
"""
Handler of the tree changing operations triggered by the drag and drop operation.
"""
# Get the requests parameters
id_son = request.form.get('id_son', 0, type=int)
id_dad = request.form.get('id_dad', 0, type=int)
id_new_dad = request.form.get('id_new_dad', 0, type=int)
score = request.form.get('score', 0, type=int)
#if id_dad == id_new_dad:
# score = score + 1
type = request.form.get('type', 'r')
# Check if collection exits.
Collection.query.get_or_404(id_son)
if id_dad > 0:
# Get only one record
cc = CollectionCollection.query.filter(
db.and_(
CollectionCollection.id_son == id_son,
CollectionCollection.id_dad == id_dad
)).one()
dad = Collection.query.get_or_404(id_dad)
dad._collection_children.remove(cc)
dad._collection_children.reorder()
else:
cc = CollectionCollection(
id_dad=id_new_dad,
id_son=id_son,
type=type)
db.session.add(cc)
if id_new_dad == 0:
db.session.delete(cc)
else:
new_dad = Collection.query.get_or_404(id_new_dad)
cc.id_dad = id_new_dad
try:
descendants = Collection.query.get(id_son).descendants_ids
ancestors = new_dad.ancestors_ids
print(descendants, ancestors)
if descendants & ancestors:
raise
except:
## Cycle has been detected.
db.session.rollback()
abort(406)
new_dad._collection_children.reorder()
new_dad._collection_children.insert(score, cc)
#FIXME add dbrecs rebuild for modified trees.
db.session.commit()
return 'done'
@blueprint.route('/collectiontree', methods=['GET', 'POST'])
@login_required
@permission_required('cfgwebsearch')
def managecollectiontree():
"""Here is where managing the tree is possible."""
collection = Collection.query.get_or_404(1)
orphans = Collection.query.filter(
Collection.id != CollectionCollection.id_dad,
id != CollectionCollection.id_son).get_or_404(1)
return dict(collection=collection, orphans=orphans)
@blueprint.route('/collection/<name>', methods=['GET', 'POST'])
@blueprint.route('/collection/view/<name>', methods=['GET', 'POST'])
@login_required
@permission_required('cfgwebsearch')
def manage_collection(name):
collection = Collection.query.filter(Collection.name == name).first_or_404()
form = CollectionForm(request.form, obj=collection)
# gets the collections translations
translations = dict((x.ln, x.value) for x in collection.collection_names)
# Creating the translations form
TranslationsFormFilled = TranslationsForm(language_list_long(),
translations)
translation_form = TranslationsFormFilled(request.form)
#for x in collection.collection_names:
# translation_form[x.ln](default = x.value)
#translation_form.populate_obj(translations)
return render_template('search/admin_collection.html',
collection=collection, form=form,
translation_form=translation_form)
@blueprint.route('/collection/update/<id>', methods=['POST'])
@login_required
@permission_required('cfgwebsearch')
def update(id):
form = CollectionForm(request.form)
if request.method == 'POST': # and form.validate():
collection = Collection.query.filter(Collection.id == id).first_or_404()
form.populate_obj(collection)
db.session.commit()
flash(_('Collection was updated'), "info")
return redirect(url_for('.index'))
@blueprint.route('/collection/new', methods=['GET', 'POST'])
@blueprint.route('/collection/add', methods=['GET', 'POST'])
#@login_required
@templated('search/admin_collection.html')
def create_collection():
form = CollectionForm()
return dict(form=form)
@blueprint.route('/collection/update_translations<id>', methods=['POST'])
@login_required
@permission_required('cfgwebsearch')
#@login_required
def update_translations(id):
"""Update translations if the value is altered or not void."""
collection = Collection.query.filter(Collection.id == id).first_or_404()
for (lang, lang_long) in language_list_long():
collection_name = Collectionname.query.filter(
db.and_(Collectionname.id_collection == id,
Collectionname.ln == lang, Collectionname.type == 'ln')).first()
if collection_name:
if collection_name.value != request.form.get(lang):
collection_name.value = request.form.get(lang)
db.session.commit()
else:
if request.form.get(lang) != '':
collection_name = Collectionname(collection, lang,
'ln', request.form.get(lang))
db.session.add(collection_name)
db.session.commit()
flash(_('Collection was updated on n languages:'), "info")
return redirect(url_for('.manage_collection', name=collection.name))
@blueprint.route('/collection/manage_portalboxes_order', methods=['GET', 'POST'])
#@login_required
def manage_portalboxes_order():
id_p = request.args.get('id', 0, type=int)
collection_id = request.args.get('id_collection', 0, type=int)
order = request.args.get('score', 0, type=int)
collection = Collection.query.filter(Collection.id == collection_id).first_or_404()
portalbox = \
CollectionPortalbox.query.filter(db.and_(
CollectionPortalbox.id_portalbox == id_p,
CollectionPortalbox.id_collection == collection_id)).first_or_404()
position = portalbox.position
p_order = portalbox.score
db.session.delete(portalbox)
#p = portalboxes.pop(portalbox)
collection.portal_boxes_ln.set(CollectionPortalbox(collection_id,
id_p,
g.ln, position,
p_order), order)
db.session.commit()
return ''
@blueprint.route('/collection/edit_portalbox', methods=['GET', 'POST'])
@login_required
@permission_required('cfgwebsearch')
def edit_portalbox():
portalbox = Portalbox.query.get(request.args.get_or_404('id', 0, type=int))
return dict(portalbox=portalbox)

Event Timeline

c4science · Help