tigrbl-base 0.4.4.dev1__tar.gz → 0.4.4.dev8__tar.gz

{tigrbl_base-0.4.4.dev1 → tigrbl_base-0.4.4.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tigrbl-base
-Version: 0.4.4.dev1
+Version: 0.4.4.dev8
 Summary: Base contract package for Tigrbl apps, routers, tables, sessions, middleware, requests, responses, bindings, and engine interfaces.
 License: Apache License
                                     Version 2.0, January 2004

{tigrbl_base-0.4.4.dev1 → tigrbl_base-0.4.4.dev8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "tigrbl-base"
-version = "0.4.4.dev1"
+version = "0.4.4.dev8"
 description = "Base contract package for Tigrbl apps, routers, tables, sessions, middleware, requests, responses, bindings, and engine interfaces."
 readme = "README.md"
 license = { file = "LICENSE" }

{tigrbl_base-0.4.4.dev1 → tigrbl_base-0.4.4.dev8}/tigrbl_base/_base/__init__.py

@@ -13,6 +13,8 @@ _EXPORTS = {
     "AliasBase": "_alias_base",
     "EngineBase": "_engine_base",
     "EngineProviderBase": "_engine_provider_base",
+    "HeaderCookiesBase": "_headers_base",
+    "HeadersBase": "_headers_base",
     "HookBase": "_hook_base",
     "RouterBase": "_router_base",
     "ForeignKeyBase": "_storage",

tigrbl_base-0.4.4.dev8/tigrbl_base/_base/_headers_base.py

@@ -0,0 +1,85 @@
+from __future__ import annotations
+
+from collections.abc import Iterable, Mapping
+from typing import Any
+
+from tigrbl_core._spec.headers_spec import HeadersSpec
+
+
+HeaderInput = Iterable[tuple[str, str]] | Mapping[str, str] | HeadersSpec | None
+
+
+def _normalize_header_name(name: Any) -> str:
+    if isinstance(name, (bytes, bytearray)):
+        return name.decode("latin-1").lower()
+    return str(name).lower()
+
+
+def _normalize_header_value(value: Any) -> str:
+    if isinstance(value, (bytes, bytearray)):
+        return value.decode("latin-1")
+    return str(value)
+
+
+def _iter_header_items(values: HeaderInput):
+    if values is None:
+        return ()
+    if isinstance(values, HeadersSpec):
+        return values.values.items()
+    if hasattr(values, "items"):
+        return values.items()  # type: ignore[union-attr]
+    return values
+
+
+class HeadersBase(dict[str, str]):
+    """Base header mapping type for response implementations."""
+
+    required: tuple[str, ...]
+    expose: tuple[str, ...]
+
+    def __init__(
+        self,
+        values: HeaderInput = None,
+        *,
+        required: tuple[str, ...] = (),
+        expose: tuple[str, ...] = (),
+    ) -> None:
+        if isinstance(values, HeadersSpec):
+            required = values.required
+            expose = values.expose
+        super().__init__(
+            {
+                _normalize_header_name(key): _normalize_header_value(value)
+                for key, value in _iter_header_items(values)
+            }
+        )
+        self.required = tuple(_normalize_header_name(name) for name in required)
+        self.expose = tuple(_normalize_header_name(name) for name in expose)
+
+    def __getitem__(self, key: str) -> str:
+        return super().__getitem__(_normalize_header_name(key))
+
+    def __contains__(self, key: object) -> bool:
+        if isinstance(key, (str, bytes, bytearray)):
+            return super().__contains__(_normalize_header_name(key))
+        return super().__contains__(key)
+
+    def get(self, key: str, default: Any = None) -> str | Any:
+        return super().get(_normalize_header_name(key), default)
+
+    @property
+    def raw_headers(self) -> list[tuple[bytes, bytes]]:
+        return [
+            (key.encode("latin-1"), value.encode("latin-1"))
+            for key, value in self.items()
+        ]
+
+    def as_spec(self) -> HeadersSpec:
+        return HeadersSpec(values=dict(self), required=self.required, expose=self.expose)
+
+
+class HeaderCookiesBase(dict[str, str]):
+    """Base cookie mapping type for response implementations."""
+
+
+__all__ = ["HeaderCookiesBase", "HeadersBase"]

{tigrbl_base-0.4.4.dev1 → tigrbl_base-0.4.4.dev8}/tigrbl_base/_base/_response_base.py

@@ -38,18 +38,13 @@ class ResponseBase(ResponseSpec):
         payload = (
             body if body is not None else (content if content is not None else b"")
         )
-        normalized_headers = {
-            str(key).lower(): str(value)
-            for key, value in (
-                headers.items() if hasattr(headers, "items") else (headers or [])
-            )
-        }
+        headers_base = HeadersBase(headers)
 
         super().__init__(
             kind=kind,
             media_type=media_type,
             status_code=status_code,
-            headers=dict(normalized_headers),
+            headers=dict(headers_base),
             envelope=envelope,
             template=template,
             filename=filename,
@@ -59,15 +54,13 @@ class ResponseBase(ResponseSpec):
             redirect_to=redirect_to,
         )
         self.status_code = status_code
-        self.headers: HeadersBase = HeadersBase(normalized_headers)
+        self.headers: HeadersBase = headers_base
         self.body = payload
         self.media_type = media_type
 
     @property
     def raw_headers(self) -> list[tuple[bytes, bytes]]:
-        return [
-            (k.encode("latin-1"), v.encode("latin-1")) for k, v in self.headers.items()
-        ]
+        return self.headers.raw_headers
 
     @property
     def headers_map(self) -> HeadersBase:

tigrbl_base-0.4.4.dev1/tigrbl_base/_base/_headers_base.py

@@ -1,12 +0,0 @@
-from __future__ import annotations
-
-
-class HeadersBase(dict[str, str]):
-    """Base header mapping type for response implementations."""
-
-
-class HeaderCookiesBase(dict[str, str]):
-    """Base cookie mapping type for response implementations."""
-
-
-__all__ = ["HeaderCookiesBase", "HeadersBase"]