mirror of
https://github.com/Kakune55/ComiPy.git
synced 2025-05-06 18:29:26 +08:00
87 lines
2.8 KiB
Python
87 lines
2.8 KiB
Python
import shutil, os, zipfile, io, cv2, numpy as np
|
|
|
|
import db.file, app_conf
|
|
|
|
app_conf = app_conf.conf()
|
|
|
|
|
|
def init():
|
|
paths = ("inputdir", "storedir", "tmpdir")
|
|
for path in paths:
|
|
try:
|
|
os.makedirs(app_conf.get("file", path))
|
|
except Exception as e:
|
|
print(e)
|
|
|
|
|
|
def auotLoadFile():
|
|
fileList = os.listdir(app_conf.get("file", "inputdir"))
|
|
for item in fileList:
|
|
if zipfile.is_zipfile(
|
|
app_conf.get("file", "inputdir") + "/" + item
|
|
): # 判断是否为压缩包
|
|
with zipfile.ZipFile(
|
|
app_conf.get("file", "inputdir") + "/" + item, "r"
|
|
) as zip_ref:
|
|
db.file.new(item, len(zip_ref.namelist())) # 添加数据库记录 移动到存储
|
|
shutil.move(
|
|
app_conf.get("file", "inputdir") + "/" + item,
|
|
app_conf.get("file", "storedir") + "/" + item,
|
|
)
|
|
print("已添加 " + item)
|
|
else:
|
|
print("不符合条件 " + item)
|
|
|
|
|
|
def raedZip(bookid: str, index: int):
|
|
bookinfo = db.file.searchByid(bookid)
|
|
zippath = app_conf.get("file", "storedir") + "/" + bookinfo[0][2]
|
|
|
|
try:
|
|
# 创建一个ZipFile对象
|
|
with zipfile.ZipFile(zippath, "r") as zip_ref:
|
|
# 获取图片文件列表
|
|
image_files = [
|
|
file
|
|
for file in zip_ref.namelist()
|
|
if file.lower().endswith((".png", ".jpg", ".jpeg", ".gif", ".bmp", ".webp"))
|
|
]
|
|
|
|
if not image_files:
|
|
return "not imgfile in zip", ""
|
|
|
|
if int(index) > len(image_files):
|
|
return "404 not found", ""
|
|
|
|
# 假设我们只提取图片文件
|
|
image_filename = image_files[int(index)]
|
|
|
|
# 读取图片数据
|
|
image_data = zip_ref.read(image_filename)
|
|
zip_ref.close()
|
|
return image_data, image_filename
|
|
|
|
except zipfile.BadZipFile: # 异常处理
|
|
return "Bad ZipFile", ""
|
|
except Exception as e:
|
|
return str(e), ""
|
|
|
|
|
|
def thumbnail(input, minSize: int = 600, encode:str="webp"):
|
|
img = cv2.imdecode(np.frombuffer(input, np.uint8), cv2.IMREAD_COLOR)
|
|
height = img.shape[0] # 图片高度
|
|
width = img.shape[1] # 图片宽度
|
|
if minSize < np.amin((height,width)):
|
|
if height > width:
|
|
newshape = (minSize, int(minSize / width * height))
|
|
else:
|
|
newshape = (int(minSize / height * width), minSize)
|
|
img = cv2.resize(img, newshape)
|
|
if encode == "webp":
|
|
success, encoded_image = cv2.imencode(".webp", img, [cv2.IMWRITE_WEBP_QUALITY, 75])
|
|
elif encode == "jpg" or "jpeg":
|
|
success, encoded_image = cv2.imencode(".jpg", img, [cv2.IMWRITE_JPEG_QUALITY, 75])
|
|
else:
|
|
return input
|
|
return encoded_image.tobytes()
|