manyi/script/hd_v3_update_bin.py

# coding=utf-8
from base import init_env

init_env(True)

from apps.web.device.models import Device, DeviceDict
from mongoengine import QuerySet
from collections import defaultdict
from apps.web.core.networking import MessageSender
from apps.web.core.helpers import ActionDeviceBuilder


TARGET_DRIVER_CODE = "100259"
TARGET_DRIVER_VERSION = "v5.1.5"
FILE_MAP = {
    "4gv1": {"fw_url": "http://121.43.232.118/uploaded/version/4g/SmartBox_6.1.157_Luat_V0036_ASR1802_720D_4gsck_100259.bin"},
    "4gv2": {"fw_url": "http://121.43.232.118/uploaded/version/4g/SmartBox_6.1.157_Luat_V0036_ASR1802_720D_4gv2_100259.bin"},
    "4gv3": {"fw_url": "http://121.43.232.118/uploaded/version/4g/SmartBox_6.1.157_Luat_V0036_ASR1802_720D_4gv3_100259.bin"},

    "4gcat1_0017": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__0017__0032.bin"},
    "4gcat1_0019": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__0019__0032.bin"},
    "4gcat1_0022": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__0022__0032.bin"},
    "4gcat1_0024": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__0024__0032.bin"},
    "4gcat1_0030": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__0030__0032.bin"},
    "4gcat1_0031": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__0031__0032.bin"},
    "4gcat1_0032": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__0032__0032.bin"},

    "4gcat1_3028": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__3028__3037.bin"},
    "4gcat1_3029": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__3029__3037.bin"},
    "4gcat1_3032": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__3032__3037.bin"},
    "4gcat1_3035": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__3035__3037.bin"},
    "4gcat1_3037": {"fw_url": "http://121.43.232.118/uploaded/version/cat1/100259__3037__3037.bin"},

}


def get_update_device_by_group(groupId):
    """
    通过地址获取设备
    :param groupId:
    :return:
    """
    groupId = groupId.strip()
    return Device.objects.filter(
        groupId=groupId
    )


def get_update_device_by_nos(nos):
    """
    通过编号获取设备
    :param nos: 设备编号字符串
    :return:
    """
    nos = nos.strip()
    logicalCodes = nos.split()
    return Device.objects.filter(
        logicalCode__in=logicalCodes
    )


def parse_devices(devs):    # type:(QuerySet) -> (int, dict)
    """
    分析设备的情况
    :param devs:
    :return:
    """
    # 符合条件的升级设备的总数量
    count = devs.count()
    devVerMap = defaultdict(list)

    for dev in devs:    # type: Device

        # 4gv1 的设备
        if dev.softVer.startswith("v50.1") or dev.hwVer in ["4gv1", "4gsck"]:
            devVerMap["4gv1"].append(dev.devNo)

        # 4gv2 的设备
        elif dev.softVer.startswith("v50.12") or dev.hwVer in ["4gv2"]:
            devVerMap["4gv2"].append(dev.devNo)

        # 4gv3 的设备
        elif dev.softVer.startswith("v50.13") or dev.hwVer in ["4gv3"]:
            devVerMap["4gv3"].append(dev.devNo)

        # 4g_v2_8910
        elif dev.softVer.startswith("v60.12") or dev.hwVer == "4gcat1":
            coreVerStr = dev.coreVer[6:10]
            devVerMap["4gcat1_{}".format(coreVerStr)].append(dev.devNo)

        # 其余的2G的设备
        else:
            devVerMap["other"].append(dev.devNo)

    return count, devVerMap


def main_show():
    """
    首页功能展示
    :return:
    """
    print """
    和动V3升级系统
    1. 输入设备组（需要输入地址组ID）
    2. 输入设备编号（需要输入设备编号）
    """


