pytbox 0.0.1__py3-none-any.whl → 0.3.1__py3-none-any.whl

pytbox/dida365.py

@@ -28,14 +28,8 @@ class Task:
     completed_time: str
     assignee: int
 
-@dataclass
-class DidaResponse:
-    code: int
-    message: str
-    data: dict
-
 
-class ProcessDidaResponse:    
+class ProcessReturnResponse:    
     @staticmethod
     def status(status):
         if status == 0:
@@ -89,7 +83,7 @@ class Dida365:
         }
         self.timeout = 10
 
-    def request(self, api_url: str=None, method: str='GET', payload: dict={}):
+    def request(self, api_url: str=None, method: str='GET', payload: dict={}) -> ReturnResponse:
         """发送请求。
         
         Args:
@@ -108,14 +102,14 @@ class Dida365:
         
         if response.status_code == 200:
             if 'complete' in api_url:
-                return DidaResponse(code=0, msg='success', data=None)
+                return ReturnResponse(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)
+                    return ReturnResponse(code=0, msg='success', data=response.json())
+                except Exception:
+                    return ReturnResponse(code=1, msg='warning', data=response)
         else:
-            return DidaResponse(code=1, msg='error', data=response.json())
+            return ReturnResponse(code=1, msg='error', data=response.json())
 
     def task_list(self, project_id: str, enhancement: bool=True):
         """获取任务列表。
@@ -142,8 +136,8 @@ class Dida365:
                         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')),
+                        priority=ProcessReturnResponse.priority(task.get('priority')),
+                        status=ProcessReturnResponse.status(task.get('status')),
                         tags=task.get('tags'),
                         completed_time=task.get('completedTime'),
                         assignee=task.get('assignee'))
@@ -157,8 +151,8 @@ class Dida365:
                         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')),
+                        priority=ProcessReturnResponse.priority(task.get('priority')),
+                        status=ProcessReturnResponse.status(task.get('status')),
                         tags=task.get('tags'),
                         completed_time=task.get('completedTime'),
                         assignee=task.get('assignee'))

pytbox/excel.py

@@ -0,0 +1,64 @@
+#!/usr/bin/env python3
+
+from openpyxl.styles import Alignment
+from openpyxl.styles import PatternFill
+from openpyxl.styles.differential import DifferentialStyle
+from openpyxl.styles import Font
+
+
+class ExcelFormat:
+    def __init__(self, ws):
+        self.ws = ws
+        
+    def set_column(self):
+        for column in self.ws.columns:
+            max_length = 0
+            column = [cell for cell in column]
+            for cell in column:
+                try:
+                    if len(str(cell.value)) > max_length:
+                        max_length = len(cell.value)
+                except:
+                    pass
+            adjusted_width = (max_length + 10)
+            self.ws.column_dimensions[column[0].column_letter].width = adjusted_width
+            
+    def set_rows_center(self):
+        # 将所有单元格的文字居中
+        for row in self.ws.iter_rows():
+            
+            for cell in row:
+                cell.alignment = Alignment(horizontal='center', vertical='center')
+        for row in self.ws.iter_rows():
+            self.ws.row_dimensions[row[0].row].height = 24
+    
+    def set_freeze_first_row(self):
+        """设置首行锁定/冻结首行"""
+        # 冻结首行，从第二行开始滚动
+        self.ws.freeze_panes = 'A2'
+    
+    def set_freeze_first_column(self):
+        """设置首列锁定/冻结首列"""
+        # 冻结首列，从第二列开始滚动
+        self.ws.freeze_panes = 'B1'
+    
+    def set_first_row_bold_color(self, font_color='FF0000FF'):
+        """设置首行字体加粗并改变颜色
+        
+        Args:
+            font_color (str): 字体颜色的十六进制代码，默认为蓝色(FF0000FF)
+                            格式：AARRGGBB 或 RRGGBB
+                            例如：'FF0000FF'(蓝色), 'FFFF0000'(红色), 'FF008000'(绿色)
+        """
+        # 遍历首行的所有单元格
+        for cell in self.ws[1]:
+            if cell.value is not None:  # 只对有内容的单元格设置样式
+                # 设置字体为粗体并改变颜色
+                cell.font = Font(bold=True, color=font_color)
+    
+    def set_freeze_first_row_and_column(self):
+        """同时冻结首行和首列"""
+        # 冻结首行首列，从第二行第二列开始滚动
+        self.ws.freeze_panes = 'B2'
+    
+    

