__rich_console__ implementation in questionContent and questionInfoTable

Author: Coderbeep

leetcode/main.py

@@ -7,9 +7,7 @@
 
 # the file can be generated when the user displsays the question
 
-# TODO: Loading animations where needed
 # TODO: handle the wrong cases in the args parser
-# TODO: loading ASCII art
 # TODO: pipes support
 # TODO: add a command to open the question in editor
 # TODO: add a command to show the solution in the terminal

leetcode/models/graphql_question_content.py

@@ -20,4 +20,8 @@ def execute(self):
 
     def show(self):
         for x in self.question_panels:
-            print(x)
+            print(x)
+    
+    def __rich_console__(self, console: Console, options):
+        for x in self.question_panels:
+            yield x

leetcode/models/graphql_question_info_table.py

@@ -59,4 +59,7 @@ def format_table(self):
 
     def show(self):
         print(self.table)
+    
+    def __rich_console__(self, console: Console, options):
+        return self.table.__rich_console__(console, options)
 

leetcode/models/graphql_question_of_today.py

@@ -80,7 +80,7 @@ def show(self):
             titleSlug = self.result.question.titleSlug
             with Loader('Fetching question content...', ''):
                 question_content = questionContent(titleSlug)
-            question_content.show()
+            print(question_content)
 
         elif self.browserFlag:
             self.show_info_table()

leetcode/models/problem_by_id_slug.py

@@ -63,6 +63,6 @@ def execute(self, args):
     def show(self):
         # TODO: make the __repr__ method for this class
         question_info_table = questionInfoTable(self.title_slug)
-        question_info_table.show()
+        print(question_info_table)
         question_content = questionContent(self.title_slug)
-        question_content.show()
+        print(question_content)

leetcode/models/styles.py

@@ -1,7 +1,7 @@
 import rich
 from rich import print
 from rich.align import VerticalAlignMethod
-from rich.console import JustifyMethod, OverflowMethod, RenderableType
+from rich.console import Console, ConsoleOptions, JustifyMethod, OverflowMethod, RenderResult, RenderableType
 from rich.style import Style, StyleType
 from rich.table import Table
 from rich.text import Text
@@ -54,6 +54,7 @@ def add_row(self, *renderables: RenderableType | None, style: StyleType | None =
                 renderables[self.status_column_index] = status_retranslate[renderables[self.status_column_index]]
             renderables = tuple(renderables)
         return super().add_row(*renderables, style=style, end_section=end_section)
+
 
 class CustomBar(Bar):
     BEGIN_BLOCK_ELEMENTS = ["█", "█", "█", "▐", "▐", "▐", "▕", "▕"]

leetcode/models/submit.py

@@ -46,7 +46,6 @@ def execute(self, args):
         except Exception as e:
             console.print(f"{e.__class__.__name__}: {e}", style=ALERT)
 
-    # TODO: inform about the file not found
     def load_code(self, filename):
         with open(filename, 'r') as file:
             code = file.read()

solution.py

@@ -0,0 +1,3 @@
+class Solution:
+    def sum(self, num1: int, num2: int) -> int:
+        return num1 + num2