|
| 1 | +from django.conf import settings |
| 2 | +from django.test.signals import setting_changed |
| 3 | + |
| 4 | +USER_SETTINGS = getattr(settings, "REST_FRAMEWORK_MICROSERVICE", None) |
| 5 | + |
| 6 | +DEFAULTS = { |
| 7 | + "TOKEN_CUSTOM_USER_CLAIMS": [], |
| 8 | + |
| 9 | +} |
| 10 | + |
| 11 | + |
| 12 | +class RestFrameworkMicroserviceSettings: |
| 13 | + def __init__(self, user_settings=None, defaults=None): |
| 14 | + self._user_settings = user_settings or {} |
| 15 | + self.defaults = defaults or DEFAULTS |
| 16 | + self._cached_attrs = set() |
| 17 | + |
| 18 | + @property |
| 19 | + def user_settings(self): |
| 20 | + if not hasattr(self, "_user_settings"): |
| 21 | + self._user_settings = getattr(settings, "REST_FRAMEWORK_MICROSERVICE", {}) |
| 22 | + return self._user_settings |
| 23 | + |
| 24 | + def __getattr__(self, attr): |
| 25 | + |
| 26 | + # check the setting is accepted |
| 27 | + if attr not in self.defaults: |
| 28 | + raise AttributeError(f"Invalid REST_FRAMEWORK_MICROSERVICE setting: {attr}") |
| 29 | + |
| 30 | + # get from user settings or default value |
| 31 | + try: |
| 32 | + val = self.user_settings[attr] |
| 33 | + except KeyError: |
| 34 | + val = self.defaults[attr] |
| 35 | + |
| 36 | + self._cached_attrs.add(attr) |
| 37 | + setattr(self, attr, val) |
| 38 | + return val |
| 39 | + |
| 40 | + def reload(self): |
| 41 | + for attr in self._cached_attrs: |
| 42 | + delattr(self, attr) |
| 43 | + self._cached_attrs.clear() |
| 44 | + if hasattr(self, "_user_settings"): |
| 45 | + delattr(self, "_user_settings") |
| 46 | + |
| 47 | + |
| 48 | +drf_stripe_settings = DrfStripeSettings(USER_SETTINGS, DEFAULTS) |
| 49 | + |
| 50 | + |
| 51 | +def reload_rest_framework_microservice_settings(*args, **kwargs): |
| 52 | + print("Reloading rest_framework_microservice settings") |
| 53 | + setting = kwargs["setting"] |
| 54 | + if setting == "REST_FRAMEWORK_MICROSERVICE": |
| 55 | + drf_stripe_settings.reload() |
| 56 | + |
| 57 | + |
| 58 | +setting_changed.connect(reload_rest_framework_microservice_settings) |
0 commit comments