make2/tools/push_cmd.py

import redis, os, sys,requests,platform,json
if platform.system() == 'Windows':
    sys.path.append('E:\\make2\\libs\\')
    #sys.path.append('libs')
else:
    sys.path.append('/data/deploy/make3d/make2/libs/')
import config, libs,libs_db,libs_db_task_rc

def main(cmd, order_id):
    if cmd == 'print': #order_id
        key = 'model:printOrder'
    elif cmd == 'repair': #reqpair_order_id
        key = 'model:IndependentRepairTeamcheckGLBQueue'
    elif cmd == 'make3d': #pid
        key = 'model:make'
    elif cmd == 'make3d10': #pid
        key = 'model:make10'
    elif cmd == 'foot': # print_id
        key = 'model:foot'
    elif cmd == 'rebuild': #pid
        key = 'model:rebuild'
    elif cmd == 'gs':
        key = 'ai:ai_build'
    elif cmd == 'gs_normals_image':
        key="model:gs_normals_image"

    if order_id == 'view':
        if cmd == "rc_model_build":
            nums = libs_db_task_rc.get_task_rc_count()
            print(f'当前{cmd}队列长度：{nums}')
            notify(f'当前{cmd}队列长度：{nums}')
        else:
            print(f"key-{key}")
            print(f'当前{cmd}队列长度：{r.llen(key)}')
            notify(f'当前{cmd}队列长度：{r.llen(key)}')
            
                
            for i in r.lrange(key, 0, -1):
                print(i)
            print(f'当前{key}队列长度：{r.llen(key)}')
    else:
        order_ids = order_id.split(',')
        for order_id in order_ids:
            r.lpush(key, order_id)
            print(f'已推送{order_id}到{key}, 当前队列长度：{r.llen(key)}')


def notify(content):

    if content == "":
        return "content 不能为空"
    
    for user_agent_id in config.notify_user_Ids:
        data = {
            'userId': user_agent_id,
            'message': content,
        }
        headers = {'Content-Type': 'application/json'}
        message_send_url = "https://mp.api.suwa3d.com/api/qyNotify/sendMessage?userId="+user_agent_id+"&message="+content
        response = requests.post(message_send_url, data=json.dumps(data), headers=headers)



def cmd(cmdName,pid):
    if pid == "view":
        key = ""
        if cmdName == 'print': #order_id
            key = 'model:printOrder'
        elif cmdName == 'repair': #repair_order_id
            key = 'model:IndependentRepairTeamcheckGLBQueue'
        elif cmdName == 'make3d': #pid
            key = 'model:make'
        elif cmdName == 'make3d10': #pid
            key = 'model:make10'
        elif cmdName == 'foot': # print_id
            key = 'model:foot'
            
        if pid == 'view':
            temp = ""
            for i in r.lrange(key, 0, -1):
                print(i)
                temp += i.decode('utf-8') + ","
            print(temp)
            print(f'当前{cmd}队列长度：{r.llen(key)}')
            notify(f'当前{cmd}队列长度：{r.llen(key)}')
            exit()
        else:            
            # for i in r.lrange(key, 0, -1):
            #     print(i)
            print(f'当前{key}队列长度：{r.llen(key)}')
            exit()

    res = requests.get(f'https://mp.api.suwa3d.com/api/infoQuery/infoByPid?pid={pid}&cmd={cmdName}')
    #获取res 的数据
    res = res.json()
    if res["code"] != 1000:
        print("查询失败",res)
        exit(1)
    data = res["data"]

    if cmdName == 'print':
        #需要去吃查询pid 对应的order_id
        #遍历data
        for i in data:
            #获取 pid 对应的数据
            tempData = data[i]
            if len(tempData) == 1:
                #pid 只有对应的一笔数据就可以直接插入了
                main(cmdName, str(tempData[0]["order_id"]))
            else:
                while True:
                    try:
                        
                        tempDataStrings = listDataToStr(tempData,cmdName)
                        order_id = int(input("pid-"+i+"对应有多个数据,请从下列中选择正确的order_id填入\r\n"+tempDataStrings+"\r\n->"))
                        # 在这里处理用户输入的整数
                        print("输入的是: ", order_id)
                        main(cmdName, str(order_id))
                        break
                    except ValueError:
                        print("输入不满足要求，请重新输入")
    elif cmdName == "repair":
        #需要去吃查询pid 对应的repair_order_id
        #遍历data
        for i in data:
            #获取 pid 对应的数据
            tempData = data[i]
            if len(tempData) == 1:
                #pid 只有对应的一笔数据就可以直接插入了
                main(cmdName, str(tempData[0]["model_repair_order_id"]))
            else:
                while True:
                    try:
                        tempDataStrings = listDataToStr(tempData,cmdName)
                        model_repair_order_id = int(input("pid-"+i+"对应有多个数据,请从下列中选择正确的model_repair_order_id填入\r\n"+tempDataStrings+"\r\n->"))
                        # 在这里处理用户输入的整数
                        print("输入的是: ", model_repair_order_id)
                        main(cmdName, str(model_repair_order_id))
                        break
                    except ValueError:
                        print("输入不满足要求，请重新输入")
    elif cmdName == 'foot':
        
        #遍历data
        for i in data:
            #获取 pid 对应的数据
            tempData = data[i]
            if len(tempData) == 1:
                print("tempData",tempData)
                #pid 只有对应的一笔数据就可以直接插入了
                main(cmdName, str(tempData[0]["print_id"]))
            else:
                while True:
                    try:
                        #tempData 转字符串
                        tempDataStrings = listDataToStr(tempData,cmdName)
                        printId = int(input("pid-"+i+"对应有多个数据,请从下列中选择正确的print_id填入\r\n"+tempDataStrings+"\r\n->"))
                        # 在这里处理用户输入的整数
                        print("输入的是: ", printId)
                        main(cmdName, str(printId))
                        break
                    except ValueError:
                        print("输入不满足要求，请重新输入")
    else:
        print("不支持的命令",cmdName)

def listDataToStr(data,cmdName):
    strMessage = ""
    for i in data:
        if cmdName == "foot":
            strMessage += i['createTime']+" print_id:"+str(i['print_id'])+"\r\n"
        elif cmdName == "repair":
            strMessage += i['createTime']+" model_repair_order_id:"+str(i['model_repair_order_id'])+"\r\n"
        else:
            strMessage += i['createTime']+" order_id:"+str(i['order_id'])+"\r\n"
    return strMessage


if __name__ == '__main__':
    r = config.redis_remote
    if len(sys.argv) == 3:
        cmdName = sys.argv[1]
        pid = sys.argv[2]
    
    elif len(sys.argv) == 2:
        cmd = sys.argv[1]
        order_id = 'view'
        if cmd == "auto_view":
            arrCmd = ["print","foot","gs","rc_model_build","gs_normals_image"]
            for cmdV in arrCmd:
                main(cmdV, order_id)
        
        exit(1)
    else:
        print('用法：python push_cmd.py <cmd> <order_id>')
        exit(1)
    
   
    #pid 可能是多个的，用逗号分隔，查询的时候接口应该也要支持多个的
    if cmdName == 'make3d' or cmdName == 'make3d10' or cmdName == 'rebuild':
        main(cmdName, pid)
    elif cmd != "":
        cmd(cmdName,pid)
    else:
        print('用法：python push_cmd.py <cmd> <order_id>')
        exit(1)