Skip to content

Commit a853a13

Browse files
authored
Improve explode_env_vars for better dict handling (#261)
1 parent e129a81 commit a853a13

File tree

2 files changed

+25
-3
lines changed

2 files changed

+25
-3
lines changed

pydantic_settings/sources.py

Lines changed: 9 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -604,6 +604,8 @@ def explode_env_vars(self, field_name: str, field: FieldInfo, env_vars: Mapping[
604604
Returns:
605605
A dictionary contains extracted values from nested env values.
606606
"""
607+
is_dict = lenient_issubclass(get_origin(field.annotation), dict)
608+
607609
prefixes = [
608610
f'{env_name}{self.env_nested_delimiter}' for _, env_name, _ in self._extract_field_info(field, field_name)
609611
]
@@ -624,11 +626,15 @@ def explode_env_vars(self, field_name: str, field: FieldInfo, env_vars: Mapping[
624626
target_field = self.next_field(target_field, last_key)
625627

626628
# check if env_val maps to a complex field and if so, parse the env_val
627-
if target_field and env_val:
628-
is_complex, allow_json_failure = self._field_is_complex(target_field)
629+
if (target_field or is_dict) and env_val:
630+
if target_field:
631+
is_complex, allow_json_failure = self._field_is_complex(target_field)
632+
else:
633+
# nested field type is dict
634+
is_complex, allow_json_failure = True, False
629635
if is_complex:
630636
try:
631-
env_val = self.decode_complex_value(last_key, target_field, env_val)
637+
env_val = self.decode_complex_value(last_key, target_field, env_val) # type: ignore
632638
except ValueError as e:
633639
if not allow_json_failure:
634640
raise e

tests/test_settings.py

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2348,3 +2348,19 @@ class Settings(BaseSettings):
23482348

23492349
s = Settings(nested_foo=NestedSettings(fooAlias='EXAMPLE'))
23502350
assert s.model_dump() == {'nested_foo': {'foo': 'EXAMPLE'}}
2351+
2352+
2353+
def test_nested_models_as_dict_value(env):
2354+
class NestedSettings(BaseModel):
2355+
foo: Dict[str, int]
2356+
2357+
class Settings(BaseSettings):
2358+
nested: NestedSettings
2359+
sub_dict: Dict[str, NestedSettings]
2360+
2361+
model_config = SettingsConfigDict(env_nested_delimiter='__')
2362+
2363+
env.set('nested__foo', '{"a": 1}')
2364+
env.set('sub_dict__bar__foo', '{"b": 2}')
2365+
s = Settings()
2366+
assert s.model_dump() == {'nested': {'foo': {'a': 1}}, 'sub_dict': {'bar': {'foo': {'b': 2}}}}

0 commit comments

Comments
 (0)