Move use_connection to sit next to functions that use them

Author: joshbmair

src/py/reactpy/reactpy/core/hooks.py

@@ -229,19 +229,6 @@ def context(
     return context
 
 
-# backend implementations should establish this context at the root of an app
-ConnectionContext: Context[Connection[Any] | None] = create_context(None)
-
-
-def use_connection() -> Connection[Any]:
-    """Get the current :class:`~reactpy.backend.types.Connection`."""
-    conn = use_context(ConnectionContext)
-    if conn is None:  # nocov
-        msg = "No backend established a connection."
-        raise RuntimeError(msg)
-    return conn
-
-
 def use_context(context: Context[_Type]) -> _Type:
     """Get the current value for the given context type.
 
@@ -263,6 +250,19 @@ def use_context(context: Context[_Type]) -> _Type:
     return provider.value
 
 
+# backend implementations should establish this context at the root of an app
+ConnectionContext: Context[Connection[Any] | None] = create_context(None)
+
+
+def use_connection() -> Connection[Any]:
+    """Get the current :class:`~reactpy.backend.types.Connection`."""
+    conn = use_context(ConnectionContext)
+    if conn is None:  # nocov
+        msg = "No backend established a connection."
+        raise RuntimeError(msg)
+    return conn
+
+
 def use_scope() -> MutableMapping[str, Any]:
     """Get the current :class:`~reactpy.backend.types.Connection`'s scope."""
     return use_connection().scope