baoshiwei
2025-04-15 776a127b8db01cd4338f4db2a84ea567a65bff9f
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
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)