在任意一个已安装的应用下,创建如下包目录:
代码语言:javascript复制app
├─management
│ ├─commands
│ │ ├─__pycache__
│ │ ├─ __init__.py
│ │ └─yourcommad.py
│ ├─__pycache__
│ └─ __init__.py
...
在commads目录下面创建我们的命令文件,命令的名称就是命令文件的名称,命令文件内部的基本结构如下:
代码语言:javascript复制#-*- coding: utf-8 -*
__author__ = 'geebos'
from django.core.management.base import BaseCommand, CommandError
from app.models import UserInfo
import os
import csv
class Command(BaseCommand):
help = '导入数据,以逗号分隔的csv文件,编码为UTF-8npython manage.py import_data 数据文件路径'
def add_arguments(self, parser):
parser.add_argument('path', type=str)
def handle(self, *args, **options):
path = options['path']
if not os.path.isfile(path):
raise CommandError('文件不存在')
print('正在导入...')
with open(path, 'r', newline='', encoding='utf-8_sig') as f:
reader = csv.DictReader(f)
for t in reader:
UserInfo.objects.create(**dict(t))
print('导入完成')
首先定义一个Commad类(注意这个类名不能改变,并且一个文件内只能有一个Commad类),这个类继承自 django.core.management.base.BaseCommand
.
help为命令的帮助文档,当输入错误命令时输出 help的内容
add_arguments
函数用来添加参数,如果没有参数可以不定义这个函数。在这个函数里通过调用 parse.add_argument
来添加参数.
handle
是命令的处理函数,可以通过 options
来取得从命令行输入的参数。