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 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 = models.IntegerField(null=False, blank=False, default=0, verbose_name="7.4", choices=choices)
	v80 = models.IntegerField(null=False, blank=False, default=0, verbose_name="8.0", choices=choices)
	v81 = models.IntegerField(null=False, blank=False, default=0, verbose_name="8.1", choices=choices)
	v82 = models.IntegerField(null=False, blank=False, default=0, verbose_name="8.2", choices=choices)
	v83 = models.IntegerField(null=False, blank=False, default=0, verbose_name="8.3", choices=choices)
	v84 = models.IntegerField(null=False, blank=False, default=0, verbose_name="8.4", choices=choices)
	v90 = models.IntegerField(null=False, blank=False, default=0, verbose_name="9.0", choices=choices)
	v91 = models.IntegerField(null=False, blank=False, default=0, verbose_name="9.1", choices=choices)

	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')]