server.py 57.5 KB
Newer Older
1
2
3
4
#!/usr/bin/env python3
import locale
locale.setlocale(locale.LC_TIME, "de_DE.utf8")

Robin Sonnabend's avatar
Robin Sonnabend committed
5
from flask import Flask, g, current_app, request, session, flash, redirect, url_for, abort, render_template, Response, send_file as flask_send_file, Markup
6
from werkzeug.utils import secure_filename
7
8
9
10
from flask_script import Manager, prompt
from flask_migrate import Migrate, MigrateCommand
#from flask_socketio import SocketIO
from celery import Celery
Robin Sonnabend's avatar
Robin Sonnabend committed
11
from sqlalchemy import or_, and_
12
13
14
from apscheduler.schedulers.background import BackgroundScheduler
from apscheduler.triggers.cron import CronTrigger
from apscheduler.triggers.interval import IntervalTrigger
Robin Sonnabend's avatar
Robin Sonnabend committed
15
import atexit
Robin Sonnabend's avatar
Robin Sonnabend committed
16
from io import StringIO, BytesIO
17
import os
Robin Sonnabend's avatar
Robin Sonnabend committed
18
from datetime import datetime
Robin Sonnabend's avatar
Robin Sonnabend committed
19
import math
20
import mimetypes
21
22

import config
23
from shared import db, date_filter, datetime_filter, date_filter_long, date_filter_short, time_filter, time_filter_short, ldap_manager, security_manager, current_user, check_login, login_required, group_required, class_filter, needs_date_test, todostate_name_filter, code_filter, indent_tab_filter
Robin Sonnabend's avatar
Robin Sonnabend committed
24
from utils import is_past, mail_manager, url_manager, get_first_unused_int, set_etherpad_text, get_etherpad_text, split_terms, optional_int_arg, fancy_join
Robin Sonnabend's avatar
Robin Sonnabend committed
25
from decorators import db_lookup, require_public_view_right, require_private_view_right, require_modify_right, require_admin_right
Robin Sonnabend's avatar
Robin Sonnabend committed
26
from models.database import ProtocolType, Protocol, DefaultTOP, TOP, LocalTOP, Document, Todo, Decision, MeetingReminder, Error, TodoMail, DecisionDocument, TodoState, Meta, DefaultMeta, DecisionCategory, Like
27
from views.forms import LoginForm, ProtocolTypeForm, DefaultTopForm, MeetingReminderForm, NewProtocolForm, DocumentUploadForm, KnownProtocolSourceUploadForm, NewProtocolSourceUploadForm, ProtocolForm, TopForm, LocalTopForm, SearchForm, DecisionSearchForm, ProtocolSearchForm, TodoSearchForm, NewProtocolFileUploadForm, NewTodoForm, TodoForm, TodoMailForm, DefaultMetaForm, MetaForm, MergeTodosForm, DecisionCategoryForm
28
from views.tables import ProtocolsTable, ProtocolTypesTable, ProtocolTypeTable, DefaultTOPsTable, MeetingRemindersTable, ErrorsTable, TodosTable, DocumentsTable, DecisionsTable, TodoTable, ErrorTable, TodoMailsTable, DefaultMetasTable, DecisionCategoriesTable
29
from legacy import import_old_todos, import_old_protocols, import_old_todomails
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

app = Flask(__name__)
app.config.from_object(config)
db.init_app(app)
migrate = Migrate(app, db)
manager = Manager(app)
manager.add_command("db", MigrateCommand)

def make_celery(app, config):
    celery = Celery(app.import_name, broker=config.CELERY_BROKER_URL)
    celery.conf.update(app.config)
    return celery
celery = make_celery(app, config)

#def make_socketio(app, config):
#    socketio = SocketIO(app)
#    return socketio
#socketio = make_socketio(app, config)

app.jinja_env.trim_blocks = True
app.jinja_env.lstrip_blocks = True
app.jinja_env.filters["datify"] = date_filter
app.jinja_env.filters["datetimify"] = datetime_filter
Robin Sonnabend's avatar
Robin Sonnabend committed
53
app.jinja_env.filters["timify"] = time_filter
54
app.jinja_env.filters["timify_short"] = time_filter_short
Robin Sonnabend's avatar
Robin Sonnabend committed
55
app.jinja_env.filters["datify_short"] = date_filter_short
Robin Sonnabend's avatar
Robin Sonnabend committed
56
app.jinja_env.filters["datify_long"] = date_filter_long
57
app.jinja_env.filters["url_complete"] = url_manager.complete
58
app.jinja_env.filters["class"] = class_filter
Robin Sonnabend's avatar
Robin Sonnabend committed
59
60
app.jinja_env.filters["todo_get_name"] = todostate_name_filter
app.jinja_env.filters["code"] = code_filter
Robin Sonnabend's avatar
Robin Sonnabend committed
61
app.jinja_env.filters["indent_tab"] = indent_tab_filter
Robin Sonnabend's avatar
Robin Sonnabend committed
62
app.jinja_env.filters["fancy_join"] = fancy_join
63
app.jinja_env.tests["auth_valid"] = security_manager.check_user
Robin Sonnabend's avatar
Robin Sonnabend committed
64
app.jinja_env.tests["needs_date"] = needs_date_test
65
66
67

import tasks

68
69
app.jinja_env.globals.update(check_login=check_login)
app.jinja_env.globals.update(current_user=current_user)
Robin Sonnabend's avatar
Robin Sonnabend committed
70
app.jinja_env.globals.update(zip=zip)
Robin Sonnabend's avatar
Robin Sonnabend committed
71
72
73
app.jinja_env.globals.update(min=min)
app.jinja_env.globals.update(max=max)
app.jinja_env.globals.update(dir=dir)
Robin Sonnabend's avatar
Robin Sonnabend committed
74
app.jinja_env.globals.update(now=datetime.now)
75

76
77
# blueprints here

78
79
@manager.command
def import_legacy():
80
    """Import the old todos and protocols from an sql dump"""
81
    filename = prompt("SQL-file")
82
    #filename = "legacy.sql"
83
84
    with open(filename, "rb") as sqlfile:
        content = sqlfile.read().decode("utf-8")
85
86
        import_old_todos(content)
        import_old_protocols(content)
87
        import_old_todomails(content)
88

89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
@manager.command
def recompile_all():
    for protocol in sorted(Protocol.query.all(), key=lambda p: p.date):
        if protocol.is_done():
            print(protocol.get_identifier())
            tasks.parse_protocol(protocol)

@manager.command
def merge_todos():
    todo_by_id = {}
    todos = Todo.query.all()
    for todo in todos:
        todo_id = todo.get_id()
        if todo_id in todo_by_id:
            todo1, todo2 = todo, todo_by_id[todo_id]
            print(todo1)
            print(todo2)
            if todo2.id > todo1.id:
                todo2, todo1 = todo1, todo2
            for protocol in todo2.protocols:
                if protocol not in todo1.protocols:
                    todo1.protocols.append(protocol)
                todo2.protocols.remove(protocol)
            db.session.delete(todo2)
            db.session.commit()
            todo_by_id[todo_id] = todo1
        else:
            todo_by_id[todo_id] = todo

@manager.command
def runserver():
120
    app.run()
121
122
    make_scheduler()

123
124
125
126
# cause uwsgi currently has a bug
def send_file(file_like, cache_timeout, as_attachment, attachment_filename):
    mimetype, _ = mimetypes.guess_type(attachment_filename)
    response = Response(file_like.read(), mimetype)
127
128
129
130
131
132
133
134
    if as_attachment:
        response.headers["Content-Disposition"] = 'attachment; filename="{}"'.format(attachment_filename)
    content_type = mimetype
    if mimetype.startswith("text/"):
        content_type = "{}; charset=utf-8".format(content_type)
    response.headers["Content-Type"] = content_type
    response.headers["Cache-Control"] = "public, max-age={}".format(cache_timeout)
    response.headers["Connection"] = "close"
135
    return response
136

137
138
@app.route("/")
def index():
Robin Sonnabend's avatar
Robin Sonnabend committed
139
140
141
    user = current_user()
    protocols = [
        protocol for protocol in Protocol.query.all()
142
143
        if protocol.protocoltype.has_public_view_right(user,
            check_networks=False)
Robin Sonnabend's avatar
Robin Sonnabend committed
144
    ]
145
    def _protocol_sort_key(protocol):
Robin Sonnabend's avatar
Robin Sonnabend committed
146
147
148
        if protocol.date is not None:
            return protocol.date
        return datetime.now().date()
Robin Sonnabend's avatar
Robin Sonnabend committed
149
    current_day = datetime.now().date()
150
    open_protocols = sorted(
Robin Sonnabend's avatar
Robin Sonnabend committed
151
152
153
154
155
        [
            protocol for protocol in protocols
            if not protocol.done
            and (protocol.date - current_day).days < config.MAX_INDEX_DAYS
        ],
156
        key=_protocol_sort_key
Robin Sonnabend's avatar
Robin Sonnabend committed
157
158
    )
    finished_protocols = sorted(
159
160
        [
            protocol for protocol in protocols
161
162
163
            if protocol.done and protocol.public
            and (protocol.has_private_view_right(user)
                or protocol.protocoltype.has_public_view_right(user, check_networks=False))
164
        ],
165
166
        key=_protocol_sort_key,
        reverse=True
Robin Sonnabend's avatar
Robin Sonnabend committed
167
168
    )
    protocol = finished_protocols[0] if len(finished_protocols) > 0 else None
169
170
    show_private = protocol.has_private_view_right(user)
    has_public_view_right = protocol.protocoltype.has_public_view_right(user)
Robin Sonnabend's avatar
Robin Sonnabend committed
171
172
173
    todos = None
    if check_login():
        todos = [
174
            todo for todo in Todo.query.all()
175
            if todo.protocoltype.has_private_view_right(user)
176
            and not todo.is_done()
Robin Sonnabend's avatar
Robin Sonnabend committed
177
        ]
178
179
        def _todo_sort_key(todo):
            protocol = todo.get_first_protocol()
180
            return protocol.date if protocol is not None and protocol.date is not None else datetime.now().date()
181
        todos = sorted(todos, key=_todo_sort_key, reverse=True)
Robin Sonnabend's avatar
Robin Sonnabend committed
182
    todos_table = TodosTable(todos) if todos is not None else None
183
    return render_template("index.html", open_protocols=open_protocols, protocol=protocol, todos=todos, show_private=show_private, has_public_view_right=has_public_view_right)
184

Robin Sonnabend's avatar
Robin Sonnabend committed
185
@app.route("/documentation")
Robin Sonnabend's avatar
Robin Sonnabend committed
186
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
187
def documentation():
Robin Sonnabend's avatar
Robin Sonnabend committed
188
189
190
    todostates = list(TodoState)
    name_to_state = TodoState.get_name_to_state()
    return render_template("documentation.html", todostates=todostates, name_to_state=name_to_state)
Robin Sonnabend's avatar
Robin Sonnabend committed
191

Robin Sonnabend's avatar
Robin Sonnabend committed
192
@app.route("/types/list")
Robin Sonnabend's avatar
Robin Sonnabend committed
193
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
194
195
196
197
198
199
200
201
def list_types():
    is_logged_in = check_login()
    user = current_user()
    types = [
        protocoltype for protocoltype in ProtocolType.query.all()
        if (protocoltype.public_group in user.groups
        or protocoltype.private_group in user.groups
        or protocoltype.is_public)]
202
    types = sorted(types, key=lambda t: t.short_name)
Robin Sonnabend's avatar
Robin Sonnabend committed
203
204
205
206
207
208
209
210
211
212
213
214
    types_table = ProtocolTypesTable(types)
    return render_template("types-list.html", types=types, types_table=types_table)

@app.route("/type/new", methods=["GET", "POST"])
@login_required
def new_type():
    form = ProtocolTypeForm()
    if form.validate_on_submit():
        user = current_user()
        if form.private_group.data not in user.groups:
            flash("Du kannst keinen internen Protokolltypen anlegen, zu dem du selbst keinen Zugang hast.", "alert-error")
        else:
215
216
            protocoltype = ProtocolType()
            form.populate_obj(protocoltype)
Robin Sonnabend's avatar
Robin Sonnabend committed
217
218
219
220
221
222
            db.session.add(protocoltype)
            db.session.commit()
            flash("Der Protokolltyp {} wurde angelegt.".format(protocoltype.name), "alert-success")
        return redirect(request.args.get("next") or url_for("list_types"))
    return render_template("type-new.html", form=form)

Robin Sonnabend's avatar
Robin Sonnabend committed
223
@app.route("/type/edit/<int:protocoltype_id>", methods=["GET", "POST"])
Robin Sonnabend's avatar
Robin Sonnabend committed
224
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
225
@db_lookup(ProtocolType)
Robin Sonnabend's avatar
Robin Sonnabend committed
226
@require_private_view_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
227
def edit_type(protocoltype):
Robin Sonnabend's avatar
Robin Sonnabend committed
228
229
230
231
232
233
234
235
    user = current_user()
    form = ProtocolTypeForm(obj=protocoltype)
    if form.validate_on_submit():
        if form.private_group.data not in user.groups:
            flash("Du kannst keinen internen Protokolltypen anlegen, zu dem du selbst keinen Zugang hast.", "alert-error")
        else:
            form.populate_obj(protocoltype)
            db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
236
            return redirect(request.args.get("next") or url_for("show_type", protocoltype_id=protocoltype.id))
