Django基础：独立模块

Django

基础：独立模块

建立模块

一般单独建立一个scripts文件夹，内部放独立模块

导入配置

!/usr/bin/env python
import os
import sys
import random
from datetime import date

# 将项目根目录添加到 sys.path 中
BASE_DIR = os.path.dirname(os.path.dirname(os.path.abspath(__file__)))  # 取出项目绝对路径
sys.path.insert(0, BASE_DIR)

# # 设置 Django 运行环境
import django

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "swiper.settings")
django.setup()

#导入Dango中所需模型
from user.models import User

运行配置

利用if判断所需运行命令

if __name__ == '__main__':
    # 解析脚本执行的参数
    if len(sys.argv) >= 2:
        command = sys.argv[1]
        if command == 'create_robots':
            create_robots(10000)
        elif command == 'create_vip':
            create_vip()
        else:
            print('未知的命令')
            sys.exit(2)
    else:
        print('请输入要执行的命令')
        sys.exit(1)

运行

python data_init.py create_rebots(windows)
data_init.py create_rebots(linux、macos)