Private
Public Access
0
0

feat(twitter_threads): thread_from_dict + download_media/render_markdown CLIs (end-to-end -m pipeline)

thread_from_dict (JSON wire boundary) round-trips thread_data.json; download_media/render_markdown gain argparse main() + __main__ so the plan Task 5.3 pipeline runs via -m; dual-import added so all modules run standalone or via -m (G5). Pipeline integration test (html->names->render) + CLI tests cover the glue without network. TDD: red -> green (4 pipeline + 29 regression = 33 passed).
This commit is contained in:
2026-07-05 17:48:07 -04:00
parent 41656d6b7c
commit ed86731701
4 changed files with 199 additions and 4 deletions
+34 -1
View File
@@ -1,10 +1,16 @@
"""Download media URLs from ThreadData via stdlib urllib into a local directory."""
from __future__ import annotations
import argparse
import json
import sys
from pathlib import Path
from urllib.request import urlopen
from urllib.error import URLError
from urllib.parse import urlparse, parse_qs
from scripts.twitter_threads.error_types import ErrorInfo, make_error, Result, PostData, ThreadData
try:
from scripts.twitter_threads.error_types import ErrorInfo, make_error, Result, PostData, ThreadData, thread_from_dict
except ImportError:
from error_types import ErrorInfo, make_error, Result, PostData, ThreadData, thread_from_dict
def _classify(url: str) -> tuple[str, str]:
parsed = urlparse(url)
@@ -54,3 +60,30 @@ def download_media(thread: ThreadData, output_dir: Path) -> Result[list[Path]]:
return Result.err(make_error("WriteError", "download_media", str(e)))
paths.append(target)
return Result.ok(paths)
def main(argv: list[str] | None = None) -> int:
parser = argparse.ArgumentParser(description="Download media for a Twitter/X thread_data.json.")
parser.add_argument("--input", type=Path, required=True, help="thread_data.json path")
parser.add_argument("--output", type=Path, required=True, help="media output directory")
args = parser.parse_args(argv)
try:
raw = args.input.read_text(encoding="utf-8")
except OSError as e:
sys.stderr.write(f"error: ReadError: {e}\n")
return 1
try:
data = json.loads(raw)
except json.JSONDecodeError as e:
sys.stderr.write(f"error: JsonParseError: {e}\n")
return 1
thread = thread_from_dict(data)
result = download_media(thread, args.output)
if not result.is_ok:
sys.stderr.write(f"error: {result.error.kind}: {result.error.detail}\n")
return 1
print(f"downloaded {len(result.data)} file(s) to {args.output}")
return 0
if __name__ == "__main__":
sys.exit(main())
+24 -1
View File
@@ -1,7 +1,7 @@
"""Shared dataclasses for the scripts.twitter_threads namespace."""
from __future__ import annotations
from dataclasses import dataclass
from typing import Generic, TypeVar
from typing import Any, Generic, TypeVar
@dataclass(frozen=True, slots=True)
class ErrorInfo:
@@ -55,3 +55,26 @@ class Result(Generic[T]):
@classmethod
def err(cls, error: ErrorInfo) -> "Result[T]":
return cls(data=None, error=error)
def thread_from_dict(d: dict[str, Any]) -> ThreadData:
posts: list[PostData] = []
for pd in d.get("posts", []):
m = pd.get("metrics", {})
metrics = PostMetrics(
reply_count=int(m.get("reply_count", 0)),
repost_count=int(m.get("repost_count", 0)),
like_count=int(m.get("like_count", 0)),
view_count=m.get("view_count"),
)
posts.append(PostData(
post_id=pd.get("post_id", ""),
author=pd.get("author", ""),
handle=pd.get("handle", ""),
text=pd.get("text", ""),
timestamp=pd.get("timestamp", ""),
media_urls=tuple(pd.get("media_urls", [])),
reply_to_id=pd.get("reply_to_id"),
quote_of_id=pd.get("quote_of_id"),
metrics=metrics,
))
return ThreadData(root_post_id=d.get("root_post_id", ""), posts=tuple(posts), source_url=d.get("source_url", ""))
+42 -2
View File
@@ -1,7 +1,13 @@
"""Render ThreadData into a Markdown file with YAML front-matter."""
from __future__ import annotations
import argparse
import json
import sys
from pathlib import Path
from scripts.twitter_threads.error_types import ErrorInfo, make_error, Result, ThreadData, PostData
try:
from scripts.twitter_threads.error_types import ErrorInfo, make_error, Result, ThreadData, PostData, thread_from_dict
except ImportError:
from error_types import ErrorInfo, make_error, Result, ThreadData, PostData, thread_from_dict
_EM_DASH = "\u2014"
@@ -68,4 +74,38 @@ def render_markdown(thread: ThreadData, media_names: dict[str, list[str]], outpu
output.write_text(content, encoding="utf-8")
except OSError as e:
return Result.err(make_error("WriteError", "render_markdown", str(e)))
return Result.ok(output)
return Result.ok(output)
def main(argv: list[str] | None = None) -> int:
parser = argparse.ArgumentParser(description="Render a Twitter/X thread_data.json to Markdown.")
parser.add_argument("--input", type=Path, required=True, help="thread_data.json path")
parser.add_argument("--media-dir", type=Path, required=True, help="media directory to link")
parser.add_argument("--output", type=Path, required=True, help="output .md path")
args = parser.parse_args(argv)
try:
raw = args.input.read_text(encoding="utf-8")
except OSError as e:
sys.stderr.write(f"error: ReadError: {e}\n")
return 1
try:
data = json.loads(raw)
except json.JSONDecodeError as e:
sys.stderr.write(f"error: JsonParseError: {e}\n")
return 1
thread = thread_from_dict(data)
media_names: dict[str, list[str]] = {}
for post in thread.posts:
if args.media_dir.exists():
files = sorted(f.name for f in args.media_dir.glob(f"{post.post_id}_*"))
if files:
media_names[post.post_id] = files
result = render_markdown(thread, media_names, args.output)
if not result.is_ok:
sys.stderr.write(f"error: {result.error.kind}: {result.error.detail}\n")
return 1
print(str(args.output))
return 0
if __name__ == "__main__":
sys.exit(main())