def wraper(*args, **kwargs):

t = threading.Thread(target=func, args=args, kwargs=kwargs)

t.daemon = True

t.start()

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)

"""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:

"""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

"""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)