create my toolbox -- first commit

toolbox/codes/common/db_operation.py

@@ -0,0 +1,80 @@
+import sqlite3
+import time
+from codes.common import clibs
+from pathlib import Path
+
+
+def db_init(db_file):
+    conn = sqlite3.connect(db_file, isolation_level=None, check_same_thread=False, cached_statements=2048, timeout=10.0)
+    cursor = conn.cursor()
+    cursor.execute("PRAGMA journal_mode=wal")
+    cursor.execute("PRAGMA wal_checkpoint=TRUNCATE")
+    cursor.execute("PRAGMA synchronous=normal")
+    cursor.execute("PRAGMA temp_store=memory")
+    cursor.execute("PRAGMA mmap_size=30000000000")
+    cursor.execute("PRAGMA cache_size=200000")
+    cursor.execute(
+        """
+        create table if not exists logs(
+            id integer primary key autoincrement,
+            timestamp DATETIME DEFAULT(STRFTIME('%Y-%m-%d %H:%M:%f', 'NOW', 'localtime')),
+            level text,
+            module text,
+            content text
+        )
+        """
+    )
+    cursor.execute(
+        """
+        create table if not exists users(
+            id integer primary key autoincrement,
+            timestamp DATETIME DEFAULT(STRFTIME('%Y-%m-%d %H:%M:%f', 'NOW', 'localtime')),
+            username text not null unique,
+            password text not null,
+            salt text not null
+        )
+        """
+    )
+    cursor.close()
+    conn.close()
+
+def db_lock(func):
+    def wrapper(*args, **kwargs):
+        try:
+            clibs.lock.acquire(True)
+            ret = func(*args, **kwargs)
+        finally:
+            clibs.lock.release()
+        return ret
+    return wrapper
+
+def db_backup():
+    t = time.strftime("%Y%m%d%H%M%S", time.localtime())
+    db_file = clibs.base_path / "assets/database/toolbox.db"
+    db_file_backup = clibs.base_path / f"assets/database/toolbox.{t}.db"
+    if not (db_file.exists() and db_file.is_file()):
+        db_init(db_file)
+    else:
+        db_file_backup.write_bytes(db_file.read_bytes())
+        db_dir = clibs.base_path / "assets/database"
+        db_list = [db for db in db_dir.glob("*.db")]
+        for db in sorted(db_list)[:-clibs.account["maximum_db_number"]]:
+            db.unlink()
+
+def db_conn():
+    db_file = clibs.base_path / "assets/database/toolbox.db"
+    conn = sqlite3.connect(db_file, isolation_level=None, check_same_thread=False, cached_statements=2048, timeout=3.0)
+    cursor = conn.cursor()
+    cursor.execute("PRAGMA journal_mode=wal")
+    cursor.execute("PRAGMA wal_checkpoint=TRUNCATE")
+    cursor.execute("PRAGMA synchronous=normal")
+    cursor.execute("PRAGMA temp_store=memory")
+    cursor.execute("PRAGMA mmap_size=30000000000")
+    cursor.execute("PRAGMA cache_size=200000")
+    return conn, cursor
+
+@db_lock
+def db_close(conn, cursor):
+    cursor.close()
+    conn.close()
+