django定期执行任务

要在django项目中定期执行任务，比如每天一定的时间点抓取数据，刷新数据库等，可以参考stackoverflow的方法，先编写一个manage.py命令，然后使用crontab来定时执行这个命令。

定制manage.py命令

app可以使用manage.py注册自己的命令，比如要在polls这个app中定制一个closepoll命令，要先向polls文件夹中添加一个management/commands的目录：

?

1 2 3 4 5 6 7 8 9 10 11

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。

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20

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的位置参数，数目为一个或者更多，类型为整数：

?

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

　closepoll命令就可以这样使用了：

?

1	python manage.py closepoll <poll_id>

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

可以添加可选参数：

?

1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16

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，窗口会有提示：

?

1	# m h  dom mon dow   command

在某月(mon)的某天(dom)或者星期几(dow)的几点(h，24小时制)几分(m)执行某个命令（command)，*表示任意时间。

?

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

这样表示每两个小时执行一次closepoll命令，为了防止manage.py文件找不到，可以写上manage.py文件的真实地址。

保存后重启cron即可：

?

1	sudo service cron restart