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)