summaryrefslogtreecommitdiff
path: root/day9/task5_vue/backend/views.py
blob: 5c39322199a60462a63aca0f0194d1a07f2d4360 (plain)
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
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
from backend.core.utils import render_template, NOT_FOUND_CODE
from backend.settings import STATIC_FILES_PATH

from backend.core.response_types import ErrorResponse, TextFileResponse, ImageResponse, HtmlResponse, JsonResponse
from backend.database.database import get_wrapper_for
from backend.database.validators import ValidationError

import csv


def return_static(query, *args):
    if args[0] in ['css', 'js']:
        try:
            return TextFileResponse(
                path=STATIC_FILES_PATH + f'/{args[0]}/{args[1]}',
                extension={'js': 'javascript', 'css': 'css'}.get(args[0])
            )

        except (PermissionError, FileNotFoundError):
            return ErrorResponse(404)

    elif args[0] == 'images':
        try:
            return ImageResponse(STATIC_FILES_PATH + f'/{args[0]}/{args[1]}', args[1].split('.')[-1])

        except (PermissionError, FileNotFoundError):
            return ErrorResponse(404)


def upload_file(query, *args):
    base_html = '<a href="/">Return to main page</a><br>%s'

    try:
        data = query['files'][0]['data'].decode('utf-8')
    except (UnicodeDecodeError, KeyError, IndexError):
        return ErrorResponse(500, base_html % '<h1>Error while reading file</h1>')

    wrapper = get_wrapper_for('mysql')

    permitted_headers = wrapper.get_column_names()

    try:
        data = list(csv.reader(data.strip().splitlines(), delimiter=';', quotechar='"'))
        if len(data[0]) != len(permitted_headers) or set(data[0]) != set(permitted_headers):
            return ErrorResponse(500, base_html % '<h1>File format error</h1>')
    except IndexError:
        return ErrorResponse(500, base_html % '<h1>File format error</h1>')

    wrapper.clear_table('table_task1')

    scheme = wrapper.schemes['table_task1']
    for row in data[1:]:
        insert_values = dict(zip(data[0], row))
        validation_results = scheme.validate(insert_values)
        if not validation_results['error']:
            wrapper.insert_one('table_task1', insert_values)
        else:
            return ErrorResponse(500)

    return HtmlResponse(base_html % '<h1>File uploaded</h1>')


def validate(query, *args):
    wrapper = get_wrapper_for('mysql')
    scheme = wrapper.schemes['table_task1']
    validation_results = scheme.validate(query)
    return JsonResponse(validation_results)


def update_post(query, *args):
    wrapper = get_wrapper_for('mysql')
    scheme = wrapper.schemes['table_task1']

    permitted_fields = wrapper.get_column_names()
    permitted_fields.remove('service_id')

    validation_results = scheme.validate(query)
    if not validation_results['error']:
        expressions = {}
        for key, value in query.items():
            if key in permitted_fields:
                expressions[key] = value

        wrapper.update('table_task1', expressions, {'service_id': query['service_id']})

    return JsonResponse(validation_results)


def delete_post(query, *args):
    wrapper = get_wrapper_for('mysql')
    scheme = wrapper.schemes['table_task1']

    try:
        scheme.fields['service_id'].validate(query['service_id'])
    except ValidationError as e:
        result = {'error': True, 'service_id': str(e)}
    else:
        result = {'error': False}
        wrapper.delete_from('table_task1', {'service_id': query['service_id']})

    return JsonResponse(result)


def add_post(query, *args):
    wrapper = get_wrapper_for('mysql')
    header_fields = wrapper.get_column_names()

    insert_data = {}

    for field_name, value in query.items():
        if field_name in header_fields:
            insert_data[field_name] = value

    scheme = wrapper.schemes['table_task1']
    validation_result = scheme.validate(query)
    if not validation_result['error']:
        wrapper.insert_one('table_task1', insert_data)

    return JsonResponse(validation_result)


def db_get(query, *args):
    wrapper = get_wrapper_for('mysql')
    table_headers = wrapper.get_column_names()

    if query['type'] == 'full':
        content = wrapper.get_data('table_task1')

        for row in content:
            # creation_time
            row[7] = ':'.join(_.rjust(2, '0') for _ in str(row[7]).split(':'))

            # creation_request_sent_date
            if row[8] is not None:
                creation_date, creation_time = str(row[8]).split()
                creation_time = ':'.join(_.rjust(2, '0') for _ in creation_time.split(':'))
                row[8] = creation_date + ' ' + creation_time

        json_content = []
        for row in content:
            new_row = []
            for col in row:
                if not isinstance(col, (float, bool, int, dict, list, tuple, str)):
                    col = str(col)
                new_row.append(col)
            json_content.append(new_row)

        return JsonResponse({'headers': table_headers, 'content': json_content})

    elif query['type'] == 'single_id':
        content = wrapper.get_rows('table_task1', {'service_id': query['service_id']})[0]

        # creation_time
        content[7] = ':'.join(_.rjust(2, '0') for _ in str(content[7]).split(':'))

        # creation_request_sent_date
        if content[8] is not None:
            creation_date, creation_time = str(content[8]).split()
            creation_time = ':'.join(_.rjust(2, '0') for _ in creation_time.split(':'))
            content[8] = creation_date + ' ' + creation_time

        if content is not None:
            json_content = []
            for col in content:
                if not isinstance(col, (float, bool, int, dict, list, tuple)):
                    col = str(col)
                json_content.append(col)

            return JsonResponse(dict(zip(table_headers, json_content)))
        else:
            return JsonResponse({})

    return ErrorResponse(NOT_FOUND_CODE)


def index_get(query, *args):
    data = render_template('index.html')
    return HtmlResponse(data)