Add ruff config and format all files

Author: gazpachoking

sdk/python/pyproject.toml

@@ -83,3 +83,24 @@ commit_args = ""
 setup_hooks = []
 pre_commit_hooks = []
 post_commit_hooks = []
+
+[tool.ruff]
+exclude = [
+  "src/datastar_py/consts.py",
+]
+[tool.ruff.lint]
+select = [
+    # pycodestyle
+    "E",
+    # Pyflakes
+    "F",
+    # pyupgrade
+    "UP",
+    # flake8-bugbear
+    "B",
+    # flake8-simplify
+    "SIM",
+    # isort
+    "I",
+]
+fixable = ["ALL"]

sdk/python/src/datastar_py/__init__.py

@@ -1,3 +1,3 @@
 from .sse import SSE_HEADERS, ServerSentEventGenerator
 
-__all__ = ["ServerSentEventGenerator", "SSE_HEADERS"]
+__all__ = ["SSE_HEADERS", "ServerSentEventGenerator"]

sdk/python/src/datastar_py/django.py

@@ -2,19 +2,28 @@
 from functools import wraps
 from typing import Any
 
-from django.http import StreamingHttpResponse as _StreamingHttpResponse, HttpRequest
+from django.http import HttpRequest
+from django.http import StreamingHttpResponse as _StreamingHttpResponse
 
 from .sse import SSE_HEADERS, ServerSentEventGenerator
 
+__all__ = [
+    "SSE_HEADERS",
+    "DatastarStreamingHttpResponse",
+    "ServerSentEventGenerator",
+    "read_signals",
+]
+
 
 class DatastarStreamingHttpResponse(_StreamingHttpResponse, ServerSentEventGenerator):
     @wraps(_StreamingHttpResponse.__init__)
     def __init__(self, *args, **kwargs):
         kwargs["headers"] = {**SSE_HEADERS, **kwargs.get("headers", {})}
         super().__init__(*args, **kwargs)
 
+
 def read_signals(request: HttpRequest) -> dict[str, Any]:
     if request.method == "GET":
-        return json.loads(request.GET.get("datastar"))
+        return json.loads(request.GET["datastar"])
     else:
-        return json.loads(request.body)
+        return json.loads(request.body)

sdk/python/src/datastar_py/fastapi.py

@@ -1,7 +1,17 @@
 from typing import Annotated, Any
 
 from fastapi import Depends
-from .starlette import DatastarStreamingResponse, ServerSentEventGenerator, read_signals
+
+from .sse import SSE_HEADERS, ServerSentEventGenerator
+from .starlette import DatastarStreamingResponse, read_signals
+
+__all__ = [
+    "SSE_HEADERS",
+    "ServerSentEventGenerator",
+    "SignalsDep",
+    "read_signals",
+    "DatastarStreamingResponse",
+]
 
 
 SignalsDep = Annotated[dict[str, Any], Depends(read_signals)]

sdk/python/src/datastar_py/fasthtml.py

@@ -1,2 +1,9 @@
-from .sse import ServerSentEventGenerator
+from .sse import SSE_HEADERS, ServerSentEventGenerator
 from .starlette import DatastarStreamingResponse, read_signals
+
+__all__ = [
+    "SSE_HEADERS",
+    "DatastarStreamingResponse",
+    "ServerSentEventGenerator",
+    "read_signals",
+]

sdk/python/src/datastar_py/quart.py

@@ -1,9 +1,17 @@
 import json
 from typing import Any
 
-from quart import make_response as _make_response, request
+from quart import make_response as _make_response
+from quart import request
 
-from .sse import ServerSentEventGenerator, SSE_HEADERS
+from .sse import SSE_HEADERS, ServerSentEventGenerator
+
+__all__ = [
+    "SSE_HEADERS",
+    "ServerSentEventGenerator",
+    "make_datastar_response",
+    "read_signals",
+]
 
 
 async def make_datastar_response(async_generator):
