django定期執行任務


要在django項目中定期執行任務,比如每天一定的時間點抓取數據,刷新數據庫等,可以參考stackoverflow的方法,先編寫一個manage.py命令,然后使用crontab來定時執行這個命令。

定制manage.py命令

app可以使用manage.py注冊自己的命令,比如要在polls這個app中定制一個closepoll命令,要先向polls文件夾中添加一個management/commands的目錄:

polls/
    __init__.py
    models.py
    management/
        __init__.py
        commands/
            __init__.py
            _private.py
            closepoll.py
    tests.py
    views.py

這樣,commands目錄下每一個不以"_"開頭的py文件都被注冊為一個manage.py命令。

python 2版本中注意management和commands目錄下都要包含一個__init__.py文件。

closepoll.py文件中必須定義一個繼承自BaseCommand的類Command。

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(self.style.SUCCESS('Successfully closed poll "%s"' % poll_id))

 在django1.8之前,manage.py命令的命令行解析是基於optparse庫的,其中位置參數會被傳給*args,而可選參數傳給**options。1.8之后,命令行解析基於argparse庫,參數都是傳給**options的。

   closepoll.py文件中添加了名為poll_id的位置參數,數目為一個或者更多,類型為整數:

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

 closepoll命令就可以這樣使用了:

python manage.py closepoll <poll_id>

 self.stdout和self.stderr可以在控制台中顯示你想要顯示的信息。

可以添加可選參數:

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()

添加一個名為--delete值為True的可選參數。

執行python manage.py closepoll 1 2 3 --delete時,將id為1,2,3的poll刪除。

執行python manage.py closepoll 1 2 3 時,--delete取值為默認值False。

使用crontab定時執行manage.py命令

使用crontab -e命令編輯cron,窗口會有提示:

# m h  dom mon dow   command

在某月(mon)的某天(dom)或者星期幾(dow)的幾點(h,24小時制)幾分(m)執行某個命令(command),*表示任意時間。

* */2 * * * python manage.py closepoll <poll_id>  

這樣表示每兩個小時執行一次closepoll命令,為了防止manage.py文件找不到,可以寫上manage.py文件的真實地址。

保存后重啟cron即可:

sudo service cron restart

 


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM