v1.1

kurocore/database.py

@@ -0,0 +1,29 @@
+import logging
+from asyncio import get_event_loop
+from peewee_async import PostgresqlDatabase, Manager
+
+from kurocore.logger import BotLogger
+
+
+class Database:
+    db: PostgresqlDatabase = PostgresqlDatabase(None)
+    manager: Manager
+
+    def __init__(self, settings):
+        db_settings = settings.database
+
+        driver_name = db_settings.driver
+        host = db_settings.host
+        name = db_settings.name
+        user = db_settings.user
+        password = db_settings.password
+        port = db_settings.port
+
+        if driver_name.lower() not in ('postgres', 'postgresql', 'psql'):
+            raise
+
+        Database.db.init(name, user=user, password=password, host=host, port=port)
+        loop = get_event_loop()
+        Database.manager = Manager(Database.db, loop=loop)
+        Database.manager.allow_sync = logging.ERROR
+        BotLogger.log.info('Connected to database')