Django--自定義 Command 命令

Django 對於命令的添加有一套規範,你能夠爲每一個app 指定命令。通俗一點講,好比在使用manage.py文件執行命令的時候,能夠自定製本身的命令,來實現命令的擴充。

commands的建立python

一、在app內建立一個management的python目錄
二、在management目錄裏面建立commands的python文件夾
三、在commands文件夾下建立任意py文件

此時py文件名就是你的自定製命令,能夠使用下面方式執行django

python manage.py 命令名

Django的Command命令是要放在一個app的management/commands目錄下的。

 python2環境中,請確保management和management/commands目錄內都包含__init__.py文件ruby

 

首先對於文件名沒什麼要求,內部須要定義一個Command類並繼承BaseCommand類或其子類。

它必須定義一個Command類並擴展自BaseCommand或其 子類。app

其中help是command功能做用簡介,handle函數是主處理程序,add_arguments函數是用來接收可選參數的函數

from django.core.management.base import BaseCommand, CommandError
from polls.models import Poll  本身的

class Command(BaseCommand):
    help = 'Closes the specified poll for voting'

    def add_arguments(self, parser):
        parser.add_argument('poll_id', nargs='+', type=int)

    def handle(self, *args, **options):
        for poll_id in options['poll_id']:
            try:
                poll = Poll.objects.get(pk=poll_id)
            except Poll.DoesNotExist:
                raise CommandError('Poll "%s" does not exist' % poll_id)

            poll.opened = False
            poll.save()

            self.stdout.write('Successfully closed poll "%s"' % poll_id)

 

可選參數測試

可以使用add_argument()方法:spa

class Command(BaseCommand):
    def add_arguments(self, parser):
        # Positional arguments
        parser.add_argument('poll_id', nargs='+', type=int)

        # Named (optional) arguments
        parser.add_argument('--delete',
            action='store_true',
            dest='delete',
            default=False,
            help='Delete poll instead of closing it')

    def handle(self, *args, **options):
        # ...
        if options['delete']:
            poll.delete()
        # ...

 帶參數的測試:命令行

# -*- coding: utf-8 -*-
# __author__ = 'dandy'
from django.core.management.base import BaseCommand

class Command(BaseCommand):

    def add_arguments(self, parser):
        parser.add_argument('aaa', nargs='+', type=int)
        parser.add_argument('--delete',
                            action='store_true',
                            dest='delete',
                            default=False,
                            help='Delete poll instead of closing it')

    def handle(self, *args, **options):
        print('test')
        print(args, options)

options裏面直接取參數就能夠了。code

方法

返回django版本號:BaseCommand.get_version() 
命令的真正邏輯。子類必須實現這個方法。:BaseCommand.handle()blog

BaseCommand的子類

class LabelCommand

 

這個管理命令接收命令行上的一個或多個參數(標籤),並對它們每個都作一些動做。

子類不用實現handle(),但必須實現handle_label(),它將會爲每一個標籤調用一次。

LabelCommand.handle_label(label, **options)

 

對label完成命令行的動做,label是命令行給出的字符串。

相關文章
相關標籤/搜索