Fix env var API key timing, add select_field param, and clean up packaging

- Defer IPDATA_API_KEY env var lookup from import time to __init__ call
  time so setting the env var after import works correctly
- Add select_field parameter to lookup() matching documented PyPI API
- Move pytest, hypothesis, python-dotenv to [test] extras_require
- Make pytricia optional via [trie] extra so core package installs
  without a C compiler

https://claude.ai/code/session_01M3wRX9tnmc4qorVWRjmqTZ

Author: claude

setup.cfg

@@ -29,9 +29,13 @@ install_requires =
     click
     click_default_group
     pyperclip
+
+[options.extras_require]
+test =
     pytest
     hypothesis
     python-dotenv
+trie =
     pytricia
 
 [options.entry_points]

src/ipdata/__init__.py

@@ -7,16 +7,20 @@
         >>> ipdata.lookup() # or ipdata.lookup("8.8.8.8") 
 """
 from .ipdata import IPData
-from .iptrie import IPTrie
+
+try:
+    from .iptrie import IPTrie
+except ImportError:
+    IPTrie = None
 
 # Configuration
 api_key = None
 endpoint = "https://api.ipdata.co/"
 default_client = None
 
 
-def lookup(resource="", fields=[]):
-    return _proxy("lookup", resource=resource, fields=fields)
+def lookup(resource="", fields=[], select_field=None):
+    return _proxy("lookup", resource=resource, fields=fields, select_field=select_field)
 
 
 def bulk(resources, fields=[]):

src/ipdata/ipdata.py

@@ -100,13 +100,15 @@ class IPData(object):
 
     def __init__(
         self,
-        api_key=os.environ.get("IPDATA_API_KEY"),
+        api_key=None,
         endpoint="https://api.ipdata.co/",
         timeout=60,
         retry_limit=7,
         retry_backoff_factor=1,
         debug=False,
     ):
+        if api_key is None:
+            api_key = os.environ.get("IPDATA_API_KEY")
         if not api_key:
             raise IPDataException("API Key not set. Set an API key via the 'IPDATA_API_KEY' environment variable or see the docs for other ways to do so.")
         # Request settings
@@ -167,17 +169,18 @@ def _validate_ip_address(self, ip):
         if request_ip.is_private or request_ip.is_reserved or request_ip.is_multicast:
             raise ValueError(f"{ip} is a reserved IP Address")
 
-    def lookup(self, resource="", fields=[]):
+    def lookup(self, resource="", fields=[], select_field=None):
         """
         Makes a GET request to the IPData API for the specified 'resource' and the given 'fields'.
 
         :param resource: Either an IP address or an ASN prefixed by "AS" eg. "AS15169"
         :param fields: A collection of API fields to be returned
+        :param select_field: A single field name to return (convenience alternative to fields)
 
         :returns: An API response as a DotDict object to allow dot notation access of fields eg. data.ip, data.company.name, data.threat.blocklists[0].name etc
 
         :raises IPDataException: if the API call fails or if there is a failure in decoding the response.
-        :raises ValueError: if 'resource' is not a string
+        :raises ValueError: if 'resource' is not a string, or if both 'select_field' and 'fields' are provided
         """
         if type(resource) is not str:
             raise ValueError(f"{resource} must be of type 'str'")
@@ -189,6 +192,13 @@ def lookup(self, resource="", fields=[]):
         if resource and not resource.startswith("AS"):
             self._validate_ip_address(resource)
 
+        if select_field is not None:
+            if fields:
+                raise ValueError("'select_field' and 'fields' are mutually exclusive. Use one or the other.")
+            if not isinstance(select_field, str):
+                raise ValueError("'select_field' must be of type 'str'.")
+            fields = [select_field]
+
         self._validate_fields(fields)
         query_params = self._query_params | {"fields": ",".join(fields)}
 

src/ipdata/iptrie.py

@@ -20,7 +20,10 @@
 from collections.abc import Iterator
 from typing import TypeVar, Generic
 
-from pytricia import PyTricia
+try:
+    from pytricia import PyTricia
+except ImportError:
+    PyTricia = None
 
 T = TypeVar("T")
 
@@ -94,6 +97,11 @@ class IPTrie(Generic[T]):
 
     def __init__(self) -> None:
         """Initialize an empty IPTrie."""
+        if PyTricia is None:
+            raise ImportError(
+                "pytricia is required for IPTrie but is not installed. "
+                "Install it with: pip install ipdata[trie]"
+            )
         self._ipv4: PyTricia = PyTricia(self.IPV4_BITS)
         self._ipv6: PyTricia = PyTricia(self.IPV6_BITS)