mp-api 0.46.2rc0__tar.gz → 0.46.2rc2__tar.gz

{mp_api-0.46.2rc0 → mp_api-0.46.2rc2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mp-api
-Version: 0.46.2rc0
+Version: 0.46.2rc2
 Summary: API Client for the Materials Project
 Author-email: The Materials Project <feedback@materialsproject.org>
 License-Expression: BSD-3-Clause-LBNL

{mp_api-0.46.2rc0 → mp_api-0.46.2rc2}/mp_api/client/contribs/_types.py

@@ -316,12 +316,36 @@ class Attachment(dict, _Component):
         return cls((k, v) for k, v in dct.items() if k in keys)
 
 
+"""
+The following type annotations are only used in type checking
+(mypy) the ContribsClient.get_all_ids function.
+This function has highly hetereogeneous output and these
+represent the intermediate and final data structures returned
+by that function.
+
+They **shouldn't** have to be used elsewhere.
+"""
+
 ComponentIdSets = dict[str, set[str]]
+"""Use only when type checking of ContribsClient.get_all_ids(...,fmt="sets")."""
+
 ProjectIdSets = dict[str, set[str] | ComponentIdSets]
+"""Use only when type checking of ContribsClient.get_all_ids(...,fmt="sets")."""
+
 AllIdSets = dict[str, ProjectIdSets]
+"""Use only when type checking of ContribsClient.get_all_ids(...,fmt="sets")."""
 
 ComponentNameMap = dict[str, dict[str, str]]
+"""Use only when type checking of ContribsClient.get_all_ids(...,fmt="map")."""
+
 IdentifierLeaf = dict[str, str | ComponentNameMap]
+"""Use only when type checking of ContribsClient.get_all_ids(...,fmt="map")."""
+
 IdentifierBranch = dict[str, IdentifierLeaf]
+"""Use only when type checking of ContribsClient.get_all_ids(...,fmt="map")."""
+
 ProjectIdMap = dict[str, IdentifierLeaf | IdentifierBranch]
+"""Use only when type checking of ContribsClient.get_all_ids(...,fmt="map")."""
+
 AllIdMap = dict[str, ProjectIdMap]
+"""Use only when type checking of ContribsClient.get_all_ids(...,fmt="map")."""

{mp_api-0.46.2rc0 → mp_api-0.46.2rc2}/mp_api/client/contribs/client.py

@@ -50,6 +50,7 @@ from urllib3.util.retry import Retry
 from mp_api.client.contribs._logger import MPCC_LOGGER, TqdmToLogger
 from mp_api.client.contribs._types import (
     Attachment,
+    ComponentIdSets,
     MPCDict,
     MPCStructure,
     Table,
@@ -71,10 +72,9 @@ if TYPE_CHECKING:
     from collections.abc import Generator, Iterable, Sequence
     from typing import Any
 
-    from mp_api.client.contribs.types import (
+    from mp_api.client.contribs._types import (
         AllIdMap,
         AllIdSets,
-        ComponentIdSets,
         IdentifierLeaf,
         ProjectIdSets,
     )
@@ -88,11 +88,15 @@ warnings.formatwarning = lambda msg, *args, **kwargs: f"{msg}\n"
 warnings.filterwarnings("default", category=DeprecationWarning, module=__name__)
 
 
-def validate_email(email_string: str):
+def validate_email(email_string: str) -> None:
     """Validate user email address.
 
     Args:
         email_string (str) : the user's email address
+    Returns:
+        None
+    Raises:
+        SwaggerValidationError on malformed email address.
     """
     if email_string.count(":") != 1:
         raise SwaggerValidationError(
@@ -107,6 +111,8 @@ def validate_email(email_string: str):
     if d > BaseDiagnosis.CATEGORIES["VALID"]:
         raise SwaggerValidationError(f"{email} {d.message}")
 
+    return None
+
 
 # TODO: mypy has some problems with putting a bare `str`
 # as a callable function in SwaggerFormat
@@ -791,7 +797,7 @@ class ContribsClient(SwaggerClient):
         ret = self.projects.queryProjects(**query).result()  # first page
         total_count, total_pages = ret["total_count"], ret["total_pages"]
 
-        if total_pages < 2:
+        if total_pages < 2 or ("page" in query):
             return (
                 _convert_to_model(  # type: ignore[return-value]
                     ret["data"],
@@ -822,7 +828,12 @@ class ContribsClient(SwaggerClient):
         ]
         responses = _run_futures(futures, total=total_count, timeout=timeout)
 
-        ret["data"].extend([resp["result"]["data"] for resp in responses.values()])
+        # NOTE: resp["result"]["data"] is a dict where each key is an int,
+        # and each value is a **list** of projects as dict
+        # Double iteration is necessary because of this nested structure
+        ret["data"].extend(
+            [entry for resp in responses.values() for entry in resp["result"]["data"]]
+        )
 
         return (
             _convert_to_model(  # type: ignore[return-value]
@@ -2109,7 +2120,7 @@ class ContribsClient(SwaggerClient):
                         digest in digests[project_name][component]
                         or digest
                         in existing.get(project_name, {})
-                        .get(component, {})
+                        .get(component, {})  # type: ignore[union-attr]
                         .get("md5s", [])
                     )
 
@@ -2227,9 +2238,9 @@ class ContribsClient(SwaggerClient):
                                 pk=project_name, _fields=["unique_identifiers"]
                             ).result()["unique_identifiers"]
                         )
-                        existing_ids = existing.get(project_name, {}).get(
-                            "identifiers", []
-                        )
+                        existing_ids: Iterable[str] = existing.get(
+                            project_name, {}
+                        ).get("identifiers", [])
                         contribs[project_name] = [
                             c
                             for c in contribs[project_name]

{mp_api-0.46.2rc0 → mp_api-0.46.2rc2}/mp_api/client/contribs/schemas.py

@@ -372,5 +372,5 @@ class QueryResult(_DictLikeAccess):
 
     total_count: int
     total_pages: int
-    data: list[BaseModel] | None = None
+    data: list[ContribData] | None = None
     has_more: bool = False

{mp_api-0.46.2rc0 → mp_api-0.46.2rc2}/mp_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mp-api
-Version: 0.46.2rc0
+Version: 0.46.2rc2
 Summary: API Client for the Materials Project
 Author-email: The Materials Project <feedback@materialsproject.org>
 License-Expression: BSD-3-Clause-LBNL