1 files changed, 15 insertions, 63 deletions

diff --git a/src/home/bot/store.py b/src/home/bot/store.py
index aeedc47..e655d8f 100644
--- a/src/home/bot/store.py
+++ b/src/home/bot/store.py
@@ -1,80 +1,32 @@
-import sqlite3
-import os.path
-import logging
+from ..database.sqlite import SQLiteBase
 
-from ..config import config
-
-logger = logging.getLogger(__name__)
-
-
-def _get_database_path() -> str:
-    return os.path.join(os.environ['HOME'], '.config', config.app_name, 'bot.db')
-
-
-class Store:
-    SCHEMA_VERSION = 1
 
+class Store(SQLiteBase):
     def __init__(self):
-        self.sqlite = sqlite3.connect(_get_database_path(), check_same_thread=False)
-
-        sqlite_version = self._get_sqlite_version()
-        logger.info(f'SQLite version: {sqlite_version}')
-
-        schema_version = self._get_schema_version()
-        logger.info(f'Schema version: {schema_version}')
-
-        if schema_version < 1:
-            self._database_init()
-        elif schema_version < Store.SCHEMA_VERSION:
-            self._database_upgrade(Store.SCHEMA_VERSION)
-
-    def __del__(self):
-        if self.sqlite:
-            self.sqlite.commit()
-            self.sqlite.close()
-
-    def _get_sqlite_version(self) -> str:
-        cursor = self.sqlite.cursor()
-        cursor.execute("SELECT sqlite_version()")
-
-        return cursor.fetchone()[0]
-        
-    def _get_schema_version(self) -> int:
-        cursor = self.sqlite.execute('PRAGMA user_version')
-        return int(cursor.fetchone()[0])
-
-    def _set_schema_version(self, v) -> None:
-        self.sqlite.execute('PRAGMA user_version={:d}'.format(v))
-        logger.info(f'Schema set to {v}')
-
-    def _database_init(self) -> None:
-        cursor = self.sqlite.cursor()
-        cursor.execute("""CREATE TABLE IF NOT EXISTS users (
-            id INTEGER PRIMARY KEY,
-            lang TEXT NOT NULL
-        )""")
-        self.sqlite.commit()
-        self._set_schema_version(1)
-
-    def _database_upgrade(self, version: int) -> None:
-        # do the upgrade here
+        super().__init__()
 
-        # self.sqlite.commit()
-        self._set_schema_version(version)
+    def schema_init(self, version: int) -> None:
+        if version < 1:
+            cursor = self.cursor()
+            cursor.execute("""CREATE TABLE IF NOT EXISTS users (
+                id INTEGER PRIMARY KEY,
+                lang TEXT NOT NULL
+            )""")
+            self.commit()
 
     def get_user_lang(self, user_id: int, default: str = 'en') -> str:
-        cursor = self.sqlite.cursor()
+        cursor = self.cursor()
         cursor.execute('SELECT lang FROM users WHERE id=?', (user_id,))
         row = cursor.fetchone()
 
         if row is None:
             cursor.execute('INSERT INTO users (id, lang) VALUES (?, ?)', (user_id, default))
-            self.sqlite.commit()
+            self.commit()
             return default
         else:
             return row[0]
 
     def set_user_lang(self, user_id: int, lang: str) -> None:
-        cursor = self.sqlite.cursor()
+        cursor = self.cursor()
         cursor.execute('UPDATE users SET lang=? WHERE id=?', (lang, user_id))
-        self.sqlite.commit()
\ No newline at end of file
+        self.commit()