函数编程05--函数原型

read_sqls(sqlfiles)

功能:读取要执行的sql语句到列表
    参数:sql文件名列表
        不指定实参时,读取所有sql文件
        需判定列表是否存在元素
        实参必须使用列表形式,可自主优化
    返回值:sql语句列表
    说明:处理sql文件中的空行、注释行、行尾换行符

创建sqls01.txt文件存放exam项目初始化语句

-- 注册接口:test02、test03
delete from user where username = 'test02'

delete from user where username = 'test03'

insert into user(id,username,password) values(3,'test03','123456')

--登陆接口:test01
delete from user where username = 'test01';

insert into user(id,username,password) values(2,'test01','123456')

project_h1目录中test.py创建read_sqls(sqlfiles)方法

目的:运用read_sqls(sqlfiles)函数读取要执行的sql语句到列表

def read_sqls(sqlfiles=[]):  # 读指定sql 命令文件的函数，读取到的sql命令存入列表返回；如果没有给实参，则读所有txt中的sql命令
    try:
        if len(sqlfiles) == 0:  # 表示没有给实参,则读取所有sql文件
            sqlfiles = [file for file in os.listdir('.') if file.endswith('.txt')]  # 所有txt文件名存入sqlfiles列表
        sqls = []  # 存sql语句的列表
        for file in sqlfiles:
            sqlfile = open(file, 'r', encoding='utf-8')  # 打开一个文件
            for row in sqlfile:  # row表示文件中每一行
                # 如果这行字符长度大于0 并且 不是以--开头的
                if len(row.strip()) > 0 and not row.startswith('--'):
                    sqls.append(row.strip())  # sql语句中的回车不存入列表
        print('读取初始化sql语句成功==' + str(sqlfiles))
        print(sqls)
        return sqls
    except Exception as e:
        print('读取初始化sql语句出错'+e)
# 调试
read_sqls()