Create sqlite3.py

sqlite3.py

@@ -0,0 +1,72 @@
+#!/usr/bin/env python3
+# -*- encoding: utf-8 -*-
+# Author:  MoeClub.org
+
+import aiosqlite
+import asyncio
+
+class SQLite3:
+    def __init__(self, name, table):
+        aiosqlite.core.LOG.disabled = True
+        self.table = table
+        self._path = os.path.join(os.path.dirname(os.path.abspath(__file__)), name)
+        self._queue = asyncio.Queue(maxsize=1)
+        self._queueInit(queue=self._queue)
+        print("Data:", self._path)
+
+    def _queueInit(self, queue):
+        try:
+            assert queue.maxsize > 0
+            for _ in range(queue.maxsize):
+                queue.put_nowait(None)
+        except:
+            pass
+
+    async def execute(self, sql, write=False):
+        value = db = cursor = field = None
+        try:
+            if write is True:
+                await self._queue.get()
+                db = await aiosqlite.connect(database=self._path, check_same_thread=False, timeout=60)
+                cursor = await db.execute(sql)
+                await db.commit()
+            else:
+                db = await aiosqlite.connect(database=self._path, check_same_thread=False, timeout=60)
+                cursor = await db.execute(sql)
+                if cursor.description:
+                    field = [i[0] for i in cursor.description]
+                records = await cursor.fetchall()
+                if field:
+                    value = [dict(zip(field, i)) for i in records]
+        except Exception as e:
+            raise Exception(e)
+        finally:
+            if cursor is not None:
+                await cursor.close()
+            if db is not None:
+                await db.close()
+            if write is True:
+                await self._queue.put(None)
+        return value
+
+    async def Init(self, tbl='(row0 text primary key, row1 text, row2 text)'):
+        try:
+            vacuum = await self.execute('''PRAGMA auto_vacuum;''')
+            if vacuum[0]['auto_vacuum'] == 0:
+                await self.execute('''PRAGMA auto_vacuum = 1;''', True)
+                await self.execute('''PRAGMA encoding = "UTF-8";''', True)
+                await self.execute('''PRAGMA journal_mode = WAL;''', True)
+                await self.execute('''PRAGMA temp_store = MEMORY;''', True)
+                await self.execute('''PRAGMA synchronous = NORMAL;''', True)
+                await self.execute('''PRAGMA page_size = 4096;''', True)
+                create_tbl = '''create table %s %s''' % (self.table, tbl)
+                await self.execute(create_tbl, True)
+        except Exception as e:
+            if not 'already exists' in str(e):
+                print("init:", e)
+
+if __name__ == "__main__":
+    SQL = SQLite3("MoeClub.db", "MoeClub")
+    asyncio.get_event_loop().run_until_complete(SQL.Init())
+    asyncio.run(SQL.execute("select * from MoeClub;"))
+