move template parameter into notebook generator itself

Author: Sorosliu1029

.gitignore

@@ -1,3 +1,5 @@
+*.ipynb
+.ipynb_checkpoints
 *.bundle.*
 lib/
 node_modules/

jupyterlab_leetcode/handlers/leetcode_handler.py

@@ -1,5 +1,4 @@
 import json
-import os
 from typing import Any, Mapping, cast, overload
 
 import tornado
@@ -356,18 +355,7 @@ async def post(self):
 
         notebook_generator = self.settings.get("notebook_generator")
         if not notebook_generator:
-            template_path = os.path.join(
-                os.path.dirname(os.path.realpath(__file__)),
-                "..",
-                "utils",
-                "notebook.template.json",
-            )
-            if not os.path.exists(template_path):
-                self.set_status(500)
-                self.finish({"message": "Notebook template not found"})
-                return
-
-            notebook_generator = NotebookGenerator(template_path)
+            notebook_generator = NotebookGenerator()
             self.settings.update(notebook_generator=notebook_generator)
 
         file_path = notebook_generator.generate(question)

jupyterlab_leetcode/utils/notebook_generator.py

@@ -1,7 +1,3 @@
-"""
-Use LeetCode's GraphQL API to generate question notebook
-"""
-
 import json
 import os
 import re
@@ -16,7 +12,12 @@ def get_folder_for(qid: int, interval: int) -> str:
 
 
 class NotebookGenerator:
-    def __init__(self, template_path):
+    def __init__(self):
+        template_path = os.path.join(
+            os.path.dirname(os.path.realpath(__file__)),
+            "notebook.template.json",
+        )
+
         with open(template_path, "rt") as f:
             self.template = json.load(f)