增加更多主题选择并修复了bug

bin/assets/themes/insert_db.py

@@ -1,6 +1,39 @@
-#!/usr/bin/env python3
-# -- coding:utf-8 --
-# @Author: markushammered@gmail.com
-# @Development Tool: PyCharm
-# @Create Time: 2022/2/8
-# @File Name: insert_db.py
+#!/usr/bin/env python3
+# -- coding:utf-8 --
+# @Author: markushammered@gmail.com
+# @Development Tool: PyCharm
+# @Create Time: 2022/2/8
+# @File Name: insert_db.py
+
+
+import os
+import base64
+from PIL import Image
+import sqlite3
+
+conn = sqlite3.connect('../theme.db')
+cursor = conn.cursor()
+
+
+def get_size(file):
+    img = Image.open(file)
+    return img.size
+
+
+dir_name = './rfck'
+lst = os.listdir(dir_name)
+c = 0
+for i in lst:
+    w, h = get_size(f'{dir_name}/{i}')
+    type_ = i.split('.')[-1]
+    if type_ == 'png':
+        type_ = 'jepg'
+    else:
+        type_ = 'gif'
+    with open(f'./{dir_name}/{i}', 'rb') as f:
+        base64_data = base64.b64encode(f.read())
+        base64_code = f'data:image/{type_};base64,' + base64_data.decode('utf-8')
+    cursor.execute('insert into rfck values(?, ?, ?, ?)', (c, base64_code, w, h))
+    conn.commit()
+    c += 1
+