manyi/script/device.py

# -*- coding: utf-8 -*-
# !/usr/bin/env python
"""
将常见的设备迁移等功能做成易用的接口

## 设备管理器人力操作步骤
命令使用步骤：

1. telnet  到memcached 11211端口

2. get <imei> 查看server

3. telnet 到server

4. 执行命令

## 设备管理器接口汇总

事件开关: {"cmd":202,"IMEI":"865024038093674","report_set":1}
网络支付: {"cmd":203,"IMEI":"865024038093674","app_pay":1}
查询设备: {"cmd":201,"IMEI":"865024038093674"}
心跳设置: {"cmd":202,"IMEI":"865024038093674","cycle_set":60}
{"cmd":202,"IMEI":"865024038093674","cycle_set":300}
恢复出厂: {"cmd":202,"IMEI":"865024038093674","factory_set":1}
改服务器：{"cmd":202,"IMEI":"865024038093674","addr_set":{"ip1":"120.26.227.50","port1":30018}}
{"cmd":202,"IMEI":"865024038093674","addr_set":{"ip1":"112.74.180.175","port1":30018}}
{"cmd":202,"IMEI":"999990000000000","addr_set":{"ip1":"192.168.0.201","port1":30018}}
{"cmd":202,"IMEI":"999990000000000","addr_set":{"ip1":"192.168.0.106","port1":30018}}
{"cmd":202,"IMEI":"865024038093674","addr_set":{"ip1":"121.41.75.233","port1":30018}}
{"cmd":202,"IMEI":"865024038093674","pulse_set":{"pwm_wid":50,"pwm_inter":400,"pwm_idle":1}}
{"cmd":202,"IMEI":"865024038093674","max_coin":5}
心跳事件： {"cmd":207,"IMEI":"865024038093674","coin":3,"total_coin":15,"pay_cnt":34,"status":0,"signal":23}
远程登录: 112.74.180.175 30010

http://www.washpayer.com/userLogin?imei=865024038219469
http://www.5tao5ai.com/userLogin?imei=865024038093674
http://www.5tao5ai.com/userLogin?l=999990000000000
http://www.5tao5ai.com/agents/index.html
http://www.5tao5ai.com/app/index.html

http://www.washpayer.com/agents/login.html
http://www.washpayer.com/app/index.html

测试服务器：
查询所有模拟设备设备
telnet 192.168.1.106 30000
{"cmd":"find"}
{"cmd":"find","device":"9999900000000*"}
{"cmd":"find","device":"999990000000*"}

投币:
{"cmd":"coin","device":"999990000000000"}
{"cmd":"coin","device":"9999900000000*"}
{"cmd":"coin","device":"999990000000*"}

工作：
{"cmd":"work","device":"999990000000000"}
{"cmd":"work","device":"9999900000000*"}
{"cmd":"work","device":"999990000000*"}

停机:
{"cmd":"stop","device":"999990000000000"}

启动：
{"cmd":"boot","device":"999990000000000"}

"""
import click
import simplejson as json
import pprint
from bidict import bidict

from base import init_env

init_env(interactive=True)

from django.conf import settings
from apps.web.device.models import Device
from apps.web.core.networking import MessageSender

SERVERS = bidict({'testing': '121.40.97.20:1883', 'production':'120.27.251.159:1883'})

@click.group()
def cli():
    """
    一系列方便操作设备的函数
    :return:
    """
    click.echo(u"欢迎进入操作设备脚本")

send = MessageSender.send

def _device_command(imei, cmd, payload, server=None):
    dev = Device.get_dev(imei)
    click.echo('executing cmd(%s), payload(%s)' % (cmd, json.dumps(payload)) )
    r = send(dev, cmd, payload)
    click.echo('result=%s' % json.dumps(r))

    return r

@cli.command()
@click.option('-i', '--imei', help=u'请输入imei')
@click.option('-l', '--logical_code', help=u'请输入逻辑码')
def get(imei=None, logical_code=None):
    if not any([imei, logical_code]):
        raise SystemExit(u'imei logical_code 必提供其一')
    if imei:
        pprint.pprint(Device.get_dev(imei))
    else:
        pprint.pprint(Device.get_dev_by_logicalCode(logical_code))


