added --no-file and -f options to cli

Author: Zohair-coder

README.md

@@ -44,6 +44,8 @@ The scraper will output a JSON file called `data.json` in the same directory as
 
 You can modify the scraper to scrape other terms by changing the `year`, `quarter`, and `college_code` variables in `src/config.py`.
 
+To view all the options that the scraper supports, run `python3 src/main.py --help` on Mac/Linux, or `python src/main.py --help` on Windows.  
+
 #### Authentication
 
 Since the term master schedule is only accessible to logged-in Drexel students, to run the scraper, you will need to provide your Drexel credentials as well as provide multi-factor authentication (MFA).

src/main.py

@@ -17,27 +17,36 @@ def main(args: argparse.Namespace) -> None:
     data = scrape(include_ratings=args.ratings, all_colleges=args.all_colleges)
 
     assert len(data) > 0, "No data found"
+    print("Found {} items".format(len(data)))
 
-    with open("data.json", "w") as f:
-        json.dump(data, f, indent=4)
+    if not args.no_file:
+        with open(args.output_file, "w") as f:
+            json.dump(data, f, indent=4)
 
-    print("Found {} items".format(len(data)))
-    print("Data written to data.json")
+        print(f"Data written to {args.output_file}")
 
     if args.db:
         print("Time taken to scrape data: {} seconds".format(time.time() - start_time))
         print()
         print("Updating database...")
         db.populate_db(data)
-
-    print("Done!")
+        print("Done!")
 
     print("--- {} seconds ---".format(time.time() - start_time))
 
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser(
-        description="Scrape data from Term Master Schedule and save it to a data.json file."
+        prog="python3 src/main.py",
+        description="Scrape data from Drexel Term Master Schedule and save it to a JSON file."
+    )
+    parser.add_argument(
+        "-o",
+        "--output-file",
+        metavar="FILE",
+        action="store",
+        default="data.json",
+        help="File to write the data to (include the .json extension in the file name)",
     )
     parser.add_argument(
         "--ratings",
@@ -49,6 +58,11 @@ def main(args: argparse.Namespace) -> None:
         action="store_true",
         help="Include all colleges in the data, not just the one in the config.py file",
     )
+    parser.add_argument(
+        "--no-file",
+        action="store_true",
+        help="Do not write the data to a file",
+    )
     parser.add_argument(
         "--db",
         action="store_true",