laipeng/reflex
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

refactor inner function

Browse Source
This commit is contained in:
Khaleel Al-Adhami 2024-11-19 16:26:45 -08:00
parent 6b4e030328
commit 7c5bb67648
1 changed files with 23 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
reflex/config.py
View File

@ -799,6 +799,23 @@ class Config(Base):
self._replace_defaults(**kwargs) self._replace_defaults(**kwargs)
def _get_config() -> Config:
"""Get the app config.
Returns:
The app config.
"""
# only import the module if it exists. If a module spec exists then
# the module exists.
spec = find_spec(constants.Config.MODULE)
if not spec:
# we need this condition to ensure that a ModuleNotFound error is not thrown when
# running unit/integration tests or during `reflex init`.
return Config(app_name="")
rxconfig = importlib.import_module(constants.Config.MODULE)
return rxconfig.config
def get_config(reload: bool = False) -> Config: def get_config(reload: bool = False) -> Config:
"""Get the app config. """Get the app config.
@ -812,30 +829,17 @@ def get_config(reload: bool = False) -> Config:
if reload and constants.Config.MODULE in sys.modules: if reload and constants.Config.MODULE in sys.modules:
del sys.modules[constants.Config.MODULE] del sys.modules[constants.Config.MODULE]
sys_path = list(sys.path) sys_path = sys.path.copy()
sys.path.clear() sys.path.clear()
sys.path.append(os.getcwd()) sys.path.append(os.getcwd())
try: try:
# only import the module if it exists. If a module spec exists then # Try to import the module with only the current directory in the path.
# the module exists. return _get_config()
spec = find_spec(constants.Config.MODULE)
if not spec:
# we need this condition to ensure that a ModuleNotFound error is not thrown when
# running unit/integration tests or during `reflex init`.
return Config(app_name="")
rxconfig = importlib.import_module(constants.Config.MODULE)
return rxconfig.config
except Exception: except Exception:
# If the module import fails, try to import with the original sys.path.
sys.path.extend(sys_path) sys.path.extend(sys_path)
# only import the module if it exists. If a module spec exists then return _get_config()
# the module exists.
spec = find_spec(constants.Config.MODULE)
if not spec:
# we need this condition to ensure that a ModuleNotFound error is not thrown when
# running unit/integration tests or during `reflex init`.
return Config(app_name="")
rxconfig = importlib.import_module(constants.Config.MODULE)
return rxconfig.config
finally: finally:
# Restore the original sys.path.
sys.path.clear() sys.path.clear()
sys.path.extend(sys_path) sys.path.extend(sys_path)