init: upload

src/utils/convert.py

@@ -0,0 +1,51 @@
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/8/30
+# @File Name: convert.py
+
+
+import os
+import base64
+import sqlite3
+import asyncio
+
+
+class SQLite:
+    def __init__(self):
+        self.conn = sqlite3.connect('../db/data.sqlite')
+        self.cursor = self.conn.cursor()
+
+    def __del__(self):
+        self.conn.commit()
+        self.cursor.close()
+        self.conn.close()
+
+    async def insert(self, _id: str, b64: str, width: int, height: int) -> bool:
+        self.cursor.execute('insert into image values ("%s", "%s", %s, %s)' % (_id, b64, width, height))
+        return True
+
+
+async def convert():
+    dirs = os.listdir('../static')
+    for i in dirs:
+        files = os.listdir(f'../static/{i}')
+        for b in files:
+            with open(f'../static/{i}/{b}', 'rb') as fp:
+                if i == 'moebooru':
+                    width = 45
+                    height = 100
+                elif i == 'lewd':
+                    width = 45
+                    height = 100
+                elif i == 'lisu':
+                    width = 66
+                    height = 152
+                else:
+                    width = 68
+                    height = 150
+                await SQLite().insert(f'{i}/{b}', base64.b64encode(fp.read()).decode('utf-8'), width, height)
+
+
+asyncio.run(convert())