def deactivate_task(uid, days=14):
"""
uid：user.id
days: 默认两周后关闭账户
"""
from djcelery.models import CrontabSchedule, PeriodicTask
from django.contrib.auth.models import User
import datetime
now = datetime.datetime.today()
deadline = now + datetime.timedelta(minutes=3)
task_date = CrontabSchedule()
task_date.minute = deadline.timetuple().tm_min
task_date.hour = deadline.timetuple().tm_hour
task_date.day_of_month = deadline.timetuple().tm_mday
task_date.month_of_year = deadline.timetuple().tm_mon
try:
from django.db import transaction
with transaction.atomic():
task_date.save()
except:
return False
user = User.objects.get(pk=uid)
name = "Deactivate_User_%s" % uid
new_task = PeriodicTask(name=name)
new_task.name = name
new_task.task = 'templates.tasks.deactivate_tempuser'
new_task.crontab = task_date
new_task.args = "[%s]" % uid
new_task.enabled = True
try:
from django.db import transaction
with transaction.atomic():
new_task.save()
except:
return False
return True
def schedule_cron(task_name, minute='*', hour='*', day_of_week='*', day_of_month='*', month_of_year='*' ,args=None, kwargs=None):
# create the periodic task and the interval
ptask_name = "%s_%s" % (task_name, datetime.now()) # create some name for the period task
crontab_schedules = CrontabSchedule.objects.filter(minute=minute, hour=hour, day_of_week=day_of_week, day_of_month=day_of_month, month_of_year=month_of_year)
if crontab_schedules: # just check if interval schedules exist like that already and reuse em
crontab_schedule = crontab_schedules[0]
else: # create a brand new interval schedule
crontab_schedule = CrontabSchedule()
crontab_schedule.minute = minute
crontab_schedule.hour = hour
crontab_schedule.day_of_week = day_of_week
crontab_schedule.day_of_month = day_of_month
crontab_schedule.month_of_year = month_of_year
crontab_schedule.save()
ptask = PeriodicTask(name=ptask_name, task=task_name, crontab=crontab_schedule)
if args:
ptask.args = args
if kwargs:
ptask.kwargs = kwargs
ptask.save()
return TaskScheduler.objects.create(periodic_task=ptask)
