manyi/script/celery/tasks.py

# -*- coding: utf-8 -*-
# !/usr/bin/env python

import getopt
import sys
import os

PROJECT_ROOT = os.path.join(os.path.abspath(os.path.split(os.path.realpath(__file__))[0] + "/.."), '..')
sys.path.insert(0, PROJECT_ROOT)

try:
    options, args = getopt.getopt(sys.argv[1:], 'e:t:', ['env=', 'task='])
except getopt.GetoptError:
    sys.exit()

system_env = 'testing'
task_name = None

for name, value in options:
    if name in ('-e', '--env'):
        system_env = value

    if name in ('-t', '--task'):
        task_name = value

if not system_env:
    print 'error system env = {env}'.format(env = system_env)
    sys.exit(2)

if not task_name:
    print 'error task name = {}'.format(task_name)
    sys.exit(2)

import os

os.environ.setdefault('DJANGO_SETTINGS_MODULE', 'configs.{env}'.format(env = system_env))

from script.base import init_env

init_env(interactive = False)

from taskmanager.mediator import task_caller_once

task_caller_once(func_name = task_name, routing_key = 'dealer_data_anylyse', queue = 'dealer_data_anylyse')