def device_show(count, verMap):
    """
    设备 版本分布 展示
    :param count:
    :param verMap:
    :return:
    """
    print """
    共查询符合条件设备 {} 台
    版本分布情况如下:
    """.format(count)
    for _ver, _devNos in verMap.items():
        print u"版本： {}， 共 {}台".format(_ver, _devNos)


def update_device(dev, ota_set):    # type:(DeviceDict, str) -> None
    print """
    设备编号： {}
    设备所属地址: {} 
    设备软件版本： {}  - {}
    设备硬件版本： {}
    设备驱动版本： {}
    设备驱动编码： {}
    在线状态： {}
    升级文件: {}
    """.format(
        dev.logicalCode,
        dev.group.groupName,
        dev.softVer,
        dev.coreVer,
        dev.hwVer,
        dev.driverVersion,
        dev.driverCode,
        dev.online,
        ota_set
    )

    isContinue = raw_input(u"是否继续升级（y/n）?")
    if isContinue != "y":
        print u"不继续升级 退出程序"
        return

    MessageSender.send(
        device=dev,
        cmd=202,
        payload={
            "IMEI": dev.devNo,
            "ota_set": ota_set
        }
    )


def send_mqtt_to_update_devices(devs):
    """
    升级设备
    :param devs:
    :return:
    """
    # 分析设备的情况
    # 再次过滤一次
    devs = devs.filter(driverVersion__ne=TARGET_DRIVER_VERSION)

    count, verMap = parse_devices(devs)
    device_show(count, verMap)

    isContinue = raw_input(u"是否继续升级（y/n）?")
    if isContinue != "y":
        print u"不继续升级 退出程序"
        return

    for _ver, _devNos in verMap.items():

        _ota_set = FILE_MAP.get(_ver, u"未知版本")

        isContinue = raw_input(u"当前处理升级版本是 {}, 共有设备{}, 升级地址 {}. 是否继续升级（y/n）?".format(_ver, len(_devNos), _ota_set))
        if isContinue != "y":
            print u"不继续升级 退出程序"
            continue

        for _devNo in _devNos:

            # 获取设备 以及相应的升级文件
            _dev = Device.get_dev(_devNo)

            # 进入升级设备
            update_device(_dev, _ota_set)


def send_mqtt_to_update_settings(devs):
    """
    更新设备配置
    :param devs:
    :return:
    """
    model_devNo = '863488053227925'
    model_device = Device.objects.get(devNo=model_devNo)

    model_box = ActionDeviceBuilder.create_action_device(Device.get_dev(model_devNo))
    model_settings = model_box.get_dev_setting()

    for _dev in devs:
        # 数据库数据保存
        device = Device.objects.get(devNo=_dev.devNo)        # type: Device
        device.washConfig = model_device.washConfig
        device.devType = model_device.devType
        device.otherConf = model_device.otherConf

        # 退款的添加
        device.autoRefundLeftMoney = True
        device.save()

        Device.invalid_device_cache(_dev.devNo)

        # 主板参数下发
        box = ActionDeviceBuilder.create_action_device(Device.get_dev(_dev.devNo))
        box.set_dev_setting(model_settings)

        # 缓存清除
        Device.invalid_device_cache(_dev.devNo)


def main():

    main_show()

    # 查找设备
    enterTypes = raw_input(u"输入选择：")
    if enterTypes == "1":
        groupId = raw_input(u"请输入地址组ID：")
        devs = get_update_device_by_group(groupId)
    elif enterTypes == "2":
        nos = raw_input(u"请输入设备编号 多个设备编号之间以空格分隔：")
        devs = get_update_device_by_nos(nos)
    else:
        print u"错误的选择! 退出程序"
        return

    handle = raw_input(u"""
        请选择需要进行的操作
        1. 升级设备
        2. 更新设备配置
    """)
    if handle == "1":
        send_mqtt_to_update_devices(devs)
    elif handle == "2":
        send_mqtt_to_update_settings(devs)
    else:
        print u"错误的选择! 退出程序"
        return


if __name__ == '__main__':
    main()