93 lines
3.0 KiB
Python
93 lines
3.0 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Copyright (c) The python-semanticversion project
|
|
# This code is distributed under the two-clause BSD License.
|
|
|
|
from __future__ import unicode_literals
|
|
|
|
import django
|
|
from django.db import models
|
|
from django.utils.translation import ugettext_lazy as _
|
|
|
|
from . import base
|
|
|
|
|
|
class SemVerField(models.CharField):
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
kwargs.setdefault('max_length', 200)
|
|
super(SemVerField, self).__init__(*args, **kwargs)
|
|
|
|
if django.VERSION[:2] < (1, 8):
|
|
def contribute_to_class(self, cls, name, **kwargs):
|
|
"""Emulate SubFieldBase for Django < 1.8"""
|
|
super(SemVerField, self).contribute_to_class(cls, name, **kwargs)
|
|
from django.db.models.fields import subclassing
|
|
setattr(cls, self.name, subclassing.Creator(self))
|
|
|
|
def from_db_value(self, value, expression, connection, context):
|
|
"""Convert from the database format.
|
|
|
|
This should be the inverse of self.get_prep_value()
|
|
"""
|
|
return self.to_python(value)
|
|
|
|
def get_prep_value(self, obj):
|
|
return None if obj is None else str(obj)
|
|
|
|
def get_db_prep_value(self, value, connection, prepared=False):
|
|
if not prepared:
|
|
value = self.get_prep_value(value)
|
|
return value
|
|
|
|
def value_to_string(self, obj):
|
|
value = self.to_python(self._get_val_from_obj(obj))
|
|
return str(value)
|
|
|
|
def run_validators(self, value):
|
|
return super(SemVerField, self).run_validators(str(value))
|
|
|
|
|
|
class VersionField(SemVerField):
|
|
default_error_messages = {
|
|
'invalid': _("Enter a valid version number in X.Y.Z format."),
|
|
}
|
|
description = _("Version")
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
self.partial = kwargs.pop('partial', False)
|
|
self.coerce = kwargs.pop('coerce', False)
|
|
super(VersionField, self).__init__(*args, **kwargs)
|
|
|
|
def deconstruct(self):
|
|
"""Handle django.db.migrations."""
|
|
name, path, args, kwargs = super(VersionField, self).deconstruct()
|
|
kwargs['partial'] = self.partial
|
|
kwargs['coerce'] = self.coerce
|
|
return name, path, args, kwargs
|
|
|
|
def to_python(self, value):
|
|
"""Converts any value to a base.Version field."""
|
|
if value is None or value == '':
|
|
return value
|
|
if isinstance(value, base.Version):
|
|
return value
|
|
if self.coerce:
|
|
return base.Version.coerce(value, partial=self.partial)
|
|
else:
|
|
return base.Version(value, partial=self.partial)
|
|
|
|
|
|
class SpecField(SemVerField):
|
|
default_error_messages = {
|
|
'invalid': _("Enter a valid version number spec list in ==X.Y.Z,>=A.B.C format."),
|
|
}
|
|
description = _("Version specification list")
|
|
|
|
def to_python(self, value):
|
|
"""Converts any value to a base.Spec field."""
|
|
if value is None or value == '':
|
|
return value
|
|
if isinstance(value, base.Spec):
|
|
return value
|
|
return base.Spec(value)
|