Add timeframe filter for company questions

Refactor functions to include a timeframe parameter when fetching questions by company. Update parsing logic to handle the new `--timeframe` argument. Document usage examples for fetching questions based on the specified timeframe.

Author: nicholasadamou

.github/README.md

@@ -99,6 +99,20 @@ Or without Docker:
 poetry run python main.py --plans leetcode-75 top-interview-150
 ```
 
+### Want to fetch the most asked questions for a specific company in the last N months or days?
+
+```sh
+docker compose run leetcode-scraper --company google --timeframe 3m
+```
+
+Or without Docker:
+
+```sh
+poetry run python main.py --company google --timeframe 3m
+```
+
+This will fetch the most asked questions at Google in the last 3 months.
+
 ## Running Tests
 
 Run the tests with the following command:

leetcode/leetcode.py

@@ -131,11 +131,14 @@ def get_problem(self, slug: str) -> Problem:
         """
         return self.problems[slug] if slug in self.problems else None
 
-    def fetch_and_store_company_problems(self, company: str) -> List[Problem]:
+    def fetch_and_store_company_problems(
+        self, company: str, timeframe: str
+    ) -> List[Problem]:
         """
         Fetch problems from LeetCode by the company tag and store them in the companies' dictionary.
 
         :param company: The company tag.
+        :param timeframe: The timeframe for questions (e.g., '30d', '3m', '6m').
         :return: A dictionary of fetched Problem objects with the company tag as the key.
         """
         with self.companies_lock:
@@ -153,8 +156,20 @@ def fetch_and_store_company_problems(self, company: str) -> List[Problem]:
 
         company_problems = []
 
+        # convert 30d, 3m, 6m to 'last-30-days', 'three-months', 'six-months'
+        if timeframe == "30d":
+            timeframe = "last-30-days"
+        elif timeframe == "3m":
+            timeframe = "three-months"
+        elif timeframe == "6m":
+            timeframe = "six-months"
+        else:
+            timeframe = "six-months"
+
         questions = _fetch_with_retries(
-            lambda: self.client.get_recent_questions_for_company(company)
+            lambda: self.client.get_recent_questions_for_company(
+                company, timeframe=timeframe
+            ),
         )
 
         if not questions:

main.py

@@ -13,7 +13,9 @@
 load_dotenv()
 
 
-def main(csrf_token, leetcode_session, plans, company):
+def main(args):
+    [csrf_token, leetcode_session, plans, company, timeframe] = args
+
     if not leetcode_session and not csrf_token:
         print(
             "Using Non-Premium LeetCode account. Some premium only data will not be returned."
@@ -42,9 +44,11 @@ def main(csrf_token, leetcode_session, plans, company):
                 print(e)
 
     if company:
-        print(f"Fetching company related problems: {company}")
+        print(f"Fetching company related problems: {company} for {timeframe}")
         try:
-            company_problems = leetcode.fetch_and_store_company_problems(company)
+            company_problems = leetcode.fetch_and_store_company_problems(
+                company, timeframe
+            )
             print(
                 "====================================================================================================="
             )
@@ -91,7 +95,13 @@ def main(csrf_token, leetcode_session, plans, company):
         default=None,
         help="Company name to filter problems by",
     )
+    parser.add_argument(
+        "--timeframe",
+        type=str,
+        default="6m",
+        help="The timeframe for questions (e.g., 'last-30-days', 'three-months', 'six-months')",
+    )
 
     args = parser.parse_args()
 
-    main(args.csrf_token, args.leetcode_session, args.plans, args.company)
+    main(args)