Robin Sonnabend's avatar
Robin Sonnabend committed
237
238
    return render_template("type-edit.html", form=form, protocoltype=protocoltype)

Robin Sonnabend's avatar
Robin Sonnabend committed
239
@app.route("/type/show/<int:protocoltype_id>")
Robin Sonnabend's avatar
Robin Sonnabend committed
240
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
241
@db_lookup(ProtocolType)
Robin Sonnabend's avatar
Robin Sonnabend committed
242
@require_private_view_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
243
def show_type(protocoltype):
Robin Sonnabend's avatar
Robin Sonnabend committed
244
245
    protocoltype_table = ProtocolTypeTable(protocoltype)
    default_tops_table = DefaultTOPsTable(protocoltype.default_tops, protocoltype)
246
    reminders_table = MeetingRemindersTable(protocoltype.reminders, protocoltype)
Robin Sonnabend's avatar
Robin Sonnabend committed
247
    metas_table = DefaultMetasTable(protocoltype.metas, protocoltype)
248
249
    categories_table = DecisionCategoriesTable(protocoltype.decisioncategories, protocoltype)
    return render_template("type-show.html", protocoltype=protocoltype, protocoltype_table=protocoltype_table, default_tops_table=default_tops_table, metas_table=metas_table, reminders_table=reminders_table, mail_active=config.MAIL_ACTIVE, categories_table=categories_table)
250

Robin Sonnabend's avatar
Robin Sonnabend committed
251
@app.route("/type/delete/<int:protocoltype_id>")
252
@login_required
253
@group_required(config.ADMIN_GROUP)
Robin Sonnabend's avatar
Robin Sonnabend committed
254
@db_lookup(ProtocolType)
Robin Sonnabend's avatar
Robin Sonnabend committed
255
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
256
def delete_type(protocoltype):
257
258
259
260
261
262
    name = protocoltype.name
    db.session.delete(protocoltype) 
    db.session.commit()
    flash("Der Protokolltype {} wurde gelöscht.".format(name), "alert-success")
    return redirect(request.args.get("next") or url_for("list_types"))

Robin Sonnabend's avatar
Robin Sonnabend committed
263
@app.route("/type/reminders/new/<int:protocoltype_id>", methods=["GET", "POST"])
264
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
265
@db_lookup(ProtocolType)
Robin Sonnabend's avatar
Robin Sonnabend committed
266
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
267
def new_reminder(protocoltype):
268
269
    form = MeetingReminderForm()
    if form.validate_on_submit():
270
271
        meetingreminder = MeetingReminder(protocoltype_id=protocoltype.id)
        form.populate_obj(meetingreminder)
Robin Sonnabend's avatar
Robin Sonnabend committed
272
        db.session.add(meetingreminder)
273
        db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
274
        return redirect(request.args.get("next") or url_for("show_type", protocoltype_id=protocoltype.id))
275
276
    return render_template("reminder-new.html", form=form, protocoltype=protocoltype)

Robin Sonnabend's avatar
Robin Sonnabend committed
277
@app.route("/type/reminder/edit/<int:meetingreminder_id>", methods=["GET", "POST"])
278
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
279
@db_lookup(MeetingReminder)
Robin Sonnabend's avatar
Robin Sonnabend committed
280
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
281
282
def edit_reminder(meetingreminder):
    form = MeetingReminderForm(obj=meetingreminder)
283
    if form.validate_on_submit():
Robin Sonnabend's avatar
Robin Sonnabend committed
284
        form.populate_obj(meetingreminder)
285
        db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
286
        return redirect(request.args.get("next") or url_for("show_type", protocoltype_id=protocoltype.id))
287
    return render_template("reminder-edit.html", form=form, meetingreminder=meetingreminder)
288

Robin Sonnabend's avatar
Robin Sonnabend committed
289
@app.route("/type/reminder/delete/<int:meetingreminder_id>")
290
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
291
@db_lookup(MeetingReminder)
Robin Sonnabend's avatar
Robin Sonnabend committed
292
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
293
294
295
def delete_reminder(meetingreminder):
    protocoltype = meetingreminder.protocoltype
    db.session.delete(meetingreminder)
296
    db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
297
    return redirect(request.args.get("next") or url_for("show_type", protocoltype_id=protocoltype.id))
298

Robin Sonnabend's avatar
Robin Sonnabend committed
299
@app.route("/type/tops/new/<int:protocoltype_id>", methods=["GET", "POST"])
Robin Sonnabend's avatar
Robin Sonnabend committed
300
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
301
@db_lookup(ProtocolType)
Robin Sonnabend's avatar
Robin Sonnabend committed
302
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
303
def new_default_top(protocoltype):
Robin Sonnabend's avatar
Robin Sonnabend committed
304
305
    form = DefaultTopForm()
    if form.validate_on_submit():
306
307
        defaulttop = DefaultTOP(protocoltype_id=protocoltype.id)
        form.populate_obj(defaulttop)
Robin Sonnabend's avatar
Robin Sonnabend committed
308
        db.session.add(defaulttop)
Robin Sonnabend's avatar
Robin Sonnabend committed
309
        db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
310
        flash("Der Standard-TOP {} wurde für dem Protokolltyp {} hinzugefügt.".format(defaulttop.name, protocoltype.name), "alert-success")
Robin Sonnabend's avatar
Robin Sonnabend committed
311
312
313
        return redirect(request.args.get("next") or url_for("index"))
    return render_template("default-top-new.html", form=form, protocoltype=protocoltype)

Robin Sonnabend's avatar
Robin Sonnabend committed
314
@app.route("/type/tops/edit/<int:protocoltype_id>/<int:defaulttop_id>", methods=["GET", "POST"])
Robin Sonnabend's avatar
Robin Sonnabend committed
315
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
316
@db_lookup(ProtocolType, DefaultTOP)
Robin Sonnabend's avatar
Robin Sonnabend committed
317
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
318
319
def edit_default_top(protocoltype, defaulttop):
    form = DefaultTopForm(obj=defaulttop)
Robin Sonnabend's avatar
Robin Sonnabend committed
320
    if form.validate_on_submit():
Robin Sonnabend's avatar
Robin Sonnabend committed
321
        form.populate_obj(defaulttop)
Robin Sonnabend's avatar
Robin Sonnabend committed
322
        db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
323
324
        return redirect(request.args.get("next") or url_for("show_type", protocoltype_id=protocoltype.id))
    return render_template("default-top-edit.html", form=form, protocoltype=protocoltype, defaulttop=defaulttop)
Robin Sonnabend's avatar
Robin Sonnabend committed
325

