简道云V2.0

app/tasks/common.py

@@ -1,10 +1,19 @@
 """
 通用后台任务模块
-包含简道云表单更新和工作流审批等通用功能
+
+本模块包含所有后台任务通用的功能，包括：
+- 简道云表单更新
+- 工作流审批
+- 获取门店ID
+- 获取会员卡列表
+
+这些功能被多个后台任务模块复用。
 """
 import logging
 import time
-from typing import Dict, Any
+import requests
+from typing import Dict, Any, List, Optional, Callable
+from tqdm import tqdm
 from app.api import API
 
 api_instance = API()
@@ -14,9 +23,15 @@ logger = logging.getLogger('app')
 def update_jiandaoyun(data: Dict[str, Any], results: str):
     """
     更新简道云表单
-    :param data: 包含表单id、应用id、数据id的字典
-    :param results: 执行结果信息
-    :return: 更新结果字典
+    
+    将后台任务的执行结果更新到简道云表单中。
+    
+    Args:
+        data: 包含表单ID(api_key)、表单ID(entry_id)、数据ID(data_id)的字典
+        results: 执行结果信息，将写入到表单的执行明细字段
+        
+    Returns:
+        Dict: 更新结果字典，{'msg': True} 表示成功，{'msg': False} 表示失败
     """
     # 定义简道云数据配置
     jiandaoyun_data = {
@@ -44,8 +59,17 @@ def update_jiandaoyun(data: Dict[str, Any], results: str):
 def approve_workflow(data: Dict[str, Any]):
     """
     获取简道云当前流程节点并直接提交
-    :param data: 包含表单id、应用id、数据id的字典
-    :return: None
+    
+    获取简道云工作流的当前待处理任务，并自动提交到下一步。
+    
+    Args:
+        data: 包含表单ID(api_key)、表单ID(entry_id)、数据ID(data_id)的字典
+        
+    Returns:
+        None
+        
+    注意:
+        如果未找到待处理任务，函数会记录错误并返回，不会抛出异常
     """
     # 获取简道云当前流程列表
     json = api_instance.workflow_instance_get(data)
@@ -92,3 +116,108 @@ def approve_workflow(data: Dict[str, Any]):
     except Exception as e:
         logger.error(f"简道云工作流任务提交失败: {e}")
 
+
+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=10&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
+