Allow overriding default response headers (#926)

Author: gazpachoking

sdk/python/src/datastar_py/django.py

@@ -23,6 +23,8 @@
 class DatastarResponse(_StreamingHttpResponse):
     """Respond with 0..N `DatastarEvent`s"""
 
+    default_headers: dict[str, str] = SSE_HEADERS.copy()
+
     def __init__(
         self,
         content: DatastarEvents = None,
@@ -34,7 +36,7 @@ def __init__(
             status = status or 204
             content = tuple()
         else:
-            headers = {**SSE_HEADERS, **(headers or {})}
+            headers = {**self.default_headers, **(headers or {})}
         if isinstance(content, DatastarEvent):
             content = (content,)
         super().__init__(content, status=status, headers=headers)

sdk/python/src/datastar_py/litestar.py

@@ -25,6 +25,8 @@
 class DatastarResponse(Stream):
     """Respond with 0..N `DatastarEvent`s"""
 
+    default_headers: dict[str, str] = SSE_HEADERS.copy()
+
     def __init__(
         self,
         content: DatastarEvents = None,
@@ -40,7 +42,7 @@ def __init__(
             status_code = status_code or 204
             content = tuple()
         else:
-            headers = {**SSE_HEADERS, **(headers or {})}
+            headers = {**self.default_headers, **(headers or {})}
         if isinstance(content, DatastarEvent):
             content = (content,)
         super().__init__(

sdk/python/src/datastar_py/quart.py

@@ -22,6 +22,8 @@
 class DatastarResponse(Response):
     """Respond with 0..N `DatastarEvent`s"""
 
+    default_headers: dict[str, str] = SSE_HEADERS.copy()
+
     def __init__(
         self,
         content: DatastarEvents = None,
@@ -31,7 +33,7 @@ def __init__(
         if not content:
             status = status or 204
         else:
-            headers = {**SSE_HEADERS, **(headers or {})}
+            headers = {**self.default_headers, **(headers or {})}
         super().__init__(content, status=status, headers=headers)
         if isgenerator(content) or isasyncgen(content):
             self.timeout = None

sdk/python/src/datastar_py/sanic.py

@@ -20,6 +20,8 @@
 
 
 class DatastarResponse(HTTPResponse):
+    default_headers: dict[str, str] = SSE_HEADERS.copy()
+
     def __init__(
         self,
         content: DatastarEvent | Collection[DatastarEvent] | None = None,
@@ -28,7 +30,9 @@ def __init__(
     ) -> None:
         if not content:
             status = status or 204
-        super().__init__(content, status=status or 200, headers={**SSE_HEADERS, **(headers or {})})
+        super().__init__(
+            content, status=status or 200, headers={**self.default_headers, **(headers or {})}
+        )
 
     async def send(
         self,

sdk/python/src/datastar_py/starlette.py

@@ -23,6 +23,8 @@
 class DatastarResponse(_StreamingResponse):
     """Respond with 0..N `DatastarEvent`s"""
 
+    default_headers: dict[str, str] = SSE_HEADERS.copy()
+
     def __init__(
         self,
         content: DatastarEvents = None,
@@ -35,7 +37,7 @@ def __init__(
             content = tuple()
         else:
             status_code = status_code or 200
-            headers = {**SSE_HEADERS, **(headers or {})}
+            headers = {**self.default_headers, **(headers or {})}
         if isinstance(content, DatastarEvent):
             content = (content,)
         super().__init__(content, status_code=status_code, headers=headers, background=background)