外部调用django模型

1.单个对象通过orm写入数据库

#!/usr/local/bin/python3
import os
import sys
import django


def save_cli_run_log(cli, state, result, user_id):
    """将数据写入数据库"""
    # 配置环境变量
    dir = os.path.dirname(os.path.abspath(__file__))
    dir = os.path.join(dir, 'autoweb')
    sys.path.insert(0, dir)
    settings_path = 'autoweb.settings'
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", settings_path)
    django.setup()
    # 调用模型
    from azure_audit.models import CliRunLogModel  # 这行不要移动到头部
    obj = CliRunLogModel()
    obj.cli = cli
    obj.state = state
    obj.result = result
    obj.user_id = user_id
    obj.save()
    return 复制代码

2.excel通过orm批量写入数据库

def excel_save_mysql():
    """调用orm将数据写入数据库"""
    # 配置环境变量
    dir = os.path.dirname(os.path.abspath(__file__))
    dir = os.path.join(dir, 'django项目名称')
    sys.path.insert(0, dir)
    settings_path = '项目名称.settings'
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", settings_path)
    django.setup()
    # 调用模型
    from Score.models import Score
    """清空数据库,谨慎使用"""
    # Score.objects.all().delete()
    # 不论execl中的字段是数值或字符，保存时会以orm的类型保存
    # 读取excel数据
    data = pd.read_excel("/home/python/Desktop/score7.xls")
    # 将空值转换为-1
    data = data.fillna({"student_code": -1})
    # 将data["student_code"]类型强转成int,再转成str
    data["student_code"] = data["student_code"].astype(int).astype(str)
    # print(data["student_code"])
    # 将"-1"替换成""
    data = data.replace("-1", "")
    data = data.replace("1", "")
    # 将所有的空值转换为""
    data = data.fillna("")

    # 将数据转为json个数
    # all_vm_json = json.dumps(all_dt_list)
    # 建立数据库链接
    connect = create_engine('mysql+pymysql://root:pwd@127.0.0.1:3306/数据库名')
    # data = pd.read_json(all_vm_json, orient='records')
    # 将数据保存到数据库; if_exists:不能使用replace; schema:数据库名;
    pd.io.sql.to_sql(data, "表名", connect, schema='数据库名', if_exists='append', index=False)
    print('数据保存成功')
    return 0复制代码