mp_api.client.routes.materials.electronic_structure.DosRester

class mp_api.client.routes.materials.electronic_structure.DosRester(api_key: str | None = None, endpoint: str | None = None, include_user_agent: bool = True, use_document_model: bool = True, session: requests.Session | None = None, headers: dict | None = None, mute_progress_bars: bool = False, db_version: str | None = None, local_dataset_cache: str | os.PathLike = PosixPath('/home/runner/mp_datasets'), force_renew: bool = False, query_builder: QueryBuilderWithCache | None = None, s3_client: Any | None = None, timeout: int = 20, **kwargs)

Bases: BaseESPropertyRester

Initialize the REST API helper class.

s3_client: boto3 S3 client object with which to connect to the object stores. timeout: Time in seconds to wait until a request timeout error is thrown

Parameters:

• api_key – A String API key for accessing the MaterialsProject REST interface. Please obtain your API key at https://www.materialsproject.org/dashboard. If this is None, the code will check if there is a “PMG_MAPI_KEY” setting. If so, it will use that environment variable. This makes easier for heavy users to simply add this environment variable to their setups and MPRester can then be called without any arguments.

• endpoint – Url of endpoint to access the MaterialsProject REST interface. Defaults to the standard Materials Project REST address at “https://api.materialsproject.org”, but can be changed to other urls implementing a similar interface.

• include_user_agent – If True, will include a user agent with the HTTP request including information on pymatgen and system version making the API request. This helps MP support pymatgen users, and is similar to what most web browsers send with each page request. Set to False to disable the user agent.

• session – requests Session object with which to connect to the API, for advanced usage only.

• use_document_model – If False, skip the creating the document model and return data as a dictionary. This can be simpler to work with but bypasses data validation and will not give auto-complete for available fields.

• headers – Custom headers for localhost connections.

• mute_progress_bars – Whether to disable progress bars.

• db_version (str) – EXPERIMENTAL, allows for accessing a different version of the database than what is currently deployed. The Materials Project cannot guarantee that all features will still work.

• local_dataset_cache – Target directory for downloading full datasets. Defaults to ‘mp_datasets’ in the user’s home directory

• force_renew – Option to overwrite existing local dataset

• query_builder – Instance of QueryBuilderWithCache to use in querying delta tables NOTE: Must be a QueryBuilderWithCache, a deltalake.QueryBuilder will be ignored.

• s3_client – boto3 S3 client object with which to connect to the object stores.ct to the object stores.ct to the object stores.

• timeout – Time in seconds to wait until a request timeout error is thrown

• **kwargs – access to legacy kwargs that may be in the process of being deprecated

__init__(api_key: str | None = None, endpoint: str | None = None, include_user_agent: bool = True, use_document_model: bool = True, session: requests.Session | None = None, headers: dict | None = None, mute_progress_bars: bool = False, db_version: str | None = None, local_dataset_cache: str | os.PathLike = PosixPath('/home/runner/mp_datasets'), force_renew: bool = False, query_builder: QueryBuilderWithCache | None = None, s3_client: Any | None = None, timeout: int = 20, **kwargs)

Initialize the REST API helper class.

s3_client: boto3 S3 client object with which to connect to the object stores. timeout: Time in seconds to wait until a request timeout error is thrown

Parameters:

• api_key – A String API key for accessing the MaterialsProject REST interface. Please obtain your API key at https://www.materialsproject.org/dashboard. If this is None, the code will check if there is a “PMG_MAPI_KEY” setting. If so, it will use that environment variable. This makes easier for heavy users to simply add this environment variable to their setups and MPRester can then be called without any arguments.

• endpoint – Url of endpoint to access the MaterialsProject REST interface. Defaults to the standard Materials Project REST address at “https://api.materialsproject.org”, but can be changed to other urls implementing a similar interface.

• include_user_agent – If True, will include a user agent with the HTTP request including information on pymatgen and system version making the API request. This helps MP support pymatgen users, and is similar to what most web browsers send with each page request. Set to False to disable the user agent.

• session – requests Session object with which to connect to the API, for advanced usage only.

• use_document_model – If False, skip the creating the document model and return data as a dictionary. This can be simpler to work with but bypasses data validation and will not give auto-complete for available fields.