Robin Sonnabend's avatar
Robin Sonnabend committed
326
@app.route("/type/tops/delete/<int:defaulttop_id>")
Robin Sonnabend's avatar
Robin Sonnabend committed
327
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
328
@db_lookup(DefaultTOP)
Robin Sonnabend's avatar
Robin Sonnabend committed
329
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
330
331
def delete_default_top(defaulttop):
    db.session.delete(defaulttop)
Robin Sonnabend's avatar
Robin Sonnabend committed
332
    db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
333
    return redirect(request.args.get("next") or url_for("show_type", protocoltype_id=protocoltype.id))
Robin Sonnabend's avatar
Robin Sonnabend committed
334

Robin Sonnabend's avatar
Robin Sonnabend committed
335
@app.route("/type/tops/move/<int:defaulttop_id>/<diff>/")
Robin Sonnabend's avatar
Robin Sonnabend committed
336
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
337
@db_lookup(DefaultTOP)
Robin Sonnabend's avatar
Robin Sonnabend committed
338
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
339
def move_default_top(defaulttop, diff):
Robin Sonnabend's avatar
Robin Sonnabend committed
340
    try:
Robin Sonnabend's avatar
Robin Sonnabend committed
341
        defaulttop.number += int(diff)
Robin Sonnabend's avatar
Robin Sonnabend committed
342
343
344
        db.session.commit()
    except ValueError:
        flash("Die angegebene Differenz ist keine Zahl.", "alert-error")
Robin Sonnabend's avatar
Robin Sonnabend committed
345
    return redirect(request.args.get("next") or url_for("show_type", protocoltype_id=defaulttop.protocoltype.id))
Robin Sonnabend's avatar
Robin Sonnabend committed
346
347

@app.route("/protocols/list")
348
349
350
def list_protocols():
    is_logged_in = check_login()
    user = current_user()
Robin Sonnabend's avatar
Robin Sonnabend committed
351
352
353
    protocoltype = None
    protocoltype_id = None
    try:
354
        protocoltype_id = int(request.args.get("protocoltype_id"))
Robin Sonnabend's avatar
Robin Sonnabend committed
355
356
    except (ValueError, TypeError):
        pass
Robin Sonnabend's avatar
Robin Sonnabend committed
357
    state_open = -1
Robin Sonnabend's avatar
Robin Sonnabend committed
358
    try:
Robin Sonnabend's avatar
Robin Sonnabend committed
359
        state_open = int(request.args.get("state_open"))
Robin Sonnabend's avatar
Robin Sonnabend committed
360
361
    except (ValueError, TypeError):
        pass
Robin Sonnabend's avatar
Robin Sonnabend committed
362
    search_term = request.args.get("search")
363
    protocoltypes = ProtocolType.get_public_protocoltypes(user, check_networks=False)
Robin Sonnabend's avatar
Robin Sonnabend committed
364
    search_form = ProtocolSearchForm(protocoltypes)
Robin Sonnabend's avatar
Robin Sonnabend committed
365
    if protocoltype_id is not None:
366
        search_form.protocoltype_id.data = protocoltype_id
Robin Sonnabend's avatar
Robin Sonnabend committed
367
        protocoltype = ProtocolType.query.filter_by(id=protocoltype_id).first()
Robin Sonnabend's avatar
Robin Sonnabend committed
368
369
    if state_open is not None:
        search_form.state_open.data = state_open
Robin Sonnabend's avatar
Robin Sonnabend committed
370
371
372
    if search_term is not None:
        search_form.search.data = search_term
    protocol_query = Protocol.query
Robin Sonnabend's avatar
Robin Sonnabend committed
373
374
375
376
377
378
379
380
381
    shall_search = search_term is not None and len(search_term.strip()) > 0
    search_terms = []
    if shall_search:
        search_terms = list(map(str.lower, split_terms(search_term)))
        for term in search_terms:
            protocol_query = protocol_query.filter(or_(
                Protocol.content_public.ilike("%{}%".format(term)),
                Protocol.content_private.ilike("%{}%".format(term))
            ))
382
    protocols = [
Robin Sonnabend's avatar
Robin Sonnabend committed
383
        protocol for protocol in protocol_query.all()
384
        if protocol.protocoltype.has_public_view_right(user, check_networks=False)
385
    ]
Robin Sonnabend's avatar
Robin Sonnabend committed
386
387
388
389
390
391
392
393
394
395
396
397
398
    def _matches_search(content):
        content = content.lower()
        for search_term in search_terms:
            if search_term.lower() not in content:
                return False
        return True
    def _matches_search_lazy(content):
        content = content.lower()
        for search_term in search_terms:
            if search_term.lower() in content:
                return True
        return False
    search_results = {} if shall_search else None
Robin Sonnabend's avatar
Robin Sonnabend committed
399
400
401
402
403
    if protocoltype_id is not None and protocoltype_id != -1:
        protocols = [
            protocol for protocol in protocols
            if protocol.protocoltype.id == protocoltype_id
        ]
Robin Sonnabend's avatar
Robin Sonnabend committed
404
405
    if state_open is not None and state_open != -1:
        protocol_done = bool(state_open)
Robin Sonnabend's avatar
Robin Sonnabend committed
406
407
        protocols = [
            protocol for protocol in protocols
408
            if (protocol.is_done() or False) == protocol_done
Robin Sonnabend's avatar
Robin Sonnabend committed
409
        ]
Robin Sonnabend's avatar
Robin Sonnabend committed
410
411
412
413
414
    if shall_search:
        protocols = [
            protocol for protocol in protocols
            if (protocol.protocoltype.has_private_view_right(user)
                and _matches_search(protocol.content_private))
415
            or (protocol.has_public_view_right(user)
Robin Sonnabend's avatar
Robin Sonnabend committed
416
417
418
419
420
421
422
423
424
425
426
427
428
429
430
431
432
433
434
435
436
437
438
439
440
441
442
                and _matches_search(protocol.content_public))
        ]
        for protocol in protocols:
            content = protocol.content_private if protocol.protocoltype.has_private_view_right(user) else protocol.content_public
            lines = content.splitlines()
            matches = [line for line in lines if _matches_search_lazy(line)]
            formatted_lines = []
            for line in matches:
                parts = []
                lower_line = line.lower()
                last_index = 0
                while last_index < len(line):
                    index_candidates = list(filter(lambda t: t[0] != -1, 
                        [(lower_line.find(term, last_index), term) for term in search_terms]))
                    if len(index_candidates) == 0:
                        parts.append((line[last_index:], False))
                        break
                    else:
                        new_index, term = min(index_candidates, key=lambda t: t[0])
                        new_end_index = new_index + len(term)
                        parts.append((line[last_index:new_index], False))
                        parts.append((line[new_index:new_end_index], True))
                        last_index = new_end_index
                formatted_lines.append("".join([
                    "<b>{}</b>".format(text) if matched else text
                    for text, matched in parts
                ]))
