Fixing regex to accommodate extra whitespace

spakin · Mar 21, 2024 · 94a096b · 94a096b
1 parent 4e2becf
commit 94a096b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/simpinkscr/simple_inkscape_scripting.py b/simpinkscr/simple_inkscape_scripting.py
@@ -3724,8 +3724,8 @@ def effect(self):
 
         # Remove an unnecessary import that may be introduced when
         # running from Visual Studio Code.
-        pattern = r"^(import (inkex|simpinkscr)|(from (inkex|simpinkscr) import.*))\r*\n*"
-        code = re.sub(pattern, "", code, flags=re.MULTILINE+re.IGNORECASE)
+        pattern = r"^[ \t]*(import\s+(inkex|simpinkscr).*|(from\s+(inkex|simpinkscr)\s+import).*)[\r\n]?"
+        code = re.sub(pattern, "", code, flags=re.MULTILINE)
 
         # Launch the user's script.
         try: