manyi/script/eventer/listener_anqi.py

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

import getopt
import logging
import sys

from typing import TYPE_CHECKING

try:
    options, args = getopt.getopt(sys.argv[1:], 'l:q:e:p:d:m:b',
                                  ['queue=', 'env=', 'port=', 'debug='])
except getopt.GetoptError as e:
    print(str(e))
    sys.exit()

platform_env = 'testing'
queue = False
port = None
debug = ''

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

    if name in ('-q', '--queue'):
        queue = value

    if name in ('-p', '--port'):
        port = int(value)

    if name in ('-d', '--debug'):
        debug = value

if not port:
    print 'no port parameter.'
    sys.exit(1)

import os

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

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

from script.base import init_env

init_env(interactive = False)

import simplejson as json
from django.conf import settings
from apps.web.common.event import do_device_event
from apps.web.core.sysparas import SysParas
from script.eventer.mqtt_context import MqttContext
from apps.web.constant import Const

if TYPE_CHECKING:
    from apps.web.device.models import DeviceDict

logger = logging.getLogger(__name__)

host = SysParas.get_private_ip(settings.MQTT_HOSTNAME)


def worker(cmd, dev, payload):
    # type: (str, DeviceDict, str)->None

    try:
        device_event = json.loads(bytes.decode(payload))
    except ValueError:
        logger.error(
            "can not loads message payload, cmd = <{}>, dev= <{}>, payload = <{}>".format(cmd, dev.devNo, payload))
        return

    if dev.devNo != device_event.get('IMEI', None):
        logger.warning('invalid msg for not same devNo({} != {})'.format(dev.devNo, device_event.get('IMEI')))
        return

    if cmd != str(device_event.get('cmd', '')):
        logger.warning('invalid msg for not same cmd({} ！= {})'.format(cmd, device_event.get('cmd', None)))
        return

    do_device_event(cmd, dev, device_event)


with MqttContext(host = host,
                 port = port,
                 user = settings.MQTT_USER,
                 password = settings.MQTT_PSWD,
                 worker = worker,
                 platform_env = platform_env,
                 queue = queue,
                 debug = debug,
                 cmds = {
                     '999': {
                         'qos': Const.MQTT_QOS,
                         'checkRegistered': True
                     }
                 },
                 logger = logger,
                 name = 'anqi_listener') as context:
    context.start()

logger.debug('exit now.')