asndb 1.0.0__tar.gz

asndb-1.0.0/PKG-INFO

@@ -0,0 +1,182 @@
+Metadata-Version: 2.4
+Name: asndb
+Version: 1.0.0
+Summary: A simple Python CLI + library for looking up ASNs by IP or AS number. Uses BBOT.IO API.
+Author: TheTechromancer
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: cachetools>=6.2.1
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: radixtarget>=4.0.1
+Requires-Dist: typer>=0.19.2
+
+# ASNDB
+
+[![Python Version](https://img.shields.io/badge/python-3.9+-blue)](https://www.python.org) ![PyPI - Version](https://img.shields.io/pypi/v/asndb) [![License](https://img.shields.io/badge/license-GPLv3-blue.svg)](https://github.com/blacklanternsecurity/radixtarget/blob/master/LICENSE) [![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff) [![Tests](https://github.com/blacklanternsecurity/asndb/actions/workflows/tests.yml/badge.svg)](https://github.com/blacklanternsecurity/asndb/actions/workflows/tests.yml)
+
+A simple Python CLI + library for instant lookup of ASN data by IP address, AS number, or organization. Uses BBOT.IO API (`asndb.api.bbot.io`).
+
+ASNs are automatically cached for instant lookups and minimal network traffic.
+
+## Installation
+
+```
+pip install asndb
+
+# or using uv
+uv add asndb
+```
+
+## Usage (CLI)
+
+Note: To avoid rate limits, export your BBOT.IO API key as an environment variable:
+```bash
+export BBOT_IO_API_KEY=<your_api_key>
+```
+
+### IP Lookup
+```bash
+asndb ip 1.1.1.1
+```
+
+Output:
+```json
+{
+  "asn": 13335,
+  "asn_name": "CLOUDFLARENET",
+  "country": "US",
+  "ip": "1.1.1.1",
+  "org": "Cloudflare, Inc.",
+  "org_id": "CLOUD14-ARIN",
+  "rir": "ARIN",
+  "subnets": [
+    "1.0.0.0/24",
+    "1.0.1.0/24"
+  ]
+}
+```
+
+### AS Number Lookup
+
+```bash
+asndb asn 13335
+```
+
+Output:
+```json
+{
+  "asn": 13335,
+  "asn_name": "CLOUDFLARENET",
+  "country": "US",
+  "ip": "1.1.1.1",
+  "org": "Cloudflare, Inc.",
+  "org_id": "CLOUD14-ARIN",
+  "rir": "ARIN",
+  "subnets": [
+    "1.0.0.0/24",
+    "1.0.1.0/24"
+  ]
+}
+```
+
+### Organization Lookup
+
+```bash
+# Look up an organization
+asndb org CLOUD14-ARIN
+```
+
+Output:
+```json
+{
+  "asns": [
+    13335,
+    14789,
+    395747,
+    394536
+  ]
+}
+```
+
+### CLI Help
+
+```
+$ uv run asndb --help
+                                                                                              
+ Usage: asndb [OPTIONS] COMMAND [ARGS]...                                                     
+                                                                                              
+╭─ Options ──────────────────────────────────────────────────────────────────────────────────╮
+│ --install-completion          Install completion for the current shell.                    │
+│ --show-completion             Show completion for the current shell, to copy it or         │
+│                               customize the installation.                                  │
+│ --help                        Show this message and exit.                                  │
+╰────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Commands ─────────────────────────────────────────────────────────────────────────────────╮
+│ ip    Lookup ASN by IP address                                                             │
+│ asn   Lookup ASN by AS number                                                              │
+│ org   Get all the ASNs for an organization, by its registered organization ID, e.g.        │
+│       GOGL-ARIN                                                                            │
+╰────────────────────────────────────────────────────────────────────────────────────────────╯
+```
+
+## Usage (Python)
+
+```python
+from asndb import ASNDB
+
+# Create a new ASNDB client
+asndb = ASNDB()
+
+
+# Look up an IP address
+asn = asndb.lookup_ip_sync("1.1.1.1")
+# {
+#   "asn": 13335,
+#   "asn_name": "CLOUDFLARENET",
+#   "country": "US",
+#   "ip": "1.1.1.1",
+#   "org": "Cloudflare, Inc.",
+#   "org_id": "CLOUD14-ARIN",
+#   "rir": "ARIN",
+#   "subnets": [
+#     "1.0.0.0/24",
+#     "1.0.1.0/24"
+#   ]
+# }
+
+# Look up an AS number
+asn = asndb.lookup_asn_sync(13335)
+# {
+#   "asn": 13335,
+#   "asn_name": "CLOUDFLARENET",
+#   "country": "US",
+#   "ip": "1.1.1.1",
+#   "org": "Cloudflare, Inc.",
+#   "org_id": "CLOUD14-ARIN",
+#   "rir": "ARIN",
+#   "subnets": [
+#     "1.0.0.0/24",
+#     "1.0.1.0/24"
+#   ]
+# }
+
+# Look up an organization
+org = asndb.lookup_org_sync("CLOUD14-ARIN")
+# {
+#   "asns": [
+#     13335,
+#     14789,
+#     395747,
+#     394536
+#   ]
+# }
+```
+
+## Environment Variables
+
+You can customize the behavior of the ASNDB client by exporting the following environment variables:
+
+- `BBOT_IO_API_KEY`: Your BBOT.IO API key.
+- `ASNDB_BASE_URL`: The base URL of the ASNDB API (default: https://asndb.api.bbot.io/v1).
+- `ASNDB_TIMEOUT`: The timeout for the ASNDB API requests (default: 60 seconds).
+- `ASNDB_CACHE_SIZE`: The size of the cache for the ASNDB API requests (default: 10000).

asndb-1.0.0/README.md

@@ -0,0 +1,170 @@
+# ASNDB
+
+[![Python Version](https://img.shields.io/badge/python-3.9+-blue)](https://www.python.org) ![PyPI - Version](https://img.shields.io/pypi/v/asndb) [![License](https://img.shields.io/badge/license-GPLv3-blue.svg)](https://github.com/blacklanternsecurity/radixtarget/blob/master/LICENSE) [![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff) [![Tests](https://github.com/blacklanternsecurity/asndb/actions/workflows/tests.yml/badge.svg)](https://github.com/blacklanternsecurity/asndb/actions/workflows/tests.yml)
+
+A simple Python CLI + library for instant lookup of ASN data by IP address, AS number, or organization. Uses BBOT.IO API (`asndb.api.bbot.io`).
+
+ASNs are automatically cached for instant lookups and minimal network traffic.
+
+## Installation
+
+```
+pip install asndb
+
+# or using uv
+uv add asndb
+```
+
+## Usage (CLI)
+
+Note: To avoid rate limits, export your BBOT.IO API key as an environment variable:
+```bash
+export BBOT_IO_API_KEY=<your_api_key>
+```
+
+### IP Lookup
+```bash
+asndb ip 1.1.1.1
+```
+
+Output:
+```json
+{
+  "asn": 13335,
+  "asn_name": "CLOUDFLARENET",
+  "country": "US",
+  "ip": "1.1.1.1",
+  "org": "Cloudflare, Inc.",
+  "org_id": "CLOUD14-ARIN",
+  "rir": "ARIN",
+  "subnets": [
+    "1.0.0.0/24",
+    "1.0.1.0/24"
+  ]
+}
+```
+
+### AS Number Lookup
+
+```bash
+asndb asn 13335
+```
+
+Output:
+```json
+{
+  "asn": 13335,
+  "asn_name": "CLOUDFLARENET",
+  "country": "US",
+  "ip": "1.1.1.1",
+  "org": "Cloudflare, Inc.",
+  "org_id": "CLOUD14-ARIN",
+  "rir": "ARIN",
+  "subnets": [
+    "1.0.0.0/24",
+    "1.0.1.0/24"
+  ]
+}
+```
+
+### Organization Lookup
+
+```bash
+# Look up an organization
+asndb org CLOUD14-ARIN
+```
+
+Output:
+```json
+{
+  "asns": [
+    13335,
+    14789,
+    395747,
+    394536
+  ]
+}
+```
+
+### CLI Help
+
+```
+$ uv run asndb --help
+                                                                                              
+ Usage: asndb [OPTIONS] COMMAND [ARGS]...                                                     
+                                                                                              
+╭─ Options ──────────────────────────────────────────────────────────────────────────────────╮
+│ --install-completion          Install completion for the current shell.                    │
+│ --show-completion             Show completion for the current shell, to copy it or         │
+│                               customize the installation.                                  │
+│ --help                        Show this message and exit.                                  │
+╰────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Commands ─────────────────────────────────────────────────────────────────────────────────╮
+│ ip    Lookup ASN by IP address                                                             │
+│ asn   Lookup ASN by AS number                                                              │
+│ org   Get all the ASNs for an organization, by its registered organization ID, e.g.        │
+│       GOGL-ARIN                                                                            │
+╰────────────────────────────────────────────────────────────────────────────────────────────╯
+```
+
+## Usage (Python)
+
+```python
+from asndb import ASNDB
+
+# Create a new ASNDB client
+asndb = ASNDB()
+
+
+# Look up an IP address
+asn = asndb.lookup_ip_sync("1.1.1.1")
+# {
+#   "asn": 13335,
+#   "asn_name": "CLOUDFLARENET",
+#   "country": "US",
+#   "ip": "1.1.1.1",
+#   "org": "Cloudflare, Inc.",
+#   "org_id": "CLOUD14-ARIN",
+#   "rir": "ARIN",
+#   "subnets": [
+#     "1.0.0.0/24",
+#     "1.0.1.0/24"
+#   ]
+# }
+
+# Look up an AS number
+asn = asndb.lookup_asn_sync(13335)
+# {
+#   "asn": 13335,
+#   "asn_name": "CLOUDFLARENET",
+#   "country": "US",
+#   "ip": "1.1.1.1",
+#   "org": "Cloudflare, Inc.",
+#   "org_id": "CLOUD14-ARIN",
+#   "rir": "ARIN",
+#   "subnets": [
+#     "1.0.0.0/24",
+#     "1.0.1.0/24"
+#   ]
+# }
+
+# Look up an organization
+org = asndb.lookup_org_sync("CLOUD14-ARIN")
+# {
+#   "asns": [
+#     13335,
+#     14789,
+#     395747,
+#     394536
+#   ]
+# }
+```
+
+## Environment Variables
+
+You can customize the behavior of the ASNDB client by exporting the following environment variables:
+
+- `BBOT_IO_API_KEY`: Your BBOT.IO API key.
+- `ASNDB_BASE_URL`: The base URL of the ASNDB API (default: https://asndb.api.bbot.io/v1).
+- `ASNDB_TIMEOUT`: The timeout for the ASNDB API requests (default: 60 seconds).
+- `ASNDB_CACHE_SIZE`: The size of the cache for the ASNDB API requests (default: 10000).

asndb-1.0.0/asndb/__init__.py

@@ -0,0 +1 @@
+from asndb.asndb import ASNDB

asndb-1.0.0/asndb/asndb.py

@@ -0,0 +1,186 @@
+import os
+import time
+import asyncio
+from contextlib import contextmanager, suppress
+
+import httpx
+import cachetools
+from radixtarget import RadixTarget
+
+
+class ASNDBError(Exception):
+    pass
+
+
+class ASNDBTimeoutError(ASNDBError):
+    pass
+
+
+class ASNDBClient:
+    BASE_URL = "https://asndb.api.bbot.io/v1"
+    DEFAULT_CACHE_SIZE = 10000
+    DEFAULT_TIMEOUT = 60
+
+    # Default record used when no ASN data can be found
+    UNKNOWN_ASN = {
+        "asn": 0,
+        "subnets": [],
+        "name": "Unknown",
+        "description": "Unknown ASN",
+        "country": "Unknown",
+    }
+
+    def __init__(self, bbot_io_api_key=None):
+        self.bbot_io_api_key = os.getenv("BBOT_IO_API_KEY", None) or bbot_io_api_key
+        self.base_url = os.getenv("ASNDB_BASE_URL", None) or self.BASE_URL
+        self.timeout = int(os.getenv("ASNDB_TIMEOUT", self.DEFAULT_TIMEOUT))
+        self.client = httpx.AsyncClient(timeout=self.timeout)
+
+        self.headers = {}
+        if self.bbot_io_api_key:
+            self.headers["Authorization"] = f"Bearer {self.bbot_io_api_key}"
+
+        self._cache_size = os.getenv("ASNDB_CACHE_SIZE", self.DEFAULT_CACHE_SIZE)
+        # IPNetwork -> ASN Number
+        self._subnet_cache = RadixTarget()
+        # ASN Number -> ASN
+        self._asn_cache = cachetools.LRUCache(maxsize=self._cache_size)
+
+        self._event_loop = asyncio.get_event_loop()
+
+    async def request(self, url, **kwargs):
+        """
+        Make a request to the ASNDB API, respecting its retry-after mechanism.
+        """
+        start = time.time()
+        retry_after = 10
+        try:
+            while True:
+                elapsed = time.time() - start
+                if elapsed > self.timeout:
+                    raise ASNDBTimeoutError(f"Timeout after {self.timeout} seconds")
+
+                try:
+                    response = await self.client.get(url, headers=self.headers, **kwargs)
+                except httpx.TimeoutException:
+                    continue
+                except httpx.HTTPError as e:
+                    raise ASNDBError(f"HTTP error: {e}")
+
+                status_code = getattr(response, "status_code", 0)
+
+                if status_code == 200:
+                    try:
+                        return response.json()
+                    except Exception as e:
+                        raise ASNDBError(f"Error parsing JSON: {e}")
+                elif status_code == 404:
+                    return None
+                elif status_code == 429:
+                    retry_after = int(response.headers.get("Retry-After", 10))
+                    await asyncio.sleep(retry_after)
+                    continue
+                else:
+                    raise ASNDBError(f"Unexpected status code: {response.status_code}: {response.text}")
+
+        except Exception as e:
+            raise ASNDBError(f"Unexpected error while requesting {url}: {e}")
+
+    async def lookup_ip(self, ip: str):
+        """
+        Given an IP address, return the ASN data and subnet.
+
+        For convenience, will put the parent subnet of the requested IP in the "subnet" key of the returned ASN data.
+        """
+        try:
+            asn_number, subnet = self._cache_get_ip(ip)
+            asn = dict(self._cache_get_asn(asn_number))
+            asn["subnet"] = subnet
+            return asn
+        except KeyError:
+            url = f"{self.base_url}/ip/{ip}"
+            asn = await self.request(url)
+            if asn:
+                self._cache_put_asn(asn)
+                asn = dict(asn)
+                with suppress(KeyError):
+                    asn_number, subnet = self._cache_get_ip(ip)
+                    asn["subnet"] = subnet
+                return asn
+        return self.UNKNOWN_ASN
+
+    def lookup_ip_sync(self, ip: str):
+        return self._event_loop.run_until_complete(self.lookup_ip(ip))
+
+    async def ip_to_subnet(self, ip: str):
+        """
+        Given an IP address, return the AS number and subnet it belongs to.
+        """
+        # first make sure we have the ASN data for the IP
+        await self.lookup_ip(ip)
+        # then get it from the cache
+        result = self._cache_get_ip(ip)
+        if result is None:
+            raise KeyError(f"IP address {ip} not found in cache")
+        asn_number, subnet = result
+        return asn_number, subnet
+
+    def ip_to_subnet_sync(self, ip: str):
+        return self._event_loop.run_until_complete(self.ip_to_subnet(ip))
+
+    async def lookup_asn(self, asn: str):
+        try:
+            return self._cache_get_asn(asn)
+        except KeyError:
+            url = f"{self.base_url}/asn/{asn}"
+            asn = await self.request(url)
+            if asn:
+                self._cache_put_asn(asn)
+                return asn
+        return self.UNKNOWN_ASN
+
+    def lookup_asn_sync(self, asn: str):
+        return self._event_loop.run_until_complete(self.lookup_asn(asn))
+
+    async def lookup_org(self, org: str):
+        url = f"{self.base_url}/org/{org}"
+        return (await self.request(url)) or []
+
+    def lookup_org_sync(self, org: str):
+        return self._event_loop.run_until_complete(self.lookup_org(org))
+
+    def _cache_get_ip(self, ip: str) -> tuple[int, str]:
+        """
+        ip -> asn, subnet
+        """
+        result = self._subnet_cache.get(ip)
+        if result is None:
+            raise KeyError(f"IP address {ip} not found in cache")
+        asn, subnet = result
+        return asn, subnet
+
+    def _cache_get_asn(self, asn: int) -> dict:
+        """
+        asn number -> asn data
+        """
+        return self._asn_cache[asn]
+
+    def _cache_put_asn(self, asn: str):
+        asn_number = int(asn["asn"])
+        for subnet in asn.get("subnets", []):
+            self._subnet_cache.add(subnet, data=(asn_number, subnet))
+        self._asn_cache[asn_number] = asn
+
+    async def cleanup(self):
+        self._asn_cache.clear()
+        await self.client.aclose()
+
+
+asndb_client = None
+
+
+def ASNDB():
+    global asndb_client
+    if asndb_client is None:
+        asndb_client = ASNDBClient()
+    return asndb_client

asndb-1.0.0/asndb/main.py

@@ -0,0 +1,57 @@
+import re
+import json
+import typer
+from contextlib import contextmanager
+
+from asndb.asndb import ASNDB
+
+
+app = typer.Typer()
+
+asn_regex = re.compile(r"^(?:AS)?(\d+)$", re.IGNORECASE)
+
+
+@contextmanager
+def asndb_cli_context():
+    client = ASNDB()
+    try:
+        yield client
+    except Exception as e:
+        typer.echo(f"Error: {e}", err=True)
+        raise typer.Exit(1)
+
+
+@app.command(help="Lookup ASN by IP address")
+def ip(
+    ip_or_asn: str = typer.Argument(..., help="IP address to lookup"),
+):
+    with asndb_cli_context() as client:
+        asn = client.lookup_ip_sync(ip_or_asn)
+        print(json.dumps(asn, indent=2))
+
+
+@app.command(help="Lookup ASN by AS number")
+def asn(
+    asn: str = typer.Argument(..., help="AS number to lookup"),
+):
+    as_number = int(asn_regex.match(asn).group(1))
+    with asndb_cli_context() as client:
+        asn = client.lookup_asn_sync(as_number)
+        print(json.dumps(asn, indent=2))
+
+
+@app.command(help="Get all the ASNs for an organization, by its registered organization ID, e.g. GOGL-ARIN")
+def org(
+    org: str = typer.Argument(..., help="Organization to lookup, e.g. GOGL-ARIN or CLOUD14-ARIN"),
+):
+    with asndb_cli_context() as client:
+        org = client.lookup_org_sync(org)
+        print(json.dumps(org, indent=2))
+
+
+def main():
+    app()
+
+
+if __name__ == "__main__":
+    main()

asndb-1.0.0/asndb.egg-info/PKG-INFO

@@ -0,0 +1,182 @@
+Metadata-Version: 2.4
+Name: asndb
+Version: 1.0.0
+Summary: A simple Python CLI + library for looking up ASNs by IP or AS number. Uses BBOT.IO API.
+Author: TheTechromancer
+Requires-Python: >=3.9
+Description-Content-Type: text/markdown
+Requires-Dist: cachetools>=6.2.1
+Requires-Dist: httpx>=0.28.1
+Requires-Dist: radixtarget>=4.0.1
+Requires-Dist: typer>=0.19.2
+
+# ASNDB
+
+[![Python Version](https://img.shields.io/badge/python-3.9+-blue)](https://www.python.org) ![PyPI - Version](https://img.shields.io/pypi/v/asndb) [![License](https://img.shields.io/badge/license-GPLv3-blue.svg)](https://github.com/blacklanternsecurity/radixtarget/blob/master/LICENSE) [![Ruff](https://img.shields.io/endpoint?url=https://raw.githubusercontent.com/astral-sh/ruff/main/assets/badge/v2.json)](https://github.com/astral-sh/ruff) [![Tests](https://github.com/blacklanternsecurity/asndb/actions/workflows/tests.yml/badge.svg)](https://github.com/blacklanternsecurity/asndb/actions/workflows/tests.yml)
+
+A simple Python CLI + library for instant lookup of ASN data by IP address, AS number, or organization. Uses BBOT.IO API (`asndb.api.bbot.io`).
+
+ASNs are automatically cached for instant lookups and minimal network traffic.
+
+## Installation
+
+```
+pip install asndb
+
+# or using uv
+uv add asndb
+```
+
+## Usage (CLI)
+
+Note: To avoid rate limits, export your BBOT.IO API key as an environment variable:
+```bash
+export BBOT_IO_API_KEY=<your_api_key>
+```
+
+### IP Lookup
+```bash
+asndb ip 1.1.1.1
+```
+
+Output:
+```json
+{
+  "asn": 13335,
+  "asn_name": "CLOUDFLARENET",
+  "country": "US",
+  "ip": "1.1.1.1",
+  "org": "Cloudflare, Inc.",
+  "org_id": "CLOUD14-ARIN",
+  "rir": "ARIN",
+  "subnets": [
+    "1.0.0.0/24",
+    "1.0.1.0/24"
+  ]
+}
+```
+
+### AS Number Lookup
+
+```bash
+asndb asn 13335
+```
+
+Output:
+```json
+{
+  "asn": 13335,
+  "asn_name": "CLOUDFLARENET",
+  "country": "US",
+  "ip": "1.1.1.1",
+  "org": "Cloudflare, Inc.",
+  "org_id": "CLOUD14-ARIN",
+  "rir": "ARIN",
+  "subnets": [
+    "1.0.0.0/24",
+    "1.0.1.0/24"
+  ]
+}
+```
+
+### Organization Lookup
+
+```bash
+# Look up an organization
+asndb org CLOUD14-ARIN
+```
+
+Output:
+```json
+{
+  "asns": [
+    13335,
+    14789,
+    395747,
+    394536
+  ]
+}
+```
+
+### CLI Help
+
+```
+$ uv run asndb --help
+                                                                                              
+ Usage: asndb [OPTIONS] COMMAND [ARGS]...                                                     
+                                                                                              
+╭─ Options ──────────────────────────────────────────────────────────────────────────────────╮
+│ --install-completion          Install completion for the current shell.                    │
+│ --show-completion             Show completion for the current shell, to copy it or         │
+│                               customize the installation.                                  │
+│ --help                        Show this message and exit.                                  │
+╰────────────────────────────────────────────────────────────────────────────────────────────╯
+╭─ Commands ─────────────────────────────────────────────────────────────────────────────────╮
+│ ip    Lookup ASN by IP address                                                             │
+│ asn   Lookup ASN by AS number                                                              │
+│ org   Get all the ASNs for an organization, by its registered organization ID, e.g.        │
+│       GOGL-ARIN                                                                            │
+╰────────────────────────────────────────────────────────────────────────────────────────────╯
+```
+
+## Usage (Python)
+
+```python
+from asndb import ASNDB
+
+# Create a new ASNDB client
+asndb = ASNDB()
+
+
+# Look up an IP address
+asn = asndb.lookup_ip_sync("1.1.1.1")
+# {
+#   "asn": 13335,
+#   "asn_name": "CLOUDFLARENET",
+#   "country": "US",
+#   "ip": "1.1.1.1",
+#   "org": "Cloudflare, Inc.",
+#   "org_id": "CLOUD14-ARIN",
+#   "rir": "ARIN",
+#   "subnets": [
+#     "1.0.0.0/24",
+#     "1.0.1.0/24"
+#   ]
+# }
+
+# Look up an AS number
+asn = asndb.lookup_asn_sync(13335)
+# {
+#   "asn": 13335,
+#   "asn_name": "CLOUDFLARENET",
+#   "country": "US",
+#   "ip": "1.1.1.1",
+#   "org": "Cloudflare, Inc.",
+#   "org_id": "CLOUD14-ARIN",
+#   "rir": "ARIN",
+#   "subnets": [
+#     "1.0.0.0/24",
+#     "1.0.1.0/24"
+#   ]
+# }
+
+# Look up an organization
+org = asndb.lookup_org_sync("CLOUD14-ARIN")
+# {
+#   "asns": [
+#     13335,
+#     14789,
+#     395747,
+#     394536
+#   ]
+# }
+```
+
+## Environment Variables
+
+You can customize the behavior of the ASNDB client by exporting the following environment variables:
+
+- `BBOT_IO_API_KEY`: Your BBOT.IO API key.
+- `ASNDB_BASE_URL`: The base URL of the ASNDB API (default: https://asndb.api.bbot.io/v1).
+- `ASNDB_TIMEOUT`: The timeout for the ASNDB API requests (default: 60 seconds).
+- `ASNDB_CACHE_SIZE`: The size of the cache for the ASNDB API requests (default: 10000).

asndb-1.0.0/asndb.egg-info/SOURCES.txt

@@ -0,0 +1,12 @@
+README.md
+pyproject.toml
+asndb/__init__.py
+asndb/asndb.py
+asndb/main.py
+asndb.egg-info/PKG-INFO
+asndb.egg-info/SOURCES.txt
+asndb.egg-info/dependency_links.txt
+asndb.egg-info/entry_points.txt
+asndb.egg-info/requires.txt
+asndb.egg-info/top_level.txt
+tests/test_asndb.py

asndb-1.0.0/asndb.egg-info/dependency_links.txt

@@ -0,0 +1 @@
+

asndb-1.0.0/asndb.egg-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+asndb = asndb.main:main

asndb-1.0.0/asndb.egg-info/requires.txt

@@ -0,0 +1,4 @@
+cachetools>=6.2.1
+httpx>=0.28.1
+radixtarget>=4.0.1
+typer>=0.19.2

asndb-1.0.0/asndb.egg-info/top_level.txt

@@ -0,0 +1 @@
+asndb

asndb-1.0.0/pyproject.toml

@@ -0,0 +1,34 @@
+[project]
+name = "asndb"
+version = "1.0.0"
+description = "A simple Python CLI + library for looking up ASNs by IP or AS number. Uses BBOT.IO API."
+readme = "README.md"
+requires-python = ">=3.9"
+authors = [
+    {name = "TheTechromancer"}
+]
+dependencies = [
+    "cachetools>=6.2.1",
+    "httpx>=0.28.1",
+    "radixtarget>=4.0.1",
+    "typer>=0.19.2",
+]
+
+[tool.uv]
+package = true
+
+[tool.ruff]
+line-length = 119
+lint.ignore = ["E402", "E713", "E721", "E741", "F401", "F403", "F405"]
+
+[dependency-groups]
+dev = [
+    "maturin>=1.9.6",
+    "pytest>=8.4.2",
+    "pytest-asyncio>=1.2.0",
+    "pytest-cov>=7.0.0",
+    "ruff>=0.14.1",
+]
+
+[project.scripts]
+asndb = "asndb.main:main"

asndb-1.0.0/setup.cfg

@@ -0,0 +1,4 @@
+[egg_info]
+tag_build = 
+tag_date = 0
+

asndb-1.0.0/tests/test_asndb.py

@@ -0,0 +1,133 @@
+import json
+import pytest
+from asndb.asndb import ASNDBClient
+
+
+class ASNDBTestClient(ASNDBClient):
+    def __init__(self, *args, **kwargs):
+        self.requested_urls = []
+        super().__init__(*args, **kwargs)
+
+    async def request(self, url, **kwargs):
+        self.requested_urls.append(url)
+        return await super().request(url, **kwargs)
+
+
+@pytest.mark.asyncio
+async def test_asndb():
+    asndb = ASNDBTestClient()
+
+    # IP lookup
+    asn = await asndb.lookup_ip("1.1.1.1")
+    assert asn["asn"] == 13335
+    assert asn["asn_name"] == "CLOUDFLARENET"
+    assert asn["subnet"] == "1.1.1.0/24"
+    assert asndb.requested_urls == ["https://asndb.api.bbot.io/v1/ip/1.1.1.1"]
+
+    # IP to subnet lookup
+    asn_number, subnet = await asndb.ip_to_subnet("1.1.1.1")
+    assert asn_number == 13335
+    assert subnet == "1.1.1.0/24"
+
+    # Bad IP
+    asn = await asndb.lookup_ip("127.0.0.1")
+    assert asn == ASNDBClient.UNKNOWN_ASN
+    assert asndb.requested_urls == [
+        "https://asndb.api.bbot.io/v1/ip/1.1.1.1",
+        "https://asndb.api.bbot.io/v1/ip/127.0.0.1",
+    ]
+
+    # IP lookup with cache
+    asn = await asndb.lookup_ip("1.1.1.2")
+    assert asn["asn"] == 13335
+    assert asn["asn_name"] == "CLOUDFLARENET"
+    assert asn["subnet"] == "1.1.1.0/24"
+    assert len(asndb.requested_urls) == 2
+
+    # ASN Lookup
+    asn = await asndb.lookup_asn(13335)
+    assert asn["asn"] == 13335
+    assert asn["asn_name"] == "CLOUDFLARENET"
+    # we should not have made a second request
+    assert len(asndb.requested_urls) == 2
+
+    # Organization Lookup
+    org = await asndb.lookup_org("CLOUD14-ARIN")
+    assert org["asns"] == [13335, 14789, 395747, 394536]
+
+
+def test_asndb_sync():
+    asndb = ASNDBTestClient()
+
+    # IP lookup
+    asn = asndb.lookup_ip_sync("1.1.1.1")
+    assert asn["asn"] == 13335
+    assert asn["asn_name"] == "CLOUDFLARENET"
+    assert asn["subnet"] == "1.1.1.0/24"
+    assert len(asndb.requested_urls) == 1
+
+    # IP to subnet lookup
+    asn_number, subnet = asndb.ip_to_subnet_sync("1.1.1.1")
+    assert asn_number == 13335
+    assert subnet == "1.1.1.0/24"
+
+    # Bad IP
+    asn = asndb.lookup_ip_sync("127.0.0.1")
+    assert asn == ASNDBClient.UNKNOWN_ASN
+    assert len(asndb.requested_urls) == 2
+
+    # IP lookup with cache
+    asn = asndb.lookup_ip_sync("1.1.1.2")
+    assert asn["asn"] == 13335
+    assert asn["subnet"] == "1.1.1.0/24"
+    assert len(asndb.requested_urls) == 2
+
+    # ASN Lookup
+    asn = asndb.lookup_asn_sync(13335)
+    assert asn["asn"] == 13335
+    assert asn["asn_name"] == "CLOUDFLARENET"
+    assert len(asndb.requested_urls) == 2
+
+    # Organization Lookup
+    org = asndb.lookup_org_sync("CLOUD14-ARIN")
+    assert org["asns"] == [13335, 14789, 395747, 394536]
+
+
+def test_asndb_cli(monkeypatch, capsys):
+    from asndb.main import main
+    import sys
+
+    # patch sys.exit to prevent the CLI from actually exiting
+    def mock_exit(code=0):
+        pass
+
+    monkeypatch.setattr(sys, "exit", mock_exit)
+
+    # patch sys.argv to include the command
+    monkeypatch.setattr("sys.argv", ["asndb", "ip", "1.1.1.1"])
+    # run the CLI
+    main()
+    # capture the output
+    out, err = capsys.readouterr()
+    out_json = json.loads(out.strip())
+    assert err == ""
+    assert out_json["asn"] == 13335
+    assert out_json["asn_name"] == "CLOUDFLARENET"
+    assert out_json["subnet"] == "1.1.1.0/24"
+
+    # look up asn
+    monkeypatch.setattr("sys.argv", ["asndb", "asn", "13335"])
+    main()
+    out, err = capsys.readouterr()
+    out_json = json.loads(out.strip())
+    assert err == ""
+    assert out_json["asn"] == 13335
+    assert out_json["asn_name"] == "CLOUDFLARENET"
+
+    # look up org
+    monkeypatch.setattr("sys.argv", ["asndb", "org", "CLOUD14-ARIN"])
+    main()
+    out, err = capsys.readouterr()
+    out_json = json.loads(out.strip())
+    assert err == ""
+    assert out_json["asns"] == [13335, 14789, 395747, 394536]