pytbox/feishu/endpoints.py

@@ -254,9 +254,9 @@ class MessageEndpoint(Endpoint):
             body=payload
         )
         if r.code == 0:
-            return ReturnResponse(code=0, message=f"{message_id} 回复 emoji [{emoji_type}] 成功")
+            return ReturnResponse(code=0, msg=f"{message_id} 回复 emoji [{emoji_type}] 成功")
         else:
-            return ReturnResponse(code=1, message=f"{message_id} 回复 emoji [{emoji_type}] 失败")
+            return ReturnResponse(code=1, msg=f"{message_id} 回复 emoji [{emoji_type}] 失败")
 
 class BitableEndpoint(Endpoint):
     
@@ -432,10 +432,10 @@ class BitableEndpoint(Endpoint):
             resp = self.parent.request(path=f'/bitable/v1/apps/{app_token}/tables/{table_id}/records/{record_id}',
                                         method='PUT',
                                         body=payload)
-            return ReturnResponse(code=resp.code, message=f"记录已存在, 进行更新", data=resp.data)
+            return ReturnResponse(code=resp.code, msg=f"记录已存在, 进行更新", data=resp.data)
         else:
             resp = self.add_record(app_token, table_id, fields)
-            return ReturnResponse(code=resp.code, message=f"记录不存在, 进行创建", data=resp.data)
+            return ReturnResponse(code=resp.code, msg=f"记录不存在, 进行创建", data=resp.data)
 
     def query_name_by_record_id(self, app_token: str=None, table_id: str=None, field_names: list=None, record_id: str='', name: str=''):
         response = self.query_record(app_token=app_token, table_id=table_id, field_names=field_names)
@@ -930,9 +930,12 @@ class ExtensionsEndpoint(Endpoint):
                     final_data = fields[name][0].get('value')[0]['text']
                 
             elif isinstance(fields[name], int):
-                # 将时间戳转换为时间字符串格式
-                final_data = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(fields[name] / 1000 ))
-                
+                if len(str(fields[name])) >= 12 and fields[name] > 10**11:
+                        final_data = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(fields[name] / 1000))
+                elif len(str(fields[name])) == 10 and 10**9 < fields[name] < 10**11:
+                    final_data = time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(fields[name]))
+                else:
+                    final_data = fields[name]
             elif isinstance(fields[name], dict):
                 if fields[name].get('type') == 1:
                     final_data = fields[name].get('value')[0]['text']
@@ -964,9 +967,9 @@ class ExtensionsEndpoint(Endpoint):
                                    body=payload)
         if response.code == 0:
             if get == 'open_id':
-                return ReturnResponse(code=0, message=f'根据用户输入的 {user_input}, 获取用户信息成功', data=response.data['user_list'][0]['user_id'])
+                return ReturnResponse(code=0, msg=f'根据用户输入的 {user_input}, 获取用户信息成功', data=response.data['user_list'][0]['user_id'])
         else:
-            return ReturnResponse(code=response.code, message=f"获取时失败, 报错请见 data 字段", data=response.data)
+            return ReturnResponse(code=response.code, msg=f"获取时失败, 报错请见 data 字段", data=response.data)
     
     def format_rich_text(self, text: str, color: Literal['red', 'green', 'yellow', 'blue'], bold: bool=False):
         if bold:

pytbox/{logger.py → log/logger.py}

@@ -1,9 +1,15 @@
 #!/usr/bin/env python3
 
 import sys
