summaryrefslogtreecommitdiff
path: root/day9/task5/main.py
blob: d5bcbb8b67e2afe43e8351c8196728181c126b34 (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
from router import route
from utils import render_template, NOT_FOUND_CODE
from config import SERVER_HOST, SERVER_PORT

import logging
from sys import stdout


@route('/update', ['POST'])
def update_post(query, *args):
    service_id = query['service_id']
    permitted_fields = db_column_names()
    permitted_fields.remove('service_id')

    query_set = []
    for field_name, value in query.items():
        if field_name in permitted_fields:
            query_set.append(f'{field_name}="{value}"')

    cursor = db.cursor()
    cursor.execute("UPDATE `table_task1` SET {} WHERE {};".format(
        ','.join(query_set), f'service_id="{service_id}"'
    ))
    cursor.close()

    return f'<h1>Database Updated</h1>'


@route('/delete', ['POST'])
def delete_post(query, *args):
    return f'<h1>DELETE: {query}</h1>'


@route('/add', ['POST'])
def add_post(query, *args):
    return f'<h1>ADD: {query}</h1>'


@route('/get', ['POST'])
def db_get(query, *args):
    table_headers = db_column_names()
    cursor = db.cursor()

    if query['type'] == 'full':
        cursor.execute('SELECT * FROM table_task1;')
        content = cursor.fetchall()

        cursor.close()

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

        return {'headers': table_headers, 'content': json_content}

    elif query['type'] == 'single_id':
        cursor.execute(f'SELECT * FROM table_task1 WHERE service_id="{query["service_id"]}";')
        content = cursor.fetchone()

        cursor.close()

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

        return dict(zip(table_headers, json_content))

    return NOT_FOUND_CODE


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


def prepare_logger():
    logger = logging.getLogger('tableApp')
    logger.setLevel(logging.INFO)

    sh = logging.StreamHandler(stdout)
    formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
    sh.setFormatter(formatter)
    logger.addHandler(sh)


if __name__ == '__main__':
    prepare_logger()

    from server import start_server
    from database import db, db_column_names

    logging.getLogger('tableApp').info(f'Starting server...')
    start_server(SERVER_HOST, SERVER_PORT)