random problem selection

Author: Coderbeep

leetcode/main.py

@@ -52,10 +52,12 @@ def main():
 
     problem_parser = subparsers.add_parser('problem', help="Display problem")
     problem_parser.set_defaults(func=ProblemInfo)
-    problem_parser.add_argument('id', type=positive_integer, help='Problem ID of the problem')
+    problem_parser.add_argument('-r', '--random', action='store_true', help='Fetch a random problem.')
+    problem_parser.add_argument('id', type=positive_integer, help='Problem ID of the problem', default=0, nargs='?')
     problem_parser.add_argument('-b', '--browser', action='store_true', help='Open the page in browser.')
     problem_parser.add_argument('-f', '--file', action='store_true', help='Create a file with the problem content.')
 
+    
     today_problem_parser = subparsers.add_parser('today', help="Display today's problem.")
     today_problem_parser.set_defaults(func=QuestionOfToday)
 

leetcode/models/graphql_problemset_question_list.py

@@ -33,11 +33,11 @@ def from_dict(cls, data):
 class TotalCount(JSONWizard):
     total: int
 class ProblemTotalCount(QueryTemplate):
-    def __init__(self):
+    def __init__(self, filters={}):
         super().__init__()
         self.graphql_query = None
         self.result = None
-        self.params = {'categorySlug': "", 'skip': 0, 'limit': 10, 'filters': {}}
+        self.params = {'categorySlug': "", 'skip': 0, 'limit': 10, 'filters': filters}
 
         self.execute()
 
@@ -50,14 +50,16 @@ def __call__(self, *args: Any, **kwds: Any) -> Any:
 
 
 class ProblemsetQuestionList(QueryTemplate):
-    def __init__(self):
+    def __init__(self, filters={}, limit=None, skip=0):
         super().__init__()
         # Instance specific variables
         self.page : int = 1
         self.max_page : int = 0
-        self.limit = self.config.user_config.get('question_list_limit')
+        self.filters = filters
+        self.limit = limit or self.config.user_config.get('question_list_limit')
+        self.skip = skip
 
-        self.params = {'categorySlug': "", 'skip': 0, 'limit': self.limit, 'filters': {}}
+        self.params = {'categorySlug': "", 'skip':self.skip, 'limit': self.limit, 'filters': self.filters}
         self.graphql_query = None
         self.result = None
 
@@ -94,11 +96,9 @@ def execute(self, args):
     def validate_page(self):
         """ Method to validate the page number. If number is too large,
             set the page number to the last page."""
-            
         count = ProblemTotalCount().__call__()
-        if self.page > -(-count // self.limit): # ceil(total / limit)
-            self.page = -(-count // self.limit)
-            self.params['skip'] = self.limit * self.page - self.limit # update the skip value
+        self.page = min(self.page, -(-count // self.limit))
+        self.params['skip'] = self.limit * self.page - self.limit # update the skip value
 
     def show(self):
         displayed : int = self.limit * self.page if self.limit * self.page < self.result.total else self.result.total
@@ -113,4 +113,10 @@ def show(self):
         table.add_column('Difficulty')
         for item in self.result.questions:
             table.add_row(item.questionId, item.title, item.status, item.difficulty)
-        print(table)
+        print(table)
+        
+    def get_data(self):
+        self.graphql_query = GraphQLQuery(self.query, self.params)
+        self.result = self.leet_API.post_query(self.graphql_query)
+        
+        return self.result['data']

leetcode/models/problem_by_id_slug.py

@@ -4,6 +4,7 @@
 from leetcode.models.graphql_get_question_detail import GetQuestionDetail
 from leetcode.models.graphql_question_content import QuestionContent
 from leetcode.models.graphql_question_info_table import QuestionInfoTable
+from leetcode.models.graphql_problemset_question_list import ProblemTotalCount, ProblemsetQuestionList
 
 
 class ProblemInfo(QueryTemplate):
@@ -53,22 +54,42 @@ def parse_args(self, args):
             self.fileFlag = True
 
     def execute(self, args):
-        self.parse_args(args)
-        try:
-            with Loader('Fetching problem info...', ''):
-                self.result = self.leet_api.get_request(self.API_URL)
-                if getattr(args, 'id'):
-                    for item in self.result.get('stat_status_pairs', []):
-                        if item['stat'].get('question_id') == args.id:
-                            self.title_slug = item['stat'].get('question__title_slug', '')
-                            break
-                    if not self.title_slug:
-                        raise ValueError("Invalid ID has been provided. Please try again.")
-            self.show()
-        except Exception as e:
-            console.print(f"{e.__class__.__name__}: {e}", style=ALERT)
-        if self.fileFlag:
-            self.create_submission_file()
+        self.parse_args(args)          
+        if getattr(args, 'random'):
+            total = ProblemTotalCount({'status': 'NOT_STARTED'}).__call__()
+        
+            from random import randint
+            with Loader('Selecting random problem...', ''):
+                choosen_number = randint(1, total)
+                while True:
+                    list_instance = ProblemsetQuestionList({'status': 'NOT_STARTED'}, limit=1, skip=choosen_number - 1)
+                    problem = list_instance.get_data()['problemsetQuestionList']['questions'][0]
+                    if not problem['paidOnly']:
+                        break
+                    choosen_number = randint(1, total)
+
+            with Loader('Fetching problem contents...', ''):
+                question_info_table = QuestionInfoTable(problem['titleSlug'])
+                question_content = QuestionContent(problem['titleSlug'])
+            console.print(question_info_table)
+            console.print(question_content)
+            
+        else:
+            try:
+                with Loader('Fetching problem info...', ''):
+                    self.result = self.leet_api.get_request(self.API_URL)
+                    if getattr(args, 'id'):
+                        for item in self.result.get('stat_status_pairs', []):
+                            if item['stat'].get('question_id') == args.id:
+                                self.title_slug = item['stat'].get('question__title_slug', '')
+                                break
+                        if not self.title_slug:
+                            raise ValueError("Invalid ID has been provided. Please try again.")
+                self.show()
+            except Exception as e:
+                console.print(f"{e.__class__.__name__}: {e}", style=ALERT)
+            if self.fileFlag:
+                self.create_submission_file()
 
     def create_submission_file(self):
         question = GetQuestionDetail(self.title_slug)
@@ -88,4 +109,4 @@ def show(self):
             question_info_table = QuestionInfoTable(self.title_slug)
             console.print(question_info_table)
             question_content = QuestionContent(self.title_slug)
-            console.print(question_content)            
+            console.print(question_content)