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)