import pandas as pd
from pandas.core.dtypes.dtypes import register_extension_dtype
from sqlalchemy import create_engine, engine
初始化数据库连接，使用pymysql模块
Mysql的用户root，密码：mysql，端口33306，数据库xxx
engine = create_engine(‘mysql+pymysql://root:mysql@localhost:3306/xxx‘)
查询语句，选出表中的所有数据
print(df.head)
for i in range(29):
sql = ‘‘‘SELECT * FROM tpcode where IMAGETYPE like ‘‘‘+str(i+1)+‘‘‘.%%‘ limit 2500; ‘‘‘
df = pd.read_sql_query(sql, engine)
print(sql)
with open(‘a.csv‘, ‘w‘, encoding=‘utf-8‘) as file:
pd.DataFrame(columns=[‘REGNO‘,‘IMAGETYPE‘]).to_csv(file, index=False) # 去除index
for i in range(29):
sql = "SELECT REGNO,IMAGETYPE FROM tpcode where IMAGETYPE like ‘"+str(i+1)+‘.‘+"%%"+"‘ limit 2500; "
print(sql)
df = pd.read_sql_query(sql, engine)
df[‘IMAGETYPE‘] = i+1
df.to_csv(file, mode=‘a+‘, index=False,header=False) #追加，不写索引和表头
print(i)
from datetime import datetime
import hashlib
from urllib.parse import unquote, quote
import requests
df = pd.read_csv(‘a.csv‘)
a = 0
for i in df[‘REGNO‘]:
a = a +1
imgurl = sign_md5s(‘BrandInfoFind/GetBrandInfoImg‘, {"RegId": i})
try:
filename = ‘image/‘ +str(i) + ‘.jpg‘
pic = requests.get(imgurl)
if pic.status_code == 200:
with open(filename, ‘wb‘) as fp:
fp.write(pic.content)
fp.close()
if a % 2500 == 0:
print("成功输出类别：",int(a/2500)+1)
except:
print(‘获取图片失败‘,i)