1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
|
# -*- coding: utf-8 -*-
# Copyright (c) 2012-2014 The python-semanticversion project
# This code is distributed under the two-clause BSD License.
from __future__ import unicode_literals
from django.db import models
from django.utils.translation import ugettext_lazy as _
from . import base
class BaseSemVerField(models.CharField):
__metaclass__ = models.SubfieldBase
def __init__(self, *args, **kwargs):
kwargs.setdefault('max_length', 200)
super(BaseSemVerField, self).__init__(*args, **kwargs)
def get_prep_value(self, obj):
return 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(BaseSemVerField, self).run_validators(str(value))
# Py2 and Py3-compatible metaclass
SemVerField = models.SubfieldBase(
str('SemVerField'), (BaseSemVerField, models.CharField), {})
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 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)
def add_south_rules():
from south.modelsinspector import add_introspection_rules
add_introspection_rules([
(
(VersionField,),
[],
{
'partial': ('partial', {'default': False}),
'coerce': ('coerce', {'default': False}),
},
),
], ["semantic_version\.django_fields"])
try: # pragma: no cover
import south
except ImportError: # pragma: no cover
south = None
if south: # pragma: no cover
add_south_rules()
|