fix conflict

Author: patrickZWY

manual_testing/mastodon_preview_test.py

@@ -1,25 +1,97 @@
-import os, sys 
+import argparse
+import os
+import sys
+from dataclasses import asdict
+from enum import StrEnum
+from pprint import pprint
+
 sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), "..")))
 
-from social_posters.mastodon import PosterMastodon
 from mastodon_manual_test import post_exceed_500_chars_limit_with_adoption_link
+from social_posters.mastodon import PosterMastodon
 
-poster = PosterMastodon.__new__(PosterMastodon)
-
-pet = post_exceed_500_chars_limit_with_adoption_link()
-post = poster.format_post(pet)
-
-main_caption, replies = poster._format_caption_thread(post)
+"""
+How to use this to see each stage of the pipeline:
+see pet information:
+python manual_tests/mastodon_preview.py --stage pet
+see post information (platform ready but not mastodon processed):
+python manual_tests/mastodon_preview.py --stage post
+see full formatting in mastodon:
+python manual_tests/mastodon_preview.py --stage debug
+see only main thread part of the formatting in mastodon:
+python manual_tests/mastodon_preview.py --stage main
+see only replies part of the formatting in mastodon:
+python manual_tests/mastodon_preview.py --stage replies
+see all stages:
+python manual_tests/mastodon_preview.py --stage all (or no arg default to all)
+"""
+class PreviewStage(StrEnum):
+    PET = "pet"
+    POST = "post"
+    DEBUG = "debug"
+    MAIN = "main"
+    REPLIES = "replies"
+    ALL = "all"
 
-print("\n" + "=" * 60)
-print("MAIN POST")
-print("=" * 60)
-print(main_caption)
-print(f"\nLength: {len(main_caption)}")
 
-for i, reply in enumerate(replies, start=1):
+def print_section(title: str) -> None:
     print("\n" + "=" * 60)
-    print(f"REPLY {i}")
+    print(title)
     print("=" * 60)
-    print(reply)
-    print(f"\nLength: {len(reply)}")
+
+
+def parse_args() -> argparse.Namespace:
+    parser = argparse.ArgumentParser(
+        description="Preview each stage of the Mastodon formatting pipeline."
+    )
+    parser.add_argument(
+        "--stage",
+        type=PreviewStage,
+        choices=list(PreviewStage),
+        default=PreviewStage.ALL,
+        help="Which construction stage to preview.",
+    )
+    return parser.parse_args()
+
+
+def should_show(selected: PreviewStage, target: PreviewStage) -> bool:
+    return selected in (target, PreviewStage.ALL)
+
+
+def main() -> None:
+    args = parse_args()
+    stage: PreviewStage = args.stage
+
+    poster = PosterMastodon.__new__(PosterMastodon)
+
+    pet = post_exceed_500_chars_limit_with_adoption_link()
+    post = poster.format_post(pet)
+
+    main_caption, replies, debug = poster._format_caption_thread_with_trace(post)
+
+    if should_show(stage, PreviewStage.PET):
+        print_section("PET")
+        pprint(pet)
+
+    if should_show(stage, PreviewStage.POST):
+        print_section("POST OBJECT")
+        pprint(post)
+
+    if should_show(stage, PreviewStage.DEBUG):
+        print_section("DEBUG PIPELINE")
+        pprint(asdict(debug))
+
+    if should_show(stage, PreviewStage.MAIN):
+        print_section("MAIN POST")
+        print(main_caption)
+        print(f"\nLength: {len(main_caption)}")
+
+    if should_show(stage, PreviewStage.REPLIES):
+        for i, reply in enumerate(replies, start=1):
+            print_section(f"REPLY {i}")
+            print(reply)
+            print(f"\nLength: {len(reply)}")
+
+
+if __name__ == "__main__":
+    main()

requirements.txt

@@ -7,6 +7,7 @@ certifi
 chardet==3.0.4
 charset-normalizer
 clarifai==2.6.2
+click==8.3.3
 configparser==3.8.1
 decorator
 EasyProcess==1.1
@@ -17,14 +18,21 @@ grpcio==1.78.0
 h11==0.16.0
 httpcore==1.0.9
 httpx==0.28.1
+hypothesis==6.152.7
 idna==2.10
+iniconfig==2.3.0
 instapy==0.6.16
 jsonschema==2.6.0
 Mastodon.py
 MeaningCloud-python==2.0.0
 outcome==1.3.0.post0
+packaging==26.2
+pip-tools==7.5.3
+pluggy==1.6.0
 plyer==2.1.0
 protobuf==3.20.3
+Pygments==2.20.0
+pyproject_hooks==1.2.0
 PySocks==1.7.1
 certifi==2026.4.22
 charset-normalizer==3.4.7
@@ -33,6 +41,10 @@ idna
 Mastodon.py==2.2.1
 python-dateutil==2.9.0.post0
 requests==2.33.1
+setuptools==82.0.0
 six==1.17.0
+sortedcontainers==2.4.0
+typing_extensions==4.15.0
 urllib3==2.6.3
 pytest
+wheel==0.47.0