Files
postgres-web/pgweb/featurematrix/models.py
2013-05-26 12:34:47 -04:00

66 lines
2.3 KiB
Python

from django.db import models
from pgweb.util.bases import PgModel
choices_map = {
0: {'str': 'No', 'class': 'no', 'bgcolor': '#ffdddd'},
1: {'str': 'Yes', 'class': 'yes', 'bgcolor': '#ddffdd'},
2: {'str': 'Obsolete', 'class': 'obs', 'bgcolor': '#ddddff'},
3: {'str': '?', 'class': 'unk', 'bgcolor': '#ffffaa'},
}
choices = [(k, v['str']) for k,v in choices_map.items()]
class FeatureGroup(PgModel, models.Model):
groupname = models.CharField(max_length=100, null=False, blank=False)
groupsort = models.IntegerField(null=False, blank=False)
purge_urls = ('/about/featurematrix/', )
def __unicode__(self):
return self.groupname
@property
def columns(self):
# Return a list of all the columns for the matrix
return [b for a,b in versions]
class FeatureMatrixField(models.IntegerField):
def __init__(self, verbose_name, visible_default=True):
super(FeatureMatrixField, self).__init__(null=False, blank=False, default=0, verbose_name=verbose_name, choices=choices)
self.visible_default = visible_default
class Feature(PgModel, models.Model):
group = models.ForeignKey(FeatureGroup, null=False, blank=False)
featurename = models.CharField(max_length=100, null=False, blank=False)
featuredescription = models.TextField(null=False, blank=True)
#WARNING! All fields that start with "v" will be considered versions!
v74 = FeatureMatrixField(verbose_name="7.4", visible_default=False)
v80 = FeatureMatrixField(verbose_name="8.0")
v81 = FeatureMatrixField(verbose_name="8.1")
v82 = FeatureMatrixField(verbose_name="8.2")
v83 = FeatureMatrixField(verbose_name="8.3")
v84 = FeatureMatrixField(verbose_name="8.4")
v90 = FeatureMatrixField(verbose_name="9.0")
v91 = FeatureMatrixField(verbose_name="9.1")
v92 = FeatureMatrixField(verbose_name="9.2")
v93 = FeatureMatrixField(verbose_name="9.3")
purge_urls = ('/about/featurematrix/.*', )
def __unicode__(self):
# To make it look good in the admin interface, just don't render it
return ''
def columns(self):
return [choices_map[getattr(self, a)] for a,b in versions]
@property
def featurelink(self):
if self.featuredescription.startswith('http://'):
return self.featuredescription
else:
return 'detail/%s/' % self.id
versions = [(f.name,f.verbose_name) for f in Feature()._meta.fields if f.name.startswith('v') and f.visible_default]