pytbox 0.0.1__py3-none-any.whl

pytbox/alicloud/sls.py

@@ -0,0 +1,93 @@
+#!/usr/bin/env python3
+
+from typing import Literal
+# 引入sls包。
+from aliyun.log import GetLogsRequest, LogItem, PutLogsRequest
+from aliyun.log import LogClient as SlsLogClient
+from aliyun.log.auth import AUTH_VERSION_4
+from ..utils.env import check_env
+
+
+
+class AliCloudSls:
+
+    def __init__(self, access_key_id: str=None, access_key_secret: str=None, project: str, logstore: str):
+        # 日志服务的服务接入点
+        self.endpoint = "cn-shanghai.log.aliyuncs.com"
+        # 创建 LogClient 实例，使用 V4 签名，根据实际情况填写 region，这里以杭州为例
+        self.client = SlsLogClient(self.endpoint, access_key_id, access_key_secret, auth_version=AUTH_VERSION_4, region='cn-shanghai')
+        self.project = project
+        self.logstore = logstore
+
+    def get_logs(self, project_name, logstore_name, query, from_time, to_time): 
+        # Project名称。
+        # project_name = "sh-prod-network-devices-log"
+        # Logstore名称
+        # logstore_name = "sh-prod-network-devices-log"
+        # 查询语句。
+        # query = "*| select dev,id from " + logstore_name
+        # query = "*"
+        # 索引。
+        logstore_index = {'line': {
+            'token': [',', ' ', "'", '"', ';', '=', '(', ')', '[', ']', '{', '}', '?', '@', '&', '<', '>', '/', ':', '\n', '\t',
+                    '\r'], 'caseSensitive': False, 'chn': False}, 'keys': {'dev': {'type': 'text',
+                                                                                    'token': [',', ' ', "'", '"', ';', '=',
+                                                                                            '(', ')', '[', ']', '{', '}',
+                                                                                            '?', '@', '&', '<', '>', '/',
+                                                                                            ':', '\n', '\t', '\r'],
+                                                                                    'caseSensitive': False, 'alias': '',
+                                                                                    'doc_value': True, 'chn': False},
+                                                                            'id': {'type': 'long', 'alias': '',
+                                                                                    'doc_value': True}}, 'log_reduce': False,
+            'max_text_len': 2048}
+
+        # from_time和to_time表示查询日志的时间范围，Unix时间戳格式。
+        # from_time = int(time.time()) - 60
+        # to_time = time.time() + 60
+        # # 通过SQL查询日志。
+        # def get_logs():
+        # print("ready to query logs from logstore %s" % logstore_name)
+        request = GetLogsRequest(project_name, logstore_name, from_time, to_time, query=query)
+        response = self.client.get_logs(request)
+        for log in response.get_logs():
+            yield log.contents
+
+    def put_logs(self, 
+                 topic: Literal['meraki_alert', 'program']='program', 
+                 level: Literal['INFO', 'WARN']='INFO', 
+                 msg: str=None, 
+                 app: str=None, 
+                 caller_filename: str=None, 
+                 caller_lineno: int=None, 
+                 caller_function: str=None, 
+                 call_full_filename: str=None
+            ):
+        log_group = []
+        log_item = LogItem()
+        contents = [
+            ('env', check_env()),
+            ('level', level),
+            ('app', app),
+            ('msg', msg),
+            ('caller_filename', caller_filename),
+            ('caller_lineno', str(caller_lineno)),
+            ('caller_function', caller_function),
+            ('call_full_filename', call_full_filename)
+        ]
+        log_item.set_contents(contents)
+        log_group.append(log_item)
+        request = PutLogsRequest(self.project, self.logstore, topic, "", log_group, compress=False)
+        self.client.put_logs(request)
+    
+    def put_logs_for_meraki(self, alert):
+        log_group = []
+        log_item = LogItem()
+        contents = alert
+        log_item.set_contents(contents)
+        log_group.append(log_item)
+        request = PutLogsRequest(self.project, self.logstore, "", "", log_group, compress=False)
+        self.client.put_logs(request)
+        
+
+if __name__ == "__main__":
+    pass

