Add support for surveys

1 files changed, 95 insertions, 0 deletions

diff --git a/pgweb/survey/models.py b/pgweb/survey/models.py
new file mode 100644
index 00000000..f0cd1608
--- /dev/null
+++ b/pgweb/survey/models.py
@@ -0,0 +1,95 @@
+from django.db import models
+from django.contrib.auth.models import User
+
+from datetime import datetime
+
+# internal text/value object
+class SurveyQuestion(object):
+	def __init__(self, value, text):
+		self.value = value
+		self.text = text
+class SurveyAnswerValues(object):
+	def __init__(self, option, votes, votespercent):
+		self.option = option
+		self.votes = votes
+		self.votespercent = votespercent
+
+class Survey(models.Model):
+	question = models.CharField(max_length=100, null=False, blank=False)
+	opt1 = models.CharField(max_length=100, null=False, blank=False)
+	opt2 = models.CharField(max_length=100, null=False, blank=False)
+	opt3 = models.CharField(max_length=100, null=False, blank=True)
+	opt4 = models.CharField(max_length=100, null=False, blank=True)
+	opt5 = models.CharField(max_length=100, null=False, blank=True)
+	opt6 = models.CharField(max_length=100, null=False, blank=True)
+	opt7 = models.CharField(max_length=100, null=False, blank=True)
+	opt8 = models.CharField(max_length=100, null=False, blank=True)
+	posted = models.DateTimeField(null=False, default=datetime.now)
+	current = models.BooleanField(null=False, default=False)
+
+	def __unicode__(self):
+		return self.question
+
+	@property
+	def questions(self):
+		for i in range (1,9):
+			v = getattr(self, "opt%s" % i)
+			if not v: break
+			yield SurveyQuestion(i, v)
+
+	@property
+	def answers(self):
+		if not hasattr(self, "_answers"):
+			self._answers = SurveyAnswer.objects.get_or_create(survey=self)[0]
+		return self._answers
+
+	@property
+	def completeanswers(self):
+		for a in self._get_complete_answers():
+			yield SurveyAnswerValues(a[0], a[1], self.totalvotes>0 and (100*a[1]/self.totalvotes) or 0)
+
+	@property
+	def totalvotes(self):
+		if not hasattr(self,"_totalvotes"):
+			self._totalvotes = 0
+			for a in self._get_complete_answers():
+				self._totalvotes = self._totalvotes + a[1]
+		return self._totalvotes
+
+	def _get_complete_answers(self):
+		for i in range(1,9):
+			q = getattr(self, "opt%s" % i)
+			if not q: break
+			n = getattr(self.answers, "tot%s" % i)
+			yield (q,n)
+
+	def save(self):
+		# Make sure only one survey at a time can be the current one
+		# (there may be some small race conditions here, but the likelyhood
+		# that two admins are editing the surveys at the same time...)
+		if self.current:
+			previous = Survey.objects.filter(current=True)
+			for p in previous:
+				if not p == self:
+					p.current = False
+					p.save() # primary key check avoids recursion
+
+		# Now that we've made any previously current ones non-current, we are
+		# free to save this one.
+		super(Survey, self).save()
+
+class SurveyAnswer(models.Model):
+	survey = models.ForeignKey(Survey, null=False, blank=False, primary_key=True)
+	tot1 = models.IntegerField(null=False, default=0)
+	tot2 = models.IntegerField(null=False, default=0)
+	tot3 = models.IntegerField(null=False, default=0)
+	tot4 = models.IntegerField(null=False, default=0)
+	tot5 = models.IntegerField(null=False, default=0)
+	tot6 = models.IntegerField(null=False, default=0)
+	tot7 = models.IntegerField(null=False, default=0)
+	tot8 = models.IntegerField(null=False, default=0)
+
+class SurveyLock(models.Model):
+	ipaddr = models.IPAddressField(null=False, blank=False)
+	time = models.DateTimeField(null=False, default=datetime.now)
+