443
            search_results[protocol] = " …<br />\n".join(formatted_lines)
Robin Sonnabend's avatar
Robin Sonnabend committed
444
    protocols = sorted(protocols, key=lambda protocol: protocol.date, reverse=True)
Robin Sonnabend's avatar
Robin Sonnabend committed
445
    page = _get_page()
Robin Sonnabend's avatar
Robin Sonnabend committed
446
447
    page_length = _get_page_length()
    page_count = int(math.ceil(len(protocols) / page_length))
Robin Sonnabend's avatar
Robin Sonnabend committed
448
449
    if page >= page_count:
        page = 0
Robin Sonnabend's avatar
Robin Sonnabend committed
450
451
    begin_index = page * page_length
    end_index = (page + 1) * page_length
452
    max_page_length_exp = math.ceil(math.log10(len(protocols))) if len(protocols) > 0 else 1
Robin Sonnabend's avatar
Robin Sonnabend committed
453
    protocols = protocols[begin_index:end_index]
Robin Sonnabend's avatar
Robin Sonnabend committed
454
    protocols_table = ProtocolsTable(protocols, search_results=search_results)
Robin Sonnabend's avatar
Robin Sonnabend committed
455
    return render_template("protocols-list.html", protocols=protocols, protocols_table=protocols_table, search_form=search_form, page=page, page_count=page_count, page_diff=config.PAGE_DIFF, protocoltype_id=protocoltype_id, search_term=search_term, state_open=state_open, page_length=page_length, max_page_length_exp=max_page_length_exp)
Robin Sonnabend's avatar
Robin Sonnabend committed
456
457
458
459
460

@app.route("/protocol/new", methods=["GET", "POST"])
@login_required
def new_protocol():
    user = current_user()
Robin Sonnabend's avatar
Robin Sonnabend committed
461
    protocoltypes = ProtocolType.get_modifiable_protocoltypes(user)
Robin Sonnabend's avatar
Robin Sonnabend committed
462
    form = NewProtocolForm(protocoltypes)
Robin Sonnabend's avatar
Robin Sonnabend committed
463
    upload_form = NewProtocolSourceUploadForm(protocoltypes)
464
    file_upload_form = NewProtocolFileUploadForm(protocoltypes)
Robin Sonnabend's avatar
Robin Sonnabend committed
465
    if form.validate_on_submit():
466
        protocoltype = ProtocolType.query.filter_by(id=form.protocoltype_id.data).first()
Robin Sonnabend's avatar
Robin Sonnabend committed
467
468
469
        if protocoltype is None or not protocoltype.has_modify_right(user):
            flash("Dir fehlen die nötigen Zugriffsrechte.", "alert-error")
            return redirect(request.args.get("next") or url_for("index"))
470
471
        protocol = Protocol(protocoltype_id=protocoltype.id)
        form.populate_obj(protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
472
473
        db.session.add(protocol)
        db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
474
        tasks.push_tops_to_calendar(protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
475
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))
476
    type_id = request.args.get("protocoltype_id")
Robin Sonnabend's avatar
Robin Sonnabend committed
477
478
    if type_id is not None:
        form.protocoltype.data = type_id
479
480
        upload_form.protocoltype.data = type_id
    return render_template("protocol-new.html", form=form, upload_form=upload_form, file_upload_form=file_upload_form, protocoltypes=protocoltypes)
Robin Sonnabend's avatar
Robin Sonnabend committed
481
482

@app.route("/protocol/show/<int:protocol_id>")
Robin Sonnabend's avatar
Robin Sonnabend committed
483
484
@db_lookup(Protocol)
def show_protocol(protocol):
Robin Sonnabend's avatar
Robin Sonnabend committed
485
486
    user = current_user()
    errors_table = ErrorsTable(protocol.errors)
487
    if not protocol.protocoltype.has_public_view_right(user, check_networks=False): # yes, feature
Robin Sonnabend's avatar
Robin Sonnabend committed
488
        flash("Die fehlen die nötigen Zugriffsrechte.", "alert-error")
489
        return redirect(request.args.get("next") or url_for("login", next=request.url))
Robin Sonnabend's avatar
Robin Sonnabend committed
490
491
    visible_documents = [
        document for document in protocol.documents
492
        if (not document.is_private and document.protocol.has_public_view_right(user))
Robin Sonnabend's avatar
Robin Sonnabend committed
493
494
        or (document.is_private and document.protocol.protocoltype.has_private_view_right(user))
    ]
495
    documents_table = DocumentsTable(visible_documents, protocol)
496
    document_upload_form = DocumentUploadForm()
Robin Sonnabend's avatar
Robin Sonnabend committed
497
    source_upload_form = KnownProtocolSourceUploadForm()
498
499
500
    time_diff = protocol.date - datetime.now().date()
    large_time_diff = not protocol.is_done() and time_diff.days > 0
    return render_template("protocol-show.html", protocol=protocol, errors_table=errors_table, documents_table=documents_table, document_upload_form=document_upload_form, source_upload_form=source_upload_form, time_diff=time_diff, large_time_diff=large_time_diff)
Robin Sonnabend's avatar
Robin Sonnabend committed
501

Robin Sonnabend's avatar
Robin Sonnabend committed
502
503
@app.route("/protocol/delete/<int:protocol_id>")
@login_required
504
@group_required(config.ADMIN_GROUP)
Robin Sonnabend's avatar
Robin Sonnabend committed
505
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
506
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
507
def delete_protocol(protocol):
Robin Sonnabend's avatar
Robin Sonnabend committed
508
    name = protocol.get_identifier()
Robin Sonnabend's avatar
Robin Sonnabend committed
509
    protocol.delete_orphan_todos()
Robin Sonnabend's avatar
Robin Sonnabend committed
510
511
512
513
514
    db.session.delete(protocol)
    db.session.commit()
    flash("Protokoll {} ist gelöscht.".format(name), "alert-success")
    return redirect(request.args.get("next") or url_for("list_protocols"))

Robin Sonnabend's avatar
Robin Sonnabend committed
515
@app.route("/protocol/etherpull/<int:protocol_id>")
Robin Sonnabend's avatar
Robin Sonnabend committed
516
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
517
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
518
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
519
def etherpull_protocol(protocol):
520
521
522
    if not config.ETHERPAD_ACTIVE:
        flash("Die Etherpadfunktion ist nicht aktiviert.", "alert-error")
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol_id))
Robin Sonnabend's avatar
Robin Sonnabend committed
523
    protocol.source = get_etherpad_text(protocol.get_identifier())
Robin Sonnabend's avatar
Robin Sonnabend committed
524
525
526
527
528
    db.session.commit()
    tasks.parse_protocol(protocol)
    flash("Das Protokoll wird kompiliert.", "alert-success")
    return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))