+
 from loguru import logger
-from .victorialog import Victorialog
 
+from .victorialog import Victorialog
+from ..database.mongo import Mongo
+from ..feishu.client import Client as FeishuClient
+from ..utils.timeutils import TimeUtils
+from ..dida365 import Dida365
+from ..alicloud.sls import AliCloudSls
 
 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>")
@@ -21,8 +27,10 @@ class AppLogger:
                  stream: str='automation', 
                  enable_victorialog: bool=False, 
                  victorialog_url: str=None,
+                 mongo: Mongo=None,
+                 feishu: FeishuClient=None,
+                 dida: Dida365=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,
@@ -40,7 +48,17 @@ class AppLogger:
         self.stream = stream
         self.victorialog = Victorialog(url=victorialog_url)
         self.enable_victorialog = enable_victorialog
-    
+        self.enable_sls = enable_sls
+        self.mongo = mongo
+        self.feishu = feishu
+        self.dida = dida
+        self.sls = AliCloudSls(
+            access_key_id=sls_access_key_id,
+            access_key_secret=sls_access_key_secret,
+            project=sls_project,
+            logstore=sls_logstore
+        )
+
     def _get_caller_info(self) -> tuple[str, int, str]:
         """
         获取调用者信息
@@ -66,6 +84,9 @@ class AppLogger:
         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)
+        if self.enable_sls:
+            self.sls.put_logs(level="DEBUG", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
+ 
     
     def info(self, message: str='', feishu_notify: bool=False):
         """记录信息级别日志"""
@@ -73,27 +94,71 @@ class AppLogger:
         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 self.enable_sls:
+            self.sls.put_logs(level="INFO", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
         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)
-    
+        if self.enable_sls:
+            self.sls.put_logs(level="WARN", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
+            
     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)
+        if self.enable_sls:
+            self.sls.put_logs(level="ERROR", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
+        
+        if self.feishu:
+            existing_message = self.mongo.collection.find_one({"message": message}, sort=[("time", -1)])
+            current_time = TimeUtils.get_now_time_mongo()
+
+            if not existing_message or TimeUtils.get_time_diff_hours(existing_message["time"], current_time) > 36:
+                self.mongo.collection.insert_one({
+                    "message": message,
+                    "time": current_time,
+                    "file_name": caller_filename,
+                    "line_number": caller_lineno,
+                    "function_name": caller_function
+                })
+                    
+                    
+                content_list = [
+                    f"{self.feishu.extensions.format_rich_text(text='app:', color='blue', bold=True)} {self.app_name}",
+                    f"{self.feishu.extensions.format_rich_text(text='message:', color='blue', bold=True)} {message}",
+                    f"{self.feishu.extensions.format_rich_text(text='file_name:', color='blue', bold=True)} {caller_filename}",
+                    f"{self.feishu.extensions.format_rich_text(text='line_number:', color='blue', bold=True)} {caller_lineno}",
+                    f"{self.feishu.extensions.format_rich_text(text='function_name:', color='blue', bold=True)} {caller_function}"
+                ]
+                
+                self.feishu.extensions.send_message_notify(
+                    title=f"自动化脚本告警: {self.app_name}",
+                    content="\n".join(content_list)
+                )
+                
+                dida_content_list = [
+                    f"**app**: {self.app_name}",
+                    f"**message**: {message}",
+                    f"**file_name**: {caller_filename}",
+                    f"**line_number**: {caller_lineno}",
+                    f"**function_name**: {caller_function}"
+                ]
+                
+                if self.dida:
+                    self.dida.task_create(
+                        project_id="65e87d2b3e73517c2cdd9d63",
+                        title=f"自动化脚本告警: {self.app_name}",
+                        content="\n".join(dida_content_list),
+                        tags=['L-程序告警', 't-问题处理']
+                    )
         
     def critical(self, message: str):
         """记录严重错误级别日志"""
@@ -101,26 +166,9 @@ class AppLogger:
         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 self.enable_sls:
+            self.sls.put_logs(level="CRITICAL", msg=message, app=self.app_name, caller_filename=caller_filename, caller_lineno=caller_lineno, caller_function=caller_function, call_full_filename=call_full_filename)
 
 # 使用示例
 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")
+    pass

pytbox/{victorialog.py → log/victorialog.py}

@@ -3,8 +3,8 @@
 from typing import Literal
 import requests
 import time
-from .utils.response import ReturnResponse
-from .utils.timeutils import TimeUtils
+from ..utils.response import ReturnResponse
+from ..utils.timeutils import TimeUtils
 
 
 

pytbox/mail/alimail.py

@@ -0,0 +1,142 @@
+#!/usr/bin/env python3
+
+from typing import Literal
+from datetime import datetime, timedelta
+
+import requests
+from ..utils.response import ReturnResponse
+from .mail_detail import MailDetail
+from ..utils.timeutils import TimeUtils
+
+
+
+class AliMail:
+    '''
+    _summary_
+    '''
+    def __init__(self, mail_address: str=None, client_id: str=None, client_secret: str=None, timeout: int=3):
+        self.email_address = mail_address
+        self.client_id = client_id
+        self.client_secret = client_secret
+        self.base_url = "https://alimail-cn.aliyuncs.com/v2"
+        self.timeout = timeout
+        self.headers = {
+                "Content-Type": "application/json",
+                "Authorization": f"bearer {self._get_access_token_by_request()}"
+            }
+        self.session = requests.Session()
+        self.session.headers.update(self.headers)
+        
+    
+    def _get_access_token_by_request(self):
+        '''
+        https://mailhelp.aliyun.com/openapi/index.html#/markdown/authorization.md
+
+        Raises:
+            ValueError: _description_
+
+        Returns:
+            _type_: _description_
+        '''
+        # 定义接口URL
+        interface_url = "https://alimail-cn.aliyuncs.com/oauth2/v2.0/token"
+        # 设置请求头，指定内容类型
+        headers = {'Content-Type': 'application/x-www-form-urlencoded'}
+        # 准备请求数据
+        data = {
+            "grant_type": "client_credentials",
+            "client_id": self.client_id,
+            "client_secret": self.client_secret
+        }
+        try:
+            response = requests.post(interface_url, headers=headers, data=data, timeout=self.timeout)
+            response_json = response.json()
+            current_time = datetime.now()
+            data = {
+                'token_type': response_json["token_type"],
+                'access_token': response_json["access_token"],
+                'expires_in':  response_json["expires_in"],
+                'expiration_time': current_time + timedelta(seconds=response_json["expires_in"])
+            }
+            return data.get("access_token")
+        except requests.RequestException as e:
+            # 处理请求失败异常
+            raise e
+        except (KeyError, ValueError) as e:
+            # 处理解析响应失败异常
+            raise e
+
+    def get_mail_folders(self):
+        response = self.session.get(
+            url=f"{self.base_url}/users/{self.email_address}/mailFolders",
+            headers=self.headers
+        )
+        return response.json().get('folders')
+
+    def get_folder_id(self, folder_name: Literal['inbox']='inbox'):
+        folders = self.get_mail_folders()
+        for folder in folders:
+            if folder.get('displayName') == folder_name:
+                return folder.get('id')
+        return None
+
+    def get_mail_detail(self, mail_id: str):
+        params = {
+            "$select": "body,toRecipients,internetMessageId,internetMessageHeaders"
+        }
+        response = self.session.get(
+            url=f"{self.base_url}/users/{self.email_address}/messages/{mail_id}",
+            headers=self.headers,
+            params=params,
+            timeout=3
+        )
+        return response.json().get('message')
+
+    def get_mail_list(self, folder_name: str='inbox', size: int=100):
+        folder_id = self.get_folder_id(folder_name=folder_name)
+        params = {
+            "size": size,
+            # "$select": "toRecipients"
+        }
+        response = self.session.get(
+            url=f"{self.base_url}/users/{self.email_address}/mailFolders/{folder_id}/messages",
+            headers=self.headers,
+            params=params,
+            timeout=3
+        )
+        messages = response.json().get("messages")
+        sent_to_list = []
+        for message in messages:
+            mail_id = message.get("id")
+            detail = self.get_mail_detail(mail_id=mail_id)
+            
+            for to_recipient in detail.get('toRecipients'):
+                sent_to_list.append(to_recipient.get('email'))
+            
+            yield MailDetail(
+                uid=message.get('id'),
+                sent_from=message.get('from').get('email'),
+                sent_to=sent_to_list,
+                date=TimeUtils.convert_str_to_datetime(time_str=message.get('sentDateTime'), app='alimail'),
+                cc="",
+                subject=message.get('subject'),
+                body_plain=message.get('summary'),
+                body_html=""
+            )
+
+    def move(self, uid: str, destination_folder: str) -> ReturnResponse:
+        params = {
+            "ids": [uid],
+            "destinationFolderId": self.get_folder_id(destination_folder)
+        }
+        r = self.session.post(
+            url=f"{self.base_url}/users/{self.email_address}/messages/move",
+            params=params
+        )
+        if r.status_code == 200:
+            return ReturnResponse(code=0, msg=f'邮件移动到 {destination_folder} 成功', data=None)
+        else:
+            return ReturnResponse(code=1, msg=f'邮件移动到 {destination_folder} 失败', data=r.json())
+
+if __name__ == '__main__':
+    ali_mail = AliMail()

pytbox/mail/client.py

@@ -0,0 +1,171 @@
+#!/usr/bin/env python3
+
+
+from contextlib import contextmanager
+import yagmail
+from imap_tools import MailBox, MailMessageFlags, AND
+from .mail_detail import MailDetail
+from ..utils.response import ReturnResponse
+
+
+class MailClient:
+    '''
+    _summary_
+    '''
+    def __init__(self, 
+                 mail_address: str=None,
+                 password: str=None
+                ):
+        
+        self.mail_address = mail_address
+        self.password = password
+
+        if '163.com' in mail_address:
+            self.smtp_address = 'smtp.163.com'
+            self.imap_address = 'imap.163.com'
+            # self.imbox_client = self._create_imbox_object()
+        elif 'foxmail.com' in mail_address:
+            self.smtp_address = 'smtp.qq.com'
+            self.imap_address = 'imap.qq.com'
+        
+        elif 'mail' in mail_address and 'cn' in mail_address:
+            self.smtp_address = "smtpdm.aliyun.com"
+            self.imap_address = ""
+            
+        else:
+            raise ValueError(f'不支持的邮箱地址: {mail_address}')
+    
+    @contextmanager
+    def get_mailbox(self, readonly=False):
+        """
+        创建并返回一个已登录的 MailBox 上下文管理器。
+        
+        使用方式:
+            with self.get_mailbox() as mailbox:
+                # 使用 mailbox 进行操作
+                pass
+        
+        Args:
+            readonly (bool): 是否以只读模式打开邮箱，只读模式下不会将邮件标记为已读
+        
+        Yields:
+            MailBox: 已登录的邮箱对象
+        """
+        mailbox = MailBox(self.imap_address).login(self.mail_address, self.password)
+        if readonly:
+            # 以只读模式选择收件箱，防止邮件被标记为已读
+            mailbox.folder.set('INBOX', readonly=True)
+        try:
+            yield mailbox
+        finally:
+            mailbox.logout()
+
+    def send_mail(self, receiver: list=None, cc: list=None, subject: str='', contents: str='', attachments: list=None, tips: str=None):
+        '''
+        _summary_
+
+        Args:
+            receiver (list, optional): _description_. Defaults to None.
+            cc (list, optional): _description_. Defaults to None.
+            subject (str, optional): _description_. Defaults to ''.
+            contents (str, optional): _description_. Defaults to ''.
+            attachments (list, optional): _description_. Defaults to None.
+        '''
+        with yagmail.SMTP(user=self.mail_address, password=self.password, port=465, host=self.smtp_address) as yag:
+            try:
+                if tips:
+                    contents = contents + '\n' + '<p style="color: red;">本邮件为系统自动发送</p>'
+                yag.send(to=receiver, cc=cc, subject=subject, contents=contents, attachments=attachments)
+                return True
+            except Exception as e:
+                return False
+            
+    def get_mail_list(self, seen: bool=False, readonly: bool=True):
+        '''
+        获取邮件
+
+        Args:
+            seen (bool, optional): 默认获取未读邮件, 如果为 True, 则获取已读邮件
+            readonly (bool, optional): 是否以只读模式获取邮件，默认为 True，防止邮件被标记为已读
+
+        Yields:
+            MailDetail: 邮件详情
+        '''
+        with self.get_mailbox(readonly=readonly) as mailbox:
+            for msg in mailbox.fetch(AND(seen=seen)):
+                yield MailDetail(
+                    uid=msg.uid,
+                    sent_from=msg.from_,
+                    sent_to=msg.to,
+                    date=msg.date,
+                    cc=msg.cc,
+                    subject=msg.subject,
+                    body_plain=msg.text,
+                    body_html=msg.html
+                )     
+            
+    def mark_as_read(self, uid):
+        """
+        标记邮件为已读。
+        
+        Args:
+            uid (str): 邮件的唯一标识符
+        """
+        try:
+            with self.get_mailbox() as mailbox:
+                # 使用 imap_tools 的 flag 方法标记邮件为已读
+                # 第一个参数是 uid，第二个参数是要设置的标志，第三个参数 True 表示添加标志
+                mailbox.flag(uid, [MailMessageFlags.SEEN], True)
+        except Exception as e:
+            return ReturnResponse(code=1, msg='邮件删除失败', data=e)
+    
+    def delete(self, uid):
+        """
+        删除邮件。
+        
+        Args:
+            uid (str): 邮件的唯一标识符
+        """
+        try:
+            with self.get_mailbox() as mailbox:
+                # 使用 imap_tools 的 delete 方法删除邮件
+                mailbox.delete(uid)
+                # log.info(f'删除邮件{uid}')
+        except Exception as e:
+            # log.error(f'删除邮件{uid}失败: {e}')
+            raise
+    
+    def move(self, uid: str, destination_folder: str) -> ReturnResponse:
+        """
+        移动邮件到指定文件夹。
+
+        注意：部分 IMAP 服务商（如 QQ 邮箱）在移动邮件时，实际上是"复制到目标文件夹并从原文件夹删除"，
+        这会导致邮件在原文件夹中消失，表现为"被删除"。但邮件会在目标文件夹中存在，并未彻底丢失。
+
+        Args:
+            uid (str): 邮件的唯一标识符。
+            destination_folder (str): 目标文件夹名称。
+
+        Returns:
+            ReturnResponse: 移动邮件结果
+
+        Raises:
+            Exception: 移动过程中底层 imap 库抛出的异常。
+        """
+        try:
+            with self.get_mailbox() as mailbox:
+                # 使用 imap_tools 的 move 方法移动邮件
+                mailbox.move(uid, destination_folder)
+                return ReturnResponse(code=0, msg=f'邮件 {uid} 移动到 {destination_folder} 成功', data=None)
+        except Exception as e:
+            return ReturnResponse(code=1, msg=f'邮件 {uid} 移动到 {destination_folder} 失败', data=e)
+
+    def get_folder_list(self):
+        '''
+        获取文件夹列表
+        '''
+        with self.get_mailbox() as mailbox:
+            return ReturnResponse(code=0, msg='获取文件夹列表成功', data=mailbox.folder.list())
+
+if __name__ == '__main__':
+    pass