nopaque/app/main/views.py
2019-08-06 17:04:38 +02:00

69 lines
2.1 KiB
Python

from datetime import datetime
from flask import current_app, flash, redirect, render_template, url_for
from flask_login import current_user, login_required
from ..models import User, Corpus
from ..tables import AdminUserTable, AdminUserItem
from . import main
from .forms import CreateCorpusForm
from ..decorators import admin_required
from .. import db
import os
@main.route('/')
def index():
return render_template('main/index.html.j2', title='Opaque')
@main.route('/corpora/<corpus>')
def corpora(corpus):
return render_template('main/corpus.html.j2', title=corpus)
@main.route('/dashboard', methods=['GET', 'POST'])
@login_required
def dashboard():
create_corpus_form = CreateCorpusForm()
print(current_user.corpora.all())
print(current_user.jobs.all())
if create_corpus_form.validate_on_submit():
app = current_app._get_current_object()
corpus = Corpus(creator=current_user._get_current_object(),
description=create_corpus_form.description.data,
title=create_corpus_form.title.data)
db.session.add(corpus)
db.session.commit()
dir = os.path.join(app.config['OPAQUE_STORAGE'],
str(corpus.user_id),
'corpora',
str(corpus.id))
try:
os.makedirs(dir)
except OSError:
flash('OSError!')
else:
for file in create_corpus_form.files.data:
file.save(os.path.join(dir, file.filename))
flash('Corpus created!')
return redirect(url_for('main.dashboard'))
return render_template(
'main/dashboard.html.j2',
title='Dashboard',
create_corpus_form=create_corpus_form
)
@main.route('/admin', methods=['GET', 'POST'])
@login_required
@admin_required
def for_admins_only():
users = User.query.order_by(User.username).all()
items = [AdminUserItem(u.username, u.email, u.role_id, u.confirmed) for u in users]
table = AdminUserTable(items)
return render_template('main/admin.html.j2', title='Administration tools',
table=table.__html__())