pytbox 0.0.1__py3-none-any.whl

pytbox/feishu/errors.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python3
+
+from enum import Enum
+from typing import Optional
+
+import httpx
+
+
+class RequestTimeoutError(Exception):
+    code = "notionhq_client_request_timeout"
+
+    def __init__(self, message: str="Request to Notion API has time out") -> None:
+        super().__init__(message)
+
+
+class HTTPResponseError(Exception):
+    def __init__(self, response: httpx.Response, message: Optional[str]=None) -> None:
+        if message is None:
+            message = (
+                f'Request to Notion API failed with status: {response.status_code}'
+            )
+        super().__init__(message)
+
+        self.status = response.status_code
+        self.headers = response.headers
+        self.body = response.text
+
+
+class APIErrorCode(str, Enum):
+    Unauthorized = "unauthorized"
+
+
+class APIResponseError(HTTPResponseError):
+
+    code: APIErrorCode
+
+    def __init__(self, response: httpx.Response, message: str, code: APIErrorCode) -> None:
+        super().__init__(response, message)
+        self.code = code
+
+
+def is_api_error_code(code: str) -> bool:
+    if isinstance(code, str):
+        return code in (error_code.value for error_code in APIErrorCode)
+    return False

pytbox/feishu/helpers.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python3
+
+from typing import Dict, Any
+
+
+def pick(base: Dict[Any, Any], *keys: str) -> Dict[Any, Any]:
+    return {key: base[key] for key in keys if key in base and base[key] is not None}

pytbox/feishu/typing.py

@@ -0,0 +1,7 @@
+#!/usr/bin/env python3
+
+
+from typing import Awaitable, TypeVar, Union
+
+T = TypeVar("T")
+SyncAsync = Union[T, Awaitable[T]]

pytbox/logger.py

@@ -0,0 +1,126 @@
+#!/usr/bin/env python3
+
+import sys
+from loguru import logger
+from .victorialog import Victorialog
+
+
+logger.remove()
+logger.add(sys.stdout, colorize=True, format="<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | <level>{level: <8}</level> | <level>{message}</level>")
+
+
+class AppLogger:
+    """
+    应用日志记录器类
+    
+    提供统一的日志记录接口，支持多种日志级别和外部服务集成。
+    自动记录调用者信息（文件名、行号、函数名）到日志中。
+    """
+    def __init__(self, 
+                 app_name: str='inbox', 
+                 stream: str='automation', 
+                 enable_victorialog: bool=False, 
+                 victorialog_url: str=None,
+                 enable_sls: bool=False,
+                 sls_url: str=None,
+                 sls_access_key_id: str=None,
+                 sls_access_key_secret: str=None,
+                 sls_project: str=None,
+                 sls_logstore: str=None,
+                 sls_topic: str=None,
+            ):
+        """
+        初始化应用日志记录器
+        
+        Args:
+            app_name: 应用名称，用于标识日志来源
+            stream: 日志流名称，用于VictoriaLogs分类
+        """
+        self.app_name = app_name
+        self.stream = stream
+        self.victorialog = Victorialog(url=victorialog_url)
+        self.enable_victorialog = enable_victorialog
+    
+    def _get_caller_info(self) -> tuple[str, int, str]:
+        """
+        获取调用者信息
+        
+        Returns:
+            tuple: (文件名, 行号, 函数名)
+        """
+        import inspect
+        stack = inspect.stack()
+        caller = stack[2]  # 索引0是当前函数，索引1是_get_caller_info，索引2是实际调用者
+        
+        # 获取调用者的文件名、行号、函数名
+        call_full_filename = caller.filename
+        caller_filename = caller.filename.split('/')[-1]
+        caller_lineno = caller.lineno
+        caller_function = caller.function
+        
+        return caller_filename, caller_lineno, caller_function, call_full_filename
+    
+    def debug(self, message: str):
+        """记录调试级别日志"""
+        caller_filename, caller_lineno, caller_function, call_full_filename = self._get_caller_info()
+        logger.debug(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
+        if self.enable_victorialog:
+            self.victorialog.send_program_log(stream=self.stream, level="DEBUG", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+    
+    def info(self, message: str='', feishu_notify: bool=False):
+        """记录信息级别日志"""
+        caller_filename, caller_lineno, caller_function, call_full_filename = self._get_caller_info()
+        logger.info(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
+        if self.enable_victorialog:
+            r = self.victorialog.send_program_log(stream=self.stream, level="INFO", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+            print(r)
+        if feishu_notify:
+            self.feishu(message)
+    
+    def warning(self, message: str):
+        """记录警告级别日志"""
+        caller_filename, caller_lineno, caller_function, call_full_filename = self._get_caller_info()
+        logger.warning(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
+        if self.enable_victorialog:
+            self.victorialog.send_program_log(stream=self.stream, level="WARN", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+    
+    def error(self, message: str):
+        """记录错误级别日志"""
+        caller_filename, caller_lineno, caller_function, call_full_filename = self._get_caller_info()
+        logger.error(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
+        if self.enable_victorialog:
+            self.victorialog.send_program_log(stream=self.stream, level="ERROR", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+        from src.library.monitor.insert_program import insert_error_message
+        # 传递清理后的消息给监控系统
+        message.replace('#', '')
+        insert_error_message(message, self.app_name, caller_filename, caller_lineno, caller_function)
+        
+    def critical(self, message: str):
+        """记录严重错误级别日志"""
+        caller_filename, caller_lineno, caller_function, call_full_filename = self._get_caller_info()
+        logger.critical(f"[{caller_filename}:{caller_lineno}:{caller_function}] {message}")
+        if self.enable_victorialog:
+            self.victorialog.send_program_log(stream=self.stream, level="CRITICAL", message=message, app_name=self.app_name, file_name=call_full_filename, line_number=caller_lineno, function_name=caller_function)
+
+
+def get_logger(app_name: str, enable_) -> AppLogger:
+    """
+    获取应用日志记录器实例
+    
+    Args:
+        app_name: 应用名称
+        log_level: 日志级别
+        enable_influx: 是否启用InfluxDB记录
+    
+    Returns:
+        AppLogger: 日志记录器实例
+    """
+    return AppLogger(app_name)
+
+
+# 使用示例
+if __name__ == "__main__":
+    log = get_logger(app_name='test')
+    log.info("That's it, beautiful and simple logging!")
+    log.warning("That's it, beautiful and simple logging!")
+    log.error("That's it, beautiful and simple logging!11")

pytbox/onepassword_connect.py

@@ -0,0 +1,92 @@
+#!/usr/bin/env python3
+
+from typing import List
+from onepasswordconnectsdk.client import new_client_from_environment
+from onepasswordconnectsdk.models import (Item, Field)
+
+
+class OnePasswordConnect:
+    '''tbd'''
+    def __init__(self, vault_id):
+        self.client = new_client_from_environment()
+        self.vault_id = vault_id
+
+    def create_item(self, name, username: str=None, password: str=None, notes: str="create by automation", tags: List=None):
+        # Create an item
+        new_item = Item(
+            title=name,
+            category="LOGIN",
+            tags=tags,
+            fields=[
+                Field(value=username, purpose="USERNAME"),
+                Field(value=password, purpose="PASSWORD"),
+                Field(value=notes, purpose="NOTES")
+            ],
+        )
+        created_item = self.client.create_item(self.vault_id, new_item)
+        return created_item
+
+    def delete_item(self, item_id):
+        return self.client.delete_item(self.vault_id, item_id)
+    
+    def get_item(self, item_id):
+        item = self.client.get_item(item_id, self.vault_id)
+        return item
+    
+    def get_item_by_title(self, title: str='', totp: bool=False):
+        '''通过title获取具体的值, 会返回一个字典'''
+        value = {}
+            
+        item = self.client.get_item_by_title(title, self.vault_id)
+
+        if totp:
+            for field in item.fields:
+                if field.totp != None:
+                    return field.totp
+        else:
+            for field in item.fields:
+                value[field.label] = field.value
+            return value
+
+    def update_item(self, item_id: str, name: str=None, username: str=None, password: str=None, tags: list=None, notes: str=None):
+        '''
+        更新单个item
+        Parms:
+            title(str): 一条数据的名称
+            name(str): 需要更新的参数值, 例如 username, password
+            value(str): 新的值
+        Returns:
+            Item object(class): 更新后的item
+        '''
+        # self.get_item_by_title(title="")
+        update_item = self.get_item(item_id=item_id)
+        
+        if name:
+            update_item.title = name
+        
+        for field in update_item.fields:
+            if field.purpose == "USERNAME":
+                field.value = username
+            if field.purpose == "PASSWORD":
+                field.value = password
+            if field.purpose == "NOTES":
+                field.value = notes
+        
+        if tags:
+            update_item.tags = tags
+        return self.client.update_item(item_id, self.vault_id, update_item)
+
+    def search_item(self, title: str=None, tag: str=None) -> list:
+        if title:
+            filter_query = f'title eq "{title}"'
+        if tag:
+            filter_query = f'tag eq "{tag}"'
+        else:
+            filter_query = None
+        return self.client.get_items(self.vault_id, filter_query=filter_query)
+
+
+    
+if __name__ == "__main__":
+    pass
+    # my1p.update_item(title='fengmao-server', name='password', value='test')

pytbox/onepassword_sa.py

@@ -0,0 +1,197 @@
+#!/usr/bin/env python3
+"""
+1Password 客户端封装类 - 异步版本（简化）
+"""
+
+import os
+from typing import Optional, Dict, Any, List, Literal
+try:
+    from onepassword.client import Client
+    from onepassword import (
+        ItemCreateParams, ItemCategory, ItemField, ItemFieldType,
+        ItemSection, Website, AutofillBehavior
+    )
+except ImportError:
+    print("⚠️  请安装 1password SDK: pip install onepassword")
+    Client = None
+
+
+class OnePasswordClient:
+    """1Password 客户端封装类"""
+    
+    def __init__(self, service_account_token: str = None, integration_name: str = "pytbox", integration_version: str = "v1.0.0", vault_id: str = None):
+        """
+        初始化 1Password 客户端
+        
+        Args:
+            service_account_token: 服务账户令牌，如果不提供则从环境变量 OP_SERVICE_ACCOUNT_TOKEN 获取
+            integration_name: 集成名称
+            integration_version: 集成版本
+            vault_id: 默认保险库ID
+        """
+        self.token = service_account_token or os.getenv("OP_SERVICE_ACCOUNT_TOKEN")
+        self.integration_name = integration_name
+        self.integration_version = integration_version
+        self.vault_id = vault_id
+        self.client: Optional[Client] = None
+        
+        if not self.token:
+            raise ValueError("未找到 1Password 服务账户令牌，请设置 OP_SERVICE_ACCOUNT_TOKEN 环境变量或传入 service_account_token 参数")
+        
+        if Client is None:
+            raise ImportError("未安装 onepassword SDK，请运行: pip install onepassword")
+        
+        # print(f"🔧 初始化1Password客户端:")
+        # print(f"  - 集成名称: {self.integration_name}")
+        # print(f"  - 默认保险库ID: {self.vault_id}")
+        # print(f"  - 令牌: {'已设置' if self.token else '未设置'}")
+    
+    async def authenticate(self) -> None:
+        """认证并初始化客户端"""
+        if not self.client:
+            self.client = await Client.authenticate(
+                auth=self.token,
+                integration_name=self.integration_name,
+                integration_version=self.integration_version
+            )
+            # print(f"✅ 1Password 客户端认证成功")
+    
+    async def ensure_authenticated(self) -> None:
+        """确保客户端已认证"""
+        if not self.client:
+            await self.authenticate()
+    
+    async def create_item(self, url, name, username, password, notes, tags: list=[]):
+        # Create an Item and add it to your vault.
+        await self.ensure_authenticated()
+        to_create = ItemCreateParams(
+            title=name,
+            category=ItemCategory.LOGIN,
+            vault_id=self.vault_id,
+            fields=[
+                ItemField(
+                    id="username",
+                    title="username",
+                    field_type=ItemFieldType.TEXT,
+                    value=username,
+                ),
+                ItemField(
+                    id="password",
+                    title="password",
+                    field_type=ItemFieldType.CONCEALED,
+                    value=password,
+                ),
+            ],
+            sections=[
+                ItemSection(id="", title=""),
+                ItemSection(id="totpsection", title=""),
+            ],
+            tags=tags,
+            notes=notes,
+            websites=[
+                Website(
+                    label="网站",
+                    url=url,
+                    autofillBehavior=AutofillBehavior.ANYWHEREONWEBSITE
+                    # autofill_behavior=AutofillBehavior.NEVER,
+                )
+            ],
+        )
+        created_item = await self.client.items.create(to_create)
+        return created_item
+    
+    async def get_item(self, item_id):
+        # Retrieve an item from your vault.
+        item = await self.client.items.get(self.vault_id, item_id)
+        return item
+    
+    async def get_vault_items(self) -> List[Dict[str, Any]]:
+        """
+        获取保险库中的所有项目
+        
+        Args:
+            vault_id: 保险库ID，如果不提供则使用默认的
+            
+        Returns:
+            List[Dict]: 项目列表
+        """
+        await self.ensure_authenticated()
+        return await self.client.items.list(value_id = self.vault_id)
+    
+    async def list_vaults(self) -> List[Dict[str, Any]]:
+        """
+        列出所有可用的保险库
+        
+        Returns:
+            List[Dict]: 保险库列表
+        """
+        await self.ensure_authenticated()
+        try:
+            vaults = await self.client.vaults.list_all()
+            print(f"📁 找到 {len(vaults)} 个保险库:")
+            for i, vault in enumerate(vaults):
+                print(f"  {i+1}. {vault.get('name', 'Unknown')} (ID: {vault.get('id', 'Unknown')})")
+            return vaults
+        except Exception as e:
+            print(f"❌ 获取保险库列表失败: {e}")
+            return []
+    
+    async def get_item_by_item_id(self, item_id: str, title: Literal['username', 'password', 'totp']) -> Dict[str, Any]:
+        """
+        根据标题获取项目
+        
+        Args:
+            title: 项目标题
+            vault_id: 保险库ID，如果不提供则使用默认的
+            
+        Returns:
+            Dict: 项目信息，如果未找到返回None
+        """
+        r = await self.get_item(item_id=item_id)
+        for field in r.fields:
+            if title == 'totp':
+                if 'TOTP' in field.id:
+                    return field.details.content.code
+            else:           
+                if field.title == title:
+                    return field.value
+    
+    async def search_items(self, query: str, vault_id: str = None) -> List[Dict[str, Any]]:
+        """
+        搜索项目
+        
+        Args:
+            query: 搜索查询
+            vault_id: 保险库ID，如果不提供则使用默认的
+            
+        Returns:
+            List[Dict]: 搜索结果
+        """
+        await self.ensure_authenticated()
+        vault = vault_id or self.vault_id
+        if not vault:
+            raise ValueError("必须提供 vault_id 或在初始化时设置默认 vault_id")
+        
+        items = await self.get_vault_items(vault)
+        # 简单的标题匹配搜索
+        return [item for item in items if query.lower() in item.get('title', '').lower()]
+    
+    async def close(self) -> None:
+        """关闭客户端连接"""
+        if self.client:
+            # 如果SDK提供关闭方法，在这里调用
+            self.client = None
+            print("🔒 1Password 客户端连接已关闭")
+    
+    async def __aenter__(self):
+        """异步上下文管理器入口"""
+        await self.authenticate()
+        return self
+    
+    async def __aexit__(self, exc_type, exc_val, exc_tb):
+        """异步上下文管理器出口"""
+        await self.close()
+
+
+if __name__ == "__main__":
+    pass

pytbox/utils/env.py

@@ -0,0 +1,30 @@
+#!/usr/bin/env python3
+
+import os
+from typing import Literal
+
+
+def get_env_by_file_exist(file_path: str) -> Literal['prod', 'dev']:
+    '''
+    检查当前的运行环境
+
+    Returns:
+        Literal['prod', 'dev']: 环境
+    '''
+    if os.path.exists(file_path):
+        return 'prod'
+    else:
+        return 'dev'
+
+
+def get_env_by_os_environment() -> Literal['dev', 'prod']:
+    '''
+    根据环境变量获取环境
+
+    Returns:
+        Literal['dev', 'prod']: 环境，dev 表示开发环境，prod 表示生产环境
+    '''
+    if os.getenv('ENV') == 'dev':
+        return 'dev'
+    else:
+        return 'prod'

pytbox/utils/response.py

@@ -0,0 +1,45 @@
+#!/usr/bin/env python3
+
+from dataclasses import dataclass
+from typing import Any, Optional
+
+
+@dataclass
+class ReturnResponse:
+    """统一响应格式类
+    
+    Attributes:
+        code: 响应状态码
+            0 - 成功 (SUCCESS)
+            1 - 一般错误 (ERROR)
+            2 - 警告 (WARNING)
+            3 - 未授权 (UNAUTHORIZED)
+            4 - 资源未找到 (NOT_FOUND)
+            5 - 请求超时 (TIMEOUT)
+            6 - 参数错误 (INVALID_PARAMS)
+            7 - 权限不足 (PERMISSION_DENIED)
+            8 - 服务不可用 (SERVICE_UNAVAILABLE)
+            9 - 数据库错误 (DATABASE_ERROR)
+            10 - 网络错误 (NETWORK_ERROR)
+        message: 响应消息描述
+        data: 响应数据，可以是任意类型
+    """
+    code: int = 0
+    msg: str = ''
+    data: Optional[Any] = None
+    
+    def is_success(self) -> bool:
+        """判断是否为成功响应
+        
+        Returns:
+            bool: code为0时返回True，否则返回False
+        """
+        return self.code == 0
+    
+    def is_error(self) -> bool:
+        """判断是否为错误响应
+        
+        Returns:
+            bool: code不为0时返回True，否则返回False
+        """
+        return self.code != 0

pytbox/utils/timeutils.py

@@ -0,0 +1,40 @@
+#!/usr/bin/env python3
+
+
+import time
+import pytz
+import datetime
+
+
+class TimeUtils:
+    
+    @staticmethod
+    def get_timestamp(now: bool=True) -> int:
+        '''
+        获取时间戳
+        
+        Args:
+            now (bool, optional): _description_. Defaults to True.
+
+        Returns:
+            _type_: _description_
+        '''
+        if now:
+            return int(time.time())
+        else:
+            return int(time.time() * 1000)
+    
+    @staticmethod
+    def get_time_object(now: bool=True):
+        '''
+        获取当前时间, 加入了时区信息, 简单是存储在 Mongo 中时格式为 ISODate
+
+        Returns:
+            current_time(class): 时间, 格式: 2024-04-23 16:48:11.591589+08:00
+        '''
+        if now:
+            return datetime.datetime.now(pytz.timezone('Asia/Shanghai'))
+    
+    @staticmethod
+    def get_utc_time():
+        return datetime.datetime.now(datetime.timezone.utc).strftime("%Y-%m-%dT%H:%M:%S.%f")[:-3] + "Z"