Initial commit

1 files changed, 23 insertions, 0 deletions

diff --git a/base.py b/base.py
new file mode 100644
index 0000000..1d2e86b
--- /dev/null
+++ b/base.py
@@ -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