Added backend for allCourseURLs setter input

crs_main.py

@@ -1,4 +1,4 @@
-# from crs_scraper.crscraper import CRScraper
+from crs_scraper.crscraper import CRScraper
 from crs_scraper.crs_data import Data
 from crs_scraper.data_sorter import DataSorter, ScheduleGenerator
 from flask import Flask, Response, jsonify, make_response, request
@@ -14,18 +14,24 @@
 
 # ------------------------------------------------------------
 login_url = "https://crs.upd.edu.ph/"
-all_course_table_schedule_url = ["https://crs.upd.edu.ph/student_registration/class_search/5670", 
-                                 "https://crs.upd.edu.ph/student_registration/class_search/18849", 
-                                 "https://crs.upd.edu.ph/student_registration/class_search/18843",
-                                 "https://crs.upd.edu.ph/student_registration/class_search/19401",
-                                 "https://crs.upd.edu.ph/student_registration/class_search/19395"
-                                 ]
+# https://crs.upd.edu.ph/student_registration/class_search/5670, https://crs.upd.edu.ph/student_registration/class_search/18849, https://crs.upd.edu.ph/student_registration/class_search/18843, https://crs.upd.edu.ph/student_registration/class_search/19401, https://crs.upd.edu.ph/student_registration/class_search/19395
+# all_course_table_schedule_url = ["https://crs.upd.edu.ph/student_registration/class_search/5670", 
+#                                  "https://crs.upd.edu.ph/student_registration/class_search/18849", 
+#                                  "https://crs.upd.edu.ph/student_registration/class_search/18843",
+#                                  "https://crs.upd.edu.ph/student_registration/class_search/19401",
+#                                  "https://crs.upd.edu.ph/student_registration/class_search/19395"
+#                                  ]
+all_course_table_schedule_url: list[str] = []
+crs_username_global = ""
+crs_password_global = ""
 # ------------------------------------------------------------
 
 
 # ------------------------------------------------------------
 @app.route('/login', methods=['POST'])
 def login() -> Response:
+    global crs_username_global, crs_password_global
+
     credentials = request.json
     print("Received credentials:", credentials)
     if credentials:
@@ -40,36 +46,113 @@ def login() -> Response:
         print("Username: ", crs_username)
         print("Password: ", crs_password)
 
+        crs_username_global = crs_username
+        crs_password_global = crs_password
+
         # Use crscraper with the login credentials
-        # crs_scraper = CRScraper(login_url, crs_username, crs_password, all_course_table_schedule_url) # This should replace the crscraper variable once the crscraper is working
-        # data = crs_scraper.main() # This should replace the data variable once the crscraper is working
+        crs_scraper = CRScraper(login_url, crs_username, crs_password, all_course_table_schedule_url)
+        crs_scraper.login_into_crs() # Just to authenticate the user
+
+        # Return a success response with a 200 status code using make_response
+        response = make_response(jsonify({"message": "Login successful", "status": "success"}), 200)
+        return response
+
+    # Return a failure response with a 400 status code using make_response
+    response = make_response(jsonify({"message": "Missing credentials", "status": "failure"}), 400)
+    return response
 
-        # For now, we simulate succesful login and scraping data
-        data_class = Data()
-        data = data_class.data()
+    #     # crs_scraper = CRScraper(login_url, crs_username, crs_password, all_course_table_schedule_url) # This should replace the crscraper variable once the crscraper is working
+    #     # data = crs_scraper.main() # This should replace the data variable once the crscraper is working
 
-        if not data:
-            # Return a failure response with a 400 status code using make_response
-            response = make_response(jsonify({"message": "Failed to retrieve course data. Please try again", "status": "failure"}), 400)
-            return response
+    #     # For now, we simulate succesful login and scraping data
+    #     data_class = Data()
+    #     data = data_class.data()
+
+    #     if not data:
+    #         # Return a failure response with a 400 status code using make_response
+    #         response = make_response(jsonify({"message": "Failed to retrieve course data. Please try again", "status": "failure"}), 400)
+    #         return response
 
