Added basic support to polls (only voting and seeing results is implemented)

src/presenters/postDisplayer.py

@@ -639,24 +639,54 @@ class displayTopicPresenter(displayPostPresenter):
 
 class displayPollPresenter(base.basePresenter):
 
-	def __init__(self, session, poll, view, interactor):
+	def __init__(self, session, poll, view, interactor, show_results=False):
 		super(displayPollPresenter, self).__init__(view=view, interactor=interactor, modulename="display_poll")
 		self.poll = poll["poll"]
 		self.session = session
-		self.load_poll()
+		self.get_poll()
+		self.load_poll(show_results)
 		self.run()
 
-	def load_poll(self):
+	def get_poll(self):
+		# Retrieve the poll again so we will have a fresh and updated object.
+		data = dict(owner_id=self.poll["owner_id"], is_board=int(self.poll["is_board"]), poll_id=self.poll["id"])
+		self.poll = self.session.vk.client.polls.getById(**data)
+
+	def load_poll(self, load_results=False):
 		user = self.session.get_user(self.poll["author_id"])
 		title = _("Poll from {user1_nom}").format(**user)
 		self.send_message("set_title", value=title)
 		self.send_message("set", control="question", value=self.poll["question"])
-		options = []
-		for i in self.poll["answers"]:
-			options.append(i["text"])
-		self.send_message("add_options", options=options, multiple=self.poll["multiple"])
-		print(self.poll)
-		self.send_message("done")
+		if len(self.poll["answer_ids"]) > 0 or ("is_closed" in self.poll and self.poll["is_closed"] == True) or load_results == True:
+			options = []
+			for i in self.poll["answers"]:
+				options.append((i["text"], i["votes"], i["rate"]))
+			self.send_message("add_options", options=options, multiple=self.poll["multiple"])
+			self.send_message("done")
+			self.send_message("disable_control", control="ok")
+		else:
+			options = []
+			for i in self.poll["answers"]:
+				options.append(i["text"])
+			self.send_message("add_options", options=options, multiple=self.poll["multiple"])
+			self.send_message("done")
+
+	def vote(self, answers):
+		ids = ""
+		for i in range(0, len(self.poll["answers"])):
+			if answers[i] == True:
+				ids = ids+"{answer_id},".format(answer_id=self.poll["answers"][i]["id"])
+				if self.poll["multiple"] == False:
+					break
+		if ids == "":
+			log.exception("An error occurred when retrieving answer IDS for the following poll: %r. Provided answer list: %r" % (self.poll, answers))
+			return
+		data = dict(owner_id=self.poll["owner_id"], poll_id=self.poll["id"], answer_ids=ids, is_board=int(self.poll["is_board"]))
+		result = self.session.vk.client.polls.addVote(**data)
+		if result == 1:
+			output.speak(_("Your vote has been added to this poll."))
+
+
 
 class displayAudioPresenter(base.basePresenter):
 	def __init__(self, session, postObject, view, interactor):