pagure-new/pagure/ui/admin.py

96 lines
2.6 KiB
Python

# -*- coding: utf-8 -*-
"""
(c) 2014-2016 - Copyright Red Hat Inc
Authors:
Pierre-Yves Chibon <pingou@pingoured.fr>
"""
# pylint: disable=no-member
from functools import wraps
import flask
import pagure.exceptions
import pagure.forms
import pagure.lib
import pagure.lib.git
from pagure import (APP, SESSION, generate_user_key_files,
is_admin, admin_session_timedout)
def admin_required(function):
""" Flask decorator to retrict access to admins of pagure.
"""
@wraps(function)
def decorated_function(*args, **kwargs):
""" Decorated function, actually does the work. """
if admin_session_timedout():
return flask.redirect(
flask.url_for('auth_login', next=flask.request.url))
elif not is_admin():
flask.flash('Access restricted', 'error')
return flask.redirect(flask.url_for('.index'))
return function(*args, **kwargs)
return decorated_function
# Application
@APP.route('/admin/')
@APP.route('/admin')
@admin_required
def admin_index():
""" Front page of the admin section of the application.
"""
form = pagure.forms.ConfirmationForm()
return flask.render_template(
'admin_index.html', form=form,
)
@APP.route('/admin/gitolite', methods=['POST'])
@admin_required
def admin_generate_acl():
""" Regenerate the gitolite ACL file. """
form = pagure.forms.ConfirmationForm()
if form.validate_on_submit():
try:
pagure.lib.git.generate_gitolite_acls()
flask.flash('Gitolite ACLs updated')
except pagure.exceptions.PagureException as err:
flask.flash(str(err), 'error')
return flask.redirect(flask.url_for('admin_index'))
@APP.route('/admin/ssh', methods=['POST'])
@admin_required
def admin_refresh_ssh():
""" Regenerate the user key files. """
form = pagure.forms.ConfirmationForm()
if form.validate_on_submit():
try:
generate_user_key_files()
pagure.lib.git.generate_gitolite_acls()
flask.flash('User key files regenerated')
except pagure.exceptions.PagureException as err:
flask.flash(str(err), 'error')
return flask.redirect(flask.url_for('admin_index'))
@APP.route('/admin/hook_token', methods=['POST'])
@admin_required
def admin_generate_hook_token():
""" Regenerate the hook_token for each projects in the DB. """
form = pagure.forms.ConfirmationForm()
if form.validate_on_submit():
pagure.lib.generate_hook_token(SESSION)
flask.flash('Hook token all re-generated')
return flask.redirect(flask.url_for('admin_index'))