imagesmake.py

#!/usr/bin/env python3
#  -*- coding: utf-8 -*-

import tkinter as tk
from tkinter import messagebox
from tkinter.filedialog import askdirectory, askopenfilename
import threading, time, re, os, requests, math
from PIL import Image, ImageStat

def thread_run(func):
    def wraper(*args, **kwargs):
        t = threading.Thread(target=func, args=args, kwargs=kwargs)
        t.daemon = True
        t.start()

    return wraper


class ImageManager():
    def __init__(self, file=None, io=None, im=None):
        self.mean = None
        self.im = None
        self.path = None

        if file:
            if not os.path.exists(file):
                raise Exception("文件不存在的!")

            self.path = file
            self.im = Image.open(file)
        if im:
            self.im = im

    def getMean(self):
        if not self.mean:
            self.mean = ImageStat.Stat(self.im.convert('L')).mean[0]

        return self.mean

    def resize(self, size=(1, 1)):
        return self.im.resize(size, Image.ANTIALIAS)
        # return self.im.resize(size)


class BasicFrame(object):
    """Frame基类，实现grid方法"""

    def __init__(self, root=None, **args):
        super(BasicFrame, self).__init__()
        self.root = None
        if root:
            self.root = tk.Frame(root)
        if args and 'root' in args.keys:
            self.root = tk.Frame(args.root)

        if not self.root:
            raise Exception("no root")

    def grid(self, row='default', column='default', **args):
        if row == 'default' or column == 'default':
            if args:
                self.root.grid(**args)
        else:
            self.root.grid(row=row, column=column)


class SearchFrame(BasicFrame):
    """docstring for SearchFrame"""

    def __init__(self, root=None, **args):
        super(SearchFrame, self).__init__(root=root, **args)
        self.entry = tk.StringVar()
        self.cnt = 0
        self.num = tk.StringVar()
        self.num.set(300)
        self.name = 0
        self.flag = False

        row = 0

        tk.Label(self.root, text='----搜索下载图片区域---').grid(row=row, column=1)

        row += 1
        tk.Label(self.root,width=10, justify='right' ,text='关键词').grid(row=row, column=0)
        tk.Entry(self.root, textvariable=self.entry, width=24).grid(row=row, column=1)
        tk.Button(self.root, width=9,text='下载图片', command=self.serach).grid(row=row, column=2)
        tk.Label(self.root, width=30, justify='right' , text='直接输入关键词，调用百度图片').grid(row=row, column=3)

        row += 1
        tk.Label(self.root, width=9, justify='right' , text='爬取量').grid(row=row, column=0)
        self.button = tk.Entry(self.root, textvariable=self.num, width=24).grid(row=row, column=1)

        row += 1

        tk.Label(self.root,   text='---生成图片设置区域---').grid(row=row, column=1)



    @thread_run
    def serach(self):
        _input = self.entry.get().replace(' ', '')
        try:
            num = int(self.num.get())
        except Exception as e:
            num = 0
            print(e)

        if self.flag:
            messagebox.showinfo("警告", "尚有后台下载任务进行中")
        elif _input == '' or num == 0:
            messagebox.showinfo("警告", "关键词或爬取量不能为空")
        else:
            self.entry.set('')
            self.prepare(_input)

    def prepare(self, keywords):
        self.flag = True
        self.keywords = keywords
        self.base_url = 'http://image.baidu.com/search/flip?tn=baiduimage&ie=utf-8&word=' + self.keywords + '&pn='

        path = "%s\\%s" % (os.path.abspath('.'), keywords)
        if not os.path.exists(path):
            os.mkdir(path)

        t = 0
        self.cnt = 0
        fail = 0
        while True:
            if fail > 10:
                if self.flag:
                    self.flag = False
                    messagebox.showinfo("提示", "下载完成，但该词条下无足够数量图片!!!")
                return

            if (self.cnt >= int(self.num.get())):
                if self.flag:
                    self.flag = False
                    messagebox.showinfo("提示", "下载完成")
                return
            try:
                url = self.base_url + str(t)
                result = requests.get(url, timeout=10)
                pic_url = re.findall('"objURL":"(.*?)",', result.text, re.S)
                if len(pic_url) == 0:
                    fail += 1
                else:
                    self.downloadPic(pic_url, path)
            except Exception as e:
                print(e)
                time.sleep(3)
            finally:
                t += 60
                time.sleep(1)

        # print("图片下载完成")

    def downloadPic(self, urls, path):

        # 每过三秒检查一次当前正在运行的线程数，超标则沉睡等待线程结束
        while threading.activeCount() > 200:
            print("线程超标---%s" % threading.activeCount())
            time.sleep(3)

        for url in urls:
            if url:
                self.download(url, path)

    @thread_run
    def download(self, url, path):
        try:
            if lock.acquire():
                if (self.cnt >= int(self.num.get())):
                    lock.release()
                    return
                self.cnt += 1
                self.name += 1
                filename = "%s\\%s_%s.%s" % (path, self.keywords, self.name, url.split('.')[-1])
                lock.release()

                res = requests.get(url, timeout=10)
                with open(filename, 'wb') as f:
                    f.write(res.content)
            # 下载完后检查是否完成下载
            if lock.acquire():
                if (self.cnt >= int(self.num.get())):
                    lock.release()
                    if self.flag:
                        self.flag = False
                        messagebox.showinfo("提示", "下载完成")
                else:
                    lock.release()
        except Exception as e:
            print(e)
            if lock.acquire():
                self.cnt -= 1
                lock.release()


