dataclasses support

Coderbeep · Coderbeep · commit 5a16a628b50e · 2023-10-02T13:31:26.000+02:00
diff --git a/leetcode/models/problem_by_id_slug.py b/leetcode/models/problem_by_id_slug.py
@@ -97,14 +97,14 @@ def _execute(self, args):
                 choosen_number = randint(1, total)
                 while True:
                     list_instance = ProblemsetQuestionList({'status': 'NOT_STARTED'}, limit=1, skip=choosen_number - 1)
-                    problem = list_instance.fetch_data()['problemsetQuestionList']['questions'][0]
-                    if not problem['paidOnly']:
+                    problem = list_instance.fetch_data(list_instance.params).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'])
+                question_info_table = QuestionInfoTable(problem.titleSlug)
+                question_content = QuestionContent(problem.titleSlug)
             console.print(question_info_table)
             console.print(question_content)
             