@@ -14,6 +22,6 @@ async def make_datastar_response(async_generator):
 
 async def read_signals() -> dict[str, Any]:
     if request.method == "GET":
-        return json.loads(request.args.get("datastar"))
+        return json.loads(request.args["datastar"])
     else:
         return await request.get_json()

sdk/python/src/datastar_py/sanic.py

@@ -4,15 +4,23 @@
 from .sse import SSE_HEADERS, ServerSentEventGenerator
 
 if TYPE_CHECKING:
-    from sanic import Request, HTTPResponse
+    from sanic import HTTPResponse, Request
+
+__all__ = [
+    "SSE_HEADERS",
+    "ServerSentEventGenerator",
+    "datastar_respond",
+    "read_signals",
+]
 
 
 async def datastar_respond(request: "Request") -> "HTTPResponse":
     response = await request.respond(headers=SSE_HEADERS)
     return response
 
+
 async def read_signals(request: "Request") -> dict[str, Any]:
     if request.method == "GET":
-        return json.loads(request.args.get("datastar"))
+        return json.loads(request.args["datastar"])
     else:
         return await request.json()

sdk/python/src/datastar_py/sse.py

@@ -60,13 +60,19 @@ def merge_fragments(
             fragments = fragments.__html__()
         data_lines = []
         if merge_mode:
-            data_lines.append(f"data: {consts.MERGE_MODE_DATALINE_LITERAL} {merge_mode}")
+            data_lines.append(
+                f"data: {consts.MERGE_MODE_DATALINE_LITERAL} {merge_mode}"
+            )
         if selector:
             data_lines.append(f"data: {consts.SELECTOR_DATALINE_LITERAL} {selector}")
         if use_view_transition:
-            data_lines.append(f"data: {consts.USE_VIEW_TRANSITION_DATALINE_LITERAL} true")
+            data_lines.append(
+                f"data: {consts.USE_VIEW_TRANSITION_DATALINE_LITERAL} true"
+            )
         else:
-            data_lines.append(f"data: {consts.USE_VIEW_TRANSITION_DATALINE_LITERAL} false")
+            data_lines.append(
+                f"data: {consts.USE_VIEW_TRANSITION_DATALINE_LITERAL} false"
+            )
 
         data_lines.extend(
             f"data: {consts.FRAGMENTS_DATALINE_LITERAL} {x}"
@@ -92,9 +98,13 @@ def remove_fragments(
         if selector:
             data_lines.append(f"data: {consts.SELECTOR_DATALINE_LITERAL} {selector}")
         if use_view_transition:
-            data_lines.append(f"data: {consts.USE_VIEW_TRANSITION_DATALINE_LITERAL} true")
+            data_lines.append(
+                f"data: {consts.USE_VIEW_TRANSITION_DATALINE_LITERAL} true"
+            )
         else:
-            data_lines.append(f"data: {consts.USE_VIEW_TRANSITION_DATALINE_LITERAL} false")
+            data_lines.append(
+                f"data: {consts.USE_VIEW_TRANSITION_DATALINE_LITERAL} false"
+            )
 
         return ServerSentEventGenerator._send(
             consts.EventType.REMOVE_FRAGMENTS,

sdk/python/src/datastar_py/starlette.py

@@ -7,6 +7,13 @@
 
 from .sse import SSE_HEADERS, ServerSentEventGenerator
 
+__all__ = [
+    "SSE_HEADERS",
+    "DatastarStreamingResponse",
+    "ServerSentEventGenerator",
+    "read_signals",
+]
+
 
 class DatastarStreamingResponse(_StreamingResponse, ServerSentEventGenerator):
     @wraps(_StreamingResponse.__init__)
@@ -17,6 +24,6 @@ def __init__(self, *args, **kwargs):
 
 async def read_signals(request: Request) -> dict[str, Any]:
     if request.method == "GET":
-        return json.loads(request.query_params.get("datastar"))
+        return json.loads(request.query_params["datastar"])
     else:
         return await request.json()