mp-api 0.46.5__tar.gz → 0.47.0rc0__tar.gz

mp_api-0.47.0rc0/PKG-INFO

@@ -0,0 +1,63 @@
+Metadata-Version: 2.5
+Name: mp-api
+Version: 0.47.0rc0
+Summary: API Client for the Materials Project
+Author-email: The Materials Project <feedback@materialsproject.org>
+License-Expression: BSD-3-Clause-LBNL
+License-File: LICENSE
+Classifier: Development Status :: 4 - Beta
+Classifier: Intended Audience :: Information Technology
+Classifier: Intended Audience :: Science/Research
+Classifier: Operating System :: OS Independent
+Classifier: Programming Language :: Python :: 3
+Classifier: Programming Language :: Python :: 3 :: Only
+Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
+Classifier: Topic :: Scientific/Engineering
+Requires-Python: >=3.12
+Requires-Dist: boto3
+Requires-Dist: deltalake<1.6.0,>=1.4.0
+Requires-Dist: emmet-core<0.88.1,>=0.88.0rc
+Requires-Dist: monty>=2024.12.10
+Requires-Dist: orjson<4,>=3.10
+Requires-Dist: pyarrow>=20.0.0
+Requires-Dist: pymatgen>2024.2.20
+Requires-Dist: requests>=2.23.0
+Requires-Dist: rich>=13.0
+Provides-Extra: all
+Requires-Dist: custodian; extra == 'all'
+Requires-Dist: emmet-core[all]<0.88.1,>=0.88.0rc; extra == 'all'
+Requires-Dist: fastmcp; extra == 'all'
+Requires-Dist: flask; extra == 'all'
+Requires-Dist: ipywidgets; extra == 'all'
+Provides-Extra: contribs
+Requires-Dist: boltons; extra == 'contribs'
+Requires-Dist: bravado; extra == 'contribs'
+Requires-Dist: cachetools; extra == 'contribs'
+Requires-Dist: filetype; extra == 'contribs'
+Requires-Dist: ipython; extra == 'contribs'
+Requires-Dist: json2html; extra == 'contribs'
+Requires-Dist: pint>=0.25.0; extra == 'contribs'
+Requires-Dist: pyisemail; extra == 'contribs'
+Requires-Dist: pymongo; extra == 'contribs'
+Requires-Dist: requests-futures; extra == 'contribs'
+Requires-Dist: swagger-spec-validator; extra == 'contribs'
+Provides-Extra: docs
+Requires-Dist: sphinx; extra == 'docs'
+Provides-Extra: mcp
+Requires-Dist: fastmcp; extra == 'mcp'
+Provides-Extra: notebook
+Requires-Dist: ipywidgets; extra == 'notebook'
+Provides-Extra: server
+Requires-Dist: flask; extra == 'server'
+Provides-Extra: test
+Requires-Dist: mypy; extra == 'test'
+Requires-Dist: pre-commit; extra == 'test'
+Requires-Dist: pytest; extra == 'test'
+Requires-Dist: pytest-asyncio; extra == 'test'
+Requires-Dist: pytest-cov; extra == 'test'
+Requires-Dist: pytest-mock; extra == 'test'
+Requires-Dist: pytest-xdist; extra == 'test'
+Requires-Dist: types-requests; extra == 'test'
+Requires-Dist: types-setuptools; extra == 'test'

{mp_api-0.46.5 → mp_api-0.47.0rc0}/mp_api/_test_utils.py

@@ -46,8 +46,8 @@ def client_search_testing(
         alt_name_dict (dict[str, str]): Alternative names for parameters used in the projection and subsequent data checking
         custom_field_tests (dict[str, Any]): Custom queries for specific fields.
         sub_doc_fields (list[str]): Prefixes for fields to check in resulting data. Useful when data to be tested is nested.
-        int_bounds (tuple[int,int]) : integer bounds to use in testing int-type query arguments
-        float_bounds (tuple[float,float]) : float bounds to use in testing float-type query arguments
+        int_bounds (tuple[int, int]) : integer bounds to use in testing int-type query arguments
+        float_bounds (tuple[float, float]) : float bounds to use in testing float-type query arguments
     """
     if search_method is None:
         return
@@ -58,7 +58,8 @@ def client_search_testing(
     for entry in param_tuples:
         param = entry[0]
 
-        if param not in excluded_params + ["return"]:
+        # `db_version` selects a dataset version rather than filtering docs
+        if param not in excluded_params + ["return", "db_version"]:
             param_type = entry[1]
             q: dict[str, Any] = {"chunk_size": 1, "num_chunks": 1}
 
@@ -139,7 +140,6 @@ def client_sort(
         if k not in ("_page", "_sort_fields", "chunk_size", "fields")
     }
     for sort_field in [sort_fields] if isinstance(sort_fields, str) else sort_fields:
-
         asc = search_method(
             _page=1,
             _sort_fields=sort_field,

{mp_api-0.46.5 → mp_api-0.47.0rc0}/mp_api/client/__init__.py

@@ -2,18 +2,23 @@
 
 from __future__ import annotations
 
-import logging
 import os
 from importlib.metadata import PackageNotFoundError, version
 
+from mp_api.client.core._display import (
+    _configure_from_settings,
+    enable_logging,
+    is_quiet,
+    quiet,
+)
 from mp_api.client.core.exceptions import MPRestError
 from mp_api.client.mprester import MPRester
 
-__all__ = ["MPRestError", "MPRester"]
+__all__ = ["MPRestError", "MPRester", "enable_logging", "is_quiet", "quiet"]
 
 try:
     __version__ = version("mp_api")
 except PackageNotFoundError:  # pragma: no cover
     __version__ = os.getenv("SETUPTOOLS_SCM_PRETEND_VERSION", "")
 
-logging.getLogger(__name__).addHandler(logging.NullHandler())
+_configure_from_settings()

mp_api-0.47.0rc0/mp_api/client/contribs/_logger.py

@@ -0,0 +1,50 @@
+"""Logger for the MPContribs client.
+
+Records go through the `mp_api.client` logger's default handler (stderr),
+see `mp_api.client.core._display`. No handlers or level are set here: the
+level is configured once, for the whole client, with `MPRESTER_LOG_LEVEL`.
+"""
+
+from __future__ import annotations
+
+import logging
+import os
+import warnings
+
+from mp_api.client.contribs.settings import MPCC_SETTINGS
+
+
+class CustomLoggerAdapter(logging.LoggerAdapter):
+    """Prefix messages with the supervisor group/process, if running under supervisor."""
+
+    def process(self, msg, kwargs):
+        prefix = self.extra.get("prefix")  # type: ignore[union-attr]
+        return f"[{prefix}] {msg}" if prefix else msg, kwargs
+
+
+def get_logger(name: str = "mp_api.client.contribs") -> CustomLoggerAdapter:
+    """Get the contribs logger.
+
+    Args:
+        name (str) : logger name, a child of `mp_api.client`
+
+    Returns:
+        CustomLoggerAdapter
+    """
+    logger = logging.getLogger(name)
+    process = os.environ.get("SUPERVISOR_PROCESS_NAME")
+    group = os.environ.get("SUPERVISOR_GROUP_NAME")
+    cfg = {"prefix": f"{group}/{process}"} if process and group else {}
+    return CustomLoggerAdapter(logger, cfg)
+
+
+if MPCC_SETTINGS.CLIENT_LOG_LEVEL is not None:
+    warnings.warn(
+        "MPCONTRIBS_CLIENT_LOG_LEVEL is deprecated and ignored, use MPRESTER_LOG_LEVEL "
+        "(which sets the level for the whole client) instead.",
+        FutureWarning,
+        stacklevel=2,
+    )
+
+
+MPCC_LOGGER = get_logger()

{mp_api-0.46.5 → mp_api-0.47.0rc0}/mp_api/client/contribs/_types.py

@@ -69,7 +69,10 @@ class Table(pd.DataFrame, _Component):
             try:
                 allowed_kwargs = getfullargspec(line_chart).args
                 attrs = {k: v for k, v in self.attrs.items() if k in allowed_kwargs}
-                return self.plot(**attrs)
+                # Plot with plotly directly, rather than changing the
+                # process-wide pandas plotting backend / plotly template.
+                attrs.setdefault("template", "simple_white")
+                return line_chart(self, **attrs)
             except Exception as e:
                 MPCC_LOGGER.error(f"Can't display table: {e}")