Make aiohttp and requests optional dependencies

Closes #104 (supersedes it)

Fixes #101

Author: akx

README.rst

@@ -17,6 +17,8 @@ To install the ``geoip2`` module, type:
 .. code-block:: bash
 
     $ pip install geoip2
+    $ pip install geoip2[aiohttp]  # Install aiohttp as well
+    $ pip install geoip2[requests]   # Install requests as well
 
 If you are not able to install from PyPI, you may also use ``pip`` from the
 source directory:

pyproject.toml

@@ -103,6 +103,14 @@ env_list = [
 ]
 skip_missing_interpreters = false
 
+[project.optional-dependencies]
+aiohttp = [
+    "aiohttp>=3.6.2,<4.0.0",
+]
+requests = [
+    "requests>=2.24.0,<3.0.0",
+]
+
 [tool.tox.env_run_base]
 runner = "uv-venv-lock-runner"
 dependency_groups = [
@@ -115,6 +123,10 @@ commands = [
 [tool.tox.env.lint]
 description = "Code linting"
 python = "3.14"
+extras = [
+    "aiohttp",
+    "requests",
+]
 dependency_groups = [
     "dev",
     "lint",

src/geoip2/webservice.py

@@ -27,10 +27,18 @@
 import json
 from typing import TYPE_CHECKING, cast
 
-import aiohttp
-import aiohttp.http
-import requests
-import requests.utils
+try:
+    import aiohttp
+    import aiohttp.http
+except ImportError:
+    aiohttp = None  # type: ignore[assignment]
+
+try:
+    import requests
+    import requests.utils
+except ImportError:
+    requests = None  # type: ignore[assignment]
+
 
 import geoip2
 import geoip2.models
@@ -52,14 +60,6 @@
     from geoip2.models import City, Country, Insights
     from geoip2.types import IPAddress
 
-_AIOHTTP_UA = (
-    f"GeoIP2-Python-Client/{geoip2.__version__} {aiohttp.http.SERVER_SOFTWARE}"
-)
-
-_REQUEST_UA = (
-    f"GeoIP2-Python-Client/{geoip2.__version__} {requests.utils.default_user_agent()}"
-)
-
 
 class BaseClient:
     """Base class for AsyncClient and Client."""
@@ -352,15 +352,23 @@ async def insights(self, ip_address: IPAddress = "me") -> Insights:
         )
 
     async def _session(self) -> aiohttp.ClientSession:
+        if aiohttp is None:
+            msg = "aiohttp is required for async mode; install `GeoIP2[aiohttp]`"
+            raise ImportError(msg)
+
         if not hasattr(self, "_existing_session"):
+            user_agent = (
+                f"GeoIP2-Python-Client/{geoip2.__version__} "
+                f"{aiohttp.http.SERVER_SOFTWARE}"
+            )
             self._existing_session = aiohttp.ClientSession(
                 headers={
                     "Accept": "application/json",
                     "Authorization": aiohttp.encode_basic_auth(
                         self._account_id,
                         self._license_key,
                     ),
-                    "User-Agent": _AIOHTTP_UA,
+                    "User-Agent": user_agent,
                 },
                 timeout=aiohttp.ClientTimeout(total=self._timeout),
             )
@@ -474,7 +482,10 @@ def __init__(  # noqa: PLR0913
         self._session = requests.Session()
         self._session.auth = (self._account_id, self._license_key)
         self._session.headers["Accept"] = "application/json"
-        self._session.headers["User-Agent"] = _REQUEST_UA
+        self._session.headers["User-Agent"] = (
+            f"GeoIP2-Python-Client/{geoip2.__version__}"
+            f" {requests.utils.default_user_agent()}"
+        )
         if proxy is None:
             self._proxies = None
         else:

tests/webservice_test.py

@@ -403,6 +403,7 @@ class TestClient(TestBaseClient):
     client: Client
 
     def setUp(self) -> None:
+        pytest.importorskip("requests")
         self.client_class = Client
         self.client = Client(42, "abcdef123456")
         self.client._base_uri = self.httpserver.url_for("/geoip/v2.1")  # noqa: SLF001
@@ -416,6 +417,7 @@ class TestAsyncClient(TestBaseClient):
     client: AsyncClient
 
     def setUp(self) -> None:
+        pytest.importorskip("aiohttp")
         self._loop = asyncio.new_event_loop()
         self.client_class = AsyncClient
         self.client = AsyncClient(42, "abcdef123456")