pcmt

Author: esoteric-ephemera

mp_api/client/core/utils.py

@@ -74,36 +74,37 @@ def validate_ids(id_list: list[str]) -> list[str]:
     # return [validate_identifier(idx,serialize=True) for idx in id_list]
     return [str(validate_identifier(idx)) for idx in id_list]
 
+
 class LazyImport:
+    """Lazily import and load an object."""
 
     __slots__ = ["_module_name", "_class_name", "_obj"]
 
-    def __init__(self, module_name : str, class_name : str,) -> None:
+    def __init__(  # noqa: D107
+        self,
+        module_name: str,
+        class_name: str,
+    ) -> None:
         self._module_name = module_name
         self._class_name = class_name
         self._obj = None
-    
+
     def __str__(self) -> str:
         return f"LazyImport of {self._module_name}.{self._class_name}"
-    
+
     def __repr__(self) -> str:
         return self.__str__()
 
     def __call__(self, *args, **kwargs):
         if self._obj is None:
             try:
-                self._obj = getattr(
-                    import_module(self._module_name),
-                    self._class_name
-                )(
+                self._obj = getattr(import_module(self._module_name), self._class_name)(
                     *args,
                     **kwargs,
                 )
             except Exception as exc:
-                raise ImportError(
-                    f"Failed to import {self._class_name}:\n{exc}"
-                )
+                raise ImportError(f"Failed to import {self._class_name}:\n{exc}")
         return self._obj
-    
-    def __getattr__(self,v):
-        return getattr(self._obj,v)
+
+    def __getattr__(self, v):
+        return getattr(self._obj, v)

mp_api/client/mprester.py

@@ -26,7 +26,7 @@
 from mp_api.client.core import BaseRester, MPRestError
 from mp_api.client.core._oxygen_evolution import OxygenEvolution
 from mp_api.client.core.settings import MAPIClientSettings
-from mp_api.client.core.utils import load_json, validate_ids, LazyImport
+from mp_api.client.core.utils import LazyImport, load_json, validate_ids
 from mp_api.client.routes import GENERIC_RESTERS
 from mp_api.client.routes.materials.materials import MATERIALS_RESTERS
 
@@ -42,16 +42,17 @@
 
 RESTER_LAYOUT = {
     **{
-        f"materials/{k}" : v
+        f"materials/{k}": v
         for k, v in MATERIALS_RESTERS.items()
-        if k not in {"materials","doi"}
+        if k not in {"materials", "doi"}
     },
     "materials/core": MATERIALS_RESTERS["materials"],
     "doi": MATERIALS_RESTERS["doi"],
     **GENERIC_RESTERS,
-    "molecules/core": LazyImport("mp_api.client.routes.molecules","MoleculeRester")        
+    "molecules/core": LazyImport("mp_api.client.routes.molecules", "MoleculeRester"),
 }
 
+
 class MPRester:
     """Access the new Materials Project API."""
 
@@ -238,7 +239,10 @@ def __init__(
         def __core_custom_getattr(_self, _attr, _rester_map):
             if _attr in RESTER_LAYOUT:
                 lazy_rester = RESTER_LAYOUT[_attr]
-                monty_disable = lazy_rester._class_name in ["TaskRester", "ProvenanceRester"]
+                monty_disable = lazy_rester._class_name in [
+                    "TaskRester",
+                    "ProvenanceRester",
+                ]
                 monty_decode = False if monty_disable else self.monty_decode
                 rester = lazy_rester(
                     api_key=api_key,

mp_api/client/routes/__init__.py

@@ -3,9 +3,10 @@
 from mp_api.client.core.utils import LazyImport
 
 GENERIC_RESTERS = {
-    k : LazyImport(f"mp_api.client.routes.{k}", v) for k, v in {
+    k: LazyImport(f"mp_api.client.routes.{k}", v)
+    for k, v in {
         "_general_store": "GeneralStoreRester",
         "_messages": "MessagesRester",
-        "_user_settings": "UserSettingsRester"
+        "_user_settings": "UserSettingsRester",
     }.items()
-}
+}

mp_api/client/routes/materials/__init__.py

@@ -1,33 +1,42 @@
 """Define routes and imports to materials resters."""
+from __future__ import annotations
 
 from mp_api.client.core.utils import LazyImport
 
-MATERIALS_RESTERS : dict[str,LazyImport] = {
-    route : LazyImport(f"mp_api.client.routes.materials.{module_name}", cls_name)
+MATERIALS_RESTERS: dict[str, LazyImport] = {
+    route: LazyImport(f"mp_api.client.routes.materials.{module_name}", cls_name)
     for route, module_name, cls_name in [
         ("absorption", "absorption", "AbsorptionRester"),
         ("alloys", "alloys", "AlloysRester"),
         ("bonds", "bonds", "BondsRester"),
         ("charge_density", "charge_density", "ChargeDensityRester"),
-        ("chemenv", "chemenv", "ChemenvRester",),
-        ("conversion_electrodes","electrodes","ConversionElectrodeRester"), 
+        (
+            "chemenv",
+            "chemenv",
+            "ChemenvRester",
+        ),
+        ("conversion_electrodes", "electrodes", "ConversionElectrodeRester"),
         ("dielectric", "dielectric", "DielectricRester"),
         ("doi", "doi", "DOIRester"),
         ("elasticity", "elasticity", "ElasticityRester"),
-        ("electronic_structure","electronic_structure","ElectronicStructureRester"),
-        ("electronic_structure_bandstructure","electronic_structure", "BandStructureRester"),
+        ("electronic_structure", "electronic_structure", "ElectronicStructureRester"),
+        (
+            "electronic_structure_bandstructure",
+            "electronic_structure",
+            "BandStructureRester",
+        ),
         ("electronic_structure_dos", "electronic_structure", "DosRester"),
-        ("eos","eos","EOSRester"),
+        ("eos", "eos", "EOSRester"),
         ("grain_boundaries", "grain_boundaries", "GrainBoundaryRester"),
-        ("insertion_electrodes","electrodes","ElectrodeRester"),
-        ("magnetism","magnetism","MagnetismRester"),
+        ("insertion_electrodes", "electrodes", "ElectrodeRester"),
+        ("magnetism", "magnetism", "MagnetismRester"),
         ("materials", "materials", "MaterialsRester"),
         ("oxidation_states", "oxidation_states", "OxidationStatesRester"),
-        ("phonon","phonon","PhononRester"),
-        ("piezoelectric","piezoelectric", "PiezoRester"),
-        ("provenance","provenance","ProvenanceRester"),
+        ("phonon", "phonon", "PhononRester"),
+        ("piezoelectric", "piezoelectric", "PiezoRester"),
+        ("provenance", "provenance", "ProvenanceRester"),
         ("robocrys", "robocrys", "RobocrysRester"),
-        ("similarity","similarity","SimilarityRester"),
+        ("similarity", "similarity", "SimilarityRester"),
         ("substrates", "substrates", "SubstratesRester"),
         ("summary", "summary", "SummaryRester"),
         ("surface_properties", "surface_properties", "SurfacePropertiesRester"),
@@ -36,4 +45,4 @@
         ("thermo", "thermo", "ThermoRester"),
         ("xas", "xas", "XASRester"),
     ]
-}
+}

mp_api/client/routes/materials/materials.py

@@ -47,12 +47,14 @@ class MaterialsRester(BaseRester):
         "chemenv",
     ]
 
-    def __getattr__(self, v : str):
+    def __getattr__(self, v: str):
         if v in self._sub_resters:
             if MATERIALS_RESTERS[v]._obj is None:
-
                 # TODO: Enable monty decoding when tasks and SNL schema is normalized
-                monty_disable = MATERIALS_RESTERS[v]._class_name in ["TaskRester", "ProvenanceRester"]
+                monty_disable = MATERIALS_RESTERS[v]._class_name in [
+                    "TaskRester",
+                    "ProvenanceRester",
+                ]
 
                 MATERIALS_RESTERS[v](
                     api_key=self.api_key,