from django.db import models


class IntegerChoicesField(models.IntegerField):

    def from_db_value(self, value, expressin, connection):
        if value is None:
            return value
        return int(value)

    def to_python(self, value):
        if value is None:
            return value
        return super().to_python(value)