check if the file is python or not

metatypedev · Dec 25, 2024 · 8f9fc5f · 8f9fc5f
1 parent 6cbfad5
commit 8f9fc5f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/typegate/engine/src/py_validation.rs b/src/typegate/engine/src/py_validation.rs
@@ -18,7 +18,11 @@ fn read_file(path: &str) -> Result<String> {
 
 #[deno_core::op2(fast)]
 pub fn op_validate(#[string] input: String) -> Result<()> {
-    let python_source = read_file(&input)?;
-    ast::Suite::parse(&python_source, "<embedded>")?;
+    if let Some(file_name) = input.split(".").last() {
+        if file_name == "py" {
+            let python_source = read_file(&input)?;
+            ast::Suite::parse(&python_source, "<embedded>")?;
+        }
+    }
     Ok(())
 }