pil example auto recommend img d (snippet)

Python pil example 'auto recommend img d'

Modules used in program:

import argparse
import time
import re
import sys
import random,os
import urllib,json,socket

Python pil example: auto recommend img d

# -*- coding: utf-8 -*-
import urllib,json,socket  
import random,os  
import sys
import re
import time
import argparse

parser = argparse.ArgumentParser(formatter_class=argparse.ArgumentDefaultsHelpFormatter)
add_arg = parser.add_argument
add_arg('--keyword', '-kw', default='车牌', help='输入要查询的关键词')
add_arg('--download_page', '-dp', default=1, type=int, help='希望下载的页数，每页60张图')
add_arg('--dir', default='E:\\test\\', help='下载图片保存的位置')
args = parser.parse_args()

valid_type = ['.png', '.jpg', '.PNG', '.JPG', '.gif', '.GIF', '.jpeg', '.JPEG']
download_page = args.download_page
socket.setdefaulttimeout(10)

keyword = args.keyword

tmpurl = 'http://image.baidu.com/search/flip?tn=baiduimage&ie=utf-8&word='+keyword+'&pn='

dir = args.dir
try:
    if not os.path.isdir(dir):
        os.mkdir(dir)
except OSError:
    print('can not make dir!')
    sys.exit()
i = 0

for dp in xrange(download_page):
    url = tmpurl+str(dp*60)
    pattern = re.compile(r'setData\(\'imgData\', (\{[\s\S]*?\})\);')
    try:
        ipdata = urllib.urlopen(url).read()
    except IOError:
        print('can not open this url!')
        sys.exit() 
    ipdata = pattern.search(ipdata)
    ipdata = ipdata.group(1)
    _regex = re.compile(r'\\(?![/u"])')  
    ipdata = _regex.sub(r"\\\\", ipdata) 
    imgData = json.loads(ipdata, strict=False)

    if imgData['data']:
        for obj in imgData['data']:
            if obj and obj['objURL']:
                try:
                    urllib.urlretrieve(obj['objURL'],'E:/test/'+str(time.time())+'.jpg')
                    continue
#                     data_img = urllib.urlopen(obj['objURL']).read()
                except IOError:
                    print('img is damaged.')
                else:
                    fPostfix = os.path.splitext(obj['objURL'])[1]
                    if fPostfix in valid_type:
                        filename = dir + os.path.basename(obj['objURL'])
                    else:
                        filename = dir + os.path.basename(obj['objURL']) + '.jpg'
                    try:
                        file_obj = open(filename, 'w')
                        file_obj.write(data_img)
                        file_obj.close()
                    except socket.timeout, e:
                        print('socket time out!')
                    else:
                        i += 1
                        print('img '+ str(i) + ' is downloaded!')
                    finally:
                        pass

print('all images have been downloaded!')

Python links

Learn Python: https://pythonbasics.org/
Python Tutorial: https://pythonprogramminglanguage.com

pil example auto recommend img d (snippet)

python auto recommend img d

Python links