import json import uuid from datetime import datetime from django.db import models from django.utils.translation import gettext_lazy as _ import logging logger = logging.getLogger(__name__) class BaseModel(models.Model): id = None uuid = models.UUIDField(primary_key=True, editable=False, default=uuid.uuid4, unique=True, db_index=True) created_at = models.DateTimeField(auto_now_add=True, db_index=True) updated_at = models.DateTimeField(auto_now=True, db_index=True) class Meta: abstract = True class ConfigValueChoices(models.TextChoices): INT = 'INT', _('Integer') FLOAT = 'FLOAT', _('Float') BOOLEAN = 'BOOLEAN', _('Boolean') STRING = 'STRING', _('String') DATE = 'DATE', _('Date') DATETIME = 'DATETIME', _('DateTime') JSON = 'JSON', _('JSON') class Config(BaseModel): value_dict = {} key = models.CharField(_('key'), max_length=255, unique=True, db_index=True) value = models.TextField(_('value'), null=True, blank=True) value_type = models.CharField(choices=ConfigValueChoices.choices, max_length=64) description = models.TextField(_('description'), null=True, blank=True) comment = models.TextField(_('comment'), null=True, blank=True) def get_value(self): return Config.get_value_of(self.key) @staticmethod def type_cast(value, value_type): try: if value_type == ConfigValueChoices.STRING.value: value = str(value) elif value_type == ConfigValueChoices.BOOLEAN.value: value = value.lower() == 'true' elif value_type == ConfigValueChoices.DATE.value: value = datetime.fromisoformat(value) elif value_type == ConfigValueChoices.DATETIME.value: value = datetime.fromisoformat(value) elif value_type == ConfigValueChoices.JSON.value: value = json.loads(value) elif value_type == ConfigValueChoices.INT.value: value = int(value) elif value_type == ConfigValueChoices.FLOAT.value: value = float(value) except Exception as e: value = None return value # TODO: revoke cache mechanism @staticmethod def get_value_of(key, default=None, casting_type=None): if not key in Config.value_dict: config, created = Config.objects.get_or_create(key=key, defaults={'value': default}) Config.value_dict[key] = Config.type_cast(config.value, casting_type or config.value_type) logger.info(f'getting value of {key} from db: {Config.value_dict[key]}') value = Config.value_dict[key] return value