from __future__ import absolute_import
import os
from celery import Celery
from django.conf import settings
from django.utils import timezone

# set the default Django settings module for the 'celery' program.
# 為celery設置環(huán)境變量
os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'voice_quality_assurance_configure.settings')
# 創(chuàng)建celery app
app = Celery('voice_quality_assurance_configure')
# Using a string here means the worker will not have to
# pickle the object when using Windows.
# 從單獨的配置模塊中加載配置
app.config_from_object('voice_quality_assurance_configure.celeryconfig')
# 設置app自動加載任務
app.autodiscover_tasks(lambda: settings.INSTALLED_APPS)
# 解決時區(qū)問題,定時任務啟動就循環(huán)輸出
app.now = timezone.now

from __future__ import absolute_import
from kombu import Queue
from django.conf import settings

# 設置代理人broker
CELERY_BROKER_URL = 'redis://127.0.0.1:6379/2'
# 指定 Backend
CELERY_RESULT_BACKEND = 'redis://127.0.0.1:6379/1'
# 指定時區(qū)，默認是 UTC
CELERY_TIMEZONE='Asia/Shanghai'
# celery 序列化與反序列化配置
CELERY_TASK_SERIALIZER = 'pickle'
CELERY_RESULT_SERIALIZER = 'pickle'
CELERY_ACCEPT_CONTENT = ['pickle', 'json']
CELERY_IGNORE_RESULT = True
# celery 的啟動工作數(shù)量設置
CELERY_WORKER_CONCURRENCY = 10
# 任務預取功能，會盡量多拿 n 個，以保證獲取的通訊成本可以壓縮。
CELERYD_PREFETCH_MULTIPLIER = 20
# 有些情況下可以防止死鎖
CELERYD_FORCE_EXECV = True
# celery 的 worker 執(zhí)行多少個任務后進行重啟操作
CELERY_WORKER_MAX_TASKS_PER_CHILD = 100
# 禁用所有速度限制，如果網(wǎng)絡資源有限，不建議開足馬力。
CELERY_DISABLE_RATE_LIMITS = True

# celery beat配置（周期性任務設置）
CELERY_ENABLE_UTC = False
CELERY_TIMEZONE = settings.TIME_ZONE
DJANGO_CELERY_BEAT_TZ_AWARE = False
CELERY_BEAT_SCHEDULER = 'django_celery_beat.schedulers:DatabaseScheduler'

from datetime import datetime, timedelta
import json
import os,django

os.environ.setdefault("DJANGO_SETTINGS_MODULE", "voice_quality_assurance_configure.settings")# project_name 項目名稱
django.setup()
from django_celery_beat.models import PeriodicTask, IntervalSchedule
schedule, created = IntervalSchedule.objects.get_or_create(every=10,period=IntervalSchedule.SECONDS,)

# 帶參數(shù)的創(chuàng)建方法，如下：
PeriodicTask.objects.create(
   interval=schedule,     # 上面創(chuàng)建10秒的間隔 interval 對象
   name='test_task',     # 設置任務的name值
   task='mission.tasks.my_task', # 指定需要周期性執(zhí)行的任務
   args=json.dumps([10, 2, 76]),
  expires=datetime.utcnow() + timedelta(seconds=30)
)

schedule, created = IntervalSchedule.objects.get_or_create(
  every=10,
  period=IntervalSchedule.SECONDS,
)

PeriodicTask.objects.create(
   interval=schedule,         # we created this above.
   name='test_task',     # simply describes this periodic task.
   task='app名.tasks.任務函數(shù)名', # name of task.)

PeriodicTask.objects.create(
   interval=schedule,         # we created this above.
   name='test'_task',     # simply describes this periodic task.
   task='app名.tasks.任務函數(shù)名', # name of task. 
　　　args=json.dumps(['arg1', 'arg2']), 
   kwargs=json.dumps({ 'be_careful': True, }), 
   expires=datetime.utcnow() + timedelta(seconds=30) )

class MonitorDeviceTask(object):
  """
  設備創(chuàng)建，增加周期性任務
  """

  def __init__(self, device_obj):
    self.device_obj = device_obj
    self.periodic_task = PeriodicTask.objects.create(
      interval=schedule,
      name='test_task',
      task='mission.tasks.my_task',
      args=json.dumps([self.device_obj.ip])
    )

  def starttask(self):
    """
    啟動任務
    """
    self.periodic_task.enabled = True
    self.periodic_task.save()

  def stoptask(self):
    """
    停止任務
    """
    self.periodic_task.enabled = False
    self.periodic_task.save()

  def deltask(self):
    """
    刪除任務
    """
    self.periodic_task.delete()  
    self.periodic_task.save()

from django_celery_beat.models import CrontabSchedule, PeriodicTask
schedule, _ = CrontabSchedule.objects.get_or_create(
   minute='30',
   hour='*',
   day_of_week='*',
   day_of_month='*',
   month_of_year='*',
   timezone=pytz.timezone('Canada/Pacific')
)