Handle html file types

sparckles · Feb 28, 2022 · 5c7f7bc · 5c7f7bc
1 parent ef5e068
commit 5c7f7bc
Show file tree

Hide file tree

Showing 2 changed files with 14 additions and 3 deletions.
diff --git a/robyn/responses.py b/robyn/responses.py
@@ -9,8 +9,11 @@ def static_file(file_path):
     """
 
     return {
-        "response_type": "static_file",
-        "file_path": file_path
+        "type": "static_file",
+        "file_path": file_path,
+        # this is a hack for now
+        "body": "",
+        "status_code": 200,
     }
 
 

diff --git a/src/processor.rs b/src/processor.rs
@@ -280,8 +280,16 @@ async fn execute_http_function(
             let res = Python::with_gil(|py| -> PyResult<HashMap<String, String>> {
                 println!("This is the result of the code {:?}", output);
 
-                let res: HashMap<String, String> =
+                let mut res: HashMap<String, String> =
                     output.into_ref(py).downcast::<PyDict>()?.extract()?;
+
+                let response_type = res.get("type").unwrap();
+
+                if response_type == "static_file" {
+                    let file_path = res.get("file_path").unwrap();
+                    let contents = read_file(file_path);
+                    res.insert("body".to_owned(), contents.to_owned());
+                }
                 Ok(res)
             })?;