"""
头条极速版

吃饭补贴
"""
import os
import random
from strategy.rpa_enum import media_type
from tools import loggerKit
from func.action_tool import \
    read_pic_base64_string, \
    make_task_click_widget, \
    make_task_click_image, \
    check_result_status, \
    make_task_continual_swipe_screen, \
    make_task_stop_app

# 头条app的id
toutiao_package_name = "com.ss.android.article.lite"
# 头条app的版本
toutiao_app_version = "9.7.0.0"


def meal_allowance_task(task_id, data):
    start_pages=0
    max_pages=50

    task_id = str(task_id)
    # 获取设备id
    device_id = data.get("deviceID")
    loggerKit.info("[头条极速版][吃饭补贴任务]请求信息:{0}{1}'.format(task_id,data))")

    """
    检查上一步执行结果状态
    """
    result_status = check_result_status(
        task_id=task_id,
        data=data,
        target_app=media_type.toutiao.value,
        target_version=toutiao_app_version,
        package_name=toutiao_package_name
    )
    if result_status is not None:
        return result_status

    """
    首页点击任务
    """
    step_index = 1
    action_dict = make_task_click_widget(
        data=data,
        step_index=step_index,
        target_app=media_type.toutiao.value,
        target_version=toutiao_app_version,
        package_name=toutiao_package_name,
        control_id='android:id/tabs',
        item_index=2,
        sleep_time=4
    )
    if action_dict is not None:
        # 返回新生成的操作json字典
        return action_dict
    
    """
    点击吃饭补贴入口
    """
    step_index += 1
    # 获取图片base64字符串
    current_dir = os.path.dirname(os.path.abspath(__file__))
    image_path = os.path.join(current_dir, "pic/toutiao_meal_allowance.png")
    encoded_string = read_pic_base64_string(image_path)
    action_dict = make_task_click_image(
        data=data,
        step_index=step_index,
        target_app=media_type.toutiao.value,
        target_version=toutiao_app_version,
        package_name=toutiao_package_name,
        pic_base64=encoded_string,
        sleep_time=4
    )
    if action_dict is not None:
        # 返回新生成的操作json字典
        return action_dict
    
    """
    持续向上滑动屏幕
    """
    random_sleep_time = random.randint(5, 10)
    step_index += 1
    action_dict = make_task_continual_swipe_screen(
        data=data,
        step_index=step_index,
        target_app=media_type.toutiao.value,
        target_version=toutiao_app_version,
        package_name=toutiao_package_name,
        direction="up",
        is_need_loop=True,
        loop_count=1000*5,
        sleep_time=random_sleep_time
    )
    if action_dict is not None:
        # 返回新生成的操作json字典
        return action_dict
    
    """
    关闭APP
    """
    step_index += 1
    return make_task_stop_app(
        task_id,
        data,
        step_index, 
        target_app=media_type.toutiao.value, 
        target_version=toutiao_app_version,
        package_name=toutiao_package_name
    )