Archived
40 lines
899 B
Python
40 lines
899 B
Python
#!/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
|
|
|