• headers – Custom headers for localhost connections.

• mute_progress_bars – Whether to disable progress bars.

• db_version (str) – EXPERIMENTAL, allows for accessing a different version of the database than what is currently deployed. The Materials Project cannot guarantee that all features will still work.

• local_dataset_cache – Target directory for downloading full datasets. Defaults to ‘mp_datasets’ in the user’s home directory

• force_renew – Option to overwrite existing local dataset

• query_builder – Instance of QueryBuilderWithCache to use in querying delta tables NOTE: Must be a QueryBuilderWithCache, a deltalake.QueryBuilder will be ignored.

• s3_client – boto3 S3 client object with which to connect to the object stores.ct to the object stores.ct to the object stores.

• timeout – Time in seconds to wait until a request timeout error is thrown

• **kwargs – access to legacy kwargs that may be in the process of being deprecated

Methods

__init__([api_key, endpoint, ...])	Initialize the REST API helper class.
count([criteria])	Return a count of total documents.
get_data_by_id(document_id[, fields])
get_dos_from_material_id(material_id[, ...])	Get the complete density of states pymatgen object associated with a Materials Project ID.
get_dos_from_task_id(task_id[, run_type, ...])	Get the density of states pymatgen object associated with a given calculation ID.
search([band_gap, efermi, element, ...])	Query density of states summary data in electronic structure docs using a variety of search criteria.
search_dos_summary(*args, **kwargs)	Deprecated.

Attributes

available_fields
delta_backed
es_rester
primary_key
query_builder
s3_client
session
suffix

search_dos_summary(*args, **kwargs)

Deprecated.

search(band_gap: tuple[float, float] | None = None, efermi: tuple[float, float] | None = None, element: Element | str | None = None, magnetic_ordering: Ordering | str | None = None, orbital: OrbitalType | str | None = None, projection_type: DOSProjectionType | str = DOSProjectionType.total, spin: Spin | str = Spin.up, num_chunks: int | None = None, chunk_size: int = 1000, all_fields: bool = True, fields: list[str] | None = None)

Query density of states summary data in electronic structure docs using a variety of search criteria.

Parameters:

• band_gap (Tuple[float,float]) – Minimum and maximum band gap in eV to consider.

• efermi (Tuple[float,float]) – Minimum and maximum fermi energy in eV to consider.

• element (Element or str) – Element for element-projected dos data.

• magnetic_ordering (Ordering or str) – Magnetic ordering of the material.

• orbital (OrbitalType or str) – Orbital for orbital-projected dos data.

• projection_type (DOSProjectionType or str) – Projection type of dos data. Default is the total dos.

• spin (Spin or str) – Spin channel of dos data. If non spin-polarized data is stored in Spin.up

• num_chunks (int) – Maximum number of chunks of data to yield. None will yield all possible.

• chunk_size (int) – Number of data entries per chunk.

• all_fields (bool) – Whether to return all fields in the document. Defaults to True.

• fields (List[str]) – List of fields in ElectronicStructureDoc to return data for. Default is material_id and last_updated if all_fields is False.

Returns:

([ElectronicStructureDoc]) List of electronic structure documents

get_dos_from_task_id(task_id: str, run_type: str | RunType | None = None, load_projections: bool = False) → Dos

Get the density of states pymatgen object associated with a given calculation ID.

Parameters:

• task_id (str) – Task ID for the density of states calculation

• run_type (str, RunType, or None) – Optional run type to query by. Will speed up query due to delta table partitioning.

• load_projections (bool) – Optionally load atom- and spin-orbital-projected DOS, if available.

Returns:

pymatgen Dos

get_dos_from_material_id(material_id: str, load_projections: bool = False) → Dos

Get the complete density of states pymatgen object associated with a Materials Project ID.

Parameters:

• material_id (str) – Materials Project ID for a material

• load_projections (bool) – Optionally load atom- and spin-orbital-projected DOS, if available.

Returns:

pymatgen Dos

count(criteria: dict | None = None) → int

Return a count of total documents.

Parameters:

criteria (dict | None) – As in .search(). Defaults to None

Returns:

Count of total results

Return type:

int

document_model

alias of ElectronicStructureDoc