python的N个小功能(连接数据库并下载相应位置的图片)
##############################################################################################################连接数据库, mysqldb.py###########################
###############################################################################
# -*- coding: utf-8 -*-
"""
Created on Fri Mar 20 10:50:56 2015
@author:sl
"""
import os
import MySQLdb
MySQL_NAME = os.getenv('MySQL_NAME', 'aaa')
MySQL_USER = os.getenv('MySQL_USER', 'bbb')
MySQLPASSWORD = os.getenv('MySQL_PASSWORD', 'ccc')
MySQL_HOST = os.getenv('MySQL_HOST', 'ddd')
MySQL_PORT = os.getenv('MySQL_PORT','eee')
def connect():
db=MySQLdb.connect(host=MySQL_HOST, user=MySQL_USER, passwd=MySQLPASSWORD, db=MySQL_NAME, charset="utf8")
return db
###############################################################################
#####################循环读取下载相应位置图片##################################
###############################################################################
# -*- coding: utf-8 -*-
"""
Created on Mon Mar 06 14:55:15 2017
@author:sl
"""
import mysqldb
#import datetime
import pandas as pd
db = mysqldb.connect()
sql='''
SELECT
aaa,
bbb as urls
FROM
ccc
GROUP BY
aaa
'''
df = pd.read_sql(sql, con=db)
db.close()
from PIL import Image
from io import BytesIO
import requests
import os
from PIL import ImageFile
ImageFile.LOAD_TRUNCATED_IMAGES = True
def download_jpg(url, name):
response = requests.get(url).content ##########获取url对应内容
img = Image.open(BytesIO(response)) ##########打开图片
img.save(name) #########保存图片
if __name__ == '__main__':
df1=df.set_index('aaa') ####以第一列为index，第二列为内容
for mulu in df['aaa']: ####得到第一列全部内容并循环
if not os.path.exists(r' E:\pp\%s'%(mulu)):
os.makedirs(r' E:\pp\%s'%(mulu))
for url in df1.loc[mulu]: ####由第一列内容对应位置得到第二列内容
#print url
bb=url.split(',')
lens=len(bb)
for i in range(lens):
download_jpg('http://aaa=%s'%(str(bb[i])),r'E:\pp\%s\%s.jpg'%(mulu,str(mulu)+str(i))) ######下载对应url区域图片，并以文件夹+序号命名
