jdy_fastapi/app/tasks/common.py

"""
通用后台任务模块

本模块包含所有后台任务通用的功能，包括：
- 简道云表单更新
- 工作流审批
- 获取门店ID
- 获取会员卡列表

这些功能被多个后台任务模块复用。
"""
import logging
import time
import requests
from typing import Dict, Any, List, Optional, Callable
from tqdm import tqdm
from app.api import API

api_instance = API()
logger = logging.getLogger('app')


def update_jiandaoyun(data: Dict[str, Any], results: str):
    """
    更新简道云表单
    
    将后台任务的执行结果更新到简道云表单中。
    
    Args:
        data: 包含表单ID(api_key)、表单ID(entry_id)、数据ID(data_id)的字典
        results: 执行结果信息，将写入到表单的执行明细字段
        
    Returns:
        Dict: 更新结果字典，{'msg': True} 表示成功，{'msg': False} 表示失败
    """
    # 定义简道云数据配置
    jiandaoyun_data = {
        'api_key': data['api_key'],
        'entry_id': data['entry_id'],
        'data_id': data['data_id'],
        "data": {
            '_widget_1731379774828': {"value": "已执行"},  # f6系统批量操作测试 是否执行成功
            '_widget_1731381334870': {"value": results}  # f6系统批量操作测试 执行明细
        }
    }

    time.sleep(1)
    print(jiandaoyun_data)

    try:
        response = api_instance.entry_data_update(jiandaoyun_data)
        logger.info(f"简道云表单更新成功: {response}")
        return {'msg': True}
    except Exception as e:
        logger.error(f"简道云表单更新失败: {e}")
        return {'msg': False}


def approve_workflow(data: Dict[str, Any]):
    """
    获取简道云当前流程节点并直接提交
    
    获取简道云工作流的当前待处理任务，并自动提交到下一步。
    
    Args:
        data: 包含表单ID(api_key)、表单ID(entry_id)、数据ID(data_id)的字典
        
    Returns:
        None
        
    注意:
        如果未找到待处理任务，函数会记录错误并返回，不会抛出异常
    """
    # 获取简道云当前流程列表
    json = api_instance.workflow_instance_get(data)
    
    # 检查返回数据是否有效
    if not json:
        logger.error("未获取到工作流实例信息")
        return
    
    # 安全地获取任务列表
    tasks = json.get('tasks', [])
    if not tasks:
        logger.error("未找到待处理任务")
        return
    
    # 将JSON字符串转换为Python字典
    username = ''
    instance_id = ''
    task_id = ''
    
    for task in tasks:
        if task.get('status') == 0:
            assignee = task.get('assignee', {})
            username = assignee.get('username', '')
            instance_id = task.get('instance_id', '')
            task_id = task.get('task_id', '')
            
            if username and instance_id and task_id:
                break
    
    if not username or not instance_id or not task_id:
        logger.error("未找到有效的待处理任务信息")
        return
    
    task_data = {
        "username": username,
        "instance_id": instance_id,
        "task_id": task_id,
    }

    try:
        response = api_instance.workflow_task_approve(task_data)
        logger.info(f"简道云工作流任务提交成功: {response}")
    except Exception as e:
        logger.error(f"简道云工作流任务提交失败: {e}")


def execute_failure_handler(data: Dict[str, Any]):
    """
    简道云失败流程通知

    函数执行失败时调用，通过钉钉webhook通知到指定人员
    """
    now = time.strftime("%Y-%m-%d %H:%M:%S", time.gmtime())
    pay_load = {
        "api_key":"6694d3c4fcb69ca9a111a6c4",
        "entry_id":"6938e011b360a1132522a62a",
        "data": {
            "_widget_1765335060501": {"value": now}, # 失败时间
            "_widget_1765335060502": {"value": data['failure_name']},  # 任务名称
            "_widget_1765335060503": {"value": data['failure_details']}  # 失败明细
        }
    }

    api_instance.data_batch_create(pay_load)


def get_operate_org_id(cookies: Dict[str, str]) -> Optional[str]:
    """
    获取操作门店ID
    
    从F6系统获取第一个门店的组织ID，用于后续操作。
    
    Args:
        cookies: 用户登录 F6 系统的 cookies 信息
        
    Returns:
        Optional[str]: 门店ID，如果获取失败返回 None
        
    注意:
        如果未获取到门店信息或门店ID为空，会记录错误日志并返回 None
    """
    org_url = "https://yunxiu.f6car.cn/hive/org/getPageOrgGroupMembers?currentPage=1&pageSize=100&name="
    
    try:
        org_res = requests.get(url=org_url, cookies=cookies)
        org_data = org_res.json().get("data", {})
        org_list = org_data.get("list", [])
        
        if not org_list or len(org_list) == 0:
            logger.error("未获取到门店信息")
            return None
        
        operate_org_id = org_list[0].get("orgId")
        if not operate_org_id:
            logger.error("门店ID为空")
            return None
        
        logger.info(f"获取门店ID成功: {operate_org_id}")
        return operate_org_id
    except Exception as e:
        logger.error(f"获取门店ID时发生错误: {e}")
        return None


def get_card_list(
    cookies: Dict[str, str], 
    operate_org_id: str, 
    extract_func: Callable[[Dict], Optional[str]] = None
) -> List[str]:
    """
    获取会员卡列表
    
    从F6系统获取指定门店的会员卡列表，支持自定义提取逻辑。
    
    Args:
        cookies: 用户登录 F6 系统的 cookies 信息
        operate_org_id: 门店ID
        extract_func: 自定义提取函数，用于从会员卡数据中提取ID
                     如果不提供，默认提取 idCustomer 字段
                     
    Returns:
        List[str]: 会员卡ID列表
        
    注意:
        - 默认每页100条数据，会自动分页获取所有数据
        - 每页请求间隔0.2秒，避免请求过快
    """
    card_list = []
    
    try:
        # 获取第一页，确定总页数
        card_url = f"https://yunxiu.f6car.cn/marketing/card/paging?useStationIdOwnOrgList={operate_org_id}&pageSize=100&pageNo=1"
        card_res = requests.get(url=card_url, cookies=cookies)
        total_card = int(card_res.json().get("data", {}).get("total", 0))
        
        if total_card == 0:
            logger.info("未找到会员卡数据")
            return card_list
        
        total_page = total_card // 100 + (total_card % 100 > 0)
        logger.info(f"会员卡总数: {total_card}, 总页数: {total_page}")
        
        # 定义默认提取函数（提取客户ID）
        if extract_func is None:
            def default_extract(card_item: Dict) -> Optional[str]:
                return card_item.get("idCustomer")
            extract_func = default_extract
        
        # 分页获取所有会员卡数据
        for page in tqdm(range(1, total_page + 1), desc="查询会员卡"):
            card_url = (f"https://yunxiu.f6car.cn/marketing/card/paging?useStationIdOwnOrgList={operate_org_id}"
                       f"&pageSize=100&pageNo={page}")
            card_res = requests.get(url=card_url, cookies=cookies)
            card_data_list = card_res.json().get("data", {}).get("data", [])
            
            # 使用提取函数提取ID
            for card_item in card_data_list:
                extracted_id = extract_func(card_item)
                if extracted_id is not None:
                    card_list.append(extracted_id)
            
            time.sleep(0.2)
        
        logger.info(f"获取会员卡列表成功，共 {len(card_list)} 条")
        return card_list
        
    except Exception as e:
        logger.error(f"获取会员卡列表时发生错误: {e}")
        return card_list