twitter scraper script

scraper/__main__.py

@@ -0,0 +1,52 @@
+import os
+import sys
+import argparse
+from twitter_scraper import Twitter_Scraper
+
+try:
+    from dotenv import load_dotenv
+
+    print("Loading .env file")
+    load_dotenv()
+    print("Loaded .env file\n")
+except Exception as e:
+    print(f"Error loading .env file: {e}")
+    sys.exit(1)
+
+
+def main():
+    try:
+        USER_UNAME = os.getenv('TWITTER_USERNAME')
+        USER_PASSWORD = os.getenv('TWITTER_PASSWORD')
+    except Exception as e:
+        print(f"Error retrieving environment variables: {e}")
+        USER_UNAME = None
+        USER_PASSWORD = None
+        sys.exit(1)
+
+    parser = argparse.ArgumentParser(description='Twitter Scraper')
+    parser.add_argument('--tweets', type=int, default=50,
+                        help='Number of tweets to scrape (default: 50)')
+    args = parser.parse_args()
+
+    if USER_UNAME is not None and USER_PASSWORD is not None:
+        try:
+            scraper = Twitter_Scraper(
+                username=USER_UNAME,
+                password=USER_PASSWORD,
+                max_tweets=args.tweets
+            )
+
+            scraper.scrape_tweets()
+            scraper.driver.close()
+            scraper.save_to_csv()
+        except KeyboardInterrupt:
+            print("\nScript Interrupted by user. Exiting...")
+            sys.exit(1)
+    else:
+        print("Missing Twitter username or password environment variables. Please check your .env file.")
+        sys.exit(1)
+
+
+if __name__ == '__main__':
+    main()