leet problem command amendment

Author: Coderbeep

leetcode/main.py

@@ -15,13 +15,9 @@
 # TODO: add a command to show the solution in the terminal
 # TODO: add a command to show the solution in the browser
 # TODO: problem with import in synced code or code to submit
-# TODO: random problem selector (from not accepted problems)
 # TODO: check the changes in question_content and apply them to the code in other files
-# TODO: use config without having to have a session
 # TODO: check all commands for errors
 # TODO: README - search - download - check - submit
-# TODO: leet problem -f      displays erors
-# TODO: submit with only id (or filename)
 
 def positive_integer(value):
     try:
@@ -62,6 +58,7 @@ def main():
     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.')
+    problem_parser.add_argument('-c', '--contents', action='store_true', help='Display contents of the question in the terminal.')
 
 
     today_problem_parser = subparsers.add_parser('today', help="Display today's problem.")
@@ -78,17 +75,18 @@ def main():
     submission_parser.set_defaults(func=SubmissionList)
 
     submission_parser = subparsers.add_parser('submit', help='Submit code answer')
-    submission_parser.add_argument('question_slug', type=str, help="Title slug of the question")
     submission_parser.add_argument('path', type=str, help='Path to the file with code answer')
     submission_parser.set_defaults(func=SendSubmission)
-        
-    
 
+    submission_parser = subparsers.add_parser('check', help='Check code answer on example test')
+    submission_parser.add_argument('path', type=str, help='Path to the file with code answer')
+    submission_parser.set_defaults(func=SendSubmission)
+        
     args = parser.parse_args()
 
     if hasattr(args, 'func'):
         command_instance = args.func()
-        command_instance._execute(args)  # call the private method __execute
+        command_instance._execute(args)
     else:
         print("Unknown command. Use 'leet --help' for available commands.")
 

leetcode/models/problem_by_id_slug.py

@@ -17,6 +17,8 @@ def __init__(self):
         # Instance specific variables
         self.browserFlag = False
         self.fileFlag = False
+        self.randomFlag = False
+        self.contentFlag = False
 
         self._question_id: int = None
         self._title_slug: str = None
@@ -107,8 +109,7 @@ def _execute(self, args):
                 question_content = QuestionContent(problem.titleSlug)
             console.print(question_info_table)
             console.print(question_content)
-            
-        else:
+        elif getattr(args, 'id') != 0:
             try:
                 with Loader('Fetching problem info...', ''):
                     self.data = self.leet_api.get_request(self.API_URL)
@@ -124,6 +125,8 @@ def _execute(self, args):
                 console.print(f"{e.__class__.__name__}: {e}", style=ALERT)
             if self.fileFlag:
                 self.create_submission_file(self.title_slug)
+        else:
+            console.print("Invalid ID has been provided. Please try again.", style=ALERT)
 
     @classmethod
     def create_submission_file(cls, title_slug: str = None) -> None:
@@ -133,9 +136,11 @@ def create_submission_file(cls, title_slug: str = None) -> None:
         Args:
             title_slug (str): The title slug of the problem.
         """
+        watermark_info = '# This file was created by pyleetcode-cli software.\n# Do NOT modify the name of the file.\n\n'
         question = GetQuestionDetail(title_slug)
         file_name = f"{question.question_id}.{question.title_slug}.py"
         with open(file_name, 'w') as file:
+            file.write(watermark_info)
             file.write(question.code_snippet)
         console.print(f"File '{file_name}' has been created.")
 
@@ -146,12 +151,15 @@ def show(self):
             link = self.config.host + f'/problems/{self.title_slug}/'
             console.print(f'Link to the problem: {link}')
             self.open_in_browser(link)
-        else:
+        elif self.contentFlag:
             question_info_table = QuestionInfoTable(self.title_slug)
             console.print(question_info_table)
             question_content = QuestionContent(self.title_slug)
             console.print(question_content)
-
+        else:
+            question_info_table = QuestionInfoTable(self.title_slug)
+            console.print(question_info_table)
+            
     def __parse_args(self, args) -> None:
         """ Parses the arguments passed to the query. 
         
@@ -161,6 +169,10 @@ def __parse_args(self, args) -> None:
             self.browserFlag = True
         if getattr(args, 'file'):
             self.fileFlag = True
+        if getattr(args, 'random'):
+            self.randomFlag = True
+        if getattr(args, 'contents'):
+            self.contentFlag = True
 
     @property
     def data(self):