Django扩展 从外部调用django model类进行数据库操作

有时候我们需要测试项目,给数据库批量添加一些数据

在manage.py同级的目录下 创建一个新py文件

# 外部文件操作Django model批量插入数据

import os


if __name__ == "__main__":

    # 配置Django环境
    os.environ.setdefault("DJANGO_SETTINGS_MODULE", "supercrm.settings")
    import django
    django.setup()

    # 导入model开始操作
    from sales import models

    source_type = (('qq', "qq群"),
                   ('referral', "内部转介绍"),
                   ('website', "官方网站"),
                   ('baidu_ads', "百度推广"),
                   ('office_direct', "直接上门"),
                   ('WoM', "口碑"),
                   ('public_class', "公开课"),
                   ('website_hy', "绘画官网"),
                   ('others', "其它"),)

    course_choices = (('cartoon', '漫画进阶'),
                      ('sketch', '素描进阶'),)
    import random

    obj_lis = []
    for i in range(251):
        mode_obj = {
            "qq": str(12514+i),
            "name": f"李狗子{i}号",
            "source": source_type[random.randint(0, 8)][0],
            "course": course_choices[random.randint(0, 1)][0],
        }

        obj = models.Customer(**mode_obj)
        obj_lis.append(obj)
    models.Customer.objects.bulk_create(obj_lis)

 

posted @ 2020-07-17 10:21  CatdeXin  阅读(384)  评论(0)    收藏  举报