mirror of
https://github.com/openai/codex.git
synced 2026-04-28 02:11:08 +03:00
36 lines
1.0 KiB
Python
36 lines
1.0 KiB
Python
"""
|
|
Simple library for loading and saving task metadata embedded as TOML front-matter
|
|
in task Markdown files.
|
|
"""
|
|
import re
|
|
import toml
|
|
from pathlib import Path
|
|
from datetime import datetime
|
|
from pydantic import BaseModel, Field
|
|
|
|
FRONTMATTER_RE = re.compile(r"^\+\+\+\s*(.*?)\s*\+\+\+", re.S | re.M)
|
|
|
|
class TaskMeta(BaseModel):
|
|
id: str
|
|
title: str
|
|
status: str
|
|
dependencies: str = Field(default="")
|
|
last_updated: datetime = Field(default_factory=datetime.utcnow)
|
|
|
|
def load_task(path: Path) -> (TaskMeta, str):
|
|
text = path.read_text(encoding='utf-8')
|
|
m = FRONTMATTER_RE.match(text)
|
|
if not m:
|
|
raise ValueError(f"No TOML frontmatter in {path}")
|
|
meta = toml.loads(m.group(1))
|
|
tm = TaskMeta(**meta)
|
|
body = text[m.end():].lstrip('\n')
|
|
return tm, body
|
|
|
|
def save_task(path: Path, meta: TaskMeta, body: str) -> None:
|
|
tm = meta.dict()
|
|
tm['last_updated'] = meta.last_updated.isoformat()
|
|
fm = toml.dumps(tm).strip()
|
|
content = f"+++\n{fm}\n+++\n\n{body.lstrip()}"
|
|
path.write_text(content, encoding='utf-8')
|