Skip to content

Commit a62da60

Browse files
committed
Update django decorator as well
Allow fasthtml to use starlette decorator
1 parent 9155103 commit a62da60

File tree

1 file changed

+2
-7
lines changed

1 file changed

+2
-7
lines changed

sdk/python/src/datastar_py/django.py

Lines changed: 2 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33

44
from django.http import StreamingHttpResponse as _StreamingHttpResponse
55

6-
from .sse import SSE_HEADERS, ServerSentEventGenerator, _async_map, _wrap_event
6+
from .sse import SSE_HEADERS, ServerSentEventGenerator, _sse_iterable_wrapper
77

88

99
class DatastarStreamingHttpResponse(_StreamingHttpResponse):
@@ -17,10 +17,5 @@ def sse_generator(generator_func):
1717
@wraps(generator_func)
1818
def _wrapper(*args, **kwargs):
1919
content = generator_func(*args, **kwargs)
20-
if isinstance(content, typing.AsyncIterable):
21-
content = _async_map(_wrap_event, content)
22-
else:
23-
content = map(_wrap_event, content)
24-
25-
return DatastarStreamingHttpResponse(content)
20+
return DatastarStreamingHttpResponse(_sse_iterable_wrapper(content))
2621
return _wrapper

0 commit comments

Comments
 (0)