mirror of
https://github.com/cita-777/metapi.git
synced 2026-10-31 19:47:31 +08:00
111 lines
3.1 KiB
Python
Executable File
111 lines
3.1 KiB
Python
Executable File
"""
|
||
任务数据访问层。
|
||
|
||
加载和遍历任务目录的唯一事实来源，替代散落在 9 个以上文件中的 task.json 解析。
|
||
|
||
提供：
|
||
load_task — 按目录路径加载单个任务
|
||
iter_active_tasks — 按排序遍历所有未归档任务
|
||
get_all_statuses — 获取用于子任务进度的 {dir_name: status} 映射
|
||
"""
|
||
|
||
from __future__ import annotations
|
||
|
||
from collections.abc import Iterator
|
||
from pathlib import Path
|
||
|
||
from .io import read_json
|
||
from .paths import FILE_TASK_JSON
|
||
from .types import TaskInfo
|
||
|
||
|
||
def load_task(task_dir: Path) -> TaskInfo | None:
|
||
"""从包含 task.json 的目录加载任务。
|
||
|
||
参数：
|
||
task_dir：任务目录的绝对路径。
|
||
|
||
返回：
|
||
task.json 存在且有效时返回 TaskInfo，否则返回 None。
|
||
"""
|
||
task_json = task_dir / FILE_TASK_JSON
|
||
if not task_json.is_file():
|
||
return None
|
||
|
||
data = read_json(task_json)
|
||
if not data:
|
||
return None
|
||
|
||
return TaskInfo(
|
||
dir_name=task_dir.name,
|
||
directory=task_dir,
|
||
title=data.get("title") or data.get("name") or "unknown",
|
||
status=data.get("status", "unknown"),
|
||
assignee=data.get("assignee", ""),
|
||
priority=data.get("priority", "P2"),
|
||
children=tuple(data.get("children", [])),
|
||
parent=data.get("parent"),
|
||
package=data.get("package"),
|
||
raw=data,
|
||
)
|
||
|
||
|
||
def iter_active_tasks(tasks_dir: Path) -> Iterator[TaskInfo]:
|
||
"""按目录名称排序遍历所有活动（未归档）任务。
|
||
|
||
跳过 "archive" 目录和不包含有效 task.json 的目录。
|
||
|
||
参数：
|
||
tasks_dir：tasks 目录路径。
|
||
|
||
生成：
|
||
每个有效任务的 TaskInfo。
|
||
"""
|
||
if not tasks_dir.is_dir():
|
||
return
|
||
|
||
for d in sorted(tasks_dir.iterdir()):
|
||
if not d.is_dir() or d.name == "archive":
|
||
continue
|
||
info = load_task(d)
|
||
if info is not None:
|
||
yield info
|
||
|
||
|
||
def get_all_statuses(tasks_dir: Path) -> dict[str, str]:
|
||
"""获取所有活动任务的 {dir_name: status} 映射。
|
||
|
||
可在不加载完整 TaskInfo 的情况下计算子任务进度。
|
||
|
||
参数：
|
||
tasks_dir：tasks 目录路径。
|
||
|
||
返回：
|
||
目录名称到 status 字符串的映射。
|
||
"""
|
||
return {t.dir_name: t.status for t in iter_active_tasks(tasks_dir)}
|
||
|
||
|
||
def children_progress(
|
||
children: tuple[str, ...] | list[str],
|
||
all_statuses: dict[str, str],
|
||
) -> str:
|
||
"""将子任务进度格式化为类似 " [2/3 done]" 的字符串。
|
||
|
||
参数：
|
||
children：子任务目录名称列表。
|
||
all_statuses：get_all_statuses() 返回的 status 映射。
|
||
|
||
返回：
|
||
格式化字符串；没有子任务时返回 ""。
|
||
"""
|
||
if not children:
|
||
return ""
|
||
# active statuses 中缺失的子任务已被归档（cmd_archive 移动目录前会将 status 设为
|
||
# completed）。将其计为完成，避免归档子任务后父任务进度倒退。
|
||
done = sum(
|
||
1 for c in children
|
||
if c not in all_statuses or all_statuses.get(c) in ("completed", "done")
|
||
)
|
||
return f" [{done}/{len(children)} done]"
|