vercel-headers-bundle 0.7.1__py3-none-any.whl

vercel/headers/__init__.py

@@ -0,0 +1,144 @@
+from __future__ import annotations
+
+import urllib.parse
+from collections.abc import Callable, Iterator, Mapping
+from contextlib import contextmanager
+from contextvars import ContextVar
+from dataclasses import dataclass
+from typing import Any, ParamSpec, Protocol, TypedDict, TypeVar
+
+__all__ = [
+    "ip_address",
+    "geolocation",
+    "Geo",
+    "HeadersContext",
+    "set_headers",
+    "get_headers",
+    "headers_from_asgi_scope",
+    "headers_from_wsgi_environ",
+]
+
+
+_cv_headers: ContextVar[Mapping[str, str] | None] = ContextVar("vercel_headers", default=None)
+P = ParamSpec("P")
+R = TypeVar("R")
+
+# Header constants (same as TS names)
+CITY_HEADER_NAME = "x-vercel-ip-city"
+COUNTRY_HEADER_NAME = "x-vercel-ip-country"
+IP_HEADER_NAME = "x-real-ip"
+LATITUDE_HEADER_NAME = "x-vercel-ip-latitude"
+LONGITUDE_HEADER_NAME = "x-vercel-ip-longitude"
+REGION_HEADER_NAME = "x-vercel-ip-country-region"
+POSTAL_CODE_HEADER_NAME = "x-vercel-ip-postal-code"
+REQUEST_ID_HEADER_NAME = "x-vercel-id"
+
+EMOJI_FLAG_UNICODE_STARTING_POSITION = 127397
+
+
+def set_headers(headers: Mapping[str, str] | None) -> None:
+    _cv_headers.set(headers)
+
+
+def get_headers() -> Mapping[str, str] | None:
+    return _cv_headers.get()
+
+
+@dataclass(frozen=True)
+class HeadersContext:
+    """Immutable snapshot of the current Vercel request headers."""
+
+    headers: Mapping[str, str] | None
+
+    @contextmanager
+    def use(self) -> Iterator[None]:
+        token = _cv_headers.set(self.headers)
+        try:
+            yield
+        finally:
+            _cv_headers.reset(token)
+
+    def run(self, func: Callable[P, R], *args: P.args, **kwargs: P.kwargs) -> R:
+        with self.use():
+            return func(*args, **kwargs)
+
+
+class _HeadersLike(Protocol):
+    def get(self, name: str) -> str | None: ...
+
+
+class _RequestLike(Protocol):
+    headers: _HeadersLike
+
+
+class Geo(TypedDict, total=False):
+    city: str | None
+    country: str | None
+    flag: str | None
+    region: str | None
+    countryRegion: str | None
+    latitude: str | None
+    longitude: str | None
+    postalCode: str | None
+
+
+def _get_header(headers: _HeadersLike, key: str) -> str | None:
+    return headers.get(key)
+
+
+def _get_header_decode(req: _RequestLike, key: str) -> str | None:
+    raw = _get_header(req.headers, key)
+    return urllib.parse.unquote(raw) if raw is not None else None
+
+
+def _get_flag(country_code: str | None) -> str | None:
+    if not country_code or len(country_code) != 2 or not country_code.isalpha():
+        return None
+    return "".join(chr(EMOJI_FLAG_UNICODE_STARTING_POSITION + ord(c)) for c in country_code.upper())
+
+
+def headers_from_asgi_scope(scope: Mapping[str, Any]) -> dict[str, str]:
+    """Return request headers decoded from an ASGI scope."""
+    return {
+        name.decode("latin-1"): value.decode("latin-1") for name, value in scope.get("headers", [])
+    }
+
+
+def headers_from_wsgi_environ(environ: Mapping[str, Any]) -> dict[str, str]:
+    """Return request headers decoded from a WSGI environ mapping."""
+    headers: dict[str, str] = {}
+    if "CONTENT_TYPE" in environ:
+        headers["Content-Type"] = str(environ["CONTENT_TYPE"])
+    if "CONTENT_LENGTH" in environ:
+        headers["Content-Length"] = str(environ["CONTENT_LENGTH"])
+    for name, value in environ.items():
+        if not name.startswith("HTTP_"):
+            continue
+        header_name = name[5:].replace("_", "-").title()
+        headers[header_name] = str(value)
+    return headers
+
+
+def ip_address(input: _RequestLike | _HeadersLike) -> str | None:
+    headers = input.headers if hasattr(input, "headers") else input
+    return _get_header(headers, IP_HEADER_NAME)
+
+
+def _region_from_request_id(request_id: str | None) -> str | None:
+    if request_id is None:
+        return "dev1"
+    return request_id.split(":")[0]
+
+
+def geolocation(request: _RequestLike) -> Geo:
+    headers = request.headers
+    return {
+        "city": _get_header_decode(request, CITY_HEADER_NAME),
+        "country": _get_header(headers, COUNTRY_HEADER_NAME),
+        "flag": _get_flag(_get_header(headers, COUNTRY_HEADER_NAME)),
+        "countryRegion": _get_header(headers, REGION_HEADER_NAME),
+        "region": _region_from_request_id(_get_header(headers, REQUEST_ID_HEADER_NAME)),
+        "latitude": _get_header(headers, LATITUDE_HEADER_NAME),
+        "longitude": _get_header(headers, LONGITUDE_HEADER_NAME),
+        "postalCode": _get_header(headers, POSTAL_CODE_HEADER_NAME),
+    }

vercel/headers/_vendor/__init__.py

@@ -0,0 +1 @@
+"""Generated vendored dependencies for vercel-headers-bundle."""

vercel/headers/version.py

@@ -0,0 +1 @@
+__version__ = "0.7.1"

vercel_headers_bundle-0.7.1.dist-info/METADATA

@@ -0,0 +1,27 @@
+Metadata-Version: 2.4
+Name: vercel-headers-bundle
+Version: 0.7.1
+Summary: Request header helpers for Vercel Python applications
+License-Expression: MIT
+License-File: LICENSE
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+
+# vercel-headers-bundle
+
+This is a version of `vercel-headers` with third-party dependencies bundled. For normal use, install the unbundled `vercel-headers` package instead: https://pypi.org/project/vercel-headers/
+
+# Headers
+
+`vercel.headers` stores request headers for Vercel Function helpers and exposes
+IP address and geolocation helpers.
+
+Register request headers once per request:
+
+```python
+from vercel.headers import set_headers
+
+set_headers(request.headers)
+```
+
+OIDC and cache helpers read the same registered header context.

vercel_headers_bundle-0.7.1.dist-info/RECORD

@@ -0,0 +1,8 @@
+vercel/headers/__init__.py,sha256=blQhLacX7FqefHOSMDiHGfjDdeW437jNzL0qv3-k6zQ,4439
+vercel/headers/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+vercel/headers/version.py,sha256=2KJZDSMOG7KS82AxYOrZ4ZihYxX0wjfUjDsIZh3L024,22
+vercel/headers/_vendor/__init__.py,sha256=eNgwm8aDiNnSnNnz-wKVqoqnn80GlSsHZMETGb4EKsk,65
+vercel_headers_bundle-0.7.1.dist-info/METADATA,sha256=g2aDNGVNXYYhZ09srrTA3CpoPU3Gi60Y4gkYd10GKso,775
+vercel_headers_bundle-0.7.1.dist-info/WHEEL,sha256=lCkmxWfQsSc9CfIClYeavTdQeEX2toPqufh9gI35EQA,87
+vercel_headers_bundle-0.7.1.dist-info/licenses/LICENSE,sha256=ZhFC5TwxPSu14bBV9cCjkAFFD_G14nuJ3EvH3ppjUso,1069
+vercel_headers_bundle-0.7.1.dist-info/RECORD,,

vercel_headers_bundle-0.7.1.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.31.0
+Root-Is-Purelib: true
+Tag: py3-none-any

vercel_headers_bundle-0.7.1.dist-info/licenses/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2026 Vercel, Inc.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.