LocalizedIntegerField

Complexity

Total Complexity

14

Size/Duplication

Total Lines	59
Duplicated Lines	0 %

Test Coverage

Coverage

50%

Importance

Changes	1
Bugs	0	Features	0

4 Methods

Rating	Name	Duplication	Size	Complexity
A	from_db_value()	0	13	3
B	_convert_localized_value()	0	16	5
B	get_prep_value()	0	17	5
A	to_python()	0	5	1

from typing import Optional, Union, Dict

The import typing could not be resolved.

from django.conf import settings

The import django.conf could not be resolved.

from django.db.utils import IntegrityError

The import django.db.utils could not be resolved.

from .field import LocalizedField
from ..value import LocalizedValue, LocalizedIntegerValue


class LocalizedIntegerField(LocalizedField):
    """Stores integers as a localized value."""

    attr_class = LocalizedIntegerValue

    @classmethod
    def from_db_value(cls, value, *_) -> Optional[LocalizedIntegerValue]:
        db_value = super().from_db_value(value)
        if db_value is None:
            return db_value

        # if we were used in an expression somehow then it might be
        # that we're returning an individual value or an array, so
        # we should not convert that into an :see:LocalizedIntegerValue
        if not isinstance(db_value, LocalizedValue):
            return db_value

        return cls._convert_localized_value(db_value)

    def to_python(self, value: Union[Dict[str, int], int, None]) -> LocalizedIntegerValue:
        """Converts the value from a database value into a Python value."""

        db_value = super().to_python(value)
        return self._convert_localized_value(db_value)

    def get_prep_value(self, value: LocalizedValue) -> dict:
        """Gets the value in a format to store into the database."""

        prepped_value = super().get_prep_value(value)
        if prepped_value is None:
            return None

        # make sure all values are proper integers
        for lang_code, _ in settings.LANGUAGES:
            try:
                if prepped_value[lang_code] is not None:
                    int(prepped_value[lang_code])
            except (TypeError, ValueError):
                raise IntegrityError('non-integer value in column "%s.%s" violates '
                                     'integer constraint' % (self.name, lang_code))

The Instance of LocalizedIntegerField does not seem to have a member named name.

        return prepped_value

    @staticmethod
    def _convert_localized_value(value: LocalizedValue) -> LocalizedIntegerValue:
        """Converts from :see:LocalizedValue to :see:LocalizedIntegerValue."""

        integer_values = {}
        for lang_code, _ in settings.LANGUAGES:
            local_value = value.get(lang_code, None)
            if local_value is None or local_value.strip() == '':
                local_value = None

            try:
                integer_values[lang_code] = int(local_value)
            except (ValueError, TypeError):
                integer_values[lang_code] = None

        return LocalizedIntegerValue(integer_values)