Handle comments

elazarg · Jun 29, 2023 · ec8e3a2 · ec8e3a2
1 parent aa6b925
commit ec8e3a2
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/examples/tmp.py b/examples/tmp.py
@@ -12,7 +12,8 @@ def func3():
     def func4():
         pass
 
-    func5()
+    func3()
     :return:
     """
+    # func3()
     pass
diff --git a/uncalled/regex_finder.py b/uncalled/regex_finder.py
@@ -2,6 +2,9 @@
 
 
 def prepare(txt: str) -> str:
+    # this is a hack to remove comments. It may remove too much, but rarely
+    txt = re.sub(r'#[^\r\n]*', '', txt)
+
     return re.sub(r'"""[^\r]*?"""' + r"|'''[^\r]*?'''", '...', txt)