manyi/taskmanager/tasks.py

# -*- coding: utf-8 -*-
#!/usr/bin/env python
"""
为了与userserver解耦，大部分情况这里的任务只做celery转换，并不涉及具体实现。
"""
from __future__ import absolute_import


from taskmanager import TraceTask
from taskmanager.operator import app

from apps.web.user.tasks import (
    report_to_user_via_wechat,
    poll_user_recharge_record,
    test_sync,
    report_to_user_low_power,
    notify_virtual_card_expired,
    send_msg_to_user_via_wechat,
    notify_insurance_order_cancel,
    notify_insurance_order_subscribe,
    pull_refund_order,
    pull_consume_order_timeout
)

from apps.web.dealer.tasks import (report_feedback_to_dealer_via_wechat,
                                   report_daily_report_to_dealer_via_wechat,
                                   report_new_payment_to_dealer_via_wechat,
                                   report_device_abnormally_offline_to_dealer_via_wechat,
                                   report_to_dealer_via_wechat,
                                   send_SIM_expired_messages,
                                   daily_check_auto_withdraw,
                                   generate_business_stats_report_by_dealer,
                                   export_charge_order_excel_from_db,
                                   export_on_points_order_excel_from_db,
                                   export_API_order_excel_from_db,
                                   export_send_coins_to_card_order_excel_from_db,
                                   export_consume_order_excel_from_db,
                                   export_group_stat_excel_from_db,
                                   export_vcard_info_excel_from_db,
                                   export_group_user_account_excel_form_db,
                                   export_aggregate_dealer_income,
                                   calc_dealer_stat_and_insert_into_db,
                                   calc_dealer_user_count,
                                   poll_dealer_recharge_record,
                                   dealer_auto_withdraw,
                                   report_offline_device_to_dealer_via_wechat,
                                   dealer_auto_charge_sim_card,
                                   batch_set_device_params,
                                   set_device_params,
                                   batch_set_server_settings,
                                   set_server_settings,
                                   push_shanghai_platform_heatbeat,
                                   export_modify_customer_balance_record_excel_from_db,)

from apps.web.management.tasks import (weekly_notify_finance_manager,
                                       whale_withdraw_order_alert,
                                       withdraw_error_alert,
                                       check_website_is_online,
                                       generate_manager_map_options,
                                       generate_biz_stats_for_manager,
                                       check_wechat_withdraw_via_bank,
                                       check_and_retry_withdraw,
                                       manager_export_charge_order_excel_from_db,
                                       manager_export_dealer_info_excel_from_db,
                                       manager_export_consume_order_excel_from_db)

from apps.web.ad.tasks import generate_ad_excel_report
from apps.web.management.tasks import generate_simCharge_excel_report, generate_dealerWithDraw_excel_report
from apps.web.device.tasks import remove_serviceProgress_periodically, set_device_deactive_for_langxin, turn_on_power_huan_dian_gui, device_offline_notify, send_to_xf_all_dev_info, \
    send_to_xf_falut, send_to_xf_fault_handle, deduct_rent_order, gen_daily_rent_order, sync_device_time_for_tcpcar, make_rpt_into_db, report_to_zhejiang_fight

from apps.dispatch.tasks import send_topic_command
from apps.web.merchant.tasks import query_merchant_status

# noinspection PyUnresolvedReferences
from apps.web.superadmin.tasks import (import_simcard_excel_to_db, export_simcard_excel_from_db, sum_customer, export_device_excel_from_db, handle_customer_complaints_yesterday)


celery_task = lambda f: app.task(f, base = TraceTask, name = 'tasks.' + f.__name__)

# : to end user
report_to_user_via_wechat = celery_task(report_to_user_via_wechat)
send_msg_to_user_via_wechat = celery_task(send_msg_to_user_via_wechat)
report_to_dealer_via_wechat = celery_task(report_to_dealer_via_wechat)
poll_user_recharge_record = celery_task(poll_user_recharge_record)
test_sync = celery_task(test_sync)
report_to_user_low_power = celery_task(report_to_user_low_power)
notify_virtual_card_expired = celery_task(notify_virtual_card_expired)
query_merchant_status = celery_task(query_merchant_status)
notify_insurance_order_subscribe = celery_task(notify_insurance_order_subscribe)
notify_insurance_order_cancel = celery_task(notify_insurance_order_cancel)
pull_refund_order = celery_task(pull_refund_order)
pull_consume_order_timeout = celery_task(pull_consume_order_timeout)

