ryg/royalnet
ryg/royalnet
1
Fork 0
mirror of https://github.com/RYGhub/royalnet.git synced 2024-11-23 19:44:20 +00:00
Code Activity
royalnet/bot.py
Stefano Pigozzi dc495d4c48 lel
2016-11-17 19:39:04 +01:00

627 lines
27 KiB
Python
Raw Blame History

# -*- coding: utf-8 -*-
import json
import time
import filemanager
import telegram
import steam
import random
import osu
import lol
import discord
import subprocess
import sm.steammatch as steammatch
import db
# Elenco di username dei membri della RYG
royalgames = json.loads(filemanager.readfile("db.json"))
# Stringa dove mettere l'elenco di champion di lol gratuiti
lolfreestring = str()
random.seed()
# 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)
if username != "Steffo":
telegram.sendmessage("Steffo dice:\n\"Me la pagherai!\"", fromuser, source)
else:
telegram.sendmessage("Sei improvvisamente impazzito o cosa?", fromuser, source)
def ciaoruozi():
print("@" + username + ": /ciaoruozi")
# Ciao Ruozi.
if username.lower() == "ruozir":
# Manda un messaggio casuale tra quelli nella lista
chosen_msg = random.sample(["Ciao me!",
"Ciao ciao ciao!",
"1 ciaoruozi = 1000€ per me",
"Ruozi si dice: #CiaoRuozi",
"Ciao eh me!",
"Ehi me, ma ciao!",
"Ma ciao me stesso!",
"Me me me! Ciao!"], 1)[0]
telegram.sendmessage(chosen_msg, sentin, source)
else:
# 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!",
"Ehi Ruozi, ma ciao!",
"Ma ciao Ruozi!",
"Ruozi ruozi ruozi! Ciao!"], 1)[0]
telegram.sendmessage(chosen_msg, sentin, source)
def ombromanto():
print("@" + username + ": /ombromanto")
# Ma chi è ombromanto?
telegram.sendmessage("Ombromanto è @Dailir!", sentin, source)
def steamplayers():
print("@" + username + ": /steamplayers")
# Informa Telegram che il messaggio è stato ricevuto e visualizza Royal Bot sta scrivendo.
telegram.sendchataction(sentin)
# 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)
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.
telegram.sendmessage("Eh, oh. Sono cose che capitano.", sentin, source)
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
# Se è stato specificato un nome utente
if len(cmd) >= 2:
# Trova la modalità
# 0 = osu!
# 1 = osu!taiko
# 2 = osu!catch
# 3 = osu!mania
# 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")
# Se è stato specificato un numero
if len(cmd) >= 2:
if cmd[1] == "tm":
telegram.sendmessage("TM è così grassa che se la lanci rotola!", sentin, source)
# Controlla che sia convertibile in un intero.
try:
m = int(cmd[1])
except ValueError:
telegram.sendmessage(chr(9888) + " Il numero specificato non è un intero.", sentin, source)
return
else:
# Imposta il numero massimo a 100.
m = 100
# Prova a generare un numero casuale.
if m == 34261891881215712181524122318242223183627453833:
telegram.sendmessage("Numero casuale da 1 a _34261891881215712181524122318242223183627453833_:\n"
"*Frank è scarso*", sentin, source)
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():
print("@" + username + ": /cv")
# Ottieni i dati dal server della Royal Games
r = discord.getwidgetdata("176353500710699008")
# 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
for member in r['members']:
m = dict()
if 'bot' not in member or not member['bot']:
# Se una persona è connessa in chat vocale
if 'channel_id' in member:
# Controlla il suo stato (esclusa, mutata, normale) e scegli l'emoji appropriata
if member['deaf'] or member['self_deaf']:
m['vemoji'] = chr(128263)
elif member['mute'] or member['self_mute']:
m['vemoji'] = chr(128264)
else:
m['vemoji'] = chr(128266)
m['channelname'] = discord.getchannelname(r, member['channel_id'])
# Altrimenti
else:
m['vemoji'] = ""
# Controlla il suo stato (online, in gioco, afk) e scegli l'emoji appropriata
if member['status'] == "online":
m['emoji'] = chr(128309)
elif member['status'] == "idle":
m['emoji'] = chr(9899)
elif member['status'] == "dnd":
m['emoji'] = chr(128308)
else:
# Stato sconosciuto. Fallback nel caso in cui vengano aggiunti nuovi stati.
m['emoji'] = chr(2573)
# Aggiungi il nome del gioco a destra del nome
if 'game' in member:
m['gamename'] = member['game']['name']
# Visualizza il nickname se presente, altrimenti visualizza l'username
if 'nick' in m:
member['username'] = m['nick']
else:
m['name'] = member['username']
tosend += "{emoji} {name}".format(emoji=m['emoji'], name=m['name'])
if 'channelname' in m:
tosend += " | {vemoji} *{channelname}*".format(vemoji=m['vemoji'], channelname=m['channelname'])
if 'gamename' in m:
tosend += " | _{gamename}_".format(gamename=m['gamename'])
tosend += "\n"
# Controlla se l'utente è royal music
elif member['id'] == "176358898851250176":
if 'game' in member:
tosend += "{emoji} *{channelname}* | {songname}\n" \
.format(emoji="\U0001F3B5", channelname=discord.getchannelname(r, member['channel_id']),
songname=member['game']['name'])
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)
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 ")
if len(cmd) > 1:
if cmd[1].isprintable():
cmd[1] = cmd[1].replace("\n", " ")
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"
"Il messaggio non è compatibile con il diario.\n"
"Probabilmente contiene emoji o caratteri speciali non visualizzabili.",
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.
tosend = str()
try:
ragelevel = int(filemanager.readfile("ragelevel.txt"))
except ValueError:
ragelevel = 0
ragelevel += 1
filemanager.writefile("ragelevel.txt", str(ragelevel))
for rage in range(0, ragelevel):
tosend += "MADDEN "
telegram.sendmessage(tosend, sentin, source)
def lolfree():
# 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()
if len(cmd) > 1:
refresh_requested = cmd[1].startswith("refresh")
else:
refresh_requested = False
# Controlla se i dati sono già stati scaricati.
global lolfreestring
if lolfreestring == "" or refresh_requested:
# 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)
def getrygimage():
# Ricevi il link alla tua immagine del profilo Royal Games.
print("@" + username + ": /getrygimage")
telegram.sendmessage("[Scarica](https://dl.dropboxusercontent.com/u/23313381/Logo/Personali/{tag}.png)"
" la tua immagine del profilo Royal Games!".format(tag=royalgames[username.lower()]['icon']),
sentin, source)
def ciaospaggia():
# Buongiorno, stellina!
print("@" + username + ": /ciaospaggia")
telegram.sendmessage("Ma buongiorno, [Stellina](https://telegram.me/doom_darth_vader)!", sentin, source)
def smecds():
# Secondo me, è colpa...
print("@" + username + ": /smecds")
accusato = random.sample(["dello stagista", "degli sposi", "di Santinelli", "di Sensei", "di Steffo", "di Spaggia",
"della sedia", "di Satana", "del Sangue (degli occhi di Adry)", "del Sale",
"del Serpente", "della Samsung", "di /smecds", "della succursale", "di succ",
"di Sans", "di [SiivaGunner](https://www.youtube.com/channel/UC9ecwl3FTG66jIKA9JRDtmg)",
"di saaaaaas", "del semaforo", "della Seriale", "di Sistemi", "della Supercell",
"di Santaclaus", "dei Sims", "dei Santi", "di SES2017", "di Salvini"], 1)[0]
telegram.sendmessage("Secondo me è colpa {accusato}...".format(accusato=accusato), sentin, source)
def version():
# Visualizza la versione di Royal Bot
# Solo per UNIX...?
print("@" + username + ": /version")
gitmsg = subprocess.check_output(["git", "describe"])
telegram.sendmessage(gitmsg, sentin, source)
def match():
# Visualizza detutti i giochi condivisi tra x persone.
print("@" + username + ": /match")
# Informa Telegram che il messaggio è stato ricevuto.
telegram.sendchataction(sentin)
tobematched = list()
if len(cmd) > 2:
del cmd[0]
for name in cmd:
userdata = db.findbyname(name.lower())
if userdata is not None and 'steam' in userdata:
if userdata['steam'] not in tobematched:
tobematched.append(userdata['steam'])
if len(tobematched) > 1:
m = list(steammatch.compare(tobematched))
if len(m) > 0:
# Prepara il messaggio
tosend = "*Giochi in comune tra questi utenti:*\n"
for game in m:
tosend += "- {game}\n".format(game=game)
# Manda il messaggio
telegram.sendmessage(tosend, sentin, source)
else:
telegram.sendmessage("*Giochi in comune tra questi utenti:*\n_nessuno_", sentin, source)
else:
telegram.sendmessage(chr(9888) + "Non sono stati specificati abbastanza utenti per eseguire l'azione.",
sentin, source)
# Alias di tutti i comandi. Scrivendo quella stringa in chat viene attivata la funzione corrispondente.
aliases = {
"ahnonlosoio": ahnonlosoio,
"ahboh": ahnonlosoio,
"ciaostefanino": ciaostefanino,
"balurage": balurage,
"madden": balurage,
"ciaoruozi": ciaoruozi,
"ciaospaggia": ciaospaggia,
"buongiornostellina": ciaospaggia,
"stellina": ciaospaggia,
"ehoh": ehoh,
"sbam": sbam,
"rekt": sbam,
"osu": osucmd,
"roll": roll,
"cv": cv,
"discord": cv,
"shrek": shrek,
"diario": diario,
"d": diario,
"leggi": leggi,
"match": match,
"lol": lolfree,
"lolfree": lolfree,
"legoflegend": lolfree,
"getrygimage": getrygimage,
"version": version,
"smecds": smecds,
"online": online,
"steam": online,
"wow": wow
}
# Ciclo principale del bot
print("Bot avviato!")
while True:
try:
# Guarda il comando .
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']
else:
# 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'):
wow()
if 'text' in msg:
# Salvatelo in una stringa
text = msg['text']
# Guarda l'ID della chat in cui è stato inviato
sentin = msg['chat']['id']
# Persona che ha inviato il messaggio
fromuser = msg['from']['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 e dividilo in comando e argomenti.
cmd = text.lower().split(" ")
# Togli il @RoyalBot alla fine del comando
cmd[0] = cmd[0].replace("@royalbot", "").replace("/", "")
# Prova ad eseguire il comando. Se non è nella lista degli alias, ignoralo.
try:
aliases[cmd[0]]()
except KeyError:
print("@" + username + ": comando inesistente")
else:
print("@" + username + " bloccato.")
except Exception as e:
telegram.sendmessage(chr(9762) + " *Errore critico:\n*"
"{0}\n\n"
"Secondo me, è colpa dello stagista.".format(repr(e)), -2141322)
print("\033[1mERRORE CRITICO:\n"
"{0}\033[0m".format(repr(e)))
View git blame Copy permalink