class GenerateFrame(BasicFrame):
    """docstring for GenerateFrame"""

    def __init__(self, root=None, **args):
        super(GenerateFrame, self).__init__(root=root, **args)
        self.path = tk.StringVar()  # 用于保存输入源目录
        # self.path = path  # 用于保存输入源目录
        self.output = tk.StringVar()  # 用于保存输出对象路径
        self.resources = []
        self.orign = None
        self.outim = None
        self.busy = False
        self.load = False

        self.width = tk.StringVar()
        self.row = tk.StringVar()
        self.column = tk.StringVar()
        self.images = []

        self.row.set('60')
        self.column.set('60')
        self.width.set('30')

        row = 0
        tk.Label(self.root,width=10, justify='right' , text='输入源路径').grid(row=row, column=0)
        tk.Entry(self.root, textvariable=self.path, width=24).grid(row=row, column=1)
        tk.Button(self.root, text='载入输入源', command=self.setResource).grid(row=row, column=2)
        tk.Label(self.root,  width=30, justify='right' , text='选择图片来源文件夹').grid(row=row, column=3)


        row += 1
        tk.Label(self.root, width=10, justify='right' ,text='输出源路径').grid(row=row, column=0)
        tk.Entry(self.root, textvariable=self.output, width=24).grid(row=row, column=1)
        tk.Button(self.root, text='选择输出源', command=self.setOrign).grid(row=row, column=2)
        tk.Label(self.root, text='选择要生成的图片源').grid(row=row, column=3)

        row += 1
        tk.Label(self.root, width=10, justify='right' ,text='图片宽度').grid(row=row, column=0)
        tk.Entry(self.root, textvariable=self.width, width=24).grid(row=row, column=1)
        tk.Button(self.root, text='生成图片', command=self.generate).grid(row=row, column=2)
        tk.Label(self.root, text='填充的图片的宽度').grid(row=row, column=3)
        row += 1
        tk.Label(self.root, width=10, justify='right' ,text='图片行数').grid(row=row, column=0)
        tk.Entry(self.root, textvariable=self.row, width=24).grid(row=row, column=1)
        # tk.Button(self.root, text = '保存图片', command = self.saveImg).grid(row=row, column=2)

        row += 1
        tk.Label(self.root, width=10, justify='right' ,text='图片列数').grid(row=row, column=0)
        tk.Entry(self.root, textvariable=self.column, width=24).grid(row=row, column=1)

    @thread_run
    def setResource(self, ):
        try:
            self.load = True
            self.resources = []
            self.path.set(askdirectory(title="载入输入源"))
            self.images = self.getImages(self.path.get())
            for img in self.images:
                self.resources.append(ImageManager(img))
        except Exception as e:
            pass
        finally:
            self.load = False

    @thread_run
    def setOrign(self):
        orign = askopenfilename(title='选择输出源', filetypes=[('jpg', '*.jpg'), ('jpeg', '*.jpeg'), ('png', '*.png')])

        self.output.set(orign)

        self.orign = ImageManager(orign)

    @thread_run
    def generate(self, ):
        if self.load:
            messagebox.showinfo("警告", "图片源加载尚未完成，请等待...")
            return
        if self.busy:
            messagebox.showinfo("警告", "上一个合成任务仍在进行中...")
            return
        try:
            self.busy = True
            width = int(self.column.get())
            height = int(self.row.get())
            length = int(self.width.get())

            self.outim = self.orign.resize((width * length, height * length))
            for i in range(height):
                for j in range(width):
                    manager = ImageManager(
                        im=self.outim.crop((i * length, j * length, (i + 1) * length, (j + 1) * length)))
                    rgion = self.getMin(manager, self.resources)
                    box = (i * length, j * length, (i + 1) * length, (j + 1) * length)
                    self.outim.paste(rgion.resize((length, length)), box)

            self.saveImg()
            messagebox.showinfo("提示", "生成图片成功！！！")

        except Exception as e:
            messagebox.showinfo("提示", "生成图片失败，请重试...")
            print(e)
        finally:
            self.busy = False

    def saveImg(self, ):
        if self.outim:
            self.outim.save('%s.png' % int(time.time()), quality=95)

    def getImages(self, path):
        tmp = []
        for root, dirs, files in os.walk(path):
            if files and len(files) > 0:
                for file in files:
                    if 'jpg' in file or 'jpeg' in file or 'png' in file:
                        tmp.append(root + '/' + file)
            if dirs and len(dirs) > 0:
                for _dir in dirs:
                    tmp += self.getImages(root + '/' + _dir)

        return tmp

    def getMin(self, point, points):
        _min = points[0]
        _mean = point.getMean()
        _dis = math.fabs(_min.getMean() - _mean)
        for p in points:
            if math.fabs(p.getMean() - _mean) < _dis:
                _min = p
                _dis = math.fabs(p.getMean() - _mean)

        return _min


if __name__ == '__main__':
    lock = threading.Lock()
    root = tk.Tk()
    root.title("图片生成器 -- 正则兮")
    root.geometry('640x280+600+200')
    root.resizable(False, False)

    serach = SearchFrame(root)
    serach.grid(0, 0)

    generate = GenerateFrame(root)
    generate.grid(1, 0)

    tk.Label(text="created BY 正则兮").grid(row=2, column=0)

    root.mainloop()