From ee2ab64614d4e0f502212df247cb8085fdf75905 Mon Sep 17 00:00:00 2001
From: Stefano Pigozzi <ste.pigozzi@gmail.com>
Date: Thu, 21 Jan 2016 22:16:47 +0100
Subject: [PATCH] =?UTF-8?q?Aggiunta=20la=20modalit=C3=A0=20votazione.=20At?=
 =?UTF-8?q?tivata=20con=20/nuovavotazione?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bot.py | 104 ++++++++++++++++++++++++++++++++++++++++++++++++++++++---
 1 file changed, 99 insertions(+), 5 deletions(-)

diff --git a/bot.py b/bot.py
index 093d0a27..4e4d4520 100644
--- a/bot.py
+++ b/bot.py
@@ -6,11 +6,85 @@ import osu
 import hearthstone
 import sys
 
+
+# Check per la modalità votazione del bot, corrisponde al numero della chat in cui è attiva la votazione
+# 0 per disattivare la votazione
+class Votazione:
+    chat = int()
+    domanda = str()
+    # 0: non votato
+    # 1: sì
+    # 2: no
+    # 3: astenuto
+    voto = {
+        'steffo': int(0),
+        'alby1': int(0),
+        'boni3099': int(0),
+        'maxsensei': int(0),
+        'cosimo03': int(0),
+        'frankrekt': int(0),
+        'heisendoc': int(0),
+        'acterryg': int(0),
+        'adry99': int(0),
+        'alleanderl': int(0),
+        'thevagginadestroyer': int(0),
+        'tiztiztiz': int(0),
+        'fultz': int(0),
+        'gotob': int(0),
+        'enribenassati': int(0),
+        'iemax': int(0),
+        'peraemela99': int(0),
+        'ilgattopardo': int(0),
+        'mrdima98': int(0),
+        'ruozir': int(0),
+        'supersmurf': int(0),
+        'tauei': int(0),
+        'voltaggio': int(0),
+    }
+
+    def __init__(self, question, askin):
+        self.domanda = question
+        self.chat = askin
+
+    def ask(self):
+        telegram.sendmessage(self.domanda, self.chat)
+
+    def register(self, utente, voto):
+        self.voto[utente] = voto
+        telegram.sendmessage("Votazione registrata!", self.chat)
+
+    def showresults(self):
+        lista = str()
+        si = 0
+        no = 0
+        astenuti = 0
+        for membro in self.voto:
+            if self.voto[membro] == 0:
+                lista += chr(9898)
+            elif self.voto[membro] == 1:
+                si += 1
+                lista += chr(128309)
+            elif self.voto[membro] == 2:
+                no += 1
+                lista += chr(128308)
+            elif self.voto[membro] == 3:
+                astenuti += 1
+                lista += chr(9899)
+            lista += " @" + membro + "\n"
+        telegram.sendmessage(self.domanda + "\n"
+                             "*Risultati:*\n"
+                             "Sì: " + str(si) + " (" + str(round(si / (si + no + astenuti) * 100, 2)) + "%)\n"
+                             "No: " + str(no) + " (" + str(round(no / (si + no + astenuti) * 100, 2)) + "%)\n"
+                             "Astenuti: " + str(astenuti) + "\n\n" + lista, self.chat)
+
+# Votazione in corso
+incorso = None
+
 # Playlist di /rage, si riempie quando è vuota
 rage = []
 
 # TODO: Rimettere gli audio di Wololo
-wololo = []
+# wololo = []
 
 # Dizionario con i nomi utenti di osu!
 # Se qualcuno cambia nome utente di Telegram, lo cambi anche QUI.
@@ -342,8 +416,28 @@ while True:
             else:
                 telegram.sendmessage(tosend, sentin)
         elif text.startswith('/shrekt'):
+            print("@" + username + ": /shrekt ")
             telegram.senddocument("BQADBAADsQADiBjiAqYN-EBXASyhAg", sentin)
-        elif text.startswith('/restart'):
-            if username == "Steffo":
-                telegram.sendmessage("Riavvio accettato.", sentin)
-                sys.exit(0)
+        elif text.startswith('/restart') and username == "Steffo":
+            print("@" + username + ": /restart ")
+            telegram.sendmessage("Riavvio accettato.", sentin)
+            sys.exit(0)
+        elif text.startswith('/nuovavotazione') and username == "Steffo":
+            print("@" + username + ": /nuovavotazione ")
+            cmd = text.split(" ", 1)
+            incorso = Votazione(cmd[1], sentin)
+        elif text.startswith('/si') and incorso is not None:
+            print("@" + username + ": /si ")
+            incorso.register(username.lower(), 1)
+        elif text.startswith('/no') and incorso is not None:
+            print("@" + username + ": /no ")
+            incorso.register(username.lower(), 2)
+        elif text.startswith('/astieniti') and incorso is not None:
+            print("@" + username + ": /astieniti ")
+            incorso.register(username.lower(), 3)
+        elif text.startswith('/domanda') and incorso is not None:
+            print("@" + username + ": /domanda ")
+            incorso.ask()
+        elif text.startswith('/risultati') and incorso is not None:
+            print("@" + username + ": /risultati ")
+            incorso.showresults()