qrpa 1.0.34__py3-none-any.whl → 1.1.50__py3-none-any.whl

qrpa/feishu_client.py

@@ -0,0 +1,410 @@
+# -*- coding: UTF-8 -*-
+import requests
+import json
+import time
+import base64, os
+import threading
+from urllib.parse import urlparse
+from typing import Optional, Dict, List, Any
+from collections import deque
+
+from .fun_base import log
+
+class FeishuClient:
+    def __init__(self, app_id: str, app_secret: str, max_requests_per_second: int = 90, skip_token_refresh: bool = False):
+        self.app_id = app_id
+        self.app_secret = app_secret
+        self.base_url = "https://open.feishu.cn/open-apis"
+        self.tenant_access_token = None
+        self.token_expire_time = 0
+        
+        # 速率限制相关参数
+        self.max_requests_per_second = max_requests_per_second
+        self.request_times = deque()  # 存储请求时间戳
+        self.rate_limit_lock = threading.Lock()  # 保证线程安全
+        
+        # 只有在不跳过token刷新时才获取token
+        if not skip_token_refresh:
+            self._refresh_token()
+
+    def _refresh_token(self):
+        url = f"{self.base_url}/auth/v3/tenant_access_token/internal"
+        payload = {
+            "app_id"    : self.app_id,
+            "app_secret": self.app_secret
+        }
+        headers = {
+            'Content-Type': 'application/json; charset=utf-8'
+        }
+
+        response = requests.post(url, headers=headers, json=payload)
+        response.raise_for_status()
+        response_data = response.json()
+
+        if response_data.get("code") == 0:
+            self.tenant_access_token = response_data.get("tenant_access_token")
+            expire_seconds = response_data.get("expire", 7200)
+            self.token_expire_time = time.time() + expire_seconds - 300
+        else:
+            raise Exception(f"获取token失败: {response_data.get('msg', '未知错误')}")
+
+    def _ensure_valid_token(self):
+        if not self.tenant_access_token or time.time() > self.token_expire_time:
+            self._refresh_token()
+    
+    def _wait_for_rate_limit(self):
+        """
+        实现请求频率限制，确保不超过90次/秒
+        使用滑动窗口算法控制请求频率
+        """
+        with self.rate_limit_lock:
+            current_time = time.time()
+            
+            # 清理1秒前的请求记录
+            while self.request_times and current_time - self.request_times[0] > 1.0:
+                self.request_times.popleft()
+            
+            # 如果当前秒内请求数已达到限制，则等待
+            if len(self.request_times) >= self.max_requests_per_second:
+                # 计算需要等待的时间
+                oldest_request_time = self.request_times[0]
+                wait_time = 1.0 - (current_time - oldest_request_time)
+                if wait_time > 0:
+                    log(f"请求频率限制: 等待 {wait_time:.3f} 秒")
+                    time.sleep(wait_time)
+                    # 重新获取当前时间并清理过期记录
+                    current_time = time.time()
+                    while self.request_times and current_time - self.request_times[0] > 1.0:
+                        self.request_times.popleft()
+            
+            # 记录当前请求时间
+            self.request_times.append(current_time)
+    
+    def get_current_request_rate(self):
+        """
+        获取当前的请求频率 (最近1秒内的请求数)
+        
+        Returns:
+            int: 最近1秒内的请求数
+        """
+        with self.rate_limit_lock:
+            current_time = time.time()
+            # 清理1秒前的请求记录
+            while self.request_times and current_time - self.request_times[0] > 1.0:
+                self.request_times.popleft()
+            return len(self.request_times)
+    
+    def set_rate_limit(self, max_requests_per_second: int):
+        """
+        动态设置请求频率限制
+        
+        Args:
+            max_requests_per_second: 每秒最大请求数
+        """
+        with self.rate_limit_lock:
+            self.max_requests_per_second = max_requests_per_second
+            log(f"请求频率限制已更新为: {max_requests_per_second} 次/秒")
+    
+    def reset_rate_limit_counter(self):
+        """
+        重置频率限制计数器
+        """
+        with self.rate_limit_lock:
+            self.request_times.clear()
+            log("频率限制计数器已重置")
+
+    def _make_request(self, method: str, url: str, **kwargs):
+        # 在发送请求前进行速率限制检查
+        self._wait_for_rate_limit()
+        
+        self._ensure_valid_token()
+
+        headers = kwargs.get('headers', {})
+        headers.update({
+            'Authorization': f'Bearer {self.tenant_access_token}',
+            'Content-Type' : 'application/json; charset=utf-8'
+        })
+        kwargs['headers'] = headers
+
+        response = requests.request(method, url, **kwargs)
+        response.raise_for_status()
+        response_data = response.json()
+
+        if response_data.get("code") == 0:
+            log('----------------------------------------------------')
+            log(f'url: {url}')
+            log(f'kwargs: {json.dumps(kwargs, ensure_ascii=False, indent=4, sort_keys=True)}')
+            log(f'data: {response_data["data"]}')
+            log('====================================================')
+            return response_data.get("data", {})
+        else:
+            raise Exception(f"API调用失败: {response_data.get('msg', '未知错误')}")
+
+    # ==================== 文件夹操作 ====================
+
+    def get_root_folder_meta(self):
+        url = f"{self.base_url}/drive/explorer/v2/root_folder/meta"
+        return self._make_request('GET', url)
+
+    def create_folder(self, name: str, parent_folder_token: str = ""):
+        url = f"{self.base_url}/drive/v1/files/create_folder"
+        payload = {
+            "name"        : name,
+            "folder_token": parent_folder_token
+        }
+        return self._make_request('POST', url, json=payload)
+
+    def list_folder_files(self, folder_token: str, page_size: int = 20, page_token: str = None):
+        url = f"{self.base_url}/drive/v1/files"
+        params = {
+            "folder_token": folder_token,
+            "page_size"   : page_size
+        }
+        if page_token:
+            params["page_token"] = page_token
+        return self._make_request('GET', url, params=params)
+
+    def delete_file(self, file_token: str, file_type: str = "file"):
+        url = f"{self.base_url}/drive/v1/files/{file_token}"
+        params = {"type": file_type}
+        return self._make_request('DELETE', url, params=params)
+
+    # ==================== 表格操作 ====================
+
+    def create_spreadsheet(self, title: str, folder_token: str):
+        url = f"{self.base_url}/sheets/v3/spreadsheets"
+        payload = {
+            "title"       : title,
+            "folder_token": folder_token
+        }
+        return self._make_request('POST', url, json=payload)
+
+    def get_spreadsheet(self, spreadsheet_token: str, user_id_type: str = "open_id"):
+        url = f"{self.base_url}/sheets/v3/spreadsheets/{spreadsheet_token}"
+        params = {"user_id_type": user_id_type}
+        return self._make_request('GET', url, params=params)
+
+    def query_sheets(self, spreadsheet_token: str):
+        url = f"{self.base_url}/sheets/v3/spreadsheets/{spreadsheet_token}/sheets/query"
+        return self._make_request('GET', url)
+
+    def get_sheet(self, spreadsheet_token: str, sheet_id: str):
+        url = f"{self.base_url}/sheets/v3/spreadsheets/{spreadsheet_token}/sheets/{sheet_id}"
+        return self._make_request('GET', url)
+
+    def add_sheet(self, spreadsheet_token: str, title: str, index: int = None):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/sheets_batch_update"
+        request_data = {
+            "addSheet": {
+                "properties": {"title": title}
+            }
+        }
+        if index is not None:
+            request_data["addSheet"]["properties"]["index"] = index
+
+        payload = {"requests": [request_data]}
+        return self._make_request('POST', url, json=payload)
+
+    def delete_sheet(self, spreadsheet_token: str, sheet_id: str):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/sheets_batch_update"
+        payload = {
+            "requests": [{
+                "deleteSheet": {"sheetId": sheet_id}
+            }]
+        }
+        return self._make_request('POST', url, json=payload)
+
+    def update_sheet_properties(self, spreadsheet_token: str, properties: dict):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/sheets_batch_update"
+        payload = {
+            "requests": [{
+                "updateSheet": {"properties": properties}
+            }]
+        }
+        return self._make_request('POST', url, json=payload)
+
+    def obtain_spreadsheet_metainfo(self, spreadsheet_token: str, extFields: str = 'protectedRange', user_id_type: str = "open_id"):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/metainfo"
+        params = {
+            "extFields"   : extFields,
+            "user_id_type": user_id_type
+        }
+        return self._make_request('GET', url, params=params)
+
+    # ==================== 数据操作 ====================
+
+    def read_sheet_data(self, spreadsheet_token: str, sheet_id: str, range_str: str = None,
+            value_render_option: str = "ToString", date_time_render_option: str = "FormattedString"):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/values/{sheet_id}"
+        params = {
+            "value_render_option"    : value_render_option,
+            "date_time_render_option": date_time_render_option
+        }
+        if range_str:
+            params["range"] = range_str
+        return self._make_request('GET', url, params=params)
+
+    def read_multiple_ranges(self, spreadsheet_token: str, sheet_id: str, ranges: list,
+            user_id_type: str = "open_id", date_time_render_option: str = "FormattedString",
+            value_render_option: str = "ToString"):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/values_batch_get"
+        full_ranges = [f"{sheet_id}!{r}" for r in ranges]
+        params = {
+            "ranges"                 : full_ranges,
+            "user_id_type"           : user_id_type,
+            "date_time_render_option": date_time_render_option,
+            "value_render_option"    : value_render_option
+        }
+        return self._make_request('GET', url, params=params)
+
+    def write_data_to_range(self, spreadsheet_token: str, sheet_id: str, range_str: str, values: list, insert_data_option: str = 'INSERT_ROWS'):
+        """写入数据到指定范围"""
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/values_append"
+        payload = {
+            "valueRange"      : {
+                "range" : f"{sheet_id}!{range_str}",
+                "values": values
+            },
+            "insertDataOption": insert_data_option
+        }
+        return self._make_request('POST', url, json=payload)
+
+    def write_data_to_multiple_ranges(self, spreadsheet_token: str, sheet_id: str, value_ranges: list):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/values_batch_update"
+        formatted_ranges = []
+        for vr in value_ranges:
+            formatted_range = vr.copy()
+            if '!' not in formatted_range['range']:
+                formatted_range['range'] = f"{sheet_id}!{formatted_range['range']}"
+            formatted_ranges.append(formatted_range)
+
+        payload = {"value_ranges": formatted_ranges}
+        return self._make_request('POST', url, json=payload)
+
+    def image_url_to_path(self, image_url: str):
+        image_dir = r'C:\Users\Administrator\Desktop\auto\image\shein'
+        file_name = os.path.basename(urlparse(image_url).path)  # 获取 URL 路径中的文件名
+        file_path = os.path.join(image_dir, file_name)  # 拼接文件路径
+        if not os.path.exists(file_path):
+            headers = {
+                "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/136.0.0.0 Safari/537.36",
+            }
+            response = requests.get(image_url, headers=headers, timeout=10)
+            response.raise_for_status()
+            with open(file_path, 'wb') as f:
+                f.write(response.content)
+        return file_path
+
+    def write_image(self, spreadsheet_token: str, range_str: str, image_url: str):
+        image_path = self.image_url_to_path(image_url)
+        with open(image_path, "rb") as image_file:
+            fb = image_file.read()
+            misssing_padding = 4 - len(fb) % 4
+            if misssing_padding:
+                fb += b'=' * misssing_padding
+            fb = base64.b64encode(fb).decode('utf-8')
+
+            name = os.path.basename(image_path)
+
+            url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/values_image"
+            payload = {
+                "range": f"{range_str}",
+                "image": fb,
+                "name" : name,
+            }
+            return self._make_request('POST', url, json=payload)
+
+    # ==================== 单元格样式操作 ====================
+
+    def merge_cells(self, spreadsheet_token: str, sheet_id: str, merge_type: str, range_str: str):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/merge_cells"
+        payload = {
+            "range"    : f"{sheet_id}!{range_str}",
+            "mergeType": merge_type
+        }
+        return self._make_request('POST', url, json=payload)
+
+    def unmerge_cells(self, spreadsheet_token: str, sheet_id: str, range_str: str):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/unmerge_cells"
+        payload = {"range": f"{sheet_id}!{range_str}"}
+        return self._make_request('POST', url, json=payload)
+
+    def batch_set_cell_style(self, spreadsheet_token: str, data_list: list):
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/styles_batch_update"
+        payload = {"data": data_list}
+        return self._make_request('PUT', url, json=payload)
+
+    # ==================== 权限管理 ====================
+
+    def batch_create_permission_members(self, token: str, type: str, members: list, need_notification: bool = True):
+        url = f"{self.base_url}/drive/v1/permissions/{token}/members/batch_create"
+        params = {
+            "type"             : type,
+            "need_notification": str(need_notification).lower()
+        }
+        payload = {"members": members}
+        return self._make_request('POST', url, params=params, json=payload)
+
+    def update_permission_member(self, token: str, type: str, member_type: str, member_id: str,
+            perm: str, need_notification: bool = True):
+        url = f"{self.base_url}/drive/v2/permissions/{token}/members/{member_id}"
+        params = {
+            "type"             : type,
+            "member_type"      : member_type,
+            "need_notification": str(need_notification).lower()
+        }
+        payload = {"perm": perm}
+        return self._make_request('PUT', url, params=params, json=payload)
+
+    def list_permission_members(self, token: str, type: str, page_size: int = 50, page_token: str = None):
+        url = f"{self.base_url}/drive/v2/permissions/{token}/members"
+        params = {
+            "type"     : type,
+            "page_size": page_size
+        }
+        if page_token:
+            params["page_token"] = page_token
+        return self._make_request('GET', url, params=params)
+
+    def transfer_permission_owner(self, token: str, type: str, member_type: str, member_id: str, need_notification: bool = True):
+        url = f"{self.base_url}/drive/v2/permissions/{token}/members/transfer_owner"
+        params = {
+            "type"             : type,
+            "need_notification": str(need_notification).lower()
+        }
+        payload = {
+            "member_type": member_type,
+            "member_id"  : member_id
+        }
+        return self._make_request('POST', url, params=params, json=payload)
+
+    def check_permission_member_auth(self, token: str, type: str, action_type: str, member_type: str, member_id: str):
+        url = f"{self.base_url}/drive/v2/permissions/{token}/members/auth"
+        params = {
+            "type"       : type,
+            "action_type": action_type,
+            "member_type": member_type,
+            "member_id"  : member_id
+        }
+        return self._make_request('GET', url, params=params)
+
+    # ==================== 表格保护范围 ====================
+
+    def add_protected_dimension(self, spreadsheet_token: str, add_protected_dimension: list, user_id_type: str = "open_id"):
+        """增加电子表格保护范围的维度信息"""
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/protected_dimension"
+        params = {
+            "user_id_type": user_id_type
+        }
+        payload = {
+            "addProtectedDimension": add_protected_dimension
+        }
+        return self._make_request('POST', url, params=params, json=payload)
+
+    def batch_delete_protected_range(self, spreadsheet_token: str, protect_ids: list):
+        """批量删除电子表格保护范围"""
+        url = f"{self.base_url}/sheets/v2/spreadsheets/{spreadsheet_token}/protected_range_batch_del"
+        payload = {
+            "protectIds": protect_ids
+        }
+        return self._make_request('DELETE', url, json=payload)