Robin Sonnabend's avatar
Robin Sonnabend committed
529
530
@app.route("/protocol/upload/known/<int:protocol_id>", methods=["POST"])
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
531
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
532
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
533
def upload_source_to_known_protocol(protocol):
534
    form = KnownProtocolSourceUploadForm()
Robin Sonnabend's avatar
Robin Sonnabend committed
535
536
537
538
539
540
541
542
543
544
545
546
547
548
549
550
551
552
553
554
    if form.validate_on_submit():
        if form.source.data is None:
            flash("Es wurde keine Datei ausgewählt.", "alert-error")
        else:
            file = form.source.data
            if file.filename == "":
                flash("Es wurde keine Datei ausgewählt.", "alert-error")
            else:
                # todo: Prüfen, ob es Text ist?
                source = file.stream.read().decode("utf-8")
                protocol.source = source
                db.session.commit()
                tasks.parse_protocol(protocol)
                flash("Das Protokoll wird kompiliert.", "alert-success")
    return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))

@app.route("/protocol/upload/new/", methods=["POST"])
@login_required
def upload_new_protocol():
    user = current_user()
555
    available_types = ProtocolType.get_modifiable_protocoltypes(user)
Robin Sonnabend's avatar
Robin Sonnabend committed
556
557
558
559
    form = NewProtocolSourceUploadForm(protocoltypes=available_types)
    if form.validate_on_submit():
        if form.source.data is None:
            flash("Es wurde keine Datei ausgewählt.", "alert-error")
560
561
562
563
564
565
            return redirect(request.args.get("fail") or url_for("new_protocol"))
        file = form.source.data
        if file.filename == "":
            flash("Es wurde keine Datei ausgewählt.", "alert-error")
            return redirect(request.args.get("fail") or url_for("new_protocol"))
        source = file.stream.read().decode("utf-8")
566
        protocoltype = ProtocolType.query.filter_by(id=form.protocoltype_id.data).first()
567
568
569
        if protocoltype is None or not protocoltype.has_modify_right(user):
            flash("Invalider Protokolltyp oder keine Rechte.", "alert-error")
            return redirect(request.args.get("fail") or url_for("new_protocol"))
570
        protocol = Protocol(protocoltype_id=protocoltype.id, source=source)
571
572
573
574
575
576
577
578
579
580
581
582
583
584
585
586
587
588
589
590
591
        db.session.add(protocol)
        db.session.commit()
        tasks.parse_protocol(protocol)
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))
    return redirect(request.args.get("fail") or url_for("new_protocol"))

@app.route("/protocol/upload/new/file/", methods=["POST"])
@login_required
def upload_new_protocol_by_file():
    user = current_user()
    available_types = ProtocolType.get_modifiable_protocoltypes(user)
    form = NewProtocolFileUploadForm(protocoltypes=available_types)
    if form.validate_on_submit():
        if form.file.data is None:
            flash("Es wurde keine Datei ausgewählt.", "alert-error")
            return redirect(request.args.get("fail") or url_for("new_protocol"))
        file = form.file.data
        if file.filename == "":
            flash("Es wurde keine Datei ausgewählt.", "alert-error")
            return redirect(request.args.get("fail") or url_for("new_protocol"))
        filename = secure_filename(file.filename)
592
        protocoltype = ProtocolType.query.filter_by(id=form.protocoltype_id.data).first()
593
594
595
        if protocoltype is None or not protocoltype.has_modify_right(user):
            flash("Invalider Protokolltyp oder keine Rechte.", "alert-error")
            return redirect(request.args.get("fail") or url_for("new_protocol"))
596
        protocol = Protocol(protocoltype_id=protocoltype.id, date=datetime.now().date(), done=True)
597
598
        db.session.add(protocol)
        db.session.commit()
599
600
601
        document = Document(protocol_id=protocol.id, name=filename,
            filename="", is_compiled=False)
        form.populate_obj(document)
602
603
604
605
606
607
608
        db.session.add(document)
        db.session.commit()
        internal_filename = "{}-{}-{}".format(protocol.id, document.id, filename)
        document.filename = internal_filename
        file.save(os.path.join(config.DOCUMENTS_PATH, internal_filename))
        db.session.commit()
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))
Robin Sonnabend's avatar
Robin Sonnabend committed
609
610
    return redirect(request.args.get("fail") or url_for("new_protocol"))

Administrator's avatar
Administrator committed
611
612
613
@app.route("/protocol/recompile/<int:protocol_id>")
@login_required
@group_required(config.ADMIN_GROUP)
Robin Sonnabend's avatar
Robin Sonnabend committed
614
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
615
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
616
def recompile_protocol(protocol):
Administrator's avatar
Administrator committed
617
618
    tasks.parse_protocol(protocol)
    return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))
619

Robin Sonnabend's avatar
Robin Sonnabend committed
620
621
@app.route("/protocol/source/<int:protocol_id>")
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
622
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
623
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
624
def get_protocol_source(protocol):
Robin Sonnabend's avatar
Robin Sonnabend committed
625
626
627
    file_like = BytesIO(protocol.source.encode("utf-8"))
    return send_file(file_like, cache_timeout=1, as_attachment=True, attachment_filename="{}.txt".format(protocol.get_identifier()))

Robin Sonnabend's avatar
Robin Sonnabend committed
628
629
@app.route("/protocol/template/<int:protocol_id>")
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
630
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
631
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
632
def get_protocol_template(protocol):
Robin Sonnabend's avatar
Robin Sonnabend committed
633
634
635
    file_like = BytesIO(protocol.get_template().encode("utf-8"))
    return send_file(file_like, cache_timeout=1, as_attachment=True, attachment_filename="{}-template.txt".format(protocol.get_identifier()))

Robin Sonnabend's avatar
Robin Sonnabend committed
636
637
@app.route("/protocol/etherpush/<int:protocol_id>")
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
638
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
639
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
640
def etherpush_protocol(protocol):
641
642
643
    if not config.ETHERPAD_ACTIVE:
        flash("Die Etherpadfunktion ist nicht aktiviert.", "alert-error")
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol_id))
644
645
646
    if not protocol.is_done():
        tasks.set_etherpad_content(protocol)
    return redirect(request.args.get("next") or protocol.get_etherpad_link())
Robin Sonnabend's avatar
Robin Sonnabend committed
647

Robin Sonnabend's avatar
Robin Sonnabend committed
648
@app.route("/protocol/update/<int:protocol_id>", methods=["GET", "POST"])
Robin Sonnabend's avatar
Robin Sonnabend committed
649
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
650
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
651
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
652
def update_protocol(protocol):
Robin Sonnabend's avatar
Robin Sonnabend committed
653
    upload_form = KnownProtocolSourceUploadForm()
