104 lines
3.9 KiB
Python
104 lines
3.9 KiB
Python
# post_to_reddit.py
|
|
|
|
import argparse
|
|
import os
|
|
import glob
|
|
from datetime import datetime, timezone
|
|
import praw
|
|
from dotenv import load_dotenv
|
|
|
|
# --- CONFIGURATION ---
|
|
IMAGE_DIR = "images"
|
|
|
|
def get_reddit_instance():
|
|
"""Initializes and returns a PRAW Reddit instance from .env credentials."""
|
|
load_dotenv()
|
|
client_id = os.getenv("REDDIT_CLIENT_ID")
|
|
client_secret = os.getenv("REDDIT_CLIENT_SECRET")
|
|
user_agent = os.getenv("REDDIT_USER_AGENT")
|
|
username = os.getenv("REDDIT_USERNAME") # <-- Add your Reddit username to .env
|
|
password = os.getenv("REDDIT_PASSWORD") # <-- Add your Reddit password to .env
|
|
|
|
if not all([client_id, client_secret, user_agent, username, password]):
|
|
print("Error: Reddit API credentials (including username/password) not found in .env file.")
|
|
return None
|
|
|
|
return praw.Reddit(
|
|
client_id=client_id,
|
|
client_secret=client_secret,
|
|
user_agent=user_agent,
|
|
username=username,
|
|
password=password
|
|
)
|
|
|
|
def find_latest_image(pattern):
|
|
"""Finds the most recent file in the IMAGE_DIR that matches a given pattern."""
|
|
try:
|
|
search_path = os.path.join(IMAGE_DIR, pattern)
|
|
list_of_files = glob.glob(search_path)
|
|
if not list_of_files:
|
|
return None
|
|
# The latest file will be the one with the highest modification time
|
|
latest_file = max(list_of_files, key=os.path.getmtime)
|
|
return latest_file
|
|
except Exception as e:
|
|
print(f"Error finding image file: {e}")
|
|
return None
|
|
|
|
def main():
|
|
"""Main function to find an image and post it to Reddit."""
|
|
parser = argparse.ArgumentParser(description="Find the latest sentiment image and post it to a subreddit.")
|
|
parser.add_argument("-s", "--subreddit", help="The source subreddit of the image to post. (Defaults to overall summary)")
|
|
parser.add_argument("-w", "--weekly", action="store_true", help="Post the weekly summary instead of the daily one.")
|
|
parser.add_argument("-t", "--target-subreddit", default="rstat", help="The subreddit to post the image to. (Default: rstat)")
|
|
args = parser.parse_args()
|
|
|
|
# --- 1. Determine filename pattern and post title ---
|
|
current_date_str = datetime.now(timezone.utc).strftime("%Y-%m-%d")
|
|
|
|
if args.subreddit:
|
|
view_type = "weekly" if args.weekly else "daily"
|
|
filename_pattern = f"{args.subreddit.lower()}_{view_type}_*.png"
|
|
post_title = f"{view_type.capitalize()} Ticker Sentiment for r/{args.subreddit} ({current_date_str})"
|
|
else:
|
|
# Default to the overall summary
|
|
if args.weekly:
|
|
print("Warning: --weekly flag has no effect for overall summary. Posting overall daily image.")
|
|
filename_pattern = "overall_summary_*.png"
|
|
post_title = f"Overall Top 10 Ticker Mentions Across Reddit ({current_date_str})"
|
|
|
|
print(f"Searching for image pattern: {filename_pattern}")
|
|
|
|
# --- 2. Find the latest image file ---
|
|
image_to_post = find_latest_image(filename_pattern)
|
|
|
|
if not image_to_post:
|
|
print(f"Error: No image found matching the pattern '{filename_pattern}'. Please run the scraper and exporter first.")
|
|
return
|
|
|
|
print(f"Found image: {image_to_post}")
|
|
|
|
# --- 3. Connect to Reddit and submit ---
|
|
reddit = get_reddit_instance()
|
|
if not reddit:
|
|
return
|
|
|
|
try:
|
|
target_sub = reddit.subreddit(args.target_subreddit)
|
|
print(f"Submitting '{post_title}' to r/{target_sub.display_name}...")
|
|
|
|
submission = target_sub.submit_image(
|
|
title=post_title,
|
|
image_path=image_to_post,
|
|
flair_id=None # Optional: You can add a flair ID here if you want
|
|
)
|
|
|
|
print("\n--- Post Successful! ---")
|
|
print(f"Post URL: {submission.shortlink}")
|
|
|
|
except Exception as e:
|
|
print(f"\nAn error occurred while posting to Reddit: {e}")
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() |