tasks.py 31.8 KB
Newer Older
1
2
3
4
5
from flask import render_template

import os
import subprocess
import shutil
6
import tempfile
7
from datetime import datetime
Robin Sonnabend's avatar
Robin Sonnabend committed
8
import traceback
9
from copy import copy
10

11
from models.database import Document, Protocol, Error, Todo, Decision, TOP, DefaultTOP, MeetingReminder, TodoMail, DecisionDocument, TodoState, OldTodo, DecisionCategory
Robin Sonnabend's avatar
Robin Sonnabend committed
12
from models.errors import DateNotMatchingException
13
from server import celery, app
14
from shared import db, escape_tex, unhyphen, date_filter, datetime_filter, date_filter_long, date_filter_short, time_filter, class_filter, KNOWN_KEYS
Robin Sonnabend's avatar
Robin Sonnabend committed
15
from utils import mail_manager, url_manager, encode_kwargs, decode_kwargs, add_line_numbers, set_etherpad_text, get_etherpad_text, footnote_hash
Robin Sonnabend's avatar
Robin Sonnabend committed
16
from protoparser import parse, ParserException, Element, Content, Text, Tag, Remark, Fork, RenderType
Robin Sonnabend's avatar
Robin Sonnabend committed
17
from wiki import WikiClient, WikiException
Robin Sonnabend's avatar
Robin Sonnabend committed
18
from calendarpush import Client as CalendarClient, CalendarException
19
from legacy import lookup_todo_id, import_old_todos
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

import config

texenv = app.create_jinja_environment()
texenv.block_start_string = r"\ENV{"
texenv.block_end_string = r"}"
texenv.variable_start_string = r"\VAR{"
texenv.variable_end_string = r"}"
texenv.comment_start_string = r"\COMMENT{"
texenv.comment_end_string = r"}"
texenv.filters["escape_tex"] = escape_tex
texenv.filters["unhyphen"] = unhyphen
texenv.trim_blocks = True
texenv.lstrip_blocks = True
texenv.filters["url_complete"] = url_manager.complete
texenv.filters["datify"] = date_filter
texenv.filters["datify_long"] = date_filter_long
37
texenv.filters["datify_short"] = date_filter_short
38
39
texenv.filters["datetimify"] = datetime_filter
texenv.filters["timify"] = time_filter
40
texenv.filters["class"] = class_filter
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
logo_template = getattr(config, "LATEX_LOGO_TEMPLATE", None)
if logo_template is not None:
    texenv.globals["logo_template"] = logo_template
latex_geometry = getattr(config, "LATEX_GEOMETRY", "vmargin=1.5cm,hmargin={1.5cm,1.2cm},bindingoffset=8mm")
texenv.globals["latex_geometry"] = latex_geometry
raw_additional_packages = getattr(config, "LATEX_ADDITIONAL_PACKAGES", None)
additional_packages = []
if raw_additional_packages is not None:
    for package in raw_additional_packages:
        if "{" not in package:
            package = "{{{}}}".format(package)
        additional_packages.append(package)
texenv.globals["additional_packages"] = additional_packages
latex_pagestyle = getattr(config, "LATEX_PAGESTYLE", None)
if latex_pagestyle is not None:
    texenv.globals["latex_pagestyle"] = latex_pagestyle
latex_header_footer = getattr(config, "LATEX_HEADER_FOOTER", False)
texenv.globals["latex_header_footer"] = latex_header_footer
59
60
61
62
63
64
65
66

mailenv = app.create_jinja_environment()
mailenv.trim_blocks = True
mailenv.lstrip_blocks = True
mailenv.filters["url_complete"] = url_manager.complete
mailenv.filters["datify"] = date_filter
mailenv.filters["datetimify"] = datetime_filter

67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
wikienv = app.create_jinja_environment()
wikienv.trim_blocks = True
wikienv.lstrip_blocks = True
wikienv.block_start_string = "<env>"
wikienv.block_end_string = "</env>"
wikienv.variable_start_string = "<var>"
wikienv.variable_end_string = "</var>"
wikienv.comment_start_string = "<comment>"
wikienv.comment_end_string = "</comment>"
wikienv.filters["datify"] = date_filter
wikienv.filters["datify_long"] = date_filter_long
wikienv.filters["datify_short"] = date_filter_short
wikienv.filters["datetimify"] = datetime_filter
wikienv.filters["timify"] = time_filter
wikienv.filters["class"] = class_filter


