feat: remove comments from python code (#103)

* feat: remove comments from python code * add tests --------- Co-authored-by: Naomi Carrigan <[email protected]>
freeCodeCamp · Dec 11, 2023 · f3996cd · f3996cd
1 parent c67c0cc
commit f3996cd
Show file tree

Hide file tree

Showing 2 changed files with 24 additions and 0 deletions.
diff --git a/lib/__tests__/python.test.ts b/lib/__tests__/python.test.ts
@@ -25,6 +25,26 @@ def c():
     }
   });
 
+  it("removeComments", () => {
+    const code = `
+a = 1
+# comment
+def b(d, e):
+  a = 2
+  # comment
+  return a #comment
+`;
+    const result = python.removeComments(code);
+    expect(result).toEqual(`
+a = 1
+
+def b(d, e):
+  a = 2
+  
+  return a 
+`);
+  });
+
   it("getBlock", () => {
     const code = `
 a = 1

diff --git a/lib/index.ts b/lib/index.ts
@@ -124,6 +124,10 @@ export module python {
     return null;
   }
 
+  export function removeComments(code: string) {
+    return code.replace(/\/\/.*|\/\*[\s\S]*?\*\/|(#.*$)/gm, "");
+  }
+
   /**
    * Gets a Python block of code matching the `blockPattern`
    * @param code - Code string to search