import toml
import os
import db_model as db
import requests
import json
from peewee import SqliteDatabase
import datetime


def config_laden():
    configfile = os.path.join(SKRIPTPFAD, "config.toml")
    with open(configfile) as file:
        return toml.loads(file.read())


SKRIPTPFAD = os.path.abspath(os.path.dirname(__file__))
CONFIG = config_laden()


def headers_erstellen():
    headers = {"Authorization": "Bearer {token}".format(token=CONFIG["token"]),
               "Prefer": "resolution=merge-duplicates"}
    return headers


def tabelle_auslesen(tabelle, headers):
    url = f"{CONFIG['url']}{tabelle}"
    r = requests.get(url, headers=headers)
    erg = json.loads(r.text)
    return erg


def main():
    db.DATABASE.initialize(SqliteDatabase(os.path.join(SKRIPTPFAD, "corona.db3")))
    db.create_tables()
    headers = headers_erstellen()

    erg = tabelle_auslesen(CONFIG['tablename_countryliste'], headers)
    for data_dict in erg:
        db.CoronaCountry.create(**data_dict)

    erg = tabelle_auslesen(CONFIG["tablename_daten"], headers)
    for data_dict in erg:
        data_dict["ts"] = datetime.datetime.strptime(data_dict["ts"], "%Y-%m-%dT%H:%M:%S")
        db.CoronaDaten.create(**data_dict)


main()