84
85
86
87
88
89
90
91
92
93
ID_FIELD_BEGINNING = "id "

def parse_protocol(protocol, **kwargs):
    parse_protocol_async.delay(protocol.id, encode_kwargs(kwargs))

@celery.task
def parse_protocol_async(protocol_id, encoded_kwargs):
    with app.app_context():
        with app.test_request_context("/"):
            try:
94
95
96
97
98
99
                kwargs = decode_kwargs(encoded_kwargs)
                protocol = Protocol.query.filter_by(id=protocol_id).first()
                if protocol is None:
                    raise Exception("No protocol given. Aborting parsing.")
                parse_protocol_async_inner(protocol, encoded_kwargs)
            except Exception as exc:
Robin Sonnabend's avatar
Robin Sonnabend committed
100
101
102
                stacktrace = traceback.format_exc()
                error = protocol.create_error("Parsing", "Exception",
                    "{}\n\n{}".format(str(exc), stacktrace))
103
104
                db.session.add(error)
                db.session.commit()
105
106
107
108
109
110

def parse_protocol_async_inner(protocol, encoded_kwargs):
    old_errors = list(protocol.errors)
    for error in old_errors:
        protocol.errors.remove(error)
    db.session.commit()
111
112
    if protocol.source is None or len(protocol.source.strip()) == 0:
        error = protocol.create_error("Parsing", "Protocol source is empty", "")
113
114
115
        db.session.add(error)
        db.session.commit()
        return
116
117
118
119
120
    if protocol.source == config.EMPTY_ETHERPAD:
        error = protocol.create_error("Parsing", "The etherpad is unmodified and does not contain a protocol.", protocol.source)
        db.session.add(error)
        db.session.commit()
        return
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
    tree = None
    try:
        tree = parse(protocol.source)
    except ParserException as exc:
        context = ""
        if exc.linenumber is not None:
            source_lines = protocol.source.splitlines()
            start_index = max(0, exc.linenumber - config.ERROR_CONTEXT_LINES)
            end_index = min(len(source_lines) - 1, exc.linenumber + config.ERROR_CONTEXT_LINES)
            context = "\n".join(source_lines[start_index:end_index])
        if exc.tree is not None:
            context += "\n\nParsed syntax tree was:\n" + str(exc.tree.dump())
        error = protocol.create_error("Parsing", str(exc), context)
        db.session.add(error)
        db.session.commit()
        return
    remarks = {element.name: element for element in tree.children if isinstance(element, Remark)}
138
    required_fields = copy(KNOWN_KEYS)
Robin Sonnabend's avatar
Robin Sonnabend committed
139
140
    for default_meta in protocol.protocoltype.metas:
        required_fields.append(default_meta.key)
141
142
143
    if not config.PARSER_LAZY:
        missing_fields = [field for field in required_fields if field not in remarks]
        if len(missing_fields) > 0:
144
            error = protocol.create_error("Parsing", "Du hast vergessen, Metadaten anzugeben.", ", ".join(missing_fields))
145
            db.session.add(error)
Robin Sonnabend's avatar
Robin Sonnabend committed
146
            db.session.commit()
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
            return
    try:
        protocol.fill_from_remarks(remarks)
    except ValueError:
        error = protocol.create_error(
            "Parsing", "Invalid fields",
            "Date or time fields are not '%d.%m.%Y' respectively '%H:%M', "
            "but rather {}".format(
            ", ".join([
                remarks["Datum"].value.strip(),
                remarks["Beginn"].value.strip(),
                remarks["Ende"].value.strip()
            ])))
        db.session.add(error)
        db.session.commit()
        return
    except DateNotMatchingException as exc:
        error = protocol.create_error("Parsing", "Date not matching",
            "This protocol's date should be {}, but the protocol source says {}.".format(date_filter(exc.original_date) if exc.original_date is not None else "not present", date_filter(exc.protocol_date) if exc.protocol_date is not None else "not present"))
        db.session.add(error)
        db.session.commit()
        return
169
170
    # tags 
    tags = tree.get_tags()
Robin Sonnabend's avatar
Robin Sonnabend committed
171
172
    elements = tree.get_visible_elements(show_private=True)
    public_elements = tree.get_visible_elements(show_private=False)
