Skip to content

Commit

Permalink
Allow the pmgrc file to be read for the MPREster settings (#708)
Browse files Browse the repository at this point in the history
* read pmgrc

* update

* update
  • Loading branch information
jmmshn committed Nov 16, 2022
1 parent 574cda5 commit 80f4686
Showing 1 changed file with 13 additions and 5 deletions.
18 changes: 13 additions & 5 deletions mp_api/client/core/settings.py
Expand Up @@ -2,9 +2,15 @@
from mp_api.client import __file__ as root_dir
from multiprocessing import cpu_count
from typing import List
from pymatgen.core import _load_pmg_settings
import os

CPU_COUNT = 8

PMG_SETTINGS = _load_pmg_settings()
_NUM_PARALLEL_REQUESTS = PMG_SETTINGS.get("MPRESTER_NUM_PARALLEL_REQUESTS", 8)
_MAX_RETRIES = PMG_SETTINGS.get("MPRESTER_MAX_RETRIES", 3)
_MUTE_PROGRESS_BAR = PMG_SETTINGS.get("MPRESTER_MUTE_PROGRESS_BARS", False)
_MAX_HTTP_URL_LENGTH = PMG_SETTINGS.get("MPRESTER_MAX_HTTP_URL_LENGTH", 2000)

try:
CPU_COUNT = cpu_count()
Expand Down Expand Up @@ -49,19 +55,21 @@ class MAPIClientSettings(BaseSettings):
)

NUM_PARALLEL_REQUESTS: int = Field(
CPU_COUNT,
_NUM_PARALLEL_REQUESTS,
description="Number of parallel requests to send.",
)

MAX_RETRIES: int = Field(3, description="Maximum number of retries for requests.")
MAX_RETRIES: int = Field(
_MAX_RETRIES, description="Maximum number of retries for requests."
)

MUTE_PROGRESS_BARS: bool = Field(
False,
_MUTE_PROGRESS_BAR,
description="Whether to mute progress bars when data is retrieved.",
)

MAX_HTTP_URL_LENGTH: int = Field(
2000,
_MAX_HTTP_URL_LENGTH,
description="Number of characters to use to define the maximum length of a given HTTP URL.",
)

Expand Down

0 comments on commit 80f4686

Please sign in to comment.