Merge pull request #5960 from Laerte/master

chore: Implement `pop` method on `BaseSettings` class
This commit is contained in:
Andrey Rakhmatullin 2023-06-24 18:23:17 +04:00 committed by GitHub
commit 85f5122965
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 43 additions and 0 deletions

View File

@ -75,6 +75,8 @@ class BaseSettings(MutableMapping):
highest priority will be retrieved.
"""
__default = object()
def __init__(self, values=None, priority="project"):
self.frozen = False
self.attributes = {}
@ -445,6 +447,18 @@ class BaseSettings(MutableMapping):
else:
p.text(pformat(self.copy_to_dict()))
def pop(self, name, default=__default):
try:
value = self.attributes[name]
except KeyError:
if default is self.__default:
raise
return SettingsAttribute(default, get_settings_priority("project"))
else:
self.__delitem__(name)
return value
class Settings(BaseSettings):
"""

View File

@ -451,6 +451,35 @@ class SettingsTest(unittest.TestCase):
self.assertIsInstance(myhandler_instance, FileDownloadHandler)
self.assertTrue(hasattr(myhandler_instance, "download_request"))
def test_pop_item_with_default_value(self):
settings = Settings()
with self.assertRaises(KeyError):
settings.pop("DUMMY_CONFIG")
dummy_config = settings.pop("DUMMY_CONFIG", "dummy_value")
self.assertEqual(
repr(dummy_config), "<SettingsAttribute value='dummy_value' priority=20>"
)
self.assertEqual(dummy_config.value, "dummy_value")
def test_pop_item_with_immutable_settings(self):
settings = Settings(
{"DUMMY_CONFIG": "dummy_value", "OTHER_DUMMY_CONFIG": "other_dummy_value"}
)
self.assertEqual(settings.pop("DUMMY_CONFIG").value, "dummy_value")
settings.freeze()
with self.assertRaises(TypeError) as error:
settings.pop("OTHER_DUMMY_CONFIG")
self.assertEqual(
str(error.exception), "Trying to modify an immutable Settings object"
)
if __name__ == "__main__":
unittest.main()