pytbox/common/__init__.py

@@ -0,0 +1,7 @@
+"""
+Common utilities and base classes
+"""
+
+from .base import BaseAPI, BaseResponse
+
+__all__ = ["BaseAPI", "BaseResponse"]

pytbox/dida365.py

@@ -0,0 +1,297 @@
+#!/usr/bin/env python3
+
+"""滴答清单API客户端。
+
+此模块提供了与滴答清单API交互的功能，包括认证和基本操作。
+"""
+import re
+import json
+from typing import Dict, Any, Literal, Union
+from dataclasses import dataclass
+import requests
+from datetime import datetime
+from .utils.response import ReturnResponse
+
+
+@dataclass
+class Task:
+    task_id: str
+    project_id: str
+    title: str
+    content: str
+    desc: str
+    start_date: str
+    due_date: str
+    priority: int
+    status: int
+    tags: list
+    completed_time: str
+    assignee: int
+
+@dataclass
+class DidaResponse:
+    code: int
+    message: str
+    data: dict
+
+
+class ProcessDidaResponse:    
+    @staticmethod
+    def status(status):
+        if status == 0:
+            return '进行中'
+        elif status == 2:
+            return '已完成'
+        else:
+            return '未识别'
+
+    @staticmethod
+    def priority(priority):
+        if priority == 1:
+            return '低优先级'
+        elif priority == 3:
+            return '中优先级'
+        elif priority == 5:
+            return '高优先级'
+        else:
+            return '未识别'
+
+
+class Dida365:
+    """滴答清单API客户端类。
+    
+    处理滴答清单API的认证和操作。
+    
+    Attributes:
+        config: 滴答清单配置实例
+        access_token: 访问令牌
+        refresh_token: 刷新令牌
+    """
+    
+    def __init__(self, access_token: str, cookie: str) -> None:
+        """初始化客户端。
+        
+        Args:
+            access_token: 滴答清单配置实例
+        """
+        self.access_token = access_token
+        self.base_url = 'https://api.dida365.com'
+        self.cookie = cookie
+        self.headers = {
+            "Authorization": f"Bearer {self.access_token}",
+            "Content-Type": "application/json",
+            "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36"
+        }
+        self.cookie_headers = {
+            "Content-Type": "application/json",
+            "User-Agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/124.0.0.0 Safari/537.36",
+            "Cookie": self.cookie
+        }
+        self.timeout = 10
+
+    def request(self, api_url: str=None, method: str='GET', payload: dict={}):
+        """发送请求。
+        
+        Args:
+            url: 请求URL
+            method: 请求方法
+        """
+        
+        url = f"{self.base_url}{api_url}"
+        response = requests.request(
+            method=method,
+            url=url, 
+            headers=self.headers,
+            data=json.dumps(payload),
+            timeout=3
+        )
+        
+        if response.status_code == 200:
+            if 'complete' in api_url:
+                return DidaResponse(code=0, msg='success', data=None)
+            else:
+                try:
+                    return DidaResponse(code=0, msg='success', data=response.json())
+                except Exception as e:
+                    return DidaResponse(code=1, msg='warning', data=response)
+        else:
+            return DidaResponse(code=1, msg='error', data=response.json())
+
+    def task_list(self, project_id: str, enhancement: bool=True):
+        """获取任务列表。
+        
+        Returns:
+            List[Dict[str, Any]]: 任务列表
+            
+        Raises:
+            requests.exceptions.RequestException: 请求失败时抛出
+            ValueError: 没有访问令牌时抛出
+        """
+        if enhancement:
+            tasks = requests.request(
+                method='GET', 
+                url=f'https://api.dida365.com/api/v2/project/{project_id}/tasks', 
+                headers=self.cookie_headers, 
+                timeout=3
+            ).json()
+            for task in tasks:
+                yield Task(task_id=task.get('id'),
+                        project_id=task.get('projectId'),
+                        title=task.get('title'),
+                        content=task.get('content'),
+                        desc=task.get('desc'),
+                        start_date=task.get('startDate'),
+                        due_date=task.get('dueDate'),
+                        priority=ProcessDidaResponse.priority(task.get('priority')),
+                        status=ProcessDidaResponse.status(task.get('status')),
+                        tags=task.get('tags'),
+                        completed_time=task.get('completedTime'),
+                        assignee=task.get('assignee'))
+        else:
+            tasks = self.request(api_url=f"/open/v1/project/{project_id}/data", method="GET")['tasks']
+            for task in tasks:
+                yield Task(task_id=task.get('id'),
+                        project_id=task.get('projectId'),
+                        title=task.get('title'),
+                        content=task.get('content'),
+                        desc=task.get('desc'),
+                        start_date=task.get('startDate'),
+                        due_date=task.get('dueDate'),
+                        priority=ProcessDidaResponse.priority(task.get('priority')),
+                        status=ProcessDidaResponse.status(task.get('status')),
+                        tags=task.get('tags'),
+                        completed_time=task.get('completedTime'),
+                        assignee=task.get('assignee'))
+
+    def task_create(self,
+               project_id,
+               title :str,
+               content :str=None, 
+               tags :list=None, 
+               priority :Literal[1, 3, 5]=1,
+               start_date: datetime=datetime.utcnow(),
+               start_time_offset: bool=True,
+               due_date: str=None,
+               kind: Union[None, Literal["NOTE"], str] = 'TEXT',
+               assignee: int=None,
+               reminder: bool=True):
+        '''
+        _summary_
+
+        Args:
+            project_id (_type_): _description_
+            title (str): _description_
+            content (str, optional): _description_. Defaults to None.
+            tags (list, optional): _description_. Defaults to None.
+            priority (Literal[1, 3, 5], optional): _description_. Defaults to 1.
+            start_date (datetime, optional): 传入 utc 时区的时间对象. Defaults to datetime.utcnow().
+            due_date (str, optional): _description_. Defaults to None.
+            kind (Union[None, Literal["NOTE"], str], optional): _description_. Defaults to 'TEXT'.
+            assignee (int, optional): _description_. Defaults to None.
+            reminder (bool, optional): _description_. Defaults to True.
+        '''
+        # 如果存在start_date，将其增加3分钟
+        if isinstance(start_date, datetime):
+            if start_time_offset:
+                if start_date.minute + 3 >= 60:
+                    minute = 59
+                else:
+                    minute = start_date.minute + 3
+                start_date_offset = start_date.replace(minute=minute)
+                start_date_format = start_date_offset.strftime('%Y-%m-%dT%H:%M:%S.000+0000')
+            else:
+                start_date_format = start_date.strftime('%Y-%m-%dT%H:%M:%S.000+0000')
+        else:
+            start_date_format = start_date.strftime('%Y-%m-%dT%H:%M:%S.000+0000')
+ 
+        payload = {
+                    "projectId": project_id,
+                    "priority": priority,
+                    "assignee": str(assignee),
+                    # "startDate": start_date_format,
+                    "title": title,
+                    "timeZone": "Asia/Shanghai",
+                    "kind": kind,
+                    "content": content,
+            }
+
+        payload['startDate'] = start_date_format
+        
+        if isinstance(due_date, datetime):
+            due_date_format = due_date.strftime('%Y-%m-%dT%H:%M:%S.000+0000')
+            payload['dueDate'] = due_date_format
+        
+        if reminder:
+            payload["reminders"] = [
+                "TRIGGER:PT0S"
+            ]
+        if tags:
+            payload['tags'] = tags
+            
+        return self.request(api_url="/open/v1/task", method="POST", payload=payload)
+
+    def task_complete(self, project_id: str, task_id: str):
+        """完成任务。
+        
+        Args:
+            project_id: 项目ID
+            task_id: 任务ID
+        """
+        return self.request(api_url=f"/open/v1/project/{project_id}/task/{task_id}/complete", method="POST")
+
+    def task_get(self, project_id, task_id):
+        return self.request(api_url = f'/open/v1/project/{project_id}/task/{task_id}')
+
+    def task_comments(self, project_id: str, task_id: str):
+        return requests.request(
+            method='GET',
+            url=f'https://api.dida365.com/api/v2/project/{project_id}/task/{task_id}/comments',
+            headers=self.cookie_headers,
+            timeout=3
+        ).json()
+
+    def task_update(self, project_id: str=None, task_id: str=None, title: str=None, content: str=None, priority: int=None, start_date: str=None, content_front: bool=False):
+        """更新任务。
+        
+        Args:
+            project_id: 项目ID
+            task_id: 任务ID
+        """
+        task_get_resp = self.task_get(project_id, task_id)
+        if task_get_resp.code == 0:
+            exists_content = task_get_resp.data['content']
+            
+            if content_front:
+                content = f'{content}\n{exists_content}'
+            else:
+                content = f'{exists_content}\n{content}'
+                
+        elif task_get_resp.code == 1:
+            return task_get_resp
+        
+        payload = {
+            "projectId": project_id,
+            "taskId": task_id,
+            "title": title,
+            "content": content,
+            "priority": priority,
+        }
+        if start_date:
+            payload["startDate"] = start_date
+            
+        return self.request(api_url=f"/open/v1/task/{task_id}", method="POST", payload=payload)
+
+    def get_projects(self) -> ReturnResponse:
+        response = requests.request(
+            method='GET',
+            url=f'{self.base_url}/open/v1/project',
+            headers=self.headers,
+            timeout=self.timeout
+        )
+        if response.status_code == 200:
+            return ReturnResponse(code=0, msg=f"获取到 {len(response.json())} 条 project", data=response.json())
+        else:
+            return ReturnResponse(code=1, msg=f"获取 project 失败: {response.status_code}", data=response.json())
+    
+if __name__ == "__main__":
+    pass

