class Poll:
def __init__(self, pid = -1):
self.dbPoll = DbPoll()
self.dbAnswer = DbAnswer()
self.dbResponse = DbResponse()
self.pid = pid
self.title = ""
self.desc = ""
if pid != -1:
if self.dbPoll.pollInDb(pid):
pInfo = self.dbPoll.getPollInfo(pid)
self.title = pInfo["title"]
self.desc = pInfo["description"]
self.pid = pInfo["id"]
def pollExists(self, pid):
return self.dbPoll.pollInDb(pid)
#???
def pollAnswers(self):
answers = self.dbAnswer.getAnswers(self.pid)
return answers
def pollInfo(self):
pInfo = self.dbPoll.getPollInfo(self.pid)
answers = self.dbAnswer.getAnswers(self.pid)
for a in answers:
del a["fk_poll"]
pInfo["answers"] = answers
return pInfo
def formated(self):
if not self.pollExists(self.pid):
return {"status":"error", "error_type":"no such poll"}
pi = self.pollInfo()
pi["status"] = "ok"
return pi
def validAnswer(self, answerId):
return self.dbAnswer.pollHasAnswer(answerId, self.pid)
def hasUserAnswer(self, userId):
return self.dbResponse.userResponseInDb(self.pid, userId)
def answer(self, userId, answerId):
if not self.hasUserAnswer(userId):
if self.validAnswer(answerId):
print("answering poll...")
self.dbResponse.putResponse(self.pid, userId, answerId)
def getResults(self):
answers = list(self.pollAnswers())
responses = list(self.dbResponse.responseForPoll(self.pid))
for a in answers:
a["num_resp"] = 0
for r in responses:
if r["fk_answer"] == a["id"]:
a["num_resp"] = a["num_resp"] + 1
return answers
@classmethod
def getPollList(self):
tmpDbPoll = DbPoll()
return tmpDbPoll.getPolls()
@classmethod
def getUnansweredPollList(self, uid):
tmpDbPoll = DbPoll()
return tmpDbPoll.getUnansweredPolls(uid)
@classmethod
def getAnsweredPollList(self, uid):
tmpDbPoll = DbPoll()
return tmpDbPoll.getAnsweredPolls(uid)