要在django項目中定期執行任務,比如每天一定的時間點抓取數據,刷新資料庫等,可以參考stackoverflow的方法,先編寫一個manage.py命令,然後使用crontab來定時執行這個命令。 定製manage.py命令 app可以使用manage.py註冊自己的命令,比如要在polls這個a ...
要在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