283 lines
9.4 KiB
Python
283 lines
9.4 KiB
Python
"""
|
||
任务数据模型
|
||
"""
|
||
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) |