MOD: Remove get_license_free from public API

Author: cjdsellers

databento/historical/api/metadata.py

@@ -9,7 +9,6 @@
     maybe_datetime_to_string,
     maybe_enum_or_str_lowercase,
     maybe_symbols_list_to_string,
-    values_list_to_string,
 )
 from databento.common.validation import validate_enum, validate_maybe_enum
 from databento.historical.api import API_VERSION
@@ -467,43 +466,3 @@ def get_cost(
         )
 
         return response.json()
-
-    def get_license_fee(
-        self,
-        dataset: Union[Dataset, str],
-        purposes: Union[List[str], str],
-    ) -> int:
-        """
-        Request the license fee in US Dollars for a dataset and access purposes
-        from Databento.
-
-        Makes a `GET /metadata.get_license_fee` HTTP request.
-
-        Parameters
-        ----------
-        dataset : Dataset or str
-            The dataset name for the request.
-        purposes : List[str] or str, {'professional trading', 'display usage', 'non-display usage', 'external redistribution', 'internal redistribution'}  # noqa
-            The licensing purposes.
-
-        Returns
-        -------
-        int
-            The license fee in US Dollars.
-
-        """
-        dataset = enum_or_str_lowercase(dataset, "dataset")
-        purposes = values_list_to_string(purposes)
-
-        params: List[Tuple[str, str]] = [
-            ("dataset", dataset),
-            ("purposes", purposes),
-        ]
-
-        response: Response = self._get(
-            url=self._base_url + ".get_license_fee",
-            params=params,
-            basic_auth=True,
-        )
-
-        return response.json()

databento/historical/api/timeseries.py

@@ -1,18 +1,13 @@
 import warnings
 from datetime import date
-from typing import Any, List, Optional, Tuple, Union
+from typing import List, Optional, Tuple, Union
 
 import pandas as pd
 from databento.common.bento import Bento
 from databento.common.enums import Compression, Dataset, Encoding, Schema, SType
-from databento.common.logging import log_debug
 from databento.common.validation import validate_enum
 from databento.historical.api import API_VERSION
 from databento.historical.http import BentoHttpAPI
-from requests import Response
-
-
-_5GB = 1024**3 * 5
 
 
 class TimeSeriesHttpAPI(BentoHttpAPI):
@@ -107,7 +102,6 @@ def stream(
             start=start,
             end=end,
             limit=limit,
-            params=params,
         )
 
         bento: Bento = self._create_bento(path=path)
@@ -204,7 +198,6 @@ async def stream_async(
             start=start,
             end=end,
             limit=limit,
-            params=params,
         )
 
         bento: Bento = self._create_bento(path=path)
@@ -225,44 +218,26 @@ def _pre_check_data_size(
         start: Optional[Union[pd.Timestamp, date, str, int]],
         end: Optional[Union[pd.Timestamp, date, str, int]],
         limit: Optional[int],
-        params: List[Tuple[str, Any]],
     ):
         if limit and limit < 10**7:
             return
 
+        # Use heuristics to check ballpark data size
         if (
             _is_large_data_size_schema(schema)
             or _is_greater_than_one_day(start, end)
             or _is_large_number_of_symbols(symbols)
         ):
-            params = params[:]  # copy
-            params.append(("mode", "historical-streaming"))
-            params.append(("instruments", str(len(symbols))))
-            log_debug(
-                "Checking estimated data size for potentially large streaming "
-                "request...",
-            )
-            response: Response = self._get(
-                url=self._gateway + f"/v{API_VERSION}/metadata.get_size_estimation",
-                params=params,
-                basic_auth=True,
-            )
-
-            size: int = response.json()["size"]
-            log_debug(
-                f"Requesting data stream for {size:,} bytes (binary uncompressed)...",
+            warnings.warn(
+                "\nThe size of the current streaming request is estimated "
+                "to be 5 GB or greater. We recommend smaller "
+                "individual streaming request sizes, or alternatively "
+                "submit a batch data request."
+                "\nYou can check the uncompressed binary size of a request "
+                "through the metadata API (from the client library, or over "
+                "HTTP).\nThis warning can be suppressed "
+                "https://docs.python.org/3/library/warnings.html",
             )
-            if size > _5GB:
-                warnings.warn(
-                    f"\nThe size of the current streaming request is estimated "
-                    f"to exceed 5GB ({_5GB:,} bytes). We recommend smaller "
-                    f"individual streaming request sizes, or alternatively "
-                    f"submit a batch data request."
-                    f"\nYou can check the uncompressed binary size of a request "
-                    f"through the metadata API (from the client library, or over "
-                    f"HTTP).\nThis warning can be suppressed "
-                    f"https://docs.python.org/3/library/warnings.html",
-                )
 
 
 def _is_large_number_of_symbols(symbols: Optional[Union[List[str], str]]):

tests/test_historical_metadata.py

@@ -295,32 +295,3 @@ def test_get_cost_sends_expected_request(self, mocker) -> None:
         ]
         assert call["timeout"] == (100, 100)
         assert isinstance(call["auth"], requests.auth.HTTPBasicAuth)
-
-    @pytest.mark.skipif(sys.version_info < (3, 8), reason="incompatible mocking")
-    def test_get_license_fee_sends_expected_request(self, mocker) -> None:
-        # Arrange
-        mocked_get = mocker.patch("requests.get")
-
-        # Act
-        self.client.metadata.get_license_fee(
-            dataset="GLBX.MDP3",
-            purposes="professional trading",
-        )
-
-        # Assert
-        call = mocked_get.call_args.kwargs
-        assert (
-            call["url"]
-            == f"https://hist.databento.com/v{db.API_VERSION}/metadata.get_license_fee"
-        )
-        assert sorted(call["headers"].keys()) == ["accept", "user-agent"]
-        assert call["headers"]["accept"] == "application/json"
-        assert all(
-            v in call["headers"]["user-agent"] for v in ("Databento/", "Python/")
-        )
-        assert call["params"] == [
-            ("dataset", "glbx.mdp3"),
-            ("purposes", "professional trading"),
-        ]
-        assert call["timeout"] == (100, 100)
-        assert isinstance(call["auth"], requests.auth.HTTPBasicAuth)