Robin Sonnabend's avatar
Robin Sonnabend committed
654
    edit_form = ProtocolForm(obj=protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
655
656
657
    if edit_form.validate_on_submit():
        edit_form.populate_obj(protocol)
        db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
658
        tasks.push_tops_to_calendar(protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
659
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))
Robin Sonnabend's avatar
Robin Sonnabend committed
660
    return render_template("protocol-update.html", upload_form=upload_form, edit_form=edit_form, protocol=protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
661

662
663
@app.route("/protocol/publish/<int:protocol_id>")
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
664
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
665
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
666
def publish_protocol(protocol):
667
668
669
670
    protocol.public = True
    db.session.commit()
    return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))

671
672
@app.route("/prococol/send/<int:protocol_id>")
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
673
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
674
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
675
def send_protocol(protocol):
676
677
678
    if not config.MAIL_ACTIVE:
        flash("Die Mailfunktion ist nicht aktiviert.", "alert-error")
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol_id))
679
680
681
682
    tasks.send_protocol(protocol)
    flash("Das Protokoll wurde versandt.", "alert-success")
    return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))

683
684
685
686
687
688
689
690
691
692
693
694
695
696
697
698
699
700
701
702
703
704
705
706
707
708
709
@app.route("/protocol/reminder/<int:protocol_id>")
@login_required
@db_lookup(Protocol)
@require_modify_right()
def send_protocol_reminder(protocol):
    if not config.MAIL_ACTIVE:
        flash("Die Mailfunktion ist nicht aktiviert.", "alert-error")
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol_id))
    meetingreminders = MeetingReminder.query.filter_by(protocoltype_id=protocol.protocoltype.id).all()
    if len(meetingreminders) == 0:
        flash("Für diesen Protokolltyp sind keine Einladungsmails konfiguriert.", "alert-error")
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol_id))
    day_difference = (protocol.date - datetime.now().date()).days
    past_reminders = [
        meetingreminder for meetingreminder in meetingreminders
        if meetingreminder.days_before > day_difference
    ]
    if len(past_reminders) == 0:
        flash("Bisher hätte keine Einladungsmail verschickt werden sollen, schicke letzte.", "alert-info")
        past_reminders = meetingreminders
    past_reminders = sorted(past_reminders, key=lambda r: r.days_before)
    choosen_reminder = past_reminders[0]
    tasks.send_reminder(choosen_reminder, protocol)
    flash("Einladungsmail ist versandt.", "alert-success")
    return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))
        

Robin Sonnabend's avatar
Robin Sonnabend committed
710
711
@app.route("/protocol/tops/new/<int:protocol_id>", methods=["GET", "POST"])
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
712
@db_lookup(Protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
713
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
714
def new_top(protocol):
Robin Sonnabend's avatar
Robin Sonnabend committed
715
716
    form = TopForm()
    if form.validate_on_submit():
717
718
        top = TOP(protocol_id=protocol.id, planned=True)
        form.populate_obj(top)
Robin Sonnabend's avatar
Robin Sonnabend committed
719
720
        db.session.add(top)
        db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
721
        tasks.push_tops_to_calendar(top.protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
722
723
724
725
726
727
728
729
730
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))
    else:
        current_numbers = list(map(lambda t: t.number, protocol.tops))
        suggested_number = get_first_unused_int(current_numbers)
        form.number.data = suggested_number
    return render_template("top-new.html", form=form, protocol=protocol)

@app.route("/protocol/top/edit/<int:top_id>", methods=["GET", "POST"])
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
731
@db_lookup(TOP)
Robin Sonnabend's avatar
Robin Sonnabend committed
732
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
733
def edit_top(top):
Robin Sonnabend's avatar
Robin Sonnabend committed
734
735
736
737
    form = TopForm(obj=top)
    if form.validate_on_submit():
        form.populate_obj(top)
        db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
738
        tasks.push_tops_to_calendar(top.protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
739
740
741
742
743
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=top.protocol.id))
    return render_template("top-edit.html", form=form, top=top)

@app.route("/protocol/top/delete/<int:top_id>")
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
744
@db_lookup(TOP)
Robin Sonnabend's avatar
Robin Sonnabend committed
745
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
746
def delete_top(top):
Robin Sonnabend's avatar
Robin Sonnabend committed
747
    name = top.name
Robin Sonnabend's avatar
Robin Sonnabend committed
748
    protocol = top.protocol
Robin Sonnabend's avatar
Robin Sonnabend committed
749
750
    db.session.delete(top)
    db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
751
    tasks.push_tops_to_calendar(protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
752
    flash("Der TOP {} wurde gelöscht.".format(name), "alert-success")
Robin Sonnabend's avatar
Robin Sonnabend committed
753
    return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))
Robin Sonnabend's avatar
Robin Sonnabend committed
754
755
756

@app.route("/protocol/top/move/<int:top_id>/<diff>")
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
757
@db_lookup(TOP)
Robin Sonnabend's avatar
Robin Sonnabend committed
758
@require_modify_right()
Robin Sonnabend's avatar
Robin Sonnabend committed
759
def move_top(top, diff):
Robin Sonnabend's avatar
Robin Sonnabend committed
760
761
762
    try:
        top.number += int(diff)
        db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
763
        tasks.push_tops_to_calendar(top.protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
764
765
766
767
    except ValueError:
        flash("Die angegebene Differenz ist keine Zahl.", "alert-error")
    return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=top.protocol.id))

768
769
770
771
772
773
774
775
776
777
778
779
@app.route("/protocol/localtop/edit/<int:localtop_id>", methods=["GET", "POST"])
@login_required
@db_lookup(LocalTOP)
@require_modify_right()
def edit_localtop(localtop):
    form = LocalTopForm(obj=localtop)
    if form.validate_on_submit():
        form.populate_obj(localtop)
        db.session.commit()
        return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=localtop.protocol.id))
    return render_template("localtop-edit.html", form=form, localtop=localtop)

Robin Sonnabend's avatar
Robin Sonnabend committed
780
781
782
783
784
785
786
787
788
def _get_page():
    try:
        page = request.args.get("page")
        if page is None:
            return 0
        return int(page)
    except ValueError:
        return 0

Robin Sonnabend's avatar
Robin Sonnabend committed
789
790
791
792
793
794
795
796
797
def _get_page_length():
    try:
        page_length = request.args.get("page_length")
        if page_length is None:
            return config.PAGE_LENGTH
        return int(page_length)
    except ValueError:
        return config.PAGE_LENGTH

798
@app.route("/todos/list")
Robin Sonnabend's avatar
Robin Sonnabend committed
799
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
800
801
def list_todos():
    user = current_user()
Robin Sonnabend's avatar
Robin Sonnabend committed
802
803
804
    protocoltype = None
    protocoltype_id = None
    try:
805
        protocoltype_id = int(request.args.get("protocoltype_id"))
