数据库重置数据

import pymysql
import yaml

class Get_sql():
    def __init__(self):
        #连接数据库
        self.db = pymysql.connect(host='127.0.0.1', port=3306, user='root', password='**', db='book')
        self.cosor=self.db.cursor()

    #清空表
    def clear_sql(self,table_name):
        sql='truncate {};'.format(table_name)
        self.cosor.execute(sql)
        self.db.commit()

    #插入数据
    def insert_sql(self,table_name,table_data):
        for key in table_data:
            table_data[key]="'"+str(table_data[key])+"'"

        key=','.join(table_data.keys())
        value=','.join(table_data.values())

        sql='insert into {}({}) values ({});'.format(table_name,key,value)
        # sql=insert_sql='insert into '+table_name+'('+key+')'+'values'+'('+value+')'
        self.cosor.execute(sql)
        self.db.commit()

    def close(self):
        self.db.close()

    def init_data(self,datas):
        for table_name,data in datas.items():
            self.clear_sql(table_name)
            for d in data:
                self.insert_sql(table_name,d)
        self.close()

if __name__ == '__main__':
    datas1={'test_api':[{'id':1,'user':'aabb','password':'123456'}]}

    # file=r'E:\python\envname\Scripts\imooc\test_aa\datas.yaml'
    # with open(file,'r') as f:
    #     datas1=yaml.load(f,Loader=yaml.FullLoader)
    # print(datas1)

    Get_sql().init_data(datas1)