pytbox/feishu/client.py

@@ -0,0 +1,183 @@
+#!/usr/bin/env python3
+
+from typing import Optional, Union, Any, Dict, Type, List
+from dataclasses import dataclass
+from .endpoints import (
+    AuthEndpoint,
+    MessageEndpoint,
+    ExtensionsEndpoint,
+    BitableEndpoint,
+    DocsEndpoint,
+    CalendarEndpoint
+)
+
+from .errors import (
+    RequestTimeoutError,
+    is_api_error_code,
+    APIResponseError,
+    HTTPResponseError
+)
+from types import TracebackType
+from abc import abstractclassmethod
+import httpx
+from httpx import Request, Response
+from .typing import SyncAsync
+
+@dataclass
+class ClientOptions:
+    auth: Optional[str] = None
+    timeout_ms: int = 60_000
+    base_url: str = "https://open.feishu.cn/open-apis"
+
+
+@dataclass
+class FeishuResponse:
+    code: int
+    data: dict
+    chat_id: str
+    message_id: str
+    msg_type: str
+    sender: dict
+    msg: dict
+    expire: int
+    tenant_access_token: str
+
+
+class BaseClient:
+
+    def __init__(self,
+                 app_id: str,
+                 app_secret: str,
+                 client: Union[httpx.Client, httpx.AsyncClient],
+            ) -> None:
+        
+        self.app_id = app_id
+        self.app_secret = app_secret
+
+        self.options = ClientOptions()
+
+        self._clients: List[Union[httpx.Client, httpx.AsyncClient]] = []
+        self.client = client
+
+        self.auth = AuthEndpoint(self)
+        self.message = MessageEndpoint(self)
+        self.bitable = BitableEndpoint(self)
+        self.docs = DocsEndpoint(self)
+        self.calendar = CalendarEndpoint(self)
+        self.extensions = ExtensionsEndpoint(self)
+        
+    @property
+    def client(self) -> Union[httpx.Client, httpx.AsyncClient]:
+        return self._clients[-1]
+    
+    @client.setter
+    def client(self, client: Union[httpx.Client, httpx.AsyncClient]) -> None:
+        client.base_url = httpx.URL(f'{self.options.base_url}/')
+        client.timeout = httpx.Timeout(timeout=self.options.timeout_ms / 1_000)
+        client.headers = httpx.Headers(
+            {
+                "User-Agent": "cc_feishu",
+            }
+        )
+        self._clients.append(client)
+
+
+    def _build_request(self,
+                       method: str,
+                       path: str,
+                       query: Optional[Dict[str, Any]] = None,
+                       body: Optional[Dict[str, Any]] = None,
+                       data: Optional[Any] = None,
+                       files: Optional[Dict[str, Any]] = None,
+                       token: Optional[str] = None) -> Request:
+        
+        headers = httpx.Headers()
+        headers['Authorization'] = f'Bearer {token}'
+        if 'image' in path:
+            headers['Content-Type'] = 'multipart/form-data'
+        
+        return self.client.build_request(
+            method=method, url=path, params=query, json=body, headers=headers, files=files, data=data
+        )
+
+    def _parse_response(self, response) -> Any:
+        response = response.json()
+        return FeishuResponse(code=response.get('code'),
+                              data=response.get('data'),
+                              chat_id=response.get('chat_id'),
+                              message_id=response.get('message_id'),
+                              msg_type=response.get('msg_type'),
+                              sender=response.get('sender'),
+                              msg=response.get('msg'),
+                              expire=response.get('expire'),
+                              tenant_access_token=response.get('tenant_access_token'))
+
+    @abstractclassmethod
+    def request(self,
+                path: str,
+                method: str,
+                query: Optional[Dict[Any, Any]] = None,
+                body: Optional[Dict[Any, Any]] = None,
+                auth: Optional[str] = None,
+                data: Optional[Any] = None,
+                ) -> SyncAsync[Any]:
+        pass
+
+
+class Client(BaseClient):
+
+    client: httpx.Client
+
+    def __init__(self,
+                 app_id: str,
+                 app_secret: str,
+                 client: Optional[httpx.Client]=None) -> None:
+        
+        if client is None:
+            client = httpx.Client()
+        super().__init__(app_id, app_secret, client)
+    
+    def __enter__(self) -> "Client":
+        self.client = httpx.Client()
+        self.client.__enter__()
+        return self
+    
+    def __exit__(self,
+                 exc_type: Type[BaseException],
+                 exc_value: BaseException,
+                 traceback: TracebackType) -> None:
+        self.client.__exit__(exc_type, exc_value, traceback)
+        del self._clients[-1]
+    
+    def close(self) -> None:
+        self.client.close()
+
+    def _get_token(self):
+        if self.auth.fetch_token_from_file():
+            return self.auth.fetch_token_from_file()
+        else:
+            self.auth.save_token_to_file()
+            return self.auth.fetch_token_from_file()
+
+    def request(self,
+                path: str,
+                method: str,
+                query: Optional[Dict[Any, Any]] = None,
+                body: Optional[Dict[Any, Any]] = None,
+                files: Optional[Dict[Any, Any]] = None,
+                token: Optional[str] = None,
+                data: Optional[Any] = None,
+                ) -> Any:
+
+        request = self._build_request(method, path, query, body, files=files, data=data, token=self._get_token())
+        try:
+            response = self._parse_response(self.client.send(request))
+
+            if 'Invalid access token for authorization' in response.msg:
+                self.auth.save_token_to_file()
+                request = self._build_request(method, path, query, body, files=files, data=data, token=self._get_token())
+                return self._parse_response(self.client.send(request))
+            else:
+                return response
+        except httpx.TimeoutException:
+            raise RequestTimeoutError()