from peewee import MySQLDatabase, Model
import yaml


with open("config.yaml", "r") as yamlfile:
    config = yaml.load(yamlfile, Loader=yaml.CLoader)
    print("Read config successful")
    config = config


db_conf = config["database"]
db = MySQLDatabase(
    db_conf["name"],
    user=db_conf["username"],
    password=db_conf["password"],
    host=db_conf["host"],
)
db.connect()


class BaseModel(Model):
    class Meta:
        database = db