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区域图片，并以文件夹+序号命名