feat(database): moving db migration scripts in lib src

Author: lx-dev

src/astraeus_common/io/database.py

@@ -7,6 +7,7 @@
 from psycopg2.extras import DictCursor
 from yoyo import read_migrations, get_backend
 
+PING_QUERY = 'SELECT 1'
 
 class Database:
     _db_host: str
@@ -22,20 +23,20 @@ def __init__(self, db_host: str, db_port: str, db_user: str, db_name: str, db_pa
         self._db_name = db_name
         self._db_password = db_password
         self._log = structlog.get_logger()
-        self.__try_connection()
+        self.__try_connection(PING_QUERY)
         self.__apply_migration()
+        self.__try_connection('SELECT * FROM ASTRAEUS.BODY')
 
     def __apply_migration(self):
         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('./db')
         backend.apply_migrations(backend.to_apply(migrations))
 
-    def __try_connection(self):
+    def __try_connection(self, query: str):
         with self.__db_connection() as connection:
             with connection.cursor(cursor_factory=DictCursor) as cursor:
-                cursor.execute('SELECT 1')
-                cursor.execute('SELECT * FROM ASTRAEUS.BODY')
+                cursor.execute(query)
 
     def __db_connection(self):
         try:

src/astraeus_common/io/db/001.initial-db-creation.sql

@@ -0,0 +1,33 @@
+CREATE SCHEMA IF NOT EXISTS "astraeus";
+
+CREATE TABLE IF NOT EXISTS "astraeus"."system"
+(
+    "key"            jsonb PRIMARY KEY,
+    "name"           text,
+    "coordinates"    jsonb,
+    "require_permit" boolean,
+    "information"    jsonb,
+    "update_time"    timestamp,
+    "primary_star"   jsonb
+);
+
+CREATE TABLE IF NOT EXISTS "astraeus"."body"
+(
+    "key"                    jsonb PRIMARY KEY,
+    "system_key"             jsonb,
+    "name"                   text,
+    "type"                   text,
+    "sub_type"               text,
+    "discovery"              jsonb,
+    "update_time"            timestamp,
+    "materials"              jsonb,
+    "solid_composition"      jsonb,
+    "atmosphere_composition" jsonb,
+    "parents"                jsonb,
+    "belts"                  jsonb,
+    "rings"                  jsonb,
+    "properties"             jsonb
+);
+
+ALTER TABLE "astraeus"."body"
+    ADD FOREIGN KEY (system_key) REFERENCES "astraeus"."system" (key);

src/astraeus_common/io/db/002.add-sync-edsm-state-following.sql

@@ -0,0 +1,10 @@
+-- depends: 001.initial-db-creation
+
+CREATE TABLE IF NOT EXISTS "astraeus"."sync_state"
+(
+    "key"            jsonb PRIMARY KEY,
+    "sync_date"      timestamp not null,
+    "type"           text      not null,
+    "sync_hash"      text,
+    "previous_state" jsonb
+);

src/astraeus_common/io/db/003.add-eddn-messages.sql

@@ -0,0 +1,11 @@
+-- depends: 001.initial-db-creation
+
+CREATE TABLE IF NOT EXISTS "astraeus"."eddn_message"
+(
+    "id"        uuid PRIMARY KEY default uuid_generate_v4(),
+    "schema"    text      not null,
+    "header"    jsonb     not null,
+    "message"   jsonb     not null,
+    "recv_date" timestamp not null,
+    "sync_date" timestamp
+);

src/astraeus_common/models/body.py

@@ -18,6 +18,59 @@ class Body:
     _rings: dict
     _properties: dict
 
+    BODY_SELECT_BY_KEY = '''
+    select key, system_key, name, type,
+    sub_type, discovery, update_time, materials,
+    solid_composition, atmosphere_composition, parents, belts,
+    rings, properties
+    from body
+    where key = %(key)s
+    '''
+
+    BODY_SELECT_BY_SYSTEM_KEY = '''
+    select key, system_key, name, type,
+    sub_type, discovery, update_time, materials,
+    solid_composition, atmosphere_composition, parents, belts,
+    rings, properties
+    from body
+    where system_key = %(system_key)s
+    '''
+
+    BODY_INSERT = '''
+    insert into body
+    (key, system_key, name, type,
+    sub_type, discovery, update_time, materials,
+    solid_composition, atmosphere_composition, parents,
+    belts, rings, properties)
+    values 
+    (%(key)s, %(system_key)s, %(name)s, %(type)s,
+    %(sub_type)s, %(discovery)s, %(update_time)s, %(materials)s,
+    %(solid_composition)s, %(atmosphere_composition)s, %(parents)s,
+    %(belts)s, %(rings)s, %(properties)s)
+    '''
+
+    BODY_UPDATE_BY_KEY = '''
+    update body
+    set system_key = %(system_key)s,
+        name = %(name)s,
+        type = %(type)s,
+        sub_type = %(sub_type)s,
+        discovery = %(discovery)s,
+        update_time = %(update_time)s,
+        materials = %(materials)s,
+        solid_composition = %(solid_composition)s,
+        atmosphere_composition = %(atmosphere_composition)s,
+        parents = %(parents)s,
+        belts = %(belts)s,
+        rings = %(rings)s,
+        properties = %(properties)s
+    where key = %(key)s
+    '''
+
+    BODY_DELETE_BY_KEY = '''
+    delete from body where key = %(key)s
+    '''
+
     @property
     def key(self) -> dict:
         return self._key