-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathbot.py
148 lines (130 loc) · 6.32 KB
/
bot.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
from telegram.ext import Updater
from telegram.ext import MessageHandler, Filters
import logging
import wordle
import settings
import pickle
from telegram import Update
from telegram.ext import CallbackContext
from telegram.ext import CommandHandler
sessions = {}
points = [-5, 100, 80, 50, 35, 10, 5]
qwerty = [chr(x+ord('a')) for x in range(26)];
def start(update: Update, context: CallbackContext):
context.bot.send_message(chat_id=update.effective_chat.id, text="Nuova partita avviata... Invia la prima parola!")
username = update.message.from_user.id
nick = update.message.chat.username
if nick is None:
return
global sessions
sol = wordle.rand_word()
sol = sol[:-1]
sessions[username] = [sol, 0, '', nick, qwerty.copy(), [], [], [], 0]; #b, n, y, g
print("Nuova partita avviata da {}, sol: {}".format(nick, sol));
def res_to_str(res):
s = '';
for c in res:
if c == 0:
s += '\u2B1B';
if c == 1:
s += '\U0001f7e8';
if c == 2:
s += '\U0001f7e9';
return s;
def update_leaderboard(id_, user, points):
with open('leaderboard.pkl', 'rb') as f:
leaderboard = pickle.load(f)
if id_ not in leaderboard:
leaderboard[id_] = [points, user];
else:
leaderboard[id_][0] += points;
leaderboard[id_][1] = user;
with open('leaderboard.pkl', 'wb') as f:
pickle.dump(leaderboard, f)
def make_keyboard(white, grey, yellow, green):
s = '';
s += "\U0001f7e9: ";
for c in green:
s += c + ' ';
s += '\n';
s+= "\U0001f7e8: ";
for c in yellow:
s += c + ' ';
s += '\n';
s+= '\u2B1C: ';
for c in white:
s+= c + ' ';
return s;
def game(update: Update, context: CallbackContext):
username = update.message.from_user.id;
msg = update.message.text.lower()
nick = update.message.chat.username
context.bot.delete_message(update.message.chat_id, update.message.message_id)
if nick is None:
return
global sessions;
if username not in sessions:
context.bot.send_message(chat_id=update.effective_chat.id, text = "Per avviare una nuova partita digita /start");
else:
if wordle.search_word(settings.dictionary,msg) == False:
context.bot.send_message(chat_id=update.effective_chat.id, text = "Parola non valida")
else :
sessions[username][1] += 1
res = res_to_str(wordle.cmp_words(msg, sessions[username][0],
sessions[username][4], sessions[username][5], sessions[username][6], sessions[username][7]))
keyboard = make_keyboard(sessions[username][4], sessions[username][5], sessions[username][6], sessions[username][7]);
if sessions[username][8] != 0:
context.bot.delete_message(update.message.chat_id, sessions[username][8])
txt = ". "+" ".join(msg.upper()) + "\n" + res;
context.bot.send_message(chat_id=update.effective_chat.id, text = txt)
sessions[username][2] += res + '\n';
if sessions[username][0] == msg:
context.bot.send_message(chat_id=update.effective_chat.id, text = "Complimenti! Digita /start per ricominciare");
final_txt ="Valhalla Wordle " + str(sessions[username][1]) + "/6" + "\n\n" + sessions[username][2]
context.bot.send_message(chat_id=update.effective_chat.id, text = final_txt);
update_leaderboard(username, nick, points[sessions[username][1]]);
print("{} ha vinto in {} tentativi!".format(nick, sessions[username][1]))
sessions.pop(username)
elif sessions[username][1] == 6:
context.bot.send_message(chat_id=update.effective_chat.id, text = "La parola giusta era {}, digita /start per ricominciare...".format(sessions[username][0]))
final_txt ="Valhalla Wordle " + "-/6" + "\n\n" + sessions[username][2]
context.bot.send_message(chat_id=update.effective_chat.id, text = final_txt);
update_leaderboard(username, nick, points[0]);
sessions.pop(username)
print("{} ha perso!".format(nick))
else:
mess = context.bot.send_message(chat_id=update.effective_chat.id, text = keyboard)
sessions[username][8] = mess.message_id;
# context.bot.send_message(chat_id=update.effective_chat.id, text = "{}/6".format(sessions[username][1]))
def show_leaderboard(update: Update, context: CallbackContext):
text = "🏆 LEADERBOARD 🏆\n\n";
with open('leaderboard.pkl', 'rb') as f:
leaderboard = pickle.load(f)
sorted_leaderboard = {k: v for k, v in sorted(leaderboard.items(), key=lambda item: item[1], reverse = True)}
for x,i in zip(sorted_leaderboard, range(len(sorted_leaderboard))):
if(i>=20):
break;
else:
text += str(i+1) + " 》 " + leaderboard[x][1]+" " + str(leaderboard[x][0]) + "\n";
context.bot.send_message(chat_id=update.effective_chat.id, text = text);
def help_(update: Update, context: CallbackContext):
text = 'Il gioco è ancora in via di sviluppo, i punteggi sono 100, 80, 50, 35, 10, 5, -5, in base a quanti tentativi sono stati necessari, buona fortuna!';
context.bot.send_message(chat_id=update.effective_chat.id, text = text);
def start_bot():
global sessions
sessions = {}
tkn = 'set your telegram token!'
updater = Updater(token=tkn, use_context=True)
dispatcher = updater.dispatcher
logging.basicConfig(format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
level=logging.INFO)
start_handler = CommandHandler('start', start)
leaderboard_handler = CommandHandler('leaderboard', show_leaderboard)
help_handler = CommandHandler('help', help_)
dispatcher.add_handler(leaderboard_handler)
dispatcher.add_handler(start_handler)
dispatcher.add_handler(help_handler)
game_handler = MessageHandler(Filters.text & (~Filters.command), game)
dispatcher.add_handler(game_handler)
updater.start_polling()
updater.idle()