anti
010568e558
Mirrors decnet/geoip/ end-to-end: paths/base/factory/lookup at the package level, iptoasn/ subpackage holds the data-source-specific fetch+parse+provider. AsnLookup is bisect-indexed over (start, end, AsnInfo) ranges with a pickled cache invalidated on raw-file mtime bump. Why iptoasn (and not bgp.tools / Team Cymru): public-domain dump, zero attribution, no UA mandate, daily refresh — keeps DECNET stealth intact (the geoip/rir module's "never identify as DECNET" comment applies the same way here). bgp.tools' ToS would have required an identifying UA, conflicting with feedback_stealth. Public surface: decnet.asn.enrich_ip(ip) -> (asn, name, source) or all-None on miss/disabled. Same shape as decnet.geoip.enrich_ip so the profiler can compose them in one call site.
40 lines
1.0 KiB
Python
40 lines
1.0 KiB
Python
"""ASN provider factory — mirror of :mod:`decnet.geoip.factory`.
|
|
|
|
Dispatch key: ``DECNET_ASN_PROVIDER`` (default ``iptoasn``). Lazy
|
|
singleton.
|
|
"""
|
|
from __future__ import annotations
|
|
|
|
import os
|
|
from typing import Optional
|
|
|
|
from decnet.asn.base import Provider
|
|
|
|
_cached: Optional[Provider] = None
|
|
_cached_key: Optional[str] = None
|
|
|
|
|
|
def get_provider() -> Provider:
|
|
"""Return the configured :class:`Provider` singleton."""
|
|
global _cached, _cached_key
|
|
key = os.environ.get("DECNET_ASN_PROVIDER", "iptoasn").lower()
|
|
if _cached is not None and _cached_key == key:
|
|
return _cached
|
|
|
|
if key == "iptoasn":
|
|
from decnet.asn.iptoasn.provider import IptoasnProvider
|
|
provider: Provider = IptoasnProvider()
|
|
else:
|
|
raise ValueError(f"Unsupported ASN provider: {key!r}")
|
|
|
|
_cached = provider
|
|
_cached_key = key
|
|
return provider
|
|
|
|
|
|
def reset_cache() -> None:
|
|
"""Forget the singleton — tests swap providers via the env var."""
|
|
global _cached, _cached_key
|
|
_cached = None
|
|
_cached_key = None
|