1. Mikhail Korobov
  2. django-colour-field


django-colour-field / colour_field / fields.py

from django.db import models
from forms import ColourFormField
import re

class ColourField(models.Field):
    Store a colour as a CSS value (ie, #012345)
    We don't subclass models.CharField, as then the django admin would use
    it's AdminCharWidget or whatever, and we want to use our own.
    __metaclass__ = models.SubfieldBase
    _south_introspects = True
    description = "A colour object"
    def to_python(self, value):
        # assert value[0] == "#" 
        # assert re.match('#[0-9a-fA-F]{6}', value)
        return value
    def get_prep_value(self, value):
        if not value.strip() and self.null:
            return None
        return value
    def value_to_string(self, value):
        return value.strip() or None
    def formfield(self, *args, **kwargs):
        defaults = {'form_class': ColourFormField}
        return super(ColourField, self).formfield(*args, **defaults)
    def db_type(self, connection):
        return 'char(7)'