159 lines
4.2 KiB
Python
159 lines
4.2 KiB
Python
"""
|
||
项目数据模型
|
||
"""
|
||
from typing import Optional, List, Dict, Any
|
||
from .base import BaseModel
|
||
from .task import Task
|
||
|
||
class Project(BaseModel):
|
||
"""项目数据模型"""
|
||
|
||
def __init__(
|
||
self,
|
||
name: str,
|
||
color: str = "#FFD324",
|
||
group_id: Optional[str] = None,
|
||
tasks: Optional[List[Task]] = None,
|
||
in_all: bool = True,
|
||
kind: str = "TASK",
|
||
view_mode: str = "list",
|
||
**kwargs
|
||
):
|
||
"""
|
||
初始化项目实例
|
||
|
||
Args:
|
||
name: 项目名称
|
||
color: 项目颜色
|
||
group_id: 所属分组ID
|
||
tasks: 项目下的任务列表
|
||
in_all: 是否显示在所有项目中
|
||
kind: 项目类型
|
||
view_mode: 视图模式
|
||
**kwargs: 其他属性
|
||
"""
|
||
self.name = name
|
||
self.color = color
|
||
self.group_id = group_id
|
||
self.tasks = tasks or []
|
||
self.in_all = in_all
|
||
self.kind = kind
|
||
self.view_mode = view_mode
|
||
super().__init__(**kwargs)
|
||
|
||
@classmethod
|
||
def from_dict(cls, data: Dict[str, Any]) -> 'Project':
|
||
"""
|
||
从API响应数据创建项目实例
|
||
|
||
Args:
|
||
data: API响应数据
|
||
|
||
Returns:
|
||
Project: 项目实例
|
||
"""
|
||
# 处理布尔值响应
|
||
if isinstance(data, bool):
|
||
return None
|
||
|
||
# 处理任务列表
|
||
tasks = []
|
||
if isinstance(data.get('tasks'), list):
|
||
tasks = [
|
||
Task.from_dict(task_data)
|
||
for task_data in data['tasks']
|
||
]
|
||
|
||
return cls(
|
||
name=data.get('name', ''),
|
||
color=data.get('color', '#FFD324'),
|
||
group_id=data.get('groupId'),
|
||
tasks=tasks,
|
||
id=data.get('id'),
|
||
view_mode=data.get('viewMode', 'list'),
|
||
sort_order=data.get('sortOrder'),
|
||
sort_type=data.get('sortType'),
|
||
in_all=data.get('inAll', True),
|
||
kind=data.get('kind', 'TASK')
|
||
)
|
||
|
||
def to_dict(self) -> Dict[str, Any]:
|
||
"""
|
||
将项目转换为API请求数据
|
||
|
||
Returns:
|
||
Dict: API请求数据
|
||
"""
|
||
# 只包含必要的字段
|
||
data = {
|
||
'name': self.name,
|
||
'color': self.color,
|
||
'inAll': self.in_all,
|
||
'kind': self.kind,
|
||
'viewMode': self.view_mode
|
||
}
|
||
|
||
# 只在有值时添加可选字段
|
||
if self.group_id:
|
||
data['groupId'] = self.group_id
|
||
if hasattr(self, 'id') and self.id:
|
||
data['id'] = self.id
|
||
if hasattr(self, 'sortOrder'):
|
||
data['sortOrder'] = self.sortOrder
|
||
|
||
return data
|
||
|
||
def add_task(self, task: Task):
|
||
"""
|
||
添加任务到项目
|
||
|
||
Args:
|
||
task: 任务实例
|
||
"""
|
||
task.project_id = getattr(self, 'id', None)
|
||
self.tasks.append(task)
|
||
|
||
def remove_task(self, task_id: str):
|
||
"""
|
||
从项目中移除任务
|
||
|
||
Args:
|
||
task_id: 任务ID
|
||
"""
|
||
self.tasks = [
|
||
task for task in self.tasks
|
||
if getattr(task, 'id', None) != task_id
|
||
]
|
||
|
||
def get_task(self, task_id: str) -> Optional[Task]:
|
||
"""
|
||
获取项目中的特定任务
|
||
|
||
Args:
|
||
task_id: 任务ID
|
||
|
||
Returns:
|
||
Task: 任务实例，如果未找到则返回None
|
||
"""
|
||
for task in self.tasks:
|
||
if getattr(task, 'id', None) == task_id:
|
||
return task
|
||
return None
|
||
|
||
def get_completed_tasks(self) -> List[Task]:
|
||
"""
|
||
获取项目中已完成的任务
|
||
|
||
Returns:
|
||
List[Task]: 已完成的任务列表
|
||
"""
|
||
return [task for task in self.tasks if task.is_completed]
|
||
|
||
def get_uncompleted_tasks(self) -> List[Task]:
|
||
"""
|
||
获取项目中未完成的任务
|
||
|
||
Returns:
|
||
List[Task]: 未完成的任务列表
|
||
"""
|
||
return [task for task in self.tasks if not task.is_completed] |