-        # Proceed with sorting and generating schedules
-        data_sorter = DataSorter(data)
-        data_sorter.sort_data()
-        # data_sorter.display_data(data_sorter.subjects_with_time)
+    #     # Proceed with sorting and generating schedules
+    #     data_sorter = DataSorter(data)
+    #     data_sorter.sort_data()
+    #     # data_sorter.display_data(data_sorter.subjects_with_time)
+
+    #     data_generator = ScheduleGenerator(data_sorter.subjects_with_time)
+    #     schedules = data_generator.generate_schedules(data_sorter.subjects_with_time)
+    #     # data_generator.display_all_possible_schedules(schedules)
+
+    #     ranked_schedules = data_generator.rank_by_probability(schedules)
+    #     # data_generator.display_all_possible_schedules(ranked_schedules)
+    #     data_generator.convert_to_csv(ranked_schedules, "schedules_ranked.csv")
 
-        data_generator = ScheduleGenerator(data_sorter.subjects_with_time)
-        schedules = data_generator.generate_schedules(data_sorter.subjects_with_time)
-        # data_generator.display_all_possible_schedules(schedules)
+    #     return jsonify({"message": "Login successful and schedule data generated", "status": "success"})
 
-        ranked_schedules = data_generator.rank_by_probability(schedules)
-        # data_generator.display_all_possible_schedules(ranked_schedules)
-        data_generator.convert_to_csv(ranked_schedules, "schedules_ranked.csv")
+    # # Return a failure response with a 400 status code using make_response
+    # response = make_response(jsonify({"message": "Missing credentials", "status": "failure"}), 400)
+    # return response
 
-        return jsonify({"message": "Login successful and schedule data generated", "status": "success"})
 
+@app.route('/set-urls', methods=['POST'])
+def set_urls() -> Response:
+    global all_course_table_schedule_url
+    data = request.get_json()
+    print("Received data:", data)
+
+    # Retrieve and store the course links submitted by the user
+    if data:
+        links = data['links']
+        all_course_table_schedule_url = [link.strip() for link in links.split(",") if link.strip()]
+
+        # Return a success response with a 200 status code using make_response
+        response = make_response(jsonify({"message": "Course links set successfully", "status": "success"}), 200)
+        return response
+
     # Return a failure response with a 400 status code using make_response
-    response = make_response(jsonify({"message": "Missing credentials", "status": "failure"}), 400)
+    response = make_response(jsonify({"message": "Missing course links", "status": "failure"}), 400)
+    return response
+
+
+@app.route('/scrape', methods=['POST'])
+def scrape() -> Response:
+    global crs_username_global, crs_password_global, all_course_table_schedule_url
+
+    # TODO: There should be a CRS scraper class that will authenticate the urls
+    if not all_course_table_schedule_url:
+        # Return a failure response with a 400 status code using make_response
+        response = make_response(jsonify({"message": "No course links set yet", "status": "failure"}), 400)
+        return response
+
+    # ----------------------------------------------------------------
+    # crs_scraper = CRScraper(login_url, crs_username_global, crs_password_global, all_course_table_schedule_url) # This should replace the crscraper variable once the crscraper is working
+    # data = crs_scraper.main() # This should replace the data variable once the crscraper is working
+    # ----------------------------------------------------------------
+
+    # For now, we simulate succesful login and scraping data
+    data_class = Data()
+    data = data_class.data()
+
+    # Scraping logic
+    if not data:
+        # Return a failure response with a 400 status code using make_response
+        response = make_response(jsonify({"message": "Failed to retrieve course data. Please try again", "status": "failure"}), 400)
+        return response
+
+    # Proceed with sorting and generating schedules
+    data_sorter = DataSorter(data)
+    data_sorter.sort_data()
+    # data_sorter.display_data(data_sorter.subjects_with
+
+    data_generator = ScheduleGenerator(data_sorter.subjects_with_time)
+    schedules = data_generator.generate_schedules(data_sorter.subjects_with_time)
+    # data_generator.display_all_possible_schedules(schedules)
+
+    ranked_schedules = data_generator.rank_by_probability(schedules)
+    # data_generator.display_all_possible_schedules(ranked_schedules)
+    data_generator.convert_to_csv(ranked_schedules, "schedules_ranked.csv")
+
+    # Return a success response with a 200 status code using make_response
+    response = make_response(jsonify({"message": "Schedule data generated", "status": "success"}), 200)
     return response