Added functionality for exporting image, and cleaning up subreddits and stale tickers.

export_image.py

@@ -0,0 +1,51 @@
+# export_image.py
+
+import argparse
+from playwright.sync_api import sync_playwright
+import time
+
+def export_subreddit_image(subreddit_name, weekly=False):
+    """
+    Launches a headless browser to take a screenshot of a subreddit's image view.
+    """
+    view_type = "weekly" if weekly else "daily"
+    print(f"Exporting {view_type} image for r/{subreddit_name}...")
+
+    # The URL our Flask app serves
+    base_url = "http://127.0.0.1:5000"
+    path = f"image/weekly/{subreddit_name}" if weekly else f"image/{subreddit_name}"
+    url = f"{base_url}/{path}"
+    
+    # Define the output filename
+    output_file = f"{subreddit_name}_{'weekly' if weekly else 'daily'}_{int(time.time())}.png"
+
+    with sync_playwright() as p:
+        browser = p.chromium.launch()
+        page = browser.new_page()
+        
+        # Set a large viewport for high-quality screenshots
+        page.set_viewport_size({"width": 1920, "height": 1080})
+        
+        print(f"Navigating to {url}...")
+        page.goto(url)
+        
+        # Important: Give the page a second to ensure all styles and fonts have loaded
+        page.wait_for_timeout(1000) 
+        
+        # Target the specific element we want to screenshot
+        element = page.locator(".image-container")
+        
+        print(f"Saving screenshot to {output_file}...")
+        element.screenshot(path=output_file)
+        
+        browser.close()
+        print("Export complete!")
+
+if __name__ == "__main__":
+    parser = argparse.ArgumentParser(description="Export subreddit sentiment images.")
+    parser.add_argument("subreddit", help="The name of the subreddit to export.")
+    parser.add_argument("--weekly", action="store_true", help="Export the weekly view instead of the daily view.")
+    args = parser.parse_args()
+
+    # NOTE: This script assumes your 'rstat-dashboard' server is already running in another terminal.
+    export_subreddit_image(args.subreddit, args.weekly)