import os from django.conf import settings def getSql(filename): """ 获取文件内所有sql :param filename: 例如:os.path.join('permission','permission_dept.sql') :return: """ abspath = os.path.abspath(os.path.join(os.path.abspath(os.path.dirname(__file__)), "..")) pwd = os.path.join(abspath, 'scripts', filename) with open(pwd, 'rb') as fp: content = fp.read().decode('utf8') if filename == "permission/permission_userprofile.sql": user_name = "_".join(settings.AUTH_USER_MODEL.lower().split(".")) content = content.replace("permission_userprofile", user_name). \ replace("userprofile", settings.AUTH_USER_MODEL.lower().split(".")[-1]) return [ele for ele in content.split('\n') if not ele.startswith('--') and ele.strip(' ')]