上级要一个365天的数据自动导入数据库并验证

Question

有大拿知道怎么写365天自动导入数据并验证的自动化测试脚本吗

Answer 1

这是一个大致的框架，可以根据自己的需求扩展定义

from apscheduler.schedulers.blocking import BlockingScheduler
import pymysql
class TestImport():
    def __init__(self):
        self.file = "xxxx.csv"

# 时间调度器
    def timer(self):
        scheduler = BlockingScheduler()
        scheduler.add_job(self.excute(), 'cron',  day=365, hour=0) #一年365天每天0点执行
# 执行器
    def excute(self):
        self.readFile()
        self.checkData()


#数据库链接
    def connectSql(self):
        mydb = pymysql.connect(
            host="localhost",
            user="yourusername",
            passwd="yourpassword",
            database="mydatabase"
        )
        cur = mydb.cursor()
        return cur

#取出文件中的数据存入数据库
    def readFile(self):
        cur = self.connectSql()
        with open(self.file,'r') as f:
            for i in f.read():
                result = cur.execute('insert into tables(%s,%s,%s,%s)'%(i[0],i[1],i[2],i[3]))
                print("已插入{}".format(result))
            f.close()
        cur.close()

#校验数据是否已导入
    def checkData(self):
        cur = self.connectSql()
        cur.execute("select * from tabale")
        for i in cur.fetchall():
            print(i)

#主程序
    def main(self):
        self.timer()


if __name__ == "__main__":
    TestImport().main()