robots-auto-action/yiche/yiche_direction_action.py

"""
易车互动
"""
import json
import time
from urllib.parse import quote

import httpx
import yaml

from func.action_func import del_key_vague
from scene.oprator.atom_data import single_click_by_control, send_text_by_control, click_by_content, stop_app, \
    start_app, single_click_by_text
from task.task_job import call_back_task_reply
from tools import redis_client, loggerKit

with open('config.yaml', 'r') as file:
    config = yaml.load(file, Loader=yaml.FullLoader)

extern_domain = config['bmp-cp']['extern_domain']

# 任务执行回调url
task_callback_url = extern_domain + config['bmp-cp']['task_callback_url']

# 调用AIGC接口
post_ai_gc_url = extern_domain + config['bmp-content-center']['comment_local_url']
get_commented_list_url = extern_domain + config['bmp-cp']['get_commented_list_url']


def yiche_direction_action(task_id, device_id, data):
    loggerKit.info('请求信息:{0}'.format(data))

    device_id = data.get("deviceID")
    perform_action_id = data.get("performActionId")
    result = data.get("result")
    inner_data = data.get("data")
    # title = inner_data.get("resourceName")
    # author = inner_data.get("subResourceName")
    extension_info = data['data']['extendInfo']
    extension_info = json.loads(extension_info)
    title = extension_info['title']
    author = extension_info['author']

    loggerKit.info("inner_data:{0}, keyword:{1}", inner_data, inner_data.get("resourceName"))

    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)
            call_back_task_reply(500, '指令执行失败', task_id, device_id, 0, None, None)
            return return_dict

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

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

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

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

        # 终止指令
        if perform_action_result == "eleNotFound":
            # 回调任务中心
            return_dict = {
                "data": "",
                "code": -2,
                "message": "未找到签到指示"
            }

            del_key_vague(device_id)
            call_back_task_reply(500, '未找到签到指示,该账号当天可能已经签到过', task_id, device_id, 0, None, None)

            return return_dict

        # 元素未找到
        if perform_action_result == "invalid operation":
            # 回调任务中心
            return_dict = {
                "data": "",
                "code": -2,
                "message": "fail, performActionResult is not success"
            }
            del_key_vague(device_id)
            call_back_task_reply(500, '任务执行失败,元素未找到', task_id, device_id, 0, None, 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="yiche",
                                                   target_version="11.11.0_c32",
                                                   package_name="com.yiche.autoeasy",
                                                   control_id="com.yiche.autoeasy:id/hbe")

            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) and perform_action_result == "success":
            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="yiche",
                                                target_version="11.11.0_c32",
                                                package_name="com.yiche.autoeasy",
                                                content=author,
                                                control_id="com.yiche.autoeasy:id/ixb")

            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("设备:{0}, action2_id:{1}", device_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):
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))
            action3_id = int(round(time.time() * 1000))
            """
            发送第3条指令
            点击搜索按钮
            """
            action3_dict = single_click_by_control(action3_id,
                                                   target_app="yiche",
                                                   target_version="11.11.0_c32",
                                                   package_name="com.yiche.autoeasy",
                                                   control_id="com.yiche.autoeasy:id/ixc",
                                                   control_ids="com.yiche.autoeasy:id/ixc")

            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):
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))
            action4_id = int(round(time.time() * 1000))
            """
            发送第4条指令
            点击tab->文章
            """
            action4_dict = single_click_by_text(action4_id,
                                                target_app="yiche",
                                                target_version="11.11.0_c32",
                                                package_name="com.yiche.autoeasy",
                                                text="文章")
            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 action4_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):
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))
            action5_id = int(round(time.time() * 1000))
            """
            发送第5条指令
            点击tab->小视频
            """
            action5_dict = single_click_by_text(action5_id,
                                                target_app="yiche",
                                                target_version="11.11.0_c32",
                                                package_name="com.yiche.autoeasy",
                                                text="小视频")
            redis_client.set(device_id + "operate", action5_id)
            redis_client.set(f"{device_id}_step5", "1")
            redis_client.delete(f"{device_id}_step4")

            return action5_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):
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))
            action6_id = int(round(time.time() * 1000))
            """
            发送第6条指令
            点击tab->摩托车
            """
            action6_dict = single_click_by_text(action6_id, target_app="yiche",
                                                target_version="11.11.0_c32",
                                                package_name="com.yiche.autoeasy",
                                                text="摩托车")

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

            return action6_dict

        step6 = redis_client.get(f"{device_id}_step6")
        if step6 is not None and int(step6) == 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))
            action7_id = int(round(time.time() * 1000))
            """
            点击tab->用户
            """
            action7_dict = single_click_by_text(action7_id,
                                                target_app="yiche",
                                                target_version="11.11.0_c32",
                                                package_name="com.yiche.autoeasy",
                                                text="用户")

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

            return action7_dict

        step7 = redis_client.get(f"{device_id}_step7")
        if step7 is not None and int(step7) == 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))
            action8_id = int(round(time.time() * 1000))
            """
            点击搜索到的博主
            """
            action8_dict = single_click_by_control(action8_id,
                                                   target_app="yiche",
                                                   target_version="11.11.0_c32",
                                                   package_name="com.yiche.autoeasy",
                                                   control_id="com.yiche.autoeasy:id/lhh",
                                                   control_ids="com.yiche.autoeasy:id/lhh")

            redis_client.set(device_id + "operate", action8_id)
            redis_client.set(f"{device_id}_step8", "1")
            redis_client.delete(f"{device_id}_step7")

            return action8_dict

        """
        根据标题搜索博主的发布的内容
        """
        step8 = redis_client.get(f"{device_id}_step8")
        if step8 is not None and int(step8) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id) \
                and perform_action_result == "success":
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))
            action9_id = int(round(time.time() * 1000))
            """
            发送第9条指令
            向下滑动若干屏幕
            根据标题搜索博主的发布的内容
            """
            action9_dict = click_by_content(action9_id,
                                            target_app="yiche",
                                            target_version="11.11.0_c32",
                                            package_name="com.yiche.autoeasy",
                                            title=title)

            redis_client.set(device_id + "operate", action9_id)
            redis_client.set(f"{device_id}_step9", "1")
            redis_client.delete(f"{device_id}_step8")
            loggerKit.info("设备:{0}, action9_id:{1}", device_id, action9_id)

            return action9_dict

        """
        搜藏搜索到的帖子
        """
        step9 = redis_client.get(f"{device_id}_step9")
        if step9 is not None and int(step9) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id) \
                and perform_action_result == "success":
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))

            action10_id = int(round(time.time() * 1000))
            """
            发送第10条指令
            搜藏搜索到的帖子
            """
            action10_dict = single_click_by_control(action10_id, target_app="yiche",
                                                    target_version="11.11.0_c32",
                                                    package_name="com.yiche.autoeasy",
                                                    control_id="com.yiche.autoeasy:id/aai",
                                                    control_ids="com.yiche.autoeasy:id/aai")
            redis_client.set(device_id + "operate", action10_id)
            redis_client.set(f"{device_id}_step10", "1")
            redis_client.delete(f"{device_id}_step9")
            loggerKit.info("设备:{0}, action10_id:{1}", device_id, action10_id)

            return action10_dict

        """
        点赞搜索到的帖子
        """
        step10 = redis_client.get(f"{device_id}_step10")
        if step10 is not None and int(step10) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id) \
                and perform_action_result == "success":
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))

            action11_id = int(round(time.time() * 1000))
            """
            发送第11条指令
            点赞搜索到的帖子
            """
            action11_dict = single_click_by_control(action11_id, target_app="yiche",
                                                    target_version="11.11.0_c32",
                                                    package_name="com.yiche.autoeasy",
                                                    control_id="com.yiche.autoeasy:id/d44",
                                                    control_ids="com.yiche.autoeasy:id/d44")
            redis_client.set(device_id + "operate", action11_id)
            redis_client.set(f"{device_id}_step12", "1")
            redis_client.delete(f"{device_id}_step10")
            loggerKit.info("设备:{0}, action11_id:{1}", device_id, action11_id)

            return action11_dict



        """
        点击评论输入框
        """
        step12 = redis_client.get(f"{device_id}_step12")
        if step12 is not None and int(step12) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id) \
                and perform_action_result == "success":
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))

            action13_id = int(round(time.time() * 1000))
            """
            发送第13条指令
            评论搜索到的帖子
            点击评论按钮
            """
            comment_response = httpx.get(get_commented_list_url + '?resourceName=' + title, timeout=120)

            # 评论实体
            comment_body = json.loads(comment_response.text)

            # 评论集合
            comment_list = comment_body.get('data')
            loggerKit.info("任务id:{0}对应标题:{1},已有评论:{2}", task_id, title, comment_list)
            existing_comment = ''

            if comment_list is not None:
                comment_list = list(set(comment_list))
                existing_comment = '||'.join(comment_list)

            loggerKit.info("任务id:{0}对应标题:{1},拼接后的评论集合:{2}", task_id, title, existing_comment)
            existing_comment = quote(existing_comment)

            request_data = {
                "mode": "1",
                "title": title,
                "platform": "yiche",
                "comment": existing_comment
            }

            loggerKit.info("任务id:{0}对应标题:{1},请求AIGC信息:{2}", task_id, title, request_data)
            response = httpx.post(post_ai_gc_url, json=request_data, timeout=120)

            # 点击评论框
            action13_dict = single_click_by_control(action13_id, target_app="yiche",
                                                    target_version="11.11.0_c32",
                                                    package_name="com.yiche.autoeasy",
                                                    control_id="com.yiche.autoeasy:id/d3p",
                                                    control_ids="com.yiche.autoeasy:id/d3p")

            if not response.is_success:
                #  调用AIGC获取评论失败
                loggerKit.info("任务id:{0}对应标题:{1},请求AIGC失败信息:{2},返回信息:{3}", task_id, title,
                               request_data, response)

                redis_client.set(device_id + "yiche" + "comments", '不错👍')
                redis_client.set(device_id + "operate", action13_id)
                redis_client.set(f"{device_id}_step13", "1")
                redis_client.delete(f"{device_id}_step12")
                loggerKit.info("设备:{0}, action13_id:{1}", device_id, action13_id)

            response_body = json.loads(response.text)

            reply_data = response_body.get('data')

            reply = reply_data.get('comment')

            if '内容太少' in reply or '对不起' in reply or reply is None or reply == '' or '提供' in reply:
                #  获取到的评论内容不符合
                loggerKit.info("任务id:{0}对应标题:{1},请求AIGC失败信息:{2},返回信息:{3}", task_id, title, request_data,
                               response)

                redis_client.set(device_id + "yiche" + "comments", '不错👍')

                redis_client.set(device_id + "operate", action13_id)
                redis_client.set(f"{device_id}_step13", "1")
                redis_client.delete(f"{device_id}_step12")

                loggerKit.info("设备:{0}, action10_id:{1}", device_id, action13_id)

                return action13_dict

            # 将需要评论的内容存入缓存
            redis_client.set(device_id + "yiche" + "comments", reply)

            redis_client.set(device_id + "operate", action13_id)
            redis_client.set(f"{device_id}_step13", "1")
            redis_client.delete(f"{device_id}_step12")

            loggerKit.info("设备:{0}, action10_id:{1}", device_id, action13_id)

            return action13_dict

        """
        输入评论内容
        """
        step13 = redis_client.get(f"{device_id}_step13")
        if step13 is not None and int(step13) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id) \
                and perform_action_result == "success":
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))

            action14_id = int(round(time.time() * 1000))
            """
            发送第14条指令
            输入评论内容
            """
            reply = redis_client.get(device_id + "yiche" + "comments")
            if reply is None:
                #  调用AIGC获取评论失败
                del_key_vague(device_id)
                return_dict = {
                    "data": "",
                    "code": -2,
                    "message": "fail, result_text is null"
                }
                call_back_task_reply(500, '获取到的内容标题为null,无法评论',
                                     task_id, device_id, 0, None, None)

                return return_dict

            action14_dict = send_text_by_control(action14_id, target_app="yiche",
                                                 target_version="11.11.0_c32",
                                                 package_name="com.yiche.autoeasy",
                                                 control_id="com.yiche.autoeasy:id/efd",
                                                 content=reply,
                                                 timeout=5)

            content_result = content_detail(title, title, reply, None)


            redis_client.set(device_id + "reply_json", json.dumps(content_result.to_dict(), ensure_ascii=False))

            redis_client.set(device_id + "operate", action14_id)
            redis_client.set(f"{device_id}_step14", "1")
            redis_client.delete(f"{device_id}_step13")
            loggerKit.info("设备:{0}, action14_id:{1}", device_id, action14_id)

            return action14_dict

        """
        点击评论发送按钮
        """
        step14 = redis_client.get(f"{device_id}_step14")
        if step14 is not None and int(step14) == 1 and last_action_id is not None and int(perform_action_id) == int(
                last_action_id) \
                and perform_action_result == "success":
            loggerKit.info("设备:{0}, last_action_id:{1}", device_id, int(last_action_id))

            action15_id = int(round(time.time() * 1000))
            """
            发送第15条指令
            点击评论发送按钮
            """
            action15_dict = single_click_by_control(action15_id, target_app="yiche",
                                                    target_version="11.11.0_c32",
                                                    package_name="com.yiche.autoeasy",
                                                    control_id="com.yiche.autoeasy:id/j09",
                                                    control_ids="com.yiche.autoeasy:id/j09",
                                                    timeout=5)
            redis_client.set(device_id + "operate", action15_id)
            redis_client.set(f"{device_id}_step15", "1")
            redis_client.delete(f"{device_id}_step14")
            loggerKit.info("设备:{0}, action11_id:{1}", device_id, action15_id)

            return action15_dict

        step15 = redis_client.get(f"{device_id}_step15")
        if step15 is not None and int(step15) == 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))
            action16_id = int(round(time.time() * 1000))
            """
            发送第16条指令
            关闭app
            """
            action16_dict = stop_app(action16_id, target_app="yiche",
                                     target_version="11.11.0_c32",
                                     package_name="com.yiche.autoeasy")
            reply_json = redis_client.get(device_id + "reply_json")

            del_key_vague(device_id)
            loggerKit.info("设备:{0}, action12_id:{1}", device_id, action16_id)
            # 回调任务中心修改任务状态
            call_back_task_reply(None, None, task_id, device_id, 1, None, reply_json)

            return action16_dict

    else:

        action0_id = int(round(time.time() * 1000))

        """
        启动指令
        启动app
        """
        action0_dict = start_app(action0_id, target_app="yiche",
                                 target_version="11.11.0",
                                 package_name="com.yiche.autoeasy")
        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


# 回复详情
class content_detail:
    def __init__(self, keyword, title, reply, account_name):
        """

        :rtype: object
        """
        # 搜索关键词
        self.keyword = keyword
        self.title = title
        self.reply = reply
        self.account_name = account_name

    def to_dict(self):
        return {
            'keyword': self.keyword,
            'title': self.title,
            'reply': self.reply,
            'accountName': self.account_name
        }