contribs client bugfixes

Author: esoteric-ephemera

mp_api/client/contribs/_types.py

@@ -316,12 +316,36 @@ def from_dict(cls, dct: dict) -> Self:
         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/client/contribs/client.py

@@ -50,6 +50,7 @@
 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 @@
     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,7 +88,7 @@
 warnings.filterwarnings("default", category=DeprecationWarning, module=__name__)
 
 
-def validate_email(email_string: str):
+def validate_email(email_string: str) -> AllIdMap:
     """Validate user email address.
 
     Args:

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