# : to dealer
report_feedback_to_dealer_via_wechat = celery_task(report_feedback_to_dealer_via_wechat)
report_daily_report_to_dealer_via_wechat = celery_task(report_daily_report_to_dealer_via_wechat)
report_new_payment_to_dealer_via_wechat = celery_task(report_new_payment_to_dealer_via_wechat)
report_device_abnormally_offline_to_dealer_via_wechat = celery_task(report_device_abnormally_offline_to_dealer_via_wechat)
send_SIM_expired_messages = celery_task(send_SIM_expired_messages)
calc_dealer_stat_and_insert_into_db = celery_task(calc_dealer_stat_and_insert_into_db)
calc_dealer_user_count = celery_task(calc_dealer_user_count)
daily_check_auto_withdraw = celery_task(daily_check_auto_withdraw)
generate_business_stats_report_by_dealer = celery_task(generate_business_stats_report_by_dealer)
export_charge_order_excel_from_db = celery_task(export_charge_order_excel_from_db)
export_consume_order_excel_from_db = celery_task(export_consume_order_excel_from_db)
export_send_coins_to_card_order_excel_from_db = celery_task(export_send_coins_to_card_order_excel_from_db)
export_on_points_order_excel_from_db = celery_task(export_on_points_order_excel_from_db)
export_API_order_excel_from_db = celery_task(export_API_order_excel_from_db)
export_group_stat_excel_from_db = celery_task(export_group_stat_excel_from_db)
export_vcard_info_excel_from_db = celery_task(export_vcard_info_excel_from_db)
export_group_user_account_excel_form_db = celery_task(export_group_user_account_excel_form_db)
export_aggregate_dealer_income = celery_task(export_aggregate_dealer_income)
report_offline_device_to_dealer_via_wechat = celery_task(report_offline_device_to_dealer_via_wechat)
push_shanghai_platform_heatbeat = celery_task(push_shanghai_platform_heatbeat)
batch_set_device_params = celery_task(batch_set_device_params)
set_device_params = celery_task(set_device_params)
batch_set_server_settings = celery_task(batch_set_server_settings)
set_server_settings = celery_task(set_server_settings)
export_modify_customer_balance_record_excel_from_db = celery_task(export_modify_customer_balance_record_excel_from_db)

poll_dealer_recharge_record = celery_task(poll_dealer_recharge_record)

# : to manager
weekly_notify_finance_manager = celery_task(weekly_notify_finance_manager)
whale_withdraw_order_alert = celery_task(whale_withdraw_order_alert)
withdraw_error_alert = celery_task(withdraw_error_alert)
check_website_is_online = celery_task(check_website_is_online)
generate_ad_excel_report = celery_task(generate_ad_excel_report)
generate_simCharge_excel_report = celery_task(generate_simCharge_excel_report)
generate_dealerWithDraw_excel_report = celery_task(generate_dealerWithDraw_excel_report)
generate_manager_map_options = celery_task(generate_manager_map_options)
generate_biz_stats_for_manager = celery_task(generate_biz_stats_for_manager)
manager_export_charge_order_excel_from_db = celery_task(manager_export_charge_order_excel_from_db)
manager_export_consume_order_excel_from_db = celery_task(manager_export_consume_order_excel_from_db)
manager_export_dealer_info_excel_from_db = celery_task(manager_export_dealer_info_excel_from_db)

# : to device
remove_serviceProgress_periodically = celery_task(remove_serviceProgress_periodically)
set_device_deactive_for_langxin = celery_task(set_device_deactive_for_langxin)
turn_on_power_huan_dian_gui = celery_task(turn_on_power_huan_dian_gui)
send_to_xf_all_dev_info = celery_task(send_to_xf_all_dev_info)
send_to_xf_falut = celery_task(send_to_xf_falut)
send_to_xf_fault_handle = celery_task(send_to_xf_fault_handle)
gen_daily_rent_order = celery_task(gen_daily_rent_order)
deduct_rent_order = celery_task(deduct_rent_order)
sync_device_time_for_tcpcar = celery_task(sync_device_time_for_tcpcar)
make_rpt_into_db = celery_task(make_rpt_into_db)
report_to_zhejiang_fight = celery_task(report_to_zhejiang_fight)

# : to super manager
import_simcard_excel_to_db = celery_task(import_simcard_excel_to_db)
export_simcard_excel_from_db = celery_task(export_simcard_excel_from_db)
sum_customer = celery_task(sum_customer)
export_device_excel_from_db = celery_task(export_device_excel_from_db)
handle_customer_complaints_yesterday = celery_task(handle_customer_complaints_yesterday)


#: bank related
check_wechat_withdraw_via_bank = celery_task(check_wechat_withdraw_via_bank)
check_and_retry_withdraw = celery_task(check_and_retry_withdraw)

dealer_auto_withdraw = celery_task(dealer_auto_withdraw)

dealer_auto_charge_sim_card = celery_task(dealer_auto_charge_sim_card)

#: dispatch
send_topic_command = celery_task(send_topic_command)

device_offline_notify = celery_task(device_offline_notify)

# : Test purpose
@celery_task
def add(x, y):
    return x + y