Update Akeno/utils/base_sqlite.py

Akeno/utils/base_sqlite.py

@@ -1,41 +1,33 @@
-import sqlite3
+import sqlitecloud
+import asyncio
+from config import SQLITE_CLOUD_URL
 
-conn = sqlite3.connect('bot_prefix.db')
+conn = sqlitecloud.connect(SQLITE_CLOUD_URL)
 cursor = conn.cursor()
 
 cursor.execute('''
-    CREATE TABLE IF NOT EXISTS prefixes (
+    CREATE TABLE IF NOT EXISTS prefixes_dev (
         user_id INTEGER PRIMARY KEY,
         prefix TEXT NOT NULL
     )
 ''')
-
 conn.commit()
-conn.close()
-
-def set_prefix_in_db(user_id: int, prefix: str):
-    conn = sqlite3.connect('bot_prefix.db')
-    cursor = conn.cursor()
 
+async def set_prefix_in_db(user_id: int, prefix: str):
     cursor.execute('''
-        INSERT INTO prefixes (user_id, prefix)
+        INSERT INTO prefixes_dev (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()
 
+async def get_prefix(user_id: int):
     cursor.execute('SELECT prefix FROM prefixes WHERE user_id=?', (user_id,))
     result = cursor.fetchone()
 
-    conn.close()
-
     if result:
         return result[0]
     else:
-        return None
+        return None
+
+conn.close()