Added better flair tools.

rstat_tool/flair_finder.py

@@ -0,0 +1,76 @@
+# rstat_tool/flair_finder.py
+# A dedicated tool to find available link flairs for a subreddit.
+
+import argparse
+import sys
+import os
+import praw
+from dotenv import load_dotenv
+from pathlib import Path
+
+def get_reddit_instance_for_flairs():
+    """
+    Initializes and returns an authenticated PRAW instance using the refresh token.
+    This is a copy of the robust authentication from the posting script.
+    """
+    # Find the .env file relative to the project root
+    env_path = Path(__file__).parent.parent / '.env'
+    load_dotenv(dotenv_path=env_path)
+    
+    client_id = os.getenv("REDDIT_CLIENT_ID")
+    client_secret = os.getenv("REDDIT_CLIENT_SECRET")
+    user_agent = os.getenv("REDDIT_USER_AGENT")
+    refresh_token = os.getenv("REDDIT_REFRESH_TOKEN")
+
+    if not all([client_id, client_secret, user_agent, refresh_token]):
+        print("Error: Reddit API credentials (including REDDIT_REFRESH_TOKEN) must be set in .env file.", file=sys.stderr)
+        return None
+    
+    return praw.Reddit(
+        client_id=client_id,
+        client_secret=client_secret,
+        user_agent=user_agent,
+        refresh_token=refresh_token
+    )
+
+def main():
+    """Main function to fetch and display flairs."""
+    parser = argparse.ArgumentParser(description="Fetch and display available post flairs for a subreddit.")
+    parser.add_argument("subreddit", help="The name of the subreddit to check.")
+    args = parser.parse_args()
+
+    reddit = get_reddit_instance_for_flairs()
+    if not reddit:
+        sys.exit(1)
+
+    print(f"\n--- Attempting to Fetch Post Flairs for r/{args.subreddit} ---")
+    
+    try:
+        # This uses PRAW's generic GET request method to hit the specific API endpoint.
+        api_path = f"/r/{args.subreddit}/api/link_flair_v2.json"
+        flairs = reddit.get(api_path, params={"raw_json": 1})
+        
+        if not flairs:
+            print("No flairs found or flair list is empty for this subreddit.")
+            return
+
+        print("\n--- Available Post Flairs ---")
+        found_count = 0
+        for flair in flairs:
+            flair_text = flair.get('text')
+            flair_id = flair.get('id')
+            if flair_text and flair_id:
+                print(f"  Flair Text: '{flair_text}'")
+                print(f"  Flair ID:   {flair_id}\n")
+                found_count += 1
+        
+        if found_count == 0:
+            print("No flairs with both text and ID were found.")
+
+    except Exception as e:
+        print(f"\nAn error occurred: {e}", file=sys.stderr)
+        print("Hint: Please ensure the subreddit exists and that your authenticated user has permission to view it.", file=sys.stderr)
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()