Python - 批量制做图种

时间 2019-12-05

标签 python 批量制做栏目 Python 繁體版

原文原文链接

什么是图种，就是一种包含压缩文件的图片。改后缀名能够正常解压，改成图片格式又能够正常预览。
简单的说就是把图片与压缩文件用二进制的方式合并起来。图片数据在前，压缩文件数据在后。python

虽然Batch脚本一条语句就能够生成，可是咱们要讲究批量和自动化嘛，因此用python来实现。多线程

1. 设置任务列表

建立一个Import文件夹，将要处理的图片和压缩包放进去。
与压缩包同名的图片将优先和压缩包合并，没有则与默认图片default.*合并，没有放入默认图片则自动生成黑色图片与之合并。
支持的图片格式：*.jpg，*.png，*.bmp
支持的压缩格式：*.rar，*.zip，*.7z

# -*- coding: utf-8 -*-
# python 3.7.2

import os
import io

from timeit import timeit

#
from PIL import Image
from concurrent.futures import ThreadPoolExecutor

# ---
ALL_TASK= []
IMPORT_FOLDER= 'Import/'
EXPORT_FOLDER= 'Export/'
# ---


def SetTaskList():
    global ALL_TASK
    ALL_TASK= []
    if not os.path.exists(EXPORT_FOLDER):
        os.makedirs(EXPORT_FOLDER)
    
    files= os.listdir(IMPORT_FOLDER)
    default_img= ['default'+i for i in ('.jpeg','.jpg','.png','.bmp')]
    ImgM= None
    ext= '.jpeg'
    # 判断是否存在默认图，没有则自动建立
    for img in default_img:
        if img in files:
            ImgM= Image.open(IMPORT_FOLDER+img)
            ext= os.path.splitext(img)[1]
            ext= '.jpeg' if ext=='.jpg' else ext
            files.remove(img)
            break
    if not ImgM:
        ImgM= Image.new('L', (32,32), (60,))
        
    # 第一行写入图片扩展名
    ImgM_Bytes= io.BytesIO()
    ImgM_Bytes.write(ext.encode()+ b'\n')
    ImgM.save(ImgM_Bytes, ext[1:])

    # 判断是否有对应的'同名字'图片，没有则与默认图合成
    for f in files:
        split= os.path.splitext(f)
        if split[1].lower() in ('.rar','.zip','.7z'):
            imgs= [split[0]+i for i in ('.jpeg','.jpg','.png','.bmp')]
            cover_image= None
            for img in imgs:
                if img in files:
                    cover_image= img
                    break
            
            ALL_TASK.append((f,cover_image or ImgM_Bytes))

2. 多线程分配任务

def AssignTask():
    pool = ThreadPoolExecutor()
    results = list(pool.map(CreateImgSeed, ALL_TASK))
    pool.shutdown()
    
    print ('{}个图种制做完毕！'.format(len(ALL_TASK)))

3. 建立图种

用二进制格式打开
最终输出文件以压缩包名字命名，保持对应图片扩展名

def CreateImgSeed(task):
    pack, img= task
    with open(IMPORT_FOLDER+pack,'rb') as p:
        if type(img)==str:
            with open(IMPORT_FOLDER+img, 'rb') as i:
                ext= os.path.splitext(img)[1]
                img_data= i.read()
        else:
            datas= img.getvalue()
            i= datas.index(b'\n')
            ext= datas[:i].decode()
            img_data= datas[i+1:]
                
        # 最终输出文件以压缩包名字命名，保持对应图片扩展名
        out_file= open(EXPORT_FOLDER + os.path.splitext(pack)[0] + ext, 'wb')
        out_file.write(img_data)
        out_file.write(p.read())
        out_file.close()

完整代码

ImgSeed.py

# -*- coding: utf-8 -*-
# python 3.7.2

import os
import io

from timeit import timeit

#
from PIL import Image
from concurrent.futures import ThreadPoolExecutor

# ---
ALL_TASK= []
IMPORT_FOLDER= 'Import/'
EXPORT_FOLDER= 'Export/'
# ---


def SetTaskList():
    global ALL_TASK
    ALL_TASK= []
    if not os.path.exists(EXPORT_FOLDER):
        os.makedirs(EXPORT_FOLDER)
    
    files= os.listdir(IMPORT_FOLDER)
    default_img= ['default'+i for i in ('.jpeg','.jpg','.png','.bmp')]
    ImgM= None
    ext= '.jpeg'
    # 判断是否存在默认图，没有则自动建立
    for img in default_img:
        if img in files:
            ImgM= Image.open(IMPORT_FOLDER+img)
            ext= os.path.splitext(img)[1]
            ext= '.jpeg' if ext=='.jpg' else ext
            files.remove(img)
            break
    if not ImgM:
        ImgM= Image.new('L', (32,32), (60,))
        
    # 第一行写入图片扩展名
    ImgM_Bytes= io.BytesIO()
    ImgM_Bytes.write(ext.encode()+ b'\n')
    ImgM.save(ImgM_Bytes, ext[1:])

    # 判断是否有对应的'同名字'图片，没有则与默认图合成
    for f in files:
        split= os.path.splitext(f)
        if split[1].lower() in ('.rar','.zip','.7z'):
            imgs= [split[0]+i for i in ('.jpeg','.jpg','.png','.bmp')]
            cover_image= None
            for img in imgs:
                if img in files:
                    cover_image= img
                    break
            
            ALL_TASK.append((f,cover_image or ImgM_Bytes))
    
    
def CreateImgSeed(task):
    pack, img= task
    with open(IMPORT_FOLDER+pack,'rb') as p:
        if type(img)==str:
            with open(IMPORT_FOLDER+img, 'rb') as i:
                ext= os.path.splitext(img)[1]
                img_data= i.read()
        else:
            datas= img.getvalue()
            i= datas.index(b'\n')
            ext= datas[:i].decode()
            img_data= datas[i+1:]
                
        # 最终输出文件以压缩包名字命名，保持对应图片扩展名
        out_file= open(EXPORT_FOLDER + os.path.splitext(pack)[0] + ext, 'wb')
        out_file.write(img_data)
        out_file.write(p.read())
        out_file.close()


def AssignTask():
    pool = ThreadPoolExecutor()
    results = list(pool.map(CreateImgSeed, ALL_TASK))
    pool.shutdown()
    
    print ('{}个图种制做完毕！'.format(len(ALL_TASK)))
    
    
def start():
    SetTaskList()
    sec = timeit(lambda:AssignTask(),number=1)
    print ('Time used: {0:.3f} sec\n'.format(sec))
    s= input('按回车键退出...\n')
    
    
    
if __name__ == '__main__':
    while True:
        s= input('开始制做图种(Y/N): ')
        if s.lower()== 'y':
            start()
            break
        elif not s or s.lower()== 'n':
            break