latch-asgi 0.2.0__py3-none-any.whl → 0.3.0__py3-none-any.whl

latch_asgi/framework/http.py

@@ -2,11 +2,10 @@ from http import HTTPStatus
 from typing import Any, Literal, TypeAlias, TypeVar, cast
 
 import opentelemetry.context as context
-import simdjson
+import orjson
 from latch_data_validation.data_validation import DataValidationError, validate
 from latch_o11y.o11y import trace_function, trace_function_with_span
 from opentelemetry.trace.span import Span
-from orjson import dumps
 
 from ..asgi_iface import (
     HTTPReceiveCallable,
@@ -44,7 +43,6 @@ def current_http_request_span() -> Span:
 
 class HTTPErrorResponse(RuntimeError):
     def __init__(self, status: HTTPStatus, data: Any, *, headers: Headers = {}):
-        super().__init__()
         self.status = status
         self.data = data
         self.headers = headers
@@ -91,13 +89,7 @@ async def receive_class(receive: HTTPReceiveCallable, cls: type[T]) -> T:
 
 @trace_function(tracer)
 async def receive_json(receive: HTTPReceiveCallable) -> Any:
-    res = await receive_data(receive)
-
-    p = simdjson.Parser()
-    try:
-        return p.parse(res, True)
-    except ValueError as e:
-        raise HTTPBadRequest("Failed to parse JSON") from e
+    return orjson.loads(await receive_data(receive))
 
 
 async def receive_data(receive: HTTPReceiveCallable):
@@ -174,7 +166,7 @@ async def send_json(
     headers: Headers = {},
 ):
     return await send_http_data(
-        send, status, dumps(data), content_type=content_type, headers=headers
+        send, status, orjson.dumps(data), content_type=content_type, headers=headers
     )
 
 

latch_asgi/framework/websocket.py

@@ -2,10 +2,10 @@ from enum import Enum
 from typing import Any, TypeVar, cast
 
 import opentelemetry.context as context
-import simdjson
 from latch_data_validation.data_validation import DataValidationError, validate
 from latch_o11y.o11y import trace_function, trace_function_with_span
 from opentelemetry.trace.span import Span
+import orjson
 
 from ..asgi_iface import (
     WebsocketAcceptEvent,
@@ -29,28 +29,26 @@ def current_websocket_request_span() -> Span:
 
 
 class WebsocketStatus(int, Enum):
+    """
+    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
+    """
+
     normal = 1000
     """
     1000 indicates a normal closure, meaning that the purpose for
     which the connection was established has been fulfilled.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     going_away = 1001
     """
     1001 indicates that an endpoint is "going away", such as a server
     going down or a browser having navigated away from a page.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     protocol_error = 1002
     """
     1002 indicates that an endpoint is terminating the connection due
     to a protocol error.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     unsupported = 1003
@@ -59,15 +57,11 @@ class WebsocketStatus(int, Enum):
     because it has received a type of data it cannot accept (e.g., an
     endpoint that understands only text data MAY send this if it
     receives a binary message).
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     reserved = 1004
     """
     Reserved.  The specific meaning might be defined in the future.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     no_status = 1005
@@ -76,8 +70,6 @@ class WebsocketStatus(int, Enum):
     Close control frame by an endpoint.  It is designated for use in
     applications expecting a status code to indicate that no status
     code was actually present.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     abnormal = 1006
@@ -87,8 +79,6 @@ class WebsocketStatus(int, Enum):
     applications expecting a status code to indicate that the
     connection was closed abnormally, e.g., without sending or
     receiving a Close control frame.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     unsupported_payload = 1007
@@ -97,8 +87,6 @@ class WebsocketStatus(int, Enum):
     because it has received data within a message that was not
     consistent with the type of the message (e.g., non-UTF-8 [RFC3629]
     data within a text message).
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     policy_violation = 1008
@@ -108,8 +96,6 @@ class WebsocketStatus(int, Enum):
     is a generic status code that can be returned when there is no
     other more suitable status code (e.g., 1003 or 1009) or if there
     is a need to hide specific details about the policy.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     too_large = 1009
@@ -117,8 +103,6 @@ class WebsocketStatus(int, Enum):
     1009 indicates that an endpoint is terminating the connection
     because it has received a message that is too big for it to
     process.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     mandatory_extension = 1010
@@ -130,8 +114,6 @@ class WebsocketStatus(int, Enum):
     are needed SHOULD appear in the /reason/ part of the Close frame.
     Note that this status code is not used by the server, because it
     can fail the WebSocket handshake instead.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     server_error = 1011
@@ -139,8 +121,6 @@ class WebsocketStatus(int, Enum):
     1011 indicates that a server is terminating the connection because
     it encountered an unexpected condition that prevented it from
     fulfilling the request.
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
     tls_handshake_fail = 1015
@@ -150,12 +130,10 @@ class WebsocketStatus(int, Enum):
     applications expecting a status code to indicate that the
     connection was closed due to a failure to perform a TLS handshake
     (e.g., the server certificate can't be verified).
-
-    https://www.rfc-editor.org/rfc/rfc6455.html#section-7.4.1
     """
 
 
-class WebsocketError(RuntimeError):
+class WebsocketErrorResponse(RuntimeError):
     def __init__(self, status: WebsocketStatus, data: Any, *, headers: Headers = {}):
         super().__init__()
         self.status = status
@@ -163,12 +141,6 @@ class WebsocketError(RuntimeError):
         self.headers = headers
 
 
-class WebsocketErrorResponse(WebsocketError): ...
-
-
-class WebsocketConnectionClosedError(RuntimeError): ...
-
-
 class WebsocketBadMessage(WebsocketErrorResponse):
     def __init__(self, data: Any, *, headers: Headers = {}):
         super().__init__(WebsocketStatus.policy_violation, data, headers=headers)
@@ -179,11 +151,13 @@ class WebsocketInternalServerError(WebsocketErrorResponse):
         super().__init__(WebsocketStatus.server_error, data, headers=headers)
 
 
+class WebsocketConnectionClosedError(RuntimeError): ...
+
+
 # >>> I/O
 
 
 async def receive_websocket_data(receive: WebsocketReceiveCallable):
-    res = b""
     with tracer.start_as_current_span("read websocket message") as s:
         msg = await receive()
 
@@ -203,19 +177,13 @@ async def receive_websocket_data(receive: WebsocketReceiveCallable):
         else:
             raise WebsocketBadMessage("empty message")
 
-        s.set_attributes({"size": len(res)})
+        s.set_attribute("size", len(res))
         return res
 
 
 @trace_function(tracer)
 async def receive_websocket_json(receive: WebsocketReceiveCallable) -> Any:
-    res = await receive_websocket_data(receive)
-
-    p = simdjson.Parser()
-    try:
-        return p.parse(res, True)
-    except ValueError as e:
-        raise WebsocketBadMessage("Failed to parse JSON") from e
+    orjson.loads(await receive_websocket_data(receive))
 
 
 @trace_function(tracer)
@@ -253,9 +221,8 @@ async def send_websocket_data(
     )
 
 
-@trace_function_with_span(tracer)
+@trace_function(tracer)
 async def accept_websocket_connection(
-    s: Span,
     send: WebsocketSendCallable,
     receive: WebsocketReceiveCallable,
     /,

latch_asgi/server.py

@@ -52,7 +52,6 @@ from .framework.websocket import (
     accept_websocket_connection,
     close_websocket_connection,
     current_websocket_request_span,
-    send_websocket_data,
     websocket_request_span_key,
 )
 
@@ -295,7 +294,7 @@ class LatchASGIServer:
                         if x["type"] != type_str(e):
                             continue
 
-                        return validate(x, e)
+                        return untraced_validate(x, e)
 
                     raise RuntimeError(
                         f"unknown websocket event type: {repr(x['type'])}"
@@ -306,7 +305,7 @@ class LatchASGIServer:
                     await send(data)
 
                 return await self.scope_websocket(
-                    validate(scope, WebsocketScope), ws_receive, ws_send
+                    untraced_validate(scope, WebsocketScope), ws_receive, ws_send
                 )
 
             if scope["type"] == "http":

latch_asgi-0.3.0.dist-info/METADATA

@@ -0,0 +1,19 @@
+Metadata-Version: 2.1
+Name: latch-asgi
+Version: 0.3.0
+Summary: ASGI python server
+Author-Email: Max Smolin <max@latch.bio>
+License: CC0-1.0
+Requires-Python: <4.0,>=3.11
+Requires-Dist: hypercorn[uvloop]<1.0.0,>=0.14.3
+Requires-Dist: latch-data-validation<1.0.0,>=0.1.3
+Requires-Dist: latch-o11y<1.0.0,>=0.1.4
+Requires-Dist: latch-config<1.0.0,>=0.1.6
+Requires-Dist: PyJWT[crypto]<3.0.0,>=2.6.0
+Requires-Dist: orjson<4.0.0,>=3.8.5
+Requires-Dist: opentelemetry-sdk<2.0.0,>=1.15.0
+Requires-Dist: opentelemetry-api<2.0.0,>=1.15.0
+Requires-Dist: opentelemetry-instrumentation-asgi<1.0,>=0.36b0
+Description-Content-Type: text/markdown
+
+# python-asgi

{latch_asgi-0.2.0.dist-info → latch_asgi-0.3.0.dist-info}/RECORD

@@ -1,3 +1,6 @@
+latch_asgi-0.3.0.dist-info/METADATA,sha256=x1WxOrLQcsLzJzLEaFCl6yCsK0AkZ8EHdhp82vIYYPo,643
+latch_asgi-0.3.0.dist-info/WHEEL,sha256=N2J68yzZqJh3mI_Wg92rwhw0rtJDFpZj9bwQIMJgaVg,90
+latch_asgi-0.3.0.dist-info/licenses/COPYING,sha256=ogEPNDSH0_dhiv_lT3ifVIdgIzHAqNA_SemnxUfPBJk,7048
 latch_asgi/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 latch_asgi/asgi_iface.py,sha256=8cltPG2YEf20RWcMpn2FE5g4wtauiUNNlrVnD4UYDqc,8397
 latch_asgi/auth.py,sha256=gXGVIzsvDt_kBlcS_Sx9939iGlNNs1i8KDOs-tpEBiw,4996
@@ -9,10 +12,8 @@ latch_asgi/context/websocket.py,sha256=NA0pJYdJI2kUnSu1AkPnf9heG8_B4M4_op7S8MPly
 latch_asgi/datadog_propagator.py,sha256=PFKaM87B8Ia_5RBNGIIDSgcgTjxhN0uJou2o8359kzs,3306
 latch_asgi/framework/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 latch_asgi/framework/common.py,sha256=My4ePrmLD8aZd7_PC-ZJPZ9CUD90j5UZvHwA0jfTmFw,157
-latch_asgi/framework/http.py,sha256=4ux_CtLiBuv_MHp9IfyuwBTgC2t3vgAuQI4_3BHB9qI,5182
-latch_asgi/framework/websocket.py,sha256=mol3zAy4ku4Fy-XFl9qD_a5CV5FDyDr8fr4nJp406Ns,8974
+latch_asgi/framework/http.py,sha256=8AMzA1FjyaFVexVPQbi1NBdbWaH6594R5AAsjoSL67Y,4992
+latch_asgi/framework/websocket.py,sha256=Xyf4EUurKPxdHGV3CXUpCMLYCE41h_zz9GjJaR3NZgw,7993
 latch_asgi/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-latch_asgi/server.py,sha256=YtG9ckKzMq8ej3Du26Xk7OcZysEUe3Gc211PgfM6QCo,12054
-latch_asgi-0.2.0.dist-info/METADATA,sha256=irapIxpHrfyhGYFg7GfXiqSJlZEjBosiV0H54adpzTo,870
-latch_asgi-0.2.0.dist-info/WHEEL,sha256=Zb28QaM1gQi8f4VCBhsUklF61CTlNYfs9YAZn-TOGFk,88
-latch_asgi-0.2.0.dist-info/RECORD,,
+latch_asgi/server.py,sha256=kMZ7ivOk7VkZ5HjrIHcveXd2ZMawnnW7QM5-JuHDXZA,12047
+latch_asgi-0.3.0.dist-info/RECORD,,

{latch_asgi-0.2.0.dist-info → latch_asgi-0.3.0.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.6.1
+Generator: pdm-backend (2.1.8)
 Root-Is-Purelib: true
 Tag: py3-none-any

latch_asgi-0.3.0.dist-info/licenses/COPYING

@@ -0,0 +1,121 @@
+Creative Commons Legal Code
+
+CC0 1.0 Universal
+
+    CREATIVE COMMONS CORPORATION IS NOT A LAW FIRM AND DOES NOT PROVIDE
+    LEGAL SERVICES. DISTRIBUTION OF THIS DOCUMENT DOES NOT CREATE AN
+    ATTORNEY-CLIENT RELATIONSHIP. CREATIVE COMMONS PROVIDES THIS
+    INFORMATION ON AN "AS-IS" BASIS. CREATIVE COMMONS MAKES NO WARRANTIES
+    REGARDING THE USE OF THIS DOCUMENT OR THE INFORMATION OR WORKS
+    PROVIDED HEREUNDER, AND DISCLAIMS LIABILITY FOR DAMAGES RESULTING FROM
+    THE USE OF THIS DOCUMENT OR THE INFORMATION OR WORKS PROVIDED
+    HEREUNDER.
+
+Statement of Purpose
+
+The laws of most jurisdictions throughout the world automatically confer
+exclusive Copyright and Related Rights (defined below) upon the creator
+and subsequent owner(s) (each and all, an "owner") of an original work of
+authorship and/or a database (each, a "Work").
+
+Certain owners wish to permanently relinquish those rights to a Work for
+the purpose of contributing to a commons of creative, cultural and
+scientific works ("Commons") that the public can reliably and without fear
+of later claims of infringement build upon, modify, incorporate in other
+works, reuse and redistribute as freely as possible in any form whatsoever
+and for any purposes, including without limitation commercial purposes.
+These owners may contribute to the Commons to promote the ideal of a free
+culture and the further production of creative, cultural and scientific
+works, or to gain reputation or greater distribution for their Work in
+part through the use and efforts of others.
+
+For these and/or other purposes and motivations, and without any
+expectation of additional consideration or compensation, the person
+associating CC0 with a Work (the "Affirmer"), to the extent that he or she
+is an owner of Copyright and Related Rights in the Work, voluntarily
+elects to apply CC0 to the Work and publicly distribute the Work under its
+terms, with knowledge of his or her Copyright and Related Rights in the
+Work and the meaning and intended legal effect of CC0 on those rights.
+
+1. Copyright and Related Rights. A Work made available under CC0 may be
+protected by copyright and related or neighboring rights ("Copyright and
+Related Rights"). Copyright and Related Rights include, but are not
+limited to, the following:
+
+  i. the right to reproduce, adapt, distribute, perform, display,
+     communicate, and translate a Work;
+ ii. moral rights retained by the original author(s) and/or performer(s);
+iii. publicity and privacy rights pertaining to a person's image or
+     likeness depicted in a Work;
+ iv. rights protecting against unfair competition in regards to a Work,
+     subject to the limitations in paragraph 4(a), below;
+  v. rights protecting the extraction, dissemination, use and reuse of data
+     in a Work;
+ vi. database rights (such as those arising under Directive 96/9/EC of the
+     European Parliament and of the Council of 11 March 1996 on the legal
+     protection of databases, and under any national implementation
+     thereof, including any amended or successor version of such
+     directive); and
+vii. other similar, equivalent or corresponding rights throughout the
+     world based on applicable law or treaty, and any national
+     implementations thereof.
+
+2. Waiver. To the greatest extent permitted by, but not in contravention
+of, applicable law, Affirmer hereby overtly, fully, permanently,
+irrevocably and unconditionally waives, abandons, and surrenders all of
+Affirmer's Copyright and Related Rights and associated claims and causes
+of action, whether now known or unknown (including existing as well as
+future claims and causes of action), in the Work (i) in all territories
+worldwide, (ii) for the maximum duration provided by applicable law or
+treaty (including future time extensions), (iii) in any current or future
+medium and for any number of copies, and (iv) for any purpose whatsoever,
+including without limitation commercial, advertising or promotional
+purposes (the "Waiver"). Affirmer makes the Waiver for the benefit of each
+member of the public at large and to the detriment of Affirmer's heirs and
+successors, fully intending that such Waiver shall not be subject to
+revocation, rescission, cancellation, termination, or any other legal or
+equitable action to disrupt the quiet enjoyment of the Work by the public
+as contemplated by Affirmer's express Statement of Purpose.
+
+3. Public License Fallback. Should any part of the Waiver for any reason
+be judged legally invalid or ineffective under applicable law, then the
+Waiver shall be preserved to the maximum extent permitted taking into
+account Affirmer's express Statement of Purpose. In addition, to the
+extent the Waiver is so judged Affirmer hereby grants to each affected
+person a royalty-free, non transferable, non sublicensable, non exclusive,
+irrevocable and unconditional license to exercise Affirmer's Copyright and
+Related Rights in the Work (i) in all territories worldwide, (ii) for the
+maximum duration provided by applicable law or treaty (including future
+time extensions), (iii) in any current or future medium and for any number
+of copies, and (iv) for any purpose whatsoever, including without
+limitation commercial, advertising or promotional purposes (the
+"License"). The License shall be deemed effective as of the date CC0 was
+applied by Affirmer to the Work. Should any part of the License for any
+reason be judged legally invalid or ineffective under applicable law, such
+partial invalidity or ineffectiveness shall not invalidate the remainder
+of the License, and in such case Affirmer hereby affirms that he or she
+will not (i) exercise any of his or her remaining Copyright and Related
+Rights in the Work or (ii) assert any associated claims and causes of
+action with respect to the Work, in either case contrary to Affirmer's
+express Statement of Purpose.
+
+4. Limitations and Disclaimers.
+
+ a. No trademark or patent rights held by Affirmer are waived, abandoned,
+    surrendered, licensed or otherwise affected by this document.
+ b. Affirmer offers the Work as-is and makes no representations or
+    warranties of any kind concerning the Work, express, implied,
+    statutory or otherwise, including without limitation warranties of
+    title, merchantability, fitness for a particular purpose, non
+    infringement, or the absence of latent or other defects, accuracy, or
+    the present or absence of errors, whether or not discoverable, all to
+    the greatest extent permissible under applicable law.
+ c. Affirmer disclaims responsibility for clearing rights of other persons
+    that may apply to the Work or any use thereof, including without
+    limitation any person's Copyright and Related Rights in the Work.
+    Further, Affirmer disclaims responsibility for obtaining any necessary
+    consents, permissions or other rights required for any use of the
+    Work.
+ d. Affirmer understands and acknowledges that Creative Commons is not a
+    party to this document and has no duty or obligation with respect to
+    this CC0 or use of the Work.

latch_asgi-0.2.0.dist-info/METADATA

@@ -1,25 +0,0 @@
-Metadata-Version: 2.1
-Name: latch-asgi
-Version: 0.2.0
-Summary: ASGI python server
-License: CC0 1.0
-Author: Max Smolin
-Author-email: max@latch.bio
-Requires-Python: >=3.11,<4.0
-Classifier: License :: Other/Proprietary License
-Classifier: Programming Language :: Python :: 3
-Classifier: Programming Language :: Python :: 3.11
-Requires-Dist: PyJWT[crypto] (>=2.6.0,<3.0.0)
-Requires-Dist: hypercorn[uvloop] (>=0.14.3,<0.15.0)
-Requires-Dist: latch-config (>=0.1.6,<0.2.0)
-Requires-Dist: latch-data-validation (>=0.1.3,<0.2.0)
-Requires-Dist: latch-o11y (>=0.1.4,<0.2.0)
-Requires-Dist: opentelemetry-api (>=1.15.0,<2.0.0)
-Requires-Dist: opentelemetry-instrumentation-asgi (>=0.36b0,<0.37)
-Requires-Dist: opentelemetry-sdk (>=1.15.0,<2.0.0)
-Requires-Dist: orjson (>=3.8.5,<4.0.0)
-Requires-Dist: pysimdjson (>=5.0.2,<6.0.0)
-Description-Content-Type: text/markdown
-
-# python-asgi
-