update: generate-snippet.py

Ryoga-exe · Oct 23, 2022 · 3ff830f · 3ff830f
1 parent df28b4a
commit 3ff830f
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/generate-snippet.py b/generate-snippet.py
@@ -23,11 +23,11 @@ def create_snippet(file_path):
     options = {}
 
     for index, line in enumerate(lines):
-        if line == '/// start':
+        if line.strip() == '/// start':
             startline = index + 1
-        if line.startswith('/// @'):
-            key = line[5:].split(' ')[0]
-            value = line[5 + len(key) + 1:]
+        if line.strip().startswith('/// @'):
+            key = line.strip()[5:].split(' ')[0]
+            value = line.strip()[5 + len(key) + 1:]
             if key == 'isFileTemplate':
                 if value == 'true':
                     options[key] = True
@@ -101,4 +101,4 @@ def write_file(snippets, directory, filename):
                         write_file(snippets, directory, filename)
 
                 print("Created", filename, "from", dir_name)
-
+