Improved logging and added option for --stdout.

rstat_tool/main.py

@@ -12,13 +12,12 @@ from dotenv import load_dotenv
 from . import database
 from .ticker_extractor import extract_tickers
 from .sentiment_analyzer import get_sentiment_score
-from .logger_setup import get_logger
+from .logger_setup import setup_logging, logger as log
 
 load_dotenv()
 MARKET_CAP_REFRESH_INTERVAL = 86400
 POST_AGE_LIMIT = 86400 
 
-log = get_logger()
 
 def load_subreddits(filepath):
     try:
@@ -133,7 +132,7 @@ def scan_subreddits(reddit, subreddits_list, post_limit=100, comment_limit=100,
             log.error(f"Could not scan r/{subreddit_name}. Error: {e}")
             
     conn.close()
-    log.info("\n--- Scan Complete ---")
+    log.critical("\n--- Scan Complete ---")
 
 
 def main():
@@ -145,16 +144,19 @@ def main():
     parser.add_argument("-d", "--days", type=int, default=1, help="Number of past days to scan for new posts.\n(Default: 1 for last 24 hours)")
     parser.add_argument("-p", "--posts", type=int, default=200, help="Max posts to check per subreddit.\n(Default: 200)")
     parser.add_argument("-c", "--comments", type=int, default=100, help="Number of comments to scan per post.\n(Default: 100)")
-    parser.add_argument("-l", "--limit", type=int, default=20, help="Number of tickers to show in the CLI report.\n(Default: 20)")
+    parser.add_argument("--stdout", action="store_true", help="Print all log messages to the console.")
+
     args = parser.parse_args()
+    
+    setup_logging(console_verbose=args.stdout)
 
     if args.subreddit:
         # If --subreddit is used, create a list with just that one.
         subreddits_to_scan = [args.subreddit]
-        log.info(f"Targeted Scan Mode: Focusing on r/{args.subreddit}")
+        log.critical(f"Targeted Scan Mode: Focusing on r/{args.subreddit}")
     else:
         # Otherwise, load from the config file.
-        log.info(f"Config Scan Mode: Loading subreddits from {args.config}")
+        log.critical(f"Config Scan Mode: Loading subreddits from {args.config}")
         # Use the correct argument name: args.config
         subreddits_to_scan = load_subreddits(args.config) 
 
@@ -175,7 +177,6 @@ def main():
         comment_limit=args.comments,
         days_to_scan=args.days
     )
-    database.generate_summary_report(limit=args.limit)
 
 if __name__ == "__main__":
     main()