添加了更多的主题, 修复了输入不存在主题时报错

This commit is contained in:
MarkusJoe
2022-02-09 16:20:53 +08:00
parent 494cc7fea2
commit baf99a905c
317 files changed
+111 -324

No files matched your search

+25
View File
@@ -0,0 +1,25 @@
import base64
from PIL import Image
import os
import sqlite3
conn = sqlite3.connect('../assets/theme.db')
cursor = conn.cursor()
file_list = os.listdir('../assets/themes/')
for i in file_list:
img = Image.open(f'../assets/themes/{i}/0.gif')
w = img.width
h = img.height
img.close()
for n in range(0, 10):
with open(f'../assets/themes/{i}/{n}.gif', 'rb') as f:
base64_data = base64.b64encode(f.read())
base64_code = f'data:image/gif;base64,' + base64_data.decode('utf-8')
try:
cursor.execute('insert into %(name)s values(?, ?, ?, ?)' % {'name': i}, (n, base64_code, w, h))
conn.commit()
except Exception:
cursor.execute('create table %(name)s (k text, v text, w integer, h integer)' % {'name': i})