Create base_sqlite.py

Akeno/utils/base_sqlite.py

@@ -0,0 +1,41 @@
+import sqlite3
+
+conn = sqlite3.connect('bot_prefix.db')
+cursor = conn.cursor()
+
+cursor.execute('''
+    CREATE TABLE IF NOT EXISTS prefixes (
+        user_id INTEGER PRIMARY KEY,
+        prefix TEXT NOT NULL
+    )
+''')
+
+conn.commit()
+conn.close()
+
+def set_prefix(user_id: int, prefix: str):
+    conn = sqlite3.connect('bot_prefix.db')
+    cursor = conn.cursor()
+
+    cursor.execute('''
+        INSERT INTO prefixes (user_id, prefix)
+        VALUES (?, ?)
+        ON CONFLICT(user_id) DO UPDATE SET prefix=excluded.prefix
+    ''', (user_id, prefix))
+
+    conn.commit()
+    conn.close()
+
+def get_prefix(user_id: int):
+    conn = sqlite3.connect('bot_prefix.db')
+    cursor = conn.cursor()
+
+    cursor.execute('SELECT prefix FROM prefixes WHERE user_id=?', (user_id,))
+    result = cursor.fetchone()
+
+    conn.close()
+
+    if result:
+        return result[0]
+    else:
+        return None