@cli.command()
@click.option('-i','--imei', help=u'请输入设备号')
@click.option('-l', '--logical_code', help=u'请输入逻辑码')
def get_info(imei=None, logical_code=None):
    """
    e.g. {"cmd":201,"IMEI":"865024038093674"}
    :param imei:
    :param logical_code
    :return:
    """
    if not any([imei, logical_code]):
        raise SystemExit(u'imei logical_code 必提供其一')
    if not imei:
        imei = Device.get_devNo_by_logicalCode(logical_code)
        if not imei:
            raise SystemExit(u'无法找到对应设备号!')
    return _device_command(imei, cmd=201, payload={"cmd":201,"IMEI":imei})

@cli.command()
@click.option('-i','--imei', help=u'请输入设备号')
@click.option('-l', '--logical_code', help=u'请输入逻辑码')
def which_env(imei=None, logical_code=None):
    """
    一个命令告诉设备在哪个环境
    :param imei:
    :param logical_code:
    :return:
    """
    if not any([imei, logical_code]):
        raise SystemExit(u'imei logical_code 必提供其一')
    if not imei:
        imei = Device.get_devNo_by_logicalCode(logical_code)
        if not imei:
            raise SystemExit(u'无法找到对应设备号!')

    for server in SERVERS.values():
        result = _device_command(imei, cmd=201, payload={"cmd":201,"IMEI":imei}, server=server)
        if result['rst'] != -1:
            click.echo(u'设备在%s环境' % (SERVERS.inv[server],))
            break
    else:
        click.echo(u'不知道设备在哪里')


@cli.command()
@click.option('-i', '--imei', help=u'请输入设备号')
@click.option('-l', '--logical_code', help=u'逻辑码')
@click.option('--ip', help=u'请输入IP地址')
@click.option('--port', help=u'请输入端口')
@click.option('-d', '--direction', help=u'转移方向')
def migrate_server(imei, logical_code=None, ip=None, port=None, direction=None):
    """
    e.g. {"cmd":202,"IMEI":"865024038093674","addr_set":{"ip1":"120.26.227.50","port1":30018}}
    testing & staging: 121.40.97.20
    production: 120.27.251.159
    :param imei:
    :param logical_code:
    :param ip:
    :param port:
    :param direction:
    :return:
    """
    direction_config = {
        'prod2test': ('121.40.97.20', 1883),
        'test2prod': ('120.27.251.159', 1883)
    }

    if not any([imei, logical_code]):
        raise SystemExit(u'imei logical_code 必提供其一')
    if not imei:
        imei = Device.get_devNo_by_logicalCode(logical_code)
        if not imei:
            raise SystemExit(u'无法找到对应设备号!')

    if direction and direction in direction_config:
        ip, port = direction_config[direction]
    else:
        ip = settings.MQTT_HOSTNAME if ip is None else ip
        port = 1883 if port is  None else port

    if not all([ip, port]):
        raise SystemExit(u'必须提供ip,port或转移方向')

    return _device_command(imei, cmd=202, payload={"cmd":202,"IMEI":imei, "addr_set":{"ip1":ip,"port1":port}})

@cli.command()
@click.option('-i', '--imei', help=u'请输入设备号')
def factory_reset(imei):
    """
    e.g. {"cmd":202,"IMEI":"865024038093674","factory_set":1}
    :param imei:
    :return:
    """
    return _device_command(imei, cmd=202, payload={"cmd":202,"IMEI":imei,"factory_set":1})

@cli.command()
@click.option('-i', '--imei', help=u'请输入设备号')
@click.option('-p', '--path', help=u'请输入文件地址')
@click.option('-v', '--version', help=u'版本号 如v1.7.5')
def push_upgrade(imei, version, path=None):
    """
    向设备推送升级包
    e.g.
    {"cmd":202,"IMEI":"868575025777667","ota_set":{"fw_url":"http://www.5tao5ai.com/uploaded/SMARTBOX_1.0.0_Luat_V0016_8955_SSL.bin"}}
    :param imei:
    :param path:
    :param version:
    :return: 
    """
    dev = Device.get_dev(imei)
    if dev['softVer'] != version:
        return _device_command(imei, cmd=202,
                                payload={
                                    'cmd':202,
                                    'IMEI':imei,
                                    'ota_set': {
                                        'fw_url': 'http://www.5tao5ai.com/uploaded/SMARTBOX_1.0.0_Luat_V0016_8955_SSL.bin'
                                    }
                                })

if __name__ == '__main__':
    cli()