fix(database): fix migration code usage

Author: lx-dev

src/astraeus_common/io/database.py

@@ -1,22 +1,27 @@
+import os
 import sys
 from typing import List
 
 import psycopg2
 import structlog
-import yoyo
 from psycopg2.extras import DictCursor
 from yoyo import read_migrations, get_backend
 
+from . import db
+
 PING_QUERY = 'SELECT 1'
 
+
 class Database:
     _db_host: str
     _db_port: str
     _db_user: str
     _db_name: str
     _db_password: str
+    _migration_folder: str
 
     def __init__(self, db_host: str, db_port: str, db_user: str, db_name: str, db_password: str):
+        self._migration_folder = os.path.dirname(os.path.abspath(db.__file__))
         self._db_host = db_host
         self._db_port = db_port
         self._db_user = db_user
@@ -28,9 +33,11 @@ def __init__(self, db_host: str, db_port: str, db_user: str, db_name: str, db_pa
         self.__try_connection('SELECT * FROM ASTRAEUS.BODY')
 
     def __apply_migration(self):
+        self._log.debug('applying yoyo migration')
         backend = get_backend(f'postgresql://{self._db_user}:{self._db_password}@'
                               f'{self._db_host}:{self._db_port}/{self._db_name}')
-        migrations = read_migrations('./db')
+        migrations = read_migrations(self._migration_folder)
+        print(migrations)
         backend.apply_migrations(backend.to_apply(migrations))
 
     def __try_connection(self, query: str):
@@ -75,4 +82,3 @@ def exec_db_write(self, query: str, params: dict) -> None:
         except psycopg2.DatabaseError as error:
             self._log.error(f'Error occur on write of {log_query} - {error}')
             connection.rollback()
-

test/test_db_migration.py

@@ -0,0 +1,4 @@
+from astraeus_common.io.database import Database
+
+if __name__ == '__main__':
+    Database('localhost', '5432', 'astraeus', 'astraeus-db', 'astraeus')