Robin Sonnabend's avatar
Robin Sonnabend committed
806
807
    except (ValueError, TypeError):
        pass
Robin Sonnabend's avatar
Robin Sonnabend committed
808
    state_open = -1
Robin Sonnabend's avatar
Robin Sonnabend committed
809
    try:
Robin Sonnabend's avatar
Robin Sonnabend committed
810
        state_open = int(request.args.get("state_open"))
Robin Sonnabend's avatar
Robin Sonnabend committed
811
812
    except (ValueError, TypeError):
        pass
Robin Sonnabend's avatar
Robin Sonnabend committed
813
    search_term = request.args.get("search")
Robin Sonnabend's avatar
Robin Sonnabend committed
814
    protocoltypes = ProtocolType.get_public_protocoltypes(user)
Robin Sonnabend's avatar
Robin Sonnabend committed
815
    search_form = TodoSearchForm(protocoltypes)
Robin Sonnabend's avatar
Robin Sonnabend committed
816
    if protocoltype_id is not None:
817
        search_form.protocoltype_id.data = protocoltype_id
Robin Sonnabend's avatar
Robin Sonnabend committed
818
        protocoltype = ProtocolType.query.filter_by(id=protocoltype_id).first()
Robin Sonnabend's avatar
Robin Sonnabend committed
819
820
    if state_open is not None:
        search_form.state_open.data = state_open
821
822
    if search_term is not None:
        search_form.search.data = search_term
Robin Sonnabend's avatar
Robin Sonnabend committed
823
824
825
826
    todos = [
        todo for todo in Todo.query.all()
        if todo.protocoltype.has_public_view_right(user)
    ]
Robin Sonnabend's avatar
Robin Sonnabend committed
827
    if protocoltype_id is not None and protocoltype_id != -1:
Robin Sonnabend's avatar
Robin Sonnabend committed
828
829
830
831
        todos = [
            todo for todo in todos
            if todo.protocoltype.id == protocoltype_id
        ]
Robin Sonnabend's avatar
Robin Sonnabend committed
832
833
    if state_open is not None and state_open != -1:
        todo_done = bool(state_open)
Robin Sonnabend's avatar
Robin Sonnabend committed
834
835
836
837
        todos = [
            todo for todo in todos
            if todo.is_done() == todo_done
        ]
Robin Sonnabend's avatar
Robin Sonnabend committed
838
    if search_term is not None and len(search_term.strip()) > 0:
Robin Sonnabend's avatar
Robin Sonnabend committed
839
840
841
        todos = [
            todo for todo in todos
            if search_term.lower() in todo.description.lower()
842
            or search_term.lower() in todo.who.lower()
Robin Sonnabend's avatar
Robin Sonnabend committed
843
        ]
Robin Sonnabend's avatar
Robin Sonnabend committed
844
    def _sort_key(todo):
Administrator's avatar
Administrator committed
845
        return (not todo.is_done(), todo.get_id())
Robin Sonnabend's avatar
Robin Sonnabend committed
846
    todos = sorted(todos, key=_sort_key, reverse=True)
Robin Sonnabend's avatar
Robin Sonnabend committed
847
    page = _get_page()
Robin Sonnabend's avatar
Robin Sonnabend committed
848
849
    page_length = _get_page_length()
    page_count = int(math.ceil(len(todos) / page_length))
Robin Sonnabend's avatar
Robin Sonnabend committed
850
851
    if page >= page_count:
        page = 0
Robin Sonnabend's avatar
Robin Sonnabend committed
852
853
    begin_index = page * page_length
    end_index = (page + 1) * page_length
854
    max_page_length_exp = math.ceil(math.log10(len(todos))) if len(todos) > 0 else 1
Robin Sonnabend's avatar
Robin Sonnabend committed
855
    todos = todos[begin_index:end_index]
Robin Sonnabend's avatar
Robin Sonnabend committed
856
    todos_table = TodosTable(todos)
Robin Sonnabend's avatar
Robin Sonnabend committed
857
    return render_template("todos-list.html", todos=todos, todos_table=todos_table, search_form=search_form, page=page, page_count=page_count, page_diff=config.PAGE_DIFF, protocoltype_id=protocoltype_id, search_term=search_term, state_open=state_open, page_length=page_length, max_page_length_exp=max_page_length_exp)
Robin Sonnabend's avatar
Robin Sonnabend committed
858

Robin Sonnabend's avatar
Robin Sonnabend committed
859
860
861
862
@app.route("/todo/new", methods=["GET", "POST"])
@login_required
def new_todo():
    user = current_user()
863
    protocoltype_id = optional_int_arg("protocoltype_id")
Robin Sonnabend's avatar
Robin Sonnabend committed
864
865
866
867
868
869
870
871
872
873
874
875
876
877
878
879
    protocol_id = optional_int_arg("protocol_id")
    protocoltype = ProtocolType.query.filter_by(id=protocoltype_id).first()
    protocol = Protocol.query.filter_by(id=protocol_id).first()
    if protocoltype is not None and protocol is not None:
        if protocol.protocoltype != protocoltype:
            flash("Ungültige Protokoll-Typ-Kombination", "alert-error")
            return redirect(request.args.get("next") or url_for("index"))
    if protocoltype is None and protocol is not None:
        protocoltype = protocol.protocoltype
    protocoltypes = ProtocolType.get_modifiable_protocoltypes(user)
    form = NewTodoForm(protocoltypes)
    if form.validate_on_submit():
        added_protocoltype = ProtocolType.query.filter_by(id=form.protocoltype_id.data).first()
        if added_protocoltype is None or not added_protocoltype.has_modify_right(user):
            flash("Invalider Protokolltyp.")
            return redirect(request.args.get("next") or url_for("index"))
880
881
        todo = Todo()
        form.populate_obj(todo)
Robin Sonnabend's avatar
Robin Sonnabend committed
882
883
884
885
886
887
888
889
890
891
892
893
894
895
896
897
898
899
        if protocol is not None:
            todo.protocols.append(protocol)
        db.session.add(todo)
        db.session.commit()
        todo.number = todo.id
        db.session.commit()
        flash("Todo wurde angelegt.", "alert-success")
        if protocol is not None:
            return redirect(request.args.get("next") or url_for("show_protocol", protocol_id=protocol.id))
        else:
            return redirect(request.args.get("next") or url_for("list_todos", protocoltype_id=protocoltype_id))
    else:
        if protocoltype is not None:
            form.protocoltype_id.data = protocoltype.id
    return render_template("todo-new.html", form=form, protocol=protocol, protocoltype=protocoltype)

@app.route("/todo/edit/<int:todo_id>", methods=["GET", "POST"])
@login_required
Robin Sonnabend's avatar
Robin Sonnabend committed
900
@db_lookup(Todo)
Robin Sonnabend's avatar
Robin Sonnabend committed