robots-auto-action/dongchedi/dongchedi_interaction.py

import time

from func.action_func import del_key_vague
from scene.oprator.atom_data import single_click_by_control, stop_app, start_app, \
    single_click_by_text, click_pic, send_text_by_control, get_content_by_control
from task.task_job import callback_task
from tools import loggerKit, redis_client
from tools.pic_base64_util import pic_to_base64

"""
懂车帝策略互动
"""


def dongchedi_interaction(task_id, mobile, data, sub_resource_name):
    loggerKit.info('请求信息:{0}'.format(data))

    device_id = data.get("deviceID")
    keyword = data.get("resourceName")
    perform_action_id = data.get("performActionId")
    result = data.get("result")

    if result is not None:
        """
        非首个指令
        """
        perform_action_result = result.get("performActionResult")
        if perform_action_result is None:
            return_dict = {
                "data": "",
                "code": -2,
                "message": "fail, performActionResult is null"
            }

            # 回调任务中心
            del_key_vague(device_id)
            callback_task(500, '指令执行失败', task_id, mobile, 0, None)
            return return_dict

        # 指令执行失败
        if perform_action_result == "failure":
            # 回调任务中心
            return_dict = {
                "data": "",
                "code": -2,
                "message": "fail, performActionResult is null"
            }

            del_key_vague(device_id)
            callback_task(500, '指令执行失败', task_id, mobile, 0, None)
            return return_dict

        # 终止指令
        if perform_action_result == "stop":
            # 回调任务中心
            return_dict = {
                "data": "",
                "code": -2,
                "message": "指令被用户终止"
            }

            del_key_vague(device_id)
            callback_task(500, '指令被用户终止', task_id, mobile, 0, None)
            return return_dict

        """
        每次操作完成后会将对应的操作唯一id存储到redis
        并且返回给手机端 手机端下次带着上个操作id来执行下一个操作
        """
        last_action_id = redis_client.get(device_id + "operate")
        step0 = redis_client.get(f"{device_id}_step0")
        if step0 is not None and int(step0) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id) and perform_action_result == "success":
            action1_id = int(round(time.time() * 1000))
            """
            发送第1条指令
            点击搜索按钮
            """
            action1_dict = single_click_by_control(action1_id, target_app="dongchedi", target_version="7.9.0",
                                                   package_name="com.ss.android.auto",
                                                   control_id="com.ss.android.auto:id/d34", timeout=5, sleep_time=5,
                                                   item_index=4)
            redis_client.set(device_id + "operate", action1_id)
            redis_client.set(f"{device_id}_step1", "1")
            redis_client.delete(f"{device_id}_step0")

            loggerKit.info("taskId:{0}, action1_id:{1}", task_id, action1_id)

            return action1_dict

        step1 = redis_client.get(f"{device_id}_step1")
        if step1 is not None and int(step1) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id):
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))

            action2_id = int(round(time.time() * 1000))
            """
            发送第2条指令
            搜索框赋值
            """
            action2_dict = send_text_by_control(action2_id, target_app="dongchedi", target_version="7.9.0",
                                                package_name="com.ss.android.auto",
                                                control_id="com.ss.android.auto:id/h5q",
                                                content=keyword,
                                                timeout=5, sleep_time=5)

            redis_client.set(device_id + "operate", action2_id)
            redis_client.set(f"{device_id}_step2", "1")
            redis_client.delete(f"{device_id}_step1")
            loggerKit.info("taskId:{0}, action2_id:{1}", task_id, action2_id)

            return action2_dict

        step2 = redis_client.get(f"{device_id}_step2")
        if step2 is not None and int(step2) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id):
            action3_id = int(round(time.time() * 1000))
            """
            发送第3条指令
            点击搜索
            """

            action3_dict = single_click_by_control(action3_id, target_app="dongchedi", target_version="7.9.0",
                                                   package_name="com.ss.android.auto",
                                                   control_id="com.ss.android.auto:id/g9e", timeout=5, sleep_time=5,
                                                   item_index=4)

            redis_client.set(device_id + "operate", action3_id)
            redis_client.set(f"{device_id}_step3", "1")
            redis_client.delete(f"{device_id}_step2")

            loggerKit.info("taskId:{0}, action3_id:{1}", task_id, action3_id)

            return action3_dict

        step3 = redis_client.get(f"{device_id}_step3")
        if step3 is not None and int(step3) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id):
            action4_id = int(round(time.time() * 1000))
            """
            发送第4条指令
            根据标题名称进行点击
            """
            action3_dict = single_click_by_text(action4_id, target_app="dongchedi", target_version="7.9.0",
                                                package_name="com.ss.android.auto",
                                                text=keyword, timeout=5, sleep_time=5, similarity=0.8)

            redis_client.set(device_id + "operate", action4_id)
            redis_client.set(f"{device_id}_step4", "1")
            redis_client.delete(f"{device_id}_step3")

            loggerKit.info("taskId:{0}, action4_id:{1}", device_id, action4_id)

            return action3_dict

        step4 = redis_client.get(f"{device_id}_step4")
        if step4 is not None and int(step4) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id):
            action5_id = int(round(time.time() * 1000))
            """
            发送第5条指令
            判断有没有点击到标题
            如果点击到标题中 进行评论点赞收藏操作
            如果没有点击到标题中 进行搜索用户操作
            """
            if perform_action_result == "invalid operation":
                action5_dict = send_text_by_control(action5_id, target_app="dongchedi", target_version="7.9.0",
                                                    package_name="com.ss.android.auto",
                                                    control_id="com.ss.android.auto:id/h5q",
                                                    content=sub_resource_name,
                                                    timeout=5, sleep_time=5)

                loggerKit.info("taskId:{0}, action25_id:{1},未根据帖子名找到对应的帖子,进行用户名搜索", device_id, action5_id)

                redis_client.set(device_id + "operate", action5_id)
                redis_client.set(f"{device_id}_step25", "1")
                redis_client.delete(f"{device_id}_step4")

                return action5_dict

            # 已经点击进入帖子 判断是不是视频。如果是视频不进行操作

            action4_dict = single_click_by_text(action5_id, target_app="dongchedi", target_version="7.9.0",
                                                package_name="com.ss.android.auto",
                                                text='视频', timeout=5, sleep_time=5)

            redis_client.set(device_id + "operate", action5_id)
            redis_client.set(f"{device_id}_step5", "1")
            redis_client.delete(f"{device_id}_step4")

            loggerKit.info("taskId:{0}, action5_id:{1},已进入帖子 判断是否未视频", device_id, action5_id)

            return action4_dict

        step5 = redis_client.get(f"{device_id}_step5")
        if step5 is not None and int(step5) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id):
            action6_id = int(round(time.time() * 1000))
            """
            发送第6条指令
            如果有视频对应文案，则返回失败 不支持视频操作
            如果没有获取到对应的帖子内容
            """
            if perform_action_result == 'success':
                # 回调任务中心
                return_dict = {
                    "data": "",
                    "code": -2,
                    "message": "fail, performActionResult is not success"
                }

                del_key_vague(device_id)

                callback_task(500, '该帖子为视频帖子,暂不支持', task_id, device_id, 0, None)

                return return_dict

            action3_dict = get_content_by_control(action6_id, target_app="dongchedi", target_version="7.9.0",
                                                package_name="com.ss.android.auto",
                                                control_id="", timeout=5, sleep_time=5)

            redis_client.set(device_id + "operate", action6_id)
            redis_client.set(f"{device_id}_step6", "1")
            redis_client.delete(f"{device_id}_step5")

            loggerKit.info("taskId:{0}, action4_id:{1}", device_id, action6_id)

            return action3_dict

        step3 = redis_client.get(f"{device_id}_step3")
        if step3 is not None and int(step3) == 1 and (last_action_id is not None
                                                      and int(perform_action_id) == int(last_action_id)):
            loggerKit.info("设备:{0}, action4_id_mem:{1}", device_id, int(last_action_id))
            action4_id = int(round(time.time() * 1000))
            """
            发送第4条指令
            关闭app
            """
            action4_dict = stop_app(action4_id, target_version="7.9.0", target_app="dongchedi", timeout=5, sleep_time=3)
            del_key_vague(device_id)
            loggerKit.info("设备:{0}, action4_id:{1}", device_id, action4_dict)
            # 回调任务中心修改任务状态
            callback_task(None, None, task_id, mobile, 1, None)

            return action4_dict
    else:
        action0_id = int(round(time.time() * 1000))

        """
        启动指令
        启动app
        """
        action0_dict = start_app(action0_id, target_version="7.9.0")
        redis_client.set(device_id + "operate", action0_id)
        redis_client.set(f"{device_id}_step0", "1")
        loggerKit.info("设备:{0}, action0_id:{1}", device_id, action0_id)

        return action0_dict