very simply version

2021-09-05 15:56:53 +08:00
parent fa851ed751
commit 25dbdcdb24
12 changed files with 373 additions and 0 deletions
--- a/game_backuper/db.py
+++ b/game_backuper/db.py
@@ -0,0 +1,99 @@
+from sqlite3 import connect
+from os.path import join
+from typing import List
+from threading import Lock
+from game_backuper.file import File
+
+
+VERSION_TABLE = '''CREATE TABLE version (
+id TEXT,
+v1 INT,
+v2 INT,
+v3 INT,
+v4 INT,
+PRIMARY KEY(id)
+);'''
+FILES_TABLE = '''CREATE TABLE files (
+id INTEGER,
+file TEXT,
+size INT,
+program TEXT,
+hash TEXT,
+PRIMARY KEY(id)
+);'''
+
+
+class Db:
+    VERSION = [1, 0, 0, 0]
+
+    def __check_database(self) -> bool:
+        self.__updateExistsTable()
+        v = self.__read_version()
+        if v is None:
+            return False
+        if v > self.VERSION:
+            raise ValueError(
+                'Database version is higher. Please update program.')
+        return True
+
+    def __create_table(self):
+        if 'version' not in self._exist_table:
+            self.db.execute(VERSION_TABLE)
+            self.__write_version()
+        if 'files' not in self._exist_table:
+            self.db.execute(FILES_TABLE)
+        self.db.commit()
+
+    def __init__(self, loc: str):
+        fn = join(loc, "data.db")
+        self.db = connect(fn, check_same_thread=False)
+        self.db.execute('VACUUM;')
+        self.db.commit()
+        ok = self.__check_database()
+        if not ok:
+            self.__create_table()
+        self._lock = Lock()
+
+    def __read_version(self) -> List[int]:
+        if 'version' not in self._exist_table:
+            return None
+        cur = self.db.execute("SELECT * FROM version WHERE id='main';")
+        for i in cur:
+            return [k for k in i if isinstance(k, int)]
+
+    def __updateExistsTable(self):
+        cur = self.db.execute('SELECT * FROM main.sqlite_master;')
+        self._exist_table = {}
+        for i in cur:
+            if i[0] == 'table':
+                self._exist_table[i[1]] = i
+
+    def __write_version(self):
+        if self.__read_version() is None:
+            self.db.execute('INSERT INTO version VALUES (?, ?, ?, ?, ?);',
+                            tuple(['main'] + self.VERSION))
+        else:
+            self.db.execute(
+                "UPDATE version SET v1=?, v2=?, v3=?, v4=? WHERE id='main';",
+                tuple(self.VERSION))
+        self.db.commit()
+
+    def add_file(self, f: File):
+        with self._lock:
+            self.db.execute('INSERT INTO files (file, size, program, hash) VALUES (?, ?, ?, ?);',  # noqa: E501
+                            (f.file, f.size, f.program, f.hash))
+            self.db.commit()
+
+    def get_file(self, prog: str, file: str) -> File:
+        with self._lock:
+            cur = self.db.execute(
+                'SELECT * FROM files WHERE program=? AND file=?;', (prog,
+                                                                    file))
+            for i in cur:
+                return File(*i)
+
+    def set_file(self, id: int, size: int, hash: str):
+        with self._lock:
+            self.db.execute('UPDATE files SET size=?, hash=? WHERE id=?;',
+                            (id,))
+            self.db.commit()