picoweb: start_response: Allows to pass headers param as a dict.

pfalcon · Jan 3, 2018 · dd68a0d · dd68a0d
1 parent 3f332ad
commit dd68a0d
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/picoweb/__init__.py b/picoweb/__init__.py
@@ -44,7 +44,14 @@ def start_response(writer, content_type="text/html", status="200", headers=None)
         yield from writer.awrite("\r\n\r\n")
         return
     yield from writer.awrite("\r\n")
-    yield from writer.awrite(headers)
+    if isinstance(headers, bytes) or isinstance(headers, str):
+        yield from writer.awrite(headers)
+    else:
+        for h, v in headers.items():
+            yield from writer.awrite(k)
+            yield from writer.awrite(": ")
+            yield from writer.awrite(v)
+            yield from writer.awrite("\r\n")
     yield from writer.awrite("\r\n")
 
 def http_error(writer, status):