""" 任务数据模型 """ from typing import Optional, List, Dict, Any from datetime import datetime import pytz from .base import BaseModel class Task(BaseModel): """任务数据模型""" def __init__( self, title: str, content: str = "", priority: int = 0, status: int = 0, start_date: Optional[str] = None, due_date: Optional[str] = None, project_id: Optional[str] = None, tags: Optional[List[str]] = None, sort_order: int = 0, time_zone: str = "Asia/Shanghai", is_floating: bool = False, is_all_day: bool = False, reminder: str = "", reminders: Optional[List[str]] = None, repeat_flag: str = "", ex_date: Optional[List[str]] = None, items: Optional[List[Dict]] = None, progress: int = 0, modified_time: Optional[str] = None, etag: Optional[str] = None, deleted: int = 0, created_time: Optional[str] = None, creator: Optional[int] = None, attachments: Optional[List[Dict]] = None, column_id: str = "", kind: str = "TEXT", img_mode: int = 0, **kwargs ): """ 初始化任务实例 Args: title: 任务标题 content: 任务内容 priority: 优先级 (0-最低, 1-低, 3-中, 5-高) status: 任务状态 (0-未完成, 2-已完成) start_date: 开始时间 due_date: 截止时间 project_id: 所属项目ID tags: 标签列表 sort_order: 排序顺序 time_zone: 时区，默认为Asia/Shanghai is_floating: 是否浮动 is_all_day: 是否全天 reminder: 提醒 reminders: 提醒列表 repeat_flag: 重复标记 ex_date: 排除日期 items: 子项目 progress: 进度 modified_time: 修改时间 etag: 标签 deleted: 删除标记 created_time: 创建时间 creator: 创建者 attachments: 附件 column_id: 列ID kind: 类型 img_mode: 图片模式 **kwargs: 其他属性 """ self.title = title self.content = content self.priority = priority self.status = status self.time_zone = time_zone self.start_date = self._parse_datetime_with_timezone(start_date) self.due_date = self._parse_datetime_with_timezone(due_date) self.project_id = project_id self.tags = tags or [] self.sort_order = sort_order self.is_floating = is_floating self.is_all_day = is_all_day self.reminder = reminder self.reminders = reminders or [] self.repeat_flag = repeat_flag self.ex_date = ex_date or [] self.items = items or [] self.progress = progress self.modified_time = self._parse_datetime_with_timezone(modified_time) self.etag = etag self.deleted = deleted self.created_time = self._parse_datetime_with_timezone(created_time) self.creator = creator self.attachments = attachments or [] self.column_id = column_id self.kind = kind self.img_mode = img_mode super().__init__(**kwargs) def _parse_datetime_with_timezone(self, date_str: Optional[str]) -> Optional[datetime]: """ 解析时间字符串，并处理时区 Args: date_str: ISO格式的时间字符串 Returns: datetime: 转换后的datetime对象（带时区信息） """ if not date_str: return None try: # 设置默认时区为北京时间 local_tz = pytz.timezone('Asia/Shanghai') # 如果是ISO格式带时区的时间 if 'T' in date_str and ('+' in date_str or 'Z' in date_str): # 将Z替换为+00:00以便解析 clean_date_str = date_str.replace('Z', '+00:00') dt = datetime.fromisoformat(clean_date_str) # 如果时间没有时区信息，假定为UTC时间 if dt.tzinfo is None: dt = pytz.UTC.localize(dt) else: # 如果是普通格式的时间字符串，直接作为北京时间处理 dt = datetime.strptime(date_str, "%Y-%m-%d %H:%M:%S") dt = local_tz.localize(dt) # 确保返回北京时间 return dt.astimezone(local_tz) except Exception as e: print(f"Warning: Failed to parse datetime {date_str}: {e}") return None @classmethod def from_dict(cls, data: Dict[str, Any]) -> 'Task': """ 从API响应数据创建任务实例 Args: data: API响应数据 Returns: Task: 任务实例 """ return cls( title=data.get('title', ''), content=data.get('content', ''), priority=data.get('priority', 0), status=data.get('status', 0), start_date=data.get('startDate'), due_date=data.get('dueDate'), project_id=data.get('projectId'), tags=data.get('tags', []), sort_order=data.get('sortOrder', 0), time_zone=data.get('timeZone', 'Asia/Shanghai'), is_floating=data.get('isFloating', False), is_all_day=data.get('isAllDay', False), reminder=data.get('reminder', ''), reminders=data.get('reminders', []), repeat_flag=data.get('repeatFlag', ''), ex_date=data.get('exDate', []), items=data.get('items', []), progress=data.get('progress', 0), modified_time=data.get('modifiedTime'), etag=data.get('etag'), deleted=data.get('deleted', 0), created_time=data.get('createdTime'), creator=data.get('creator'), attachments=data.get('attachments', []), column_id=data.get('columnId', ''), kind=data.get('kind', 'TEXT'), img_mode=data.get('imgMode', 0), id=data.get('id'), created=data.get('created'), modified=data.get('modified') ) def to_dict(self) -> Dict[str, Any]: """ 将任务转换为API请求数据，时间会被转换为UTC时区 Returns: Dict: API请求数据 """ data = { 'title': self.title, 'content': self.content, 'priority': self.priority, 'status': self.status, 'sortOrder': self.sort_order, 'timeZone': 'Asia/Shanghai', # 固定使用北京时区 'isFloating': self.is_floating, 'isAllDay': self.is_all_day, 'reminder': self.reminder, 'reminders': self.reminders, 'repeatFlag': self.repeat_flag, 'exDate': self.ex_date, 'items': self.items, 'progress': self.progress, 'kind': self.kind, 'imgMode': self.img_mode } # 转换时间为UTC时区（用于API请求） if self.start_date: utc_start = self.start_date.astimezone(pytz.UTC) data['startDate'] = utc_start.strftime("%Y-%m-%dT%H:%M:%S.000Z") if self.due_date: utc_due = self.due_date.astimezone(pytz.UTC) data['dueDate'] = utc_due.strftime("%Y-%m-%dT%H:%M:%S.000Z") if self.modified_time: utc_modified = self.modified_time.astimezone(pytz.UTC) data['modifiedTime'] = utc_modified.strftime("%Y-%m-%dT%H:%M:%S.000Z") if self.created_time: utc_created = self.created_time.astimezone(pytz.UTC) data['createdTime'] = utc_created.strftime("%Y-%m-%dT%H:%M:%S.000Z") if self.tags: data['tags'] = self.tags if self.project_id: data['projectId'] = self.project_id if self.etag: data['etag'] = self.etag if self.creator: data['creator'] = self.creator if self.attachments: data['attachments'] = self.attachments if self.column_id: data['columnId'] = self.column_id if hasattr(self, 'id') and self.id: data['id'] = self.id return data @property def is_completed(self) -> bool: """任务是否已完成""" return self.status == 2 @property def is_overdue(self) -> bool: """任务是否已过期""" if not self.due_date: return False return self.due_date < datetime.now() def complete(self): """将任务标记为已完成""" self.status = 2 def uncomplete(self): """将任务标记为未完成""" self.status = 0 def add_tag(self, tag: str): """ 添加标签 Args: tag: 标签名称 """ if tag not in self.tags: self.tags.append(tag) def remove_tag(self, tag: str): """ 移除标签 Args: tag: 标签名称 """ if tag in self.tags: self.tags.remove(tag)