1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00
royalnet/bot.py

547 lines
24 KiB
Python
Raw Normal View History

2015-10-07 08:49:14 +00:00
# -*- coding: utf-8 -*-
import json
2016-02-01 16:31:03 +00:00
import time
import filemanager
import telegram
import steam
2015-10-08 16:39:56 +00:00
import random
2015-11-16 21:55:47 +00:00
import osu
import lol
import discord
2016-02-14 15:19:26 +00:00
2016-02-01 16:31:03 +00:00
# Elenco di username dei membri della RYG
royalgames = json.loads(filemanager.readfile("db.json"))
2016-01-21 19:47:25 +00:00
2016-03-04 13:40:19 +00:00
# Stringa dove mettere l'elenco di champion di lol gratuiti
lolfreestring = str()
2015-11-23 16:18:08 +00:00
random.seed()
2016-08-24 06:40:21 +00:00
# Spostiamo qui le funzioni del bot, così è un po' più leggibile
def wow():
print("@" + username + ": WOW!")
telegram.sendmessage("Wow. Impressionante.", sentin, source)
def ahnonlosoio():
print("@" + username + ": /ahnonlosoio")
# Rispondi con Ah, non lo so nemmeno io.
telegram.sendmessage("Ah, non lo so nemmeno io!", sentin, source)
def ciaostefanino():
print("@" + username + ": /ciaostefanino")
# Rispondi salutando Stefanino.
telegram.sendmessage("Ciao Stefanino!", sentin, source)
def ciaoruozi():
print("@" + username + ": /ciaoruozi")
# Ciao Ruozi.
if username.lower() == "ruozir":
2016-09-21 13:35:07 +00:00
# Manda un messaggio casuale tra quelli nella lista
chosen_msg = random.sample(["Ciao me!",
"Ciao ciao ciao!",
"1 ciaoruozi = 1000 euro per me",
"La mi dico: #CiaoRuozi",
"Ciao eh me stesso!"], 1)
telegram.sendmessage(chosen_msg, sentin, source)
2016-08-24 06:40:21 +00:00
else:
2016-09-21 13:35:07 +00:00
# Manda un messaggio casuale tra quelli nella lista
chosen_msg = random.sample(["Ciao Ruozi!",
"Ciao ciao ciao!",
"1 ciaoruozi = 1 prayer",
"La RYG dice: #CiaoRuozi",
"Ciao eh Ruozi!"], 1)
telegram.sendmessage(chosen_msg, sentin, source)
2016-08-24 06:40:21 +00:00
2016-09-07 15:46:36 +00:00
def ombromanto():
print("@" + username + ": /ombromanto")
# Ma chi è ombromanto?
2016-09-21 13:36:44 +00:00
telegram.sendmessage("Ombromanto è @Dailir!", sentin, source)
2016-09-07 15:46:36 +00:00
def steamplayers():
print("@" + username + ": /steamplayers")
2016-08-24 06:40:21 +00:00
# Informa Telegram che il messaggio è stato ricevuto e visualizza Royal Bot sta scrivendo.
telegram.sendchataction(sentin)
cmd = text.split(" ")
# Se è stato specificato un AppID...
if len(cmd) >= 2:
n = steam.getnumberofcurrentplayers(cmd[1])
# Se viene ricevuta una risposta...
if n is None:
telegram.sendmessage(chr(9888) + " L'app specificata non esiste!", sentin, source)
else:
name = steam.getschemaforgame(cmd[1])['game']['gameName']
telegram.sendmessage("In questo momento, *{n}* persone stanno giocando a "
"[{name}](https://steamdb.info/app/{id}/graphs/)."
.format(n=str(n), name=name, id=cmd[1]), sentin, source)
2016-08-24 06:40:21 +00:00
else:
telegram.sendmessage(chr(9888) + ' Non hai specificato un AppID!\n'
'La sintassi corretta è /playing <AppID>.', sentin, source)
def ehoh():
print("@" + username + ": /ehoh")
# Rispondi con Eh, oh. Sono cose che capitano.
2016-09-21 13:57:15 +00:00
telegram.sendmessage("Eh, oh. Sono cose che capitano. ¯\_(ツ)_/¯", sentin, source)
2016-08-24 06:40:21 +00:00
def sbam():
print("@" + username + ": /sbam")
# Manda l'audio contenente gli sbam di tutti i membri Royal Games.
telegram.senddocument('BQADAgADBwMAAh8GgAGSsR4rwmk_LwI', sentin)
def osucmd():
print("@" + username + ": /osu")
# Visualizza il punteggio più recente di osu!
# Informa Telegram che il messaggio è stato ricevuto.
telegram.sendchataction(sentin)
# Trova il nome utente specificato
cmd = text.split(' ', 1)
# Se è stato specificato un nome utente
if len(cmd) >= 2:
# Trova la modalità
# 0 = osu!
# 1 = osu!taiko
# 2 = osu!catch
# 3 = osu!mania
cmd = text.split(' ', 2)
# Se è stata specificata una modalità
if len(cmd) >= 3:
# Modalità specificata
mode = int(cmd[2])
else:
# Imposta la modalità a osu!
mode = 0
# Prova a mandare una richiesta ai server di osu per l'ultima canzone giocata
try:
r = osu.getuserrecent(cmd[1], mode)
# Se la funzione restituisce un errore, riferisci su Telegram l'errore e previeni il crash.
except NameError:
telegram.sendmessage(chr(9888) + " Errore nella richiesta ai server di Osu!", sentin,
source)
# Se tutto va bene, continua!
else:
# Se ci sono delle mod attive...
if "enabled_mods" in r:
mods = osu.listmods(r['enabled_mods'])
else:
mods = ""
# Specifica cosa vuole dire il grado F e il grado X
if r['rank'] == 'F':
r['rank'] = 'Failed'
elif r['rank'] == 'X':
r['rank'] = 'Unranked'
if mode == 0:
# Visualizza le informazioni relative alla modalità osu!
telegram.sendmessage("*osu!*\n"
"[Beatmap {0}](https://osu.ppy.sh/b/{0})\n"
"*{1}*\n"
"{2}\n"
"*Punti*: {3}\n"
"*Combo* x{4}\n"
"*300*: {5}\n"
"*100*: {6}\n"
"*50*: {7}\n"
"*Awesome*: {8}\n"
"*Good*: {9}\n"
"*Miss*: {10}"
.format(r['beatmap_id'],
r['rank'],
mods,
r['score'],
r['maxcombo'],
r['count300'],
r['count100'],
r['count50'],
r['countgeki'],
r['countkatu'],
r['countmiss']), sentin, source)
elif mode == 1:
# Visualizza le informazioni relative alla modalità osu!taiko
telegram.sendmessage("*osu!taiko*\n"
"[Beatmap {0}](https://osu.ppy.sh/b/{0})\n"
"*{1}*\n"
"{2}\n"
"*Punti*: {3}\n"
"*Combo* x{4}\n"
"*Great*: {5}\n"
"*Good*: {6}\n"
"_Large_ *Great*: {7}\n"
"_Large_ *Good*: {8}\n"
"*Miss*: {9}"
.format(r['beatmap_id'],
r['rank'],
mods,
r['score'],
r['maxcombo'],
r['count300'],
r['count100'],
r['countgeki'],
r['countkatu'],
r['countmiss']), sentin, source)
elif mode == 2:
# TODO: Cos'è successo qui?
# Visualizza le informazioni relative alla modalità osu!catch
telegram.sendmessage("*osu!catch*\n"
"[Beatmap " + r['beatmap_id'] + "](" + 'https://osu.ppy.sh/b/' + r[
'beatmap_id'] +
")\n*" + r['rank'] + "*\n" + mods +
"\n*Punti*: " + r['score'] + "\n"
"*Combo* x" + r['maxcombo'] + "\n"
"*Fruit*: " +
r['count300'] + "\n"
"*Droplet* _tick_: " + r['count100'] + "\n"
"*Droplet* _trail_: " +
r['count50'] + "\n"
"*Miss*: " + r['countmiss'], sentin, source)
elif mode == 3:
# TODO: Cos'è successo qui?
# Visualizza le informazioni relative alla modalità osu!mania
telegram.sendmessage("*osu!mania*\n" +
"[Beatmap " + r['beatmap_id'] + "](" + 'https://osu.ppy.sh/b/' + r[
'beatmap_id'] + ")\n*" + r['rank'] + "*\n" + mods +
"\n*Punti*: " + r['score'] + "\n"
"*Combo* x" + r['maxcombo'] + "\n"
"_Rainbow_ *300*: " +
r['countgeki'] + "\n"
"*300*: " + r['count300'] + "\n"
"*100*: " + r[
'count100'] + "\n"
"*200*: " + r['countkatu'] + "\n"
"*50*: " + r[
'count50'] + "\n"
"*Miss*: " + r['countmiss'], sentin, source)
else:
# TODO: Mettere a posto sto schifo.
if "osu" in royalgames[username.lower()]:
r = osu.getuserrecent(royalgames[username.lower()]['osu'], 0)
if "enabled_mods" in r:
mods = osu.listmods(r['enabled_mods'])
else:
mods = ""
telegram.sendmessage("*osu!*\n"
"[Beatmap {0}](https://osu.ppy.sh/b/{0})\n"
"*{1}*\n"
"{2}\n"
"*Punti*: {3}\n"
"*Combo* x{4}\n"
"*300*: {5}\n"
"*100*: {6}\n"
"*50*: {7}\n"
"*Awesome*: {8}\n"
"*Good*: {9}\n"
"*Miss*: {10}"
.format(r['beatmap_id'],
r['rank'],
mods,
r['score'],
r['maxcombo'],
r['count300'],
r['count100'],
r['count50'],
r['countgeki'],
r['countkatu'],
r['countmiss']), sentin, source)
def roll():
print("@" + username + ": /roll")
cmd = text.split(' ', 1)
# Se è stato specificato un numero
if len(cmd) >= 2:
# Controlla che sia convertibile in un intero.
try:
m = int(cmd[1])
except ValueError:
2016-09-21 14:08:12 +00:00
if cmd[1] == "tm":
telegram.sendmessage("TM è così grassa che se la lanci rotola!")
else:
telegram.sendmessage(chr(9888) + " Il numero specificato non è un intero.", sentin, source)
2016-08-24 06:43:26 +00:00
return
2016-08-24 06:40:21 +00:00
else:
# Imposta il numero massimo a 100.
m = 100
# Prova a generare un numero casuale.
if m == 34261891881215712181524122318242223183627453833:
2016-09-21 14:08:12 +00:00
telegram.sendmessage("Numero casuale da 1 a _34261891881215712181524122318242223183627453833_:\n"
"*Frank è scarso*", sentin, source)
2016-08-24 06:40:21 +00:00
else:
try:
n = random.randrange(m) + 1
except ValueError:
telegram.sendmessage(chr(9888) + " Il numero specificato non è maggiore o uguale a 0.",
sentin, source)
# Se tutto va bene visualizza il numero generato
else:
telegram.sendmessage("Numero casuale da 1 a " + str(m) + ":\n*" + str(n) + "*", sentin,
source)
def cv():
2016-09-04 18:37:13 +00:00
print("@" + username + ": /cv")
2016-09-21 14:39:27 +00:00
# Ottieni i dati dal server della Royal Games
2016-08-24 06:40:21 +00:00
r = discord.getwidgetdata("176353500710699008")
2016-09-21 14:39:27 +00:00
# Elenco di tutte le persone online su Discord
tosend = "*Online su Discord, nel server {servername}:*\n".format(servername=r['name'])
# Qui inizia il codice peggiore di sempre
2016-08-24 06:40:21 +00:00
for member in r['members']:
m = dict()
if 'bot' not in member or not member['bot']:
2016-09-21 14:39:27 +00:00
# Se una persona è connessa in chat vocale
2016-08-24 06:40:21 +00:00
if 'channel_id' in member:
2016-09-21 14:39:27 +00:00
# Controlla il suo stato (esclusa, mutata, normale) e scegli l'emoji appropriata
2016-08-24 06:40:21 +00:00
if member['deaf'] or member['self_deaf']:
m['emoji'] = chr(128263)
elif member['mute'] or member['self_mute']:
m['emoji'] = chr(128264)
else:
m['emoji'] = chr(128266)
2016-09-04 18:24:58 +00:00
m['channelname'] = discord.getchannelname(r, member['channel_id'])
2016-09-21 14:39:27 +00:00
# Altrimenti
2016-08-24 06:40:21 +00:00
else:
2016-09-21 14:39:27 +00:00
# Controlla il suo stato (online, in gioco, afk) e scegli l'emoji appropriata
2016-08-24 06:40:21 +00:00
if member['status'] == "online":
2016-09-21 14:39:27 +00:00
# Qui giace il pallino rosso se uno era in game, propongo di metterlo per quando uno sta streamando
m['emoji'] = chr(128309)
2016-08-24 06:40:21 +00:00
elif member['status'] == "idle":
m['emoji'] = chr(9899)
2016-09-21 14:39:27 +00:00
# Aggiungi il nome del gioco a destra del nome
2016-08-24 06:40:21 +00:00
if 'game' in member:
m['gamename'] = member['game']['name']
2016-09-21 14:39:27 +00:00
# Visualizza il nickname se presente, altrimenti visualizza l'username
if 'nick' in m:
member['username'] = m['nick']
else:
m['name'] = member['username']
2016-08-24 06:40:21 +00:00
if 'gamename' in m and 'channelname' in m:
tosend += "{emoji} *{channelname}* {name} | _{gamename}_\n".format(**m)
elif 'gamename' in m:
tosend += "{emoji} {name} | _{gamename}_\n".format(**m)
elif 'channelname' in m:
tosend += "{emoji} *{channelname}* {name}\n".format(**m)
else:
tosend += "{emoji} {name}\n".format(**m)
2016-09-04 18:02:41 +00:00
# Controlla se l'utente è royal music
2016-09-04 18:24:58 +00:00
elif member['id'] == "176358898851250176":
2016-09-04 18:02:41 +00:00
if 'game' in member:
2016-09-21 14:39:27 +00:00
tosend += "{emoji} *{channelname}* {songname}\n" \
2016-09-04 18:35:45 +00:00
.format(emoji="\U0001F3B5", channelname=discord.getchannelname(r, member['channel_id']),
songname=member['game']['name'])
2016-08-24 06:40:21 +00:00
telegram.sendmessage(tosend, sentin, source)
def online():
# Elenco di tutte le persone online su Steam
print("@" + username + ": /online ")
# Informa Telegram che il messaggio è stato ricevuto.
telegram.sendchataction(sentin)
cmd = text.split(" ")
if len(cmd) >= 2:
if cmd[1].lower() == "help":
telegram.sendmessage(chr(128309) + " Online\n" +
chr(128308) + " In gioco | Occupato\n" +
chr(9899) + " Assente | Inattivo\n" +
chr(128310) + " Disponibile per scambiare\n" +
chr(128311) + " Disponibile per giocare", sentin, source)
else:
# Stringa utilizzata per ottenere informazioni su tutti gli utenti in una sola richiesta a steam
userids = str()
for membro in royalgames:
if "steam" in royalgames[membro]:
userids += str(royalgames[membro]["steam"]) + ','
tosend = "*Su Steam ora:*\n"
r = steam.getplayersummaries(userids)
for player in r:
# In gioco
if 'gameextrainfo' in player:
tosend += chr(128308) + " _" + player['gameextrainfo'] + "_ |"
elif 'gameid' in player:
tosend += chr(128308) + " _" + player['gameid'] + "_ |"
# Online
elif player['personastate'] == 1:
tosend += chr(128309)
# Occupato
elif player['personastate'] == 2:
tosend += chr(128308)
# Assente o Inattivo
elif player['personastate'] == 3 or player['personastate'] == 4:
tosend += chr(9899)
# Disponibile per scambiare
elif player['personastate'] == 5:
tosend += chr(128310)
# Disponibile per giocare
elif player['personastate'] == 6:
tosend += chr(128311)
if player['personastate'] != 0:
tosend += " " + player['personaname'] + "\n"
else:
telegram.sendmessage(tosend, sentin, source)
def shrek():
# Manda l'audio So much to do, so much to see
print("@" + username + ": /shrekt ")
telegram.senddocument("BQADBAADsQADiBjiAqYN-EBXASyhAg", sentin)
def diario():
# Aggiungi una riga al diario Royal Games
print("@" + username + ": /diario ")
cmd = text.split(" ", 1)
if len(cmd) > 1:
if cmd[1].isprintable():
cmd[1] = cmd[1].replace("\n", " ").encode("unicode_escape")
2016-08-24 06:40:21 +00:00
fdiario = filemanager.readfile("diario.txt")
fdiario += str(int(time.time())) + "|" + cmd[1] + "\n"
filemanager.writefile("diario.txt", fdiario)
telegram.sendmessage("Aggiunto al diario RYG.", sentin, source)
else:
telegram.sendmessage(chr(9888) + " Errore nella scrittura del messaggio.\n"
"Visto che non ho la minima idea delle assurdita che ho scritto qui, "
"potresti provare a togliere le emoji.", sentin, source)
else:
telegram.sendmessage(chr(9888) + " Non hai scritto niente sul diario!\n"
"Sintassi corretta: /diario _quello che vuoi scrivere_",
sentin, source)
def leggi():
# Leggi dal diario Royal Games
print("@" + username + ": /leggi")
telegram.sendmessage("[Apri il diario RYG](http://royal.steffo.me/diario.htm)!", sentin, source)
def balurage():
print("@" + username + ": /balurage")
# Rispondi commentando l'E3.
telegram.sendmessage("MADDEN MADDEN MADDEN MADDEN MADDEN MADDEN MADDEN MADDEN MADDEN",
sentin, source)
def lolfree():
global lolfreestring
# Visualizza i campioni gratuiti di LoL di questa settimana
print("@" + username + ": /lolfree")
# Informa Telegram che il messaggio è stato ricevuto.
telegram.sendchataction(sentin)
ora = time.gmtime()
cmd = text.split(" ", 1)
if len(cmd) > 1:
refresh = cmd[1].startswith("refresh")
else:
refresh = False
# Controlla se i dati sono già stati scaricati.
if lolfreestring is None or refresh:
# Crea un nuovo set di dati.
print("Aggiornamento champ gratuiti di League of Legends...")
lolfreestring = "Champion gratuiti del `" + str(ora.tm_mday) + "/" + str(ora.tm_mon) + "/" + \
str(ora.tm_year) + " " + str(ora.tm_hour) + ":" + str(ora.tm_min) + "`\n"
r = lol.getfreerotation()
for champion in r:
staticdata = lol.getchampionstaticdata(champion['id'])
lolfreestring += "*" + staticdata['name'] + "* " + staticdata['title'] + '\n'
print("Completato.")
telegram.sendmessage(lolfreestring, sentin, source)
2016-01-14 18:37:37 +00:00
# Ciclo principale del bot
2015-10-04 17:48:45 +00:00
print("Bot avviato!")
2015-11-13 17:15:27 +00:00
while True:
2016-08-10 18:25:03 +00:00
try:
# Guarda il comando ricevuto.
msg = telegram.getupdates()
# Se il messaggio non è una notifica di servizio...
if 'edit' in msg:
if msg['edit']:
if 'text' in msg['edit_data']:
# Salvatelo in una stringa
text = msg['edit_data']['text']
# Guarda l'ID della chat in cui è stato inviato
sentin = msg['edit_data']['chat']['id']
# ID del messaggio ricevuto
source = msg['edit_data']['message_id']
if 'username' in msg['edit_data']['from']:
# Salva l'username se esiste
username = msg['edit_data']['from']['username']
2016-06-16 13:50:45 +00:00
else:
2016-08-10 18:25:03 +00:00
# Altrimenti, salva l'userID
username = str(msg['edit_data']['from']['id'])
# Se sei un membro della Royal Games
if username.lower() in royalgames:
# Riconosci il comando.
if text.startswith('wow'):
2016-08-24 06:40:21 +00:00
wow()
2016-08-10 18:25:03 +00:00
if 'text' in msg:
# Salvatelo in una stringa
text = msg['text']
# Guarda l'ID della chat in cui è stato inviato
sentin = msg['chat']['id']
# ID del messaggio ricevuto
source = msg['message_id']
# Nome da visualizzare nella console per capire chi accidenti è che invia messaggi strani
if 'username' in msg['from']:
# Salva l'username se esiste
username = msg['from']['username']
else:
# Altrimenti, salva l'userID
username = str(msg['from']['id'])
# Se sei un membro della Royal Games
if username.lower() in royalgames:
# Riconosci il comando.
# Viene usato startswith perchè il comando potrebbe anche essere inviato in forma /ciao@RoyalBot.
if text.startswith('/ahnonlosoio'):
2016-08-24 06:40:21 +00:00
ahnonlosoio()
2016-08-10 18:25:03 +00:00
elif text.startswith('/ciaostefanino'):
2016-08-24 06:40:21 +00:00
ciaostefanino()
elif text.startswith('/balurage') or text.startswith("/madden"):
balurage()
2016-08-10 18:25:03 +00:00
elif text.startswith('/potatogift'):
2016-08-24 06:40:21 +00:00
potatogift()
2016-08-10 18:25:03 +00:00
elif text.startswith('/ciaoruozi'):
2016-08-24 06:40:21 +00:00
ciaoruozi()
2016-08-10 18:25:03 +00:00
elif text.startswith('/ehoh'):
2016-08-24 06:40:21 +00:00
ehoh()
2016-08-10 18:25:03 +00:00
elif text.startswith('/playing'):
2016-08-24 06:40:21 +00:00
playing()
2016-08-10 18:25:03 +00:00
elif text.startswith('/saldi'):
2016-08-24 06:40:21 +00:00
saldi()
2016-08-10 18:25:03 +00:00
elif text.startswith('/sbam'):
2016-08-24 06:40:21 +00:00
sbam()
2016-08-10 18:25:03 +00:00
elif text.startswith('/osu'):
2016-08-24 06:40:21 +00:00
osucmd()
2016-08-10 18:25:03 +00:00
elif text.startswith('/roll'):
2016-08-24 06:40:21 +00:00
roll()
2016-08-10 18:25:03 +00:00
elif text.startswith('/automah'):
2016-08-24 06:40:21 +00:00
automah()
2016-08-10 18:25:03 +00:00
elif text.startswith('/cv'):
2016-08-24 06:40:21 +00:00
cv()
2016-08-10 18:25:03 +00:00
elif text.startswith('/online'):
2016-08-24 06:40:21 +00:00
online()
elif text.startswith('/shrek'):
shrek()
2016-08-10 18:25:03 +00:00
elif text.startswith('/diario'):
2016-08-24 06:40:21 +00:00
diario()
2016-08-10 18:25:03 +00:00
elif text.startswith('/leggi'):
2016-08-24 06:40:21 +00:00
leggi()
2016-08-10 18:25:03 +00:00
elif text.startswith('/lolfree'):
2016-08-24 06:40:21 +00:00
lolfree()
2016-09-07 15:46:36 +00:00
elif text.startswith('/ombromanto'):
ombromanto()
2016-08-10 18:25:03 +00:00
elif text.startswith('/crash'):
# Crasha il bot. Mi sembra geniale.
if username == 'Steffo':
raise Exception("Ho appena fatto crashare tutto apposta. Sono un genio.")
else:
print("@" + username + " bloccato.")
except Exception as e:
telegram.sendmessage(chr(9762) + " *ERRORE CRITICO:\n*"
"{0}\n".format(repr(e)), -2141322)
print("ERRORE CRITICO:\n"
2016-08-24 06:40:21 +00:00
"{0}".format(repr(e)))