Skip to content

Commit 567935a

Browse files
committed
refactor: improve type hints and formatting in keyring provider
1 parent 57e73af commit 567935a

File tree

1 file changed

+11
-13
lines changed

1 file changed

+11
-13
lines changed
Lines changed: 11 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
1-
from typing import TYPE_CHECKING, Mapping
1+
from collections.abc import Mapping
2+
from typing import TYPE_CHECKING
23

34
from pydantic_settings import BaseSettings, EnvSettingsSource
45

@@ -7,7 +8,8 @@
78
else:
89
keyring = None
910

10-
def import_keyring():
11+
12+
def import_keyring() -> None:
1113
global keyring
1214
if keyring is not None:
1315
return
@@ -16,7 +18,7 @@ def import_keyring():
1618

1719
return
1820
except ImportError as e:
19-
raise ImportError("Keyring is not installed, run `pip install keyring`") from e
21+
raise ImportError('Keyring is not installed, run `pip install keyring`') from e
2022

2123

2224
class KeyringConfigSettingsSource(EnvSettingsSource):
@@ -27,20 +29,16 @@ def __init__(
2729
case_sensitive: bool | None = None,
2830
env_prefix: str | None = None,
2931
):
30-
self.keyring_service = (
31-
keyring_service if case_sensitive else keyring_service.lower()
32-
)
33-
super().__init__(
34-
settings_cls, case_sensitive=case_sensitive, env_prefix=env_prefix
35-
)
32+
self.keyring_service = keyring_service if case_sensitive else keyring_service.lower()
33+
super().__init__(settings_cls, case_sensitive=case_sensitive, env_prefix=env_prefix)
3634

3735
def _load_env_vars(self) -> Mapping[str, str | None]:
3836
import_keyring()
3937

4038
prefix = self.env_prefix
4139
if not self.case_sensitive:
42-
prefix = self.env_prefix.lower()
43-
env_vars = {}
40+
prefix = self.env_prefix.lower()
41+
env_vars: dict[str, str | None] = {}
4442
for field in self.settings_cls.model_fields.keys():
4543
if not self.case_sensitive:
4644
field = field.lower()
@@ -54,9 +52,9 @@ def _load_env_vars(self) -> Mapping[str, str | None]:
5452
return env_vars
5553

5654
def __repr__(self) -> str:
57-
return f"{self.__class__.__name__}(keyring_service={self.keyring_service})"
55+
return f'{self.__class__.__name__}(keyring_service={self.keyring_service})'
5856

5957

6058
__all__ = [
61-
"KeyringConfigSettingsSource",
59+
'KeyringConfigSettingsSource',
6260
]

0 commit comments

Comments
 (0)