diff options
Diffstat (limited to 'base.py')
-rw-r--r-- | base.py | 23 |
1 files changed, 23 insertions, 0 deletions
@@ -0,0 +1,23 @@ +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 |