From 5784b8ab8cbdf87fecdd083e726977aeb8fabfe6 Mon Sep 17 00:00:00 2001
From: Stefano Pigozzi <ste.pigozzi@gmail.com>
Date: Tue, 11 Aug 2015 00:42:45 +0200
Subject: [PATCH] Aggiunto supporto per l'API di osu

---
 royalbot.py | 36 ++++++++++++++++++++++++++++++++++--
 1 file changed, 34 insertions(+), 2 deletions(-)

diff --git a/royalbot.py b/royalbot.py
index a034671e..d162dca1 100644
--- a/royalbot.py
+++ b/royalbot.py
@@ -88,9 +88,20 @@ def getSteamStatus(steamid):
 		'key': steamtoken,
 		'steamids': steamid,
 	}
-	#Manda la richiesta ai server di Telegram e convertila in un dizionario
+	#Manda la richiesta ai server di Steam e convertila in un dizionario
 	r = requests.get("http://api.steampowered.com/ISteamUser/GetPlayerSummaries/v0002/", params=parametri).json()
 	return r
+	
+def getOsuStatus(osuid, mode):
+	#Parametri della richiesta
+	parametri = {
+		'k': osutoken,
+		'u': osuid,
+		'm': mode,
+	}
+	#Manda la richiesta ai server di Osu e convertila in un dizionario
+	r = requests.get("https://osu.ppy.sh/api/get_user", params=parametri).json()
+	return r
 
 #Caricamento delle API Keys
 token = readFile('telegramapi.txt')
@@ -153,4 +164,25 @@ while(True):
 					else:
 						sendMessage(name + " e' " + text + ".", msg['chat']['id'], msg['from']['id'])
 				else:
-					sendMessage(unichr(9888) + " Lo SteamID o l'username non esiste!", msg['chat']['id'], msg['from']['id'])
\ No newline at end of file
+					sendMessage(unichr(9888) + " Lo SteamID o l'username non esiste!", msg['chat']['id'], msg['from']['id'])
+		#Trova i punteggi di una persona su osu!
+		if(msg['text'].startswith("/osu")):
+			if(msg['text'] == "/osu"):
+				sendMessage(unichr(9888) + " Non hai specificato un PlayerID o un username di osu! o Telegram!", msg['chat']['id'], msg['from']['id'])
+			else:
+				#Persona selezionata
+				selezione = msg['text'][5:]
+				#Ricevi i dati di Osu
+				osu = getOsuStatus(selezione, 0)
+				taiko = getOsuStatus(selezione, 1)
+				ctb = getOsuStatus(selezione, 2)
+				osumania = getOsuStatus(selezione, 3)
+				#Trova l'username della persona.
+				name = osu[0]['username']
+				#Trova i pp in ogni modalità
+				osupp = math.floor(osu[0]['pp_raw'])
+				taikopp = math.floor(taiko[0]['pp_raw'])
+				ctbpp = math.floor(ctb[0]['pp_raw'])
+				osumaniapp = math.floor(osumania[0]['pp_raw'])
+				#Manda il messaggio
+				sendMessage(name + " ha:" + unichr(10) + osupp + "pp su Osu!" + unichr(10) + taikopp + "pp su Taiko" + unichr(10) + ctbpp + "pp su Catch the Beat" + unichr(10) + osumaniapp + "pp su Osu!mania")
\ No newline at end of file