173
174
175
176
177
178
179
180
181
    for tag in tags:
        if tag.name not in Tag.KNOWN_TAGS:
            error = protocol.create_error("Parsing", "Invalid tag",
                "The tag in line {} has the kind '{}', which is "
                "not defined. This is probably an error mit a missing "
                "semicolon.".format(tag.linenumber, tag.name))
            db.session.add(error)
            db.session.commit()
            return
182
    # todos
183
184
185
    old_todo_number_map = {}
    for todo in protocol.todos:
        old_todo_number_map[todo.description] = todo.get_id()
186
187
188
189
    protocol.delete_orphan_todos()
    db.session.commit()
    old_todos = list(protocol.todos)
    todo_tags = [tag for tag in tags if tag.name == "todo"]
190
    raw_todos = []
191
192
193
194
195
    for todo_tag in todo_tags:
        if len(todo_tag.values) < 2:
            error = protocol.create_error("Parsing", "Invalid todo-tag",
                "The todo tag in line {} needs at least "
                "information on who and what, "
196
197
                "but has less than that. This is probably "
                "a missing semicolon.".format(todo_tag.linenumber))
198
            db.session.add(error)
199
            db.session.commit()
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
            return
        who = todo_tag.values[0]
        what = todo_tag.values[1]
        field_id = None
        field_state = None
        field_date = None
        for other_field in todo_tag.values[2:]:
            other_field = other_field.strip()
            if len(other_field) == 0:
                continue
            if other_field.startswith(ID_FIELD_BEGINNING):
                try:
                    field_id = int(other_field[len(ID_FIELD_BEGINNING):])
                except ValueError:
                    error = protocol.create_error("Parsing", "Non-numerical todo ID",
                    "The todo in line {} has a nonnumerical ID, but needs "
                    "something like \"id 1234\"".format(todo_tag.linenumber))
217
218
219
                    db.session.add(error)
                    db.session.commit()
                    return
220
221
222
223
224
225
226
            else:
                try:
                    field_state = TodoState.from_name(other_field)
                except ValueError:
                    try:
                        field_date = datetime.strptime(other_field, "%d.%m.%Y")
                    except ValueError:
227
                        try:
228
                            field_state, field_date = TodoState.from_name_with_date(other_field.strip(), protocol=protocol)
229
230
                        except ValueError:
                            try:
231
                                field_state = TodoState.from_name_lazy(other_field)
232
233
234
                            except ValueError:
                                error = protocol.create_error("Parsing",
                                "Invalid field",
235
236
                                "The todo in line {} has the field '{}', but "
                                "this does neither match a date (\"%d.%m.%Y\") "
237
238
239
240
241
                                "nor a state.".format(
                                    todo_tag.linenumber, other_field))
                                db.session.add(error)
                                db.session.commit()
                                return
242
243
244
245
246
247
248
249
250
        raw_todos.append((who, what, field_id, field_state, field_date, todo_tag))
    for (_, _, field_id, _, _, _) in raw_todos:
        if field_id is not None:
            old_todos = [todo for todo in old_todos
                if todo.id != field_id]
    for todo in old_todos:
        protocol.todos.remove(todo)
    db.session.commit()
    for (who, what, field_id, field_state, field_date, todo_tag) in raw_todos:
251
252
253
254
255
256
257
258
        if field_state is None:
            field_state = TodoState.open
        if field_state.needs_date() and field_date is None:
            error = protocol.create_error("Parsing",
                "Todo missing date",
                "The todo in line {} has a state that needs a date, "
                "but the todo does not have one.".format(todo_tag.linenumber))
            db.session.add(error)
259
            db.session.commit()
260
261
262
263
264
265
266
267
268
269
270
271
            return
        who = who.strip()
        what = what.strip()
        todo = None
        if field_id is not None:
            todo = Todo.query.filter_by(number=field_id).first()
            if todo is None and not config.PARSER_LAZY:
                error = protocol.create_error("Parsing",
                "Invalid Todo ID",
                "The todo in line {} has the ID {}, but there is no "
                "Todo with that ID.".format(todo_tag.linenumber, field_id))
                db.session.add(error)
272
                db.session.commit()
273
                return
274
        if todo is None and field_id is None and what in old_todo_number_map:
275
            todo = Todo(protocoltype_id=protocol.protocoltype.id,
276
277
278
279
                who=who, description=what, state=field_state,
                date=field_date, number=old_todo_number_map[what])
            db.session.add(todo)
            db.session.commit()
280
281
282
283
284
285
        if todo is None:
            protocol_key = protocol.get_identifier()
            old_candidates = OldTodo.query.filter(
                OldTodo.protocol_key == protocol_key).all()
            if len(old_candidates) == 0:
                # new protocol
286
                todo = Todo(protocoltype_id=protocol.protocoltype.id,
287
288
289
290
291
292
293
294
295
296
297
                    who=who, description=what, state=field_state,
                    date=field_date)
                db.session.add(todo)
                db.session.commit()
                todo.number = field_id or todo.id
                db.session.commit()
            else:
                # old protocol
                number = field_id or lookup_todo_id(old_candidates, who, what)
                todo = Todo.query.filter_by(number=number).first()
                if todo is None:
298
                    todo = Todo(protocoltype_id=protocol.protocoltype.id,
299
300
301
302
303
                        who=who, description=what, state=field_state,
                        date=field_date, number=number)
                    db.session.add(todo)
                    db.session.commit()
        todo.protocols.append(protocol)
304
305
306
307
308
309
310
311
312
313
        is_newest_protocol = True
        for other_protocol in todo.protocols:
            if other_protocol.date > protocol.date:
                is_newest_protocol = False
                break
        if is_newest_protocol:
            todo.state = field_state
            todo.date = field_date
            todo.who = who
            todo.description = what
314
315
316
        db.session.commit()
        todo_tag.todo = todo
    # Decisions
Robin Sonnabend's avatar
Robin Sonnabend committed
317
318
319
320
321
322
323
324
325
326
    decision_tags = [tag for tag in tags if tag.name == "beschluss"]
    for decision_tag in decision_tags:
        if decision_tag not in public_elements:
            error = protocol.create_error("Parsing", "Decision in private context.",
                "The decision in line {} is in a private context, but decisions are "
                "and have to be public. Please move it to a public spot.".format(
                decision_tag.linenumber))
            db.session.add(error)
            db.session.commit()
            return
327
328
329
330
    old_decisions = list(protocol.decisions)
    for decision in old_decisions:
        protocol.decisions.remove(decision)
    db.session.commit()
331
    decisions_to_render = []
332
333
334
335
336
    for decision_tag in decision_tags:
        if len(decision_tag.values) == 0:
            error = protocol.create_error("Parsing", "Empty decision found.",
                "The decision in line {} is empty.".format(decision_tag.linenumber))
            db.session.add(error)
Robin Sonnabend's avatar
Robin Sonnabend committed
337
            db.session.commit()
338
            return
339
        decision_content = decision_tag.values[0]
340
341
        decision_categories = []
        for decision_category_name in decision_tag.values[1:]:
342
343
344
345
346
347
348
349
350
351
352
353
354
355
356
357
358
359
360
            decision_category = DecisionCategory.query.filter_by(protocoltype_id=protocol.protocoltype.id, name=decision_category_name).first()
            if decision_category is None:
                category_candidates = DecisionCategory.query.filter_by(protocoltype_id=protocol.protocoltype.id).all()
                category_names = [
                    "'{}'".format(category.name)
                    for category in category_candidates
                ]
                error = protocol.create_error("Parsing",
                    "Unknown decision category",
                    "The decision in line {} has the category {}, "
                    "but there is no such category. "
                    "Known categories are {}".format(
                        decision_tag.linenumber,
                        decision_category_name,
                        ", ".join(category_names)))
                db.session.add(error)
                db.session.commit()
                return
            else:
361
                decision_categories.append(decision_category)
362
        decision = Decision(protocol_id=protocol.id,
363
            content=decision_content)
364
365
        db.session.add(decision)
        db.session.commit()
366
367
        for decision_category in decision_categories:
            decision.categories.append(decision_category)
368
369
370
        decision_tag.decision = decision
        decisions_to_render.append((decision, decision_tag))
    for decision, decision_tag in decisions_to_render:
371
372
373
        decision_top = decision_tag.fork.get_top()
        decision_content = texenv.get_template("decision.tex").render(
            render_type=RenderType.latex, decision=decision,
Robin Sonnabend's avatar
Robin Sonnabend committed
374
            protocol=protocol, top=decision_top, show_private=True)
375
376
        maxdepth = decision_top.get_maxdepth()
        compile_decision(decision_content, decision, maxdepth=maxdepth)
Robin Sonnabend's avatar
Robin Sonnabend committed
377
378
379
380
381
382

    # Footnotes
    footnote_tags = [tag for tag in tags if tag.name == "footnote"]
    public_footnote_tags = [tag for tag in footnote_tags if tag in public_elements]

    # TOPs
383
384
385
386
387
    old_tops = list(protocol.tops)
    for top in old_tops:
        protocol.tops.remove(top)
    tops = []
    for index, fork in enumerate((child for child in tree.children if isinstance(child, Fork))):
388
389
        top = TOP(protocol_id=protocol.id, name=fork.name, number=index,
            planned=False)
390
391
        db.session.add(top)
    db.session.commit()
392

Robin Sonnabend's avatar
Robin Sonnabend committed
393
394
    # render
    private_render_kwargs = {
395
        "protocol": protocol,
Robin Sonnabend's avatar
Robin Sonnabend committed
396
397
        "tree": tree,
        "footnotes": footnote_tags,
398
    }
Robin Sonnabend's avatar
Robin Sonnabend committed
399
400
401
402
    public_render_kwargs = copy(private_render_kwargs)
    public_render_kwargs["footnotes"] = public_footnote_tags
    render_kwargs = {True: private_render_kwargs, False: public_render_kwargs}
    
403
404
    maxdepth = tree.get_maxdepth()
    privacy_states = [False]
Robin Sonnabend's avatar
Robin Sonnabend committed
405
406
    content_private = render_template("protocol.txt", render_type=RenderType.plaintext, show_private=True, **private_render_kwargs)
    content_public = render_template("protocol.txt", render_type=RenderType.plaintext, show_private=False, **public_render_kwargs)
407
408
409
410
    if content_private != content_public:
        privacy_states.append(True)
    protocol.content_private = content_private
    protocol.content_public = content_public
411
    protocol.content_html_private = render_template("protocol.html",
Robin Sonnabend's avatar
Robin Sonnabend committed
412
        render_type=RenderType.html, show_private=True, **private_render_kwargs)
413
    protocol.content_html_public = render_template("protocol.html",
Robin Sonnabend's avatar
Robin Sonnabend committed
414
        render_type=RenderType.html, show_private=False, **public_render_kwargs)
415

416
    for show_private in privacy_states:
Robin Sonnabend's avatar
Robin Sonnabend committed
417
418
419
420
        latex_source = texenv.get_template("protocol.tex").render(
            render_type=RenderType.latex,
            show_private=show_private,
            **render_kwargs[show_private])
421
        compile(latex_source, protocol, show_private=show_private, maxdepth=maxdepth)
Robin Sonnabend's avatar
Robin Sonnabend committed
422

423
    if protocol.protocoltype.use_wiki:
Robin Sonnabend's avatar
Robin Sonnabend committed
424
425
426
427
428
429
430
431
432
433
        show_private = not protocol.protocoltype.wiki_only_public
        wiki_source = wikienv.get_template("protocol.wiki").render(
            render_type=RenderType.wikitext,
            show_private=show_private,
            **render_kwargs[show_private]
        ).replace("\n\n\n", "\n\n")
        wiki_infobox_source = wikienv.get_template("infobox.wiki").render(
            protocoltype=protocol.protocoltype)
        push_to_wiki(protocol, wiki_source, wiki_infobox_source,
            "Automatisch generiert vom Protokollsystem 3.0")
434
435
    protocol.done = True
    db.session.commit()
436

437
438
def push_to_wiki(protocol, content, infobox_content, summary):
    push_to_wiki_async.delay(protocol.id, content, infobox_content, summary)
Robin Sonnabend's avatar
Robin Sonnabend committed
439
440

@celery.task
441
def push_to_wiki_async(protocol_id, content, infobox_content, summary):
Robin Sonnabend's avatar
Robin Sonnabend committed
442
443
444
    with WikiClient() as wiki_client, app.app_context():
        protocol = Protocol.query.filter_by(id=protocol_id).first()
        try:
445
446
447
448
            wiki_client.edit_page(
                title=protocol.protocoltype.get_wiki_infobox_title(),
                content=infobox_content,
                summary="Automatisch generiert vom Protokollsystem 3.")
Robin Sonnabend's avatar
Robin Sonnabend committed
449
450
451
452
453
454
455
            wiki_client.edit_page(
                title=protocol.get_wiki_title(),
                content=content,
                summary="Automatisch generiert vom Protokollsystem 3.")
        except WikiException as exc:
            error = protocol.create_error("Pushing to Wiki", "Pushing to Wiki failed.", str(exc))

456
457
def compile(content, protocol, show_private, maxdepth):
   compile_async.delay(content, protocol.id, show_private=show_private, maxdepth=maxdepth)
Robin Sonnabend's avatar
Robin Sonnabend committed
458

459
460
def compile_decision(content, decision, maxdepth):
    compile_async.delay(content, decision.id, use_decision=True, maxdepth=maxdepth)
461
462

@celery.task
463
def compile_async(content, protocol_id, show_private=False, use_decision=False, maxdepth=5):
464
    with tempfile.TemporaryDirectory() as compile_dir, app.app_context():
Robin Sonnabend's avatar
Robin Sonnabend committed
465
466
467
468
469
470
471
        decision = None
        protocol = None
        if use_decision:
            decision = Decision.query.filter_by(id=protocol_id).first()
            protocol = decision.protocol
        else:
            protocol = Protocol.query.filter_by(id=protocol_id).first()
472
473
474
        try:
            current = os.getcwd()
            protocol_source_filename = "protocol.tex"
475
            protocol_target_filename = "protocol.pdf"
476
            protocol_class_filename = "protokoll2.cls"
477
478
479
            log_filename = "protocol.log"
            with open(os.path.join(compile_dir, protocol_source_filename), "w") as source_file:
                source_file.write(content)
480
481
            protocol2_class_source = texenv.get_template(protocol_class_filename).render(fonts=config.FONTS, maxdepth=maxdepth, bulletpoints=config.LATEX_BULLETPOINTS)
            with open(os.path.join(compile_dir, protocol_class_filename), "w") as protocol2_class_file:
482
                protocol2_class_file.write(protocol2_class_source)
483
484
485
486
487
488
489
490
491
492
            os.chdir(compile_dir)
            command = [
                "/usr/bin/xelatex",
                "-halt-on-error",
                "-file-line-error",
                protocol_source_filename
            ]
            subprocess.check_call(command, universal_newlines=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
            subprocess.check_call(command, universal_newlines=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
            os.chdir(current)
Robin Sonnabend's avatar
Robin Sonnabend committed
493
494
495
496
497
            document = None
            if not use_decision:
                for old_document in [document for document in protocol.documents if document.is_compiled and document.is_private == show_private]:
                    protocol.documents.remove(old_document)
                db.session.commit()
498
499
500
501
502
503
504
505
                document = Document(protocol_id=protocol.id,
                    name="protokoll{}_{}_{}.pdf".format(
                        "_intern" if show_private else "",
                        protocol.protocoltype.short_name,
                        date_filter_short(protocol.date)),
                    filename="",
                    is_compiled=True,
                    is_private=show_private)
Robin Sonnabend's avatar
Robin Sonnabend committed
506
            else:
507
508
509
510
511
512
                document = DecisionDocument(decision_id=decision.id,
                    name="beschluss_{}_{}_{}.pdf".format(
                        protocol.protocoltype.short_name,
                        date_filter_short(protocol.date),
                        decision.id),
                    filename="")
513
514
            db.session.add(document)
            db.session.commit()
Robin Sonnabend's avatar
Robin Sonnabend committed
515
            target_filename = "compiled-{}-{}.pdf".format(document.id, "internal" if show_private else "public")
Robin Sonnabend's avatar
Robin Sonnabend committed
516
517
            if use_decision:
                target_filename = "decision-{}-{}-{}.pdf".format(protocol.id, decision.id, document.id)
518
            document.filename = target_filename
519
            shutil.copy(os.path.join(compile_dir, protocol_target_filename), os.path.join(config.DOCUMENTS_PATH, target_filename))
520
            db.session.commit()
Administrator's avatar
Administrator committed
521
            shutil.copy(os.path.join(compile_dir, log_filename), "/tmp/proto-tex.log")
522
523
524
525
526
        except subprocess.SubprocessError:
            log = ""
            total_log_filename = os.path.join(compile_dir, log_filename)
            if os.path.isfile(total_log_filename):
                with open(total_log_filename, "r") as log_file:
527
                    log = "Log:\n\n" + add_line_numbers(log_file.read())
528
529
            else:
                log = "Logfile not found."
530
531
532
533
            total_source_filename = os.path.join(compile_dir, protocol_source_filename)
            if os.path.isfile(total_source_filename):
                with open(total_source_filename, "r") as source_file:
                    log += "\n\nSource:\n\n" + add_line_numbers(source_file.read())
534
535
536
537
            total_class_filename = os.path.join(compile_dir, protocol_class_filename)
            if os.path.isfile(total_class_filename):
                with open(total_class_filename, "r") as class_file:
                    log += "\n\nClass:\n\n" + add_line_numbers(class_file.read())
538
539
540
541
542
            error = protocol.create_error("Compiling", "Compiling LaTeX failed", log)
            db.session.add(error)
            db.session.commit()
        finally:
            os.chdir(current)
543

Robin Sonnabend's avatar
Robin Sonnabend committed
544
545
546
547
548
549
550
551
552
553
554
555
556
557
558
559
560
561
562
563
564
565
566
567
568
569
570
def print_file(filename, protocol):
    if config.PRINTING_ACTIVE:
        print_file_async.delay(filename, protocol.id)

@celery.task
def print_file_async(filename, protocol_id):
    with app.app_context():
        protocol = Protocol.query.filter_by(id=protocol_id).first()
        if protocol.protocoltype.printer is None:
            error = protocol.create_error("Printing", "No printer configured.", "You don't have any printer configured for the protocoltype {}. Please do so before printing a protocol.".format(protocol.protocoltype.name))
        try:
            command = [
                "/usr/bin/lpr",
                "-H", config.PRINTING_SERVER,
                "-P", protocol.protocoltype.printer,
                "-U", config.PRINTING_USER,
                "-T", protocol.get_identifier(),
            ]
            for option in config.PRINTING_PRINTERS[protocol.protocoltype.printer]:
                command.extend(["-o", '"{}"'.format(option) if " " in option else option])
            command.append(filename)
            subprocess.check_call(command, universal_newlines=True, stdout=subprocess.DEVNULL, stderr=subprocess.DEVNULL)
        except subprocess.SubprocessError:
            error = protocol.create_error("Printing", "Printing {} failed.".format(protocol.get_identifier()), "")
            db.session.add(error)
            db.session.commit()

Robin Sonnabend's avatar
Robin Sonnabend committed
571
572
def send_reminder(reminder, protocol):
    send_reminder_async.delay(reminder.id, protocol.id)
573
574

@celery.task
Robin Sonnabend's avatar
Robin Sonnabend committed
575
def send_reminder_async(reminder_id, protocol_id):
576
    with app.app_context():
Robin Sonnabend's avatar
Robin Sonnabend committed
577
578
        reminder = MeetingReminder.query.filter_by(id=reminder_id).first()
        protocol = Protocol.query.filter_by(id=protocol_id).first()
579
        reminder_text = render_template("reminder-mail.txt", reminder=reminder, protocol=protocol)
Robin Sonnabend's avatar
Robin Sonnabend committed
580
        if reminder.send_public:
581
            print("sending public reminder mail to {}".format(protocol.protocoltype.public_mail))
Robin Sonnabend's avatar
Robin Sonnabend committed
582
583
            send_mail(protocol, protocol.protocoltype.public_mail, "Tagesordnung der {}".format(protocol.protocoltype.name), reminder_text)
        if reminder.send_private:
584
            print("sending private reminder mail to {}".format(protocol.protocoltype.private_mail))
Robin Sonnabend's avatar
Robin Sonnabend committed
585
586
            send_mail(protocol, protocol.protocoltype.private_mail, "Tagesordnung der {}".format(protocol.protocoltype.name), reminder_text)

587
def send_protocol_private(protocol):
588
    send_protocol_async.delay(protocol.id, show_private=True)
Robin Sonnabend's avatar
Robin Sonnabend committed
589
    send_todomails_async.delay(protocol.id)
590

591
592
593
def send_protocol_public(protocol):
    send_protocol_async.delay(protocol.id, show_private=False)

594
595
596
597
598
599
@celery.task
def send_protocol_async(protocol_id, show_private):
    with app.app_context():
        protocol = Protocol.query.filter_by(id=protocol_id).first()
        to_addr = protocol.protocoltype.private_mail if show_private else protocol.protocoltype.public_mail
        subject = "{}{}-Protokoll vom {}".format("Internes " if show_private else "", protocol.protocoltype.short_name, date_filter(protocol.date))
600
        mail_content = render_template("protocol-mail.txt", protocol=protocol, show_private=show_private)
601
602
603
604
605
606
        appendix = [(document.name, document.as_file_like())
            for document in protocol.documents
            if show_private or not document.is_private
        ]
        send_mail(protocol, to_addr, subject, mail_content, appendix)

Robin Sonnabend's avatar
Robin Sonnabend committed
607
608
609
610
@celery.task
def send_todomails_async(protocol_id):
    with app.app_context():
        protocol = Protocol.query.filter_by(id=protocol_id).first()
611
        all_todos = [
Administrator's avatar
Administrator committed
612
            todo for todo in Todo.query.all()
613
            if not todo.is_done()
Administrator's avatar
Administrator committed
614
            and todo.protocoltype == protocol.protocoltype
615
        ]
Robin Sonnabend's avatar
Robin Sonnabend committed
616
617
618
619
620
621
        users = {user for todo in all_todos for user in todo.get_users()}
        grouped_todos = {
            user: [todo for todo in all_todos if user in todo.get_users()]
            for user in users
        }
        subject = "Du hast noch was zu tun!"
622
623
624
625
626
627
628
629
630
        todomail_providers = getattr(config, "ADDITIONAL_TODOMAIL_PROVIDERS", None)
        additional_todomails = {}
        if todomail_providers:
            for provider in todomail_providers:
                todomail_dict = provider()
                for key in todomail_dict:
                    if key not in additional_todomails:
                        name, mail = todomail_dict[key]
                        additional_todomails[key] = TodoMail(name, mail)
Robin Sonnabend's avatar
Robin Sonnabend committed
631
632
        for user in users:
            todomail = TodoMail.query.filter(TodoMail.name.ilike(user)).first()
633
634
635
            if todomail is None:
                if user in additional_todomails:
                    todomail = additional_todomails[user]
Robin Sonnabend's avatar
Robin Sonnabend committed
636
637
638
639
640
641
642
643
            if todomail is None:
                error = protocol.create_error("Sending Todomail", "Sending Todomail failed.", "User {} has no Todo-Mail-Assignment.".format(user))
                db.session.add(error)
                db.session.commit()
                continue
            to_addr = todomail.get_formatted_mail()
            mail_content = render_template("todo-mail.txt", protocol=protocol, todomail=todomail, todos=grouped_todos[user])
            send_mail(protocol, to_addr, subject, mail_content)
644
645

def send_mail(protocol, to_addr, subject, content, appendix=None):
Robin Sonnabend's avatar
Robin Sonnabend committed
646
    if to_addr is not None and len(to_addr.strip()) > 0:
647
        send_mail_async.delay(protocol.id, to_addr, subject, content, appendix)
648

Robin Sonnabend's avatar
Robin Sonnabend committed
649
@celery.task
650
def send_mail_async(protocol_id, to_addr, subject, content, appendix):
Robin Sonnabend's avatar
Robin Sonnabend committed
651
652
653
    with app.app_context():
        protocol = Protocol.query.filter_by(id=protocol_id).first()
        try:
654
            mail_manager.send(to_addr, subject, content, appendix)
Robin Sonnabend's avatar
Robin Sonnabend committed
655
656
657
658
        except Exception as exc:
            error = protocol.create_error("Sending Mail", "Sending mail failed", str(exc))
            db.session.add(error)
            db.session.commit()
659

Robin Sonnabend's avatar
Robin Sonnabend committed
660
661
662
663
664
665
666
667
668
669
670
671
672
673
674
675
676
677
678
679
680
def push_tops_to_calendar(protocol):
    push_tops_to_calendar_async.delay(protocol.id)

@celery.task
def push_tops_to_calendar_async(protocol_id):
    if not config.CALENDAR_ACTIVE:
        return
    with app.app_context():
        protocol = Protocol.query.filter_by(id=protocol_id).first()
        if protocol.protocoltype.calendar == "":
            return
        description = render_template("calendar-tops.txt", protocol=protocol)
        try:
            client = CalendarClient(protocol.protocoltype.calendar)
            client.set_event_at(begin=protocol.get_datetime(),
                name=protocol.protocoltype.short_name, description=description)
        except CalendarException as exc:
            error = Protocol.create_error("Calendar",
                "Pushing TOPs to Calendar failed", str(exc))
            db.session.add(error)
            db.session.commit()
681
682

def set_etherpad_content(protocol):
Robin Sonnabend's avatar
Robin Sonnabend committed
683
    set_etherpad_content_async.delay(protocol.id)
684
685
686
687
688

@celery.task
def set_etherpad_content_async(protocol_id):
    with app.app_context():
        protocol = Protocol.query.filter_by(id=protocol_id).first()
Administrator's avatar
Administrator committed
689
690
        identifier = protocol.get_identifier()
        return set_etherpad_text(identifier, protocol.get_template())
691