import mysql.connector
|
from mysql.connector import Error
|
from pachong import imgdata_set
|
from quchong import quchongmethod
|
import shutil
|
|
|
# 连接mysql 查询数据库
|
def mysql_connect():
|
try:
|
conn = mysql.connector.connect(host='localhost',
|
database='herb',
|
user='root',
|
password='123456')
|
if conn.is_connected():
|
print('Connected to MySQL database')
|
return conn
|
except Error as e:
|
print(e)
|
|
# 查询dry_herb_info表
|
def mysql_select(conn):
|
cursor = conn.cursor()
|
sql = "SELECT name, pinyin FROM dry_herb_info where pinyin is not null and pinyin <> ''"
|
cursor.execute(sql)
|
result = cursor.fetchall()
|
return result
|
|
|
# main
|
if __name__ == '__main__':
|
conn = mysql_connect()
|
|
result = mysql_select(conn)
|
|
for row in result:
|
name = row[0]
|
pinyin = row[1]
|
print(name)
|
print(pinyin)
|
imgdata_set('E:/pachong/'+pinyin, name + '饮片', 2)
|
quchongmethod('E:/pachong/'+pinyin, 'E:/pachong/2/'+pinyin)
|
#删除临时文件夹
|
|
shutil.rmtree('E:/pachong/2/'+pinyin)
|