TimeManager/form_parser.py at master · Goshective/TimeManager · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
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
120
121
122
123
124
125
from flask import redirect
from sqlalchemy import and_, or_, not_
from datetime import datetime

from data.act_names import Activities_names
from data.records import Records
from data.pictures import Pictures
from data import db_session


def add_activity(form, current_user):
    if form.validate_on_submit():
        with db_session.create_session() as db_sess:
            clone_detecting = db_sess.query(Activities_names).filter(
                and_(Activities_names.user == current_user, or_(
                Activities_names.name == form.name.data, and_(not_(
                form.none_color.data),
                Activities_names.color == form.color.data)))).all()
            if clone_detecting:
                return 1, "Такое имя или цвет активности уже существуют"
            db_sess.add(current_user)
            activity = Activities_names(name=form.name.data)
            if not form.none_color.data:
                activity.color = form.color.data
            current_user.act_names.append(activity)
            db_sess.commit()
        return 0, redirect('/settings')
    else:
        return 1, parser_activity_error(form.errors)

def change_activity(form, current_user):
    if not form.validate_on_submit():
        return 1, parser_activity_error(form.errors)
    with db_session.create_session() as db_sess:
        clone_detecting = db_sess.query(Activities_names).filter(
            and_(Activities_names.user == current_user, or_(
            Activities_names.name == form.name.data,
            and_((form.choose_color_mode.data == '3'),
            Activities_names.color == form.color.data),
            ), Activities_names.id != int(form.choose_names.data))).all()
        if clone_detecting:
            return 1, "Такое имя или цвет активности уже существуют"
        db_sess.add(current_user)
        change_detector = False
        if form.choose_color_mode.data == "2":
            db_sess.query(Activities_names).filter(
                Activities_names.id == int(form.choose_names.data)
                ).update({"color": None})
            change_detector = True
        elif form.choose_color_mode.data == "3":
            db_sess.query(Activities_names).filter(
                Activities_names.id == int(form.choose_names.data)
                ).update({"color": form.color.data})
            change_detector = True
        if form.name.data:
            db_sess.query(Activities_names).filter(
                Activities_names.id == int(form.choose_names.data)
                ).update({"name": form.name.data})
            change_detector = True
        if not change_detector:
            return 1, "Вы не выбрали поля для изменения"
        db_sess.commit()
    return 0, redirect('/settings')

def delete_activity(form, current_user):
    if not form.validate_on_submit():
        return 1, parser_activity_error(form.errors)
    with db_session.create_session() as db_sess:
        db_sess.add(current_user)
        db_sess.query(Records).filter_by(name_id=int(form.choose_names.data)).delete()
        db_sess.query(Activities_names).filter_by(id=int(form.choose_names.data)).delete()
        db_sess.commit()
    return 0, redirect('/settings')

def add_photo(form, current_user):
    if not form.validate_on_submit():
        return 1, parser_photo_error(form.errors)
    with db_session.create_session() as db_sess:
        db_sess.add(current_user)
        lst = form.photo.data.stream.read()
        if len(lst) > 1024 ** 2:
            return 1, "Размер файла слишком большой"
        else:
            if current_user.picture:
                current_user.picture.data = lst
            else:
                current_user.picture =  Pictures(data=lst)
            db_sess.commit()
    return 0, redirect('/settings')

def add_record(form, current_user):
    if not form.validate_on_submit():
        return 1, parser_record_error(form.errors)

    if form.work_hour.data < 0 or form.work_min.data < 0 or (form.work_hour.data == form.work_min.data == 0):
        return 1, "Некорректное значение времени"
    record = Records()
    record.name_id = form.activity.data
    if form.date.data == datetime.now().date():
        record.created_date = datetime.now()
    else:
        record.created_date = form.date.data

    record.work_hours = form.work_hour.data
    record.work_min = form.work_min.data
    current_user.records.append(record)
    return 0, "ok"

def delete_record(item_id, current_user):
    if current_user.is_authenticated:
        with db_session.create_session() as db_sess:
            db_sess.query(Records).filter(Records.id == item_id,
                                          Records.user == current_user).delete()
            db_sess.commit()

def parser_activity_error(errors):
    return errors

def parser_record_error(errors):
    return errors

def parser_photo_error(errors):
    if 'photo' in errors:
        return errors['photo'][0]
    return errors