forked from test/mau_mau_bot
move result builders and utility functions to seperate files
This commit is contained in:
Jannes Höke
parent
ab615f354c
commit
1844d1d40a
3 changed files with 178 additions and 171 deletions
174
bot.py
174
bot.py
|
|
@ -1,11 +1,8 @@
|
||||||
import logging
|
import logging
|
||||||
from datetime import datetime
|
from datetime import datetime
|
||||||
from random import randint
|
from random import randint
|
||||||
from uuid import uuid4
|
|
||||||
|
|
||||||
from telegram import InlineQueryResultArticle, ParseMode, Message, Chat, \
|
from telegram import ParseMode, Message, Chat, InlineKeyboardMarkup, \
|
||||||
Emoji, InputTextMessageContent, \
|
|
||||||
InlineQueryResultCachedSticker as Sticker, InlineKeyboardMarkup, \
|
|
||||||
InlineKeyboardButton
|
InlineKeyboardButton
|
||||||
from telegram.ext import Updater, InlineQueryHandler, \
|
from telegram.ext import Updater, InlineQueryHandler, \
|
||||||
ChosenInlineResultHandler, CommandHandler, MessageHandler, Filters, \
|
ChosenInlineResultHandler, CommandHandler, MessageHandler, Filters, \
|
||||||
|
|
@ -14,9 +11,10 @@ from telegram.ext.dispatcher import run_async
|
||||||
from telegram.utils.botan import Botan
|
from telegram.utils.botan import Botan
|
||||||
|
|
||||||
from game_manager import GameManager
|
from game_manager import GameManager
|
||||||
import card as c
|
|
||||||
from credentials import TOKEN, BOTAN_TOKEN
|
from credentials import TOKEN, BOTAN_TOKEN
|
||||||
from start_bot import start_bot
|
from start_bot import start_bot
|
||||||
|
from results import *
|
||||||
|
from utils import *
|
||||||
|
|
||||||
logging.basicConfig(
|
logging.basicConfig(
|
||||||
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
format='%(asctime)s - %(name)s - %(levelname)s - %(message)s',
|
||||||
|
|
@ -59,36 +57,6 @@ help_text = "Follow these steps:\n\n" \
|
||||||
" and buy an UNO card game.\n"
|
" and buy an UNO card game.\n"
|
||||||
|
|
||||||
|
|
||||||
def list_subtract(list1, list2):
|
|
||||||
""" Helper function to subtract two lists and return the sorted result """
|
|
||||||
list1 = list1.copy()
|
|
||||||
|
|
||||||
for x in list2:
|
|
||||||
list1.remove(x)
|
|
||||||
|
|
||||||
return list(sorted(list1))
|
|
||||||
|
|
||||||
|
|
||||||
def display_name(user):
|
|
||||||
""" Get the current players name including their username, if possible """
|
|
||||||
user_name = user.first_name
|
|
||||||
if user.username:
|
|
||||||
user_name += ' (@' + user.username + ')'
|
|
||||||
return user_name
|
|
||||||
|
|
||||||
|
|
||||||
def display_color(color):
|
|
||||||
""" Convert a color code to actual color name """
|
|
||||||
if color == "r":
|
|
||||||
return Emoji.HEAVY_BLACK_HEART + " Red"
|
|
||||||
if color == "b":
|
|
||||||
return Emoji.BLUE_HEART + " Blue"
|
|
||||||
if color == "g":
|
|
||||||
return Emoji.GREEN_HEART + " Green"
|
|
||||||
if color == "y":
|
|
||||||
return Emoji.YELLOW_HEART + " Yellow"
|
|
||||||
|
|
||||||
|
|
||||||
@run_async
|
@run_async
|
||||||
def send_async(bot, *args, **kwargs):
|
def send_async(bot, *args, **kwargs):
|
||||||
if 'timeout' not in kwargs:
|
if 'timeout' not in kwargs:
|
||||||
|
|
@ -399,142 +367,6 @@ def reply_to_query(bot, update):
|
||||||
switch_pm_text=switch, switch_pm_parameter='select')
|
switch_pm_text=switch, switch_pm_parameter='select')
|
||||||
|
|
||||||
|
|
||||||
def add_choose_color(results):
|
|
||||||
for color in c.COLORS:
|
|
||||||
results.append(
|
|
||||||
InlineQueryResultArticle(
|
|
||||||
id=color,
|
|
||||||
title="Choose Color",
|
|
||||||
description=display_color(color),
|
|
||||||
input_message_content=
|
|
||||||
InputTextMessageContent(display_color(color))
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def add_other_cards(playable, player, results, game):
|
|
||||||
if not playable:
|
|
||||||
playable = list()
|
|
||||||
|
|
||||||
players = player_list(game)
|
|
||||||
|
|
||||||
results.append(
|
|
||||||
InlineQueryResultArticle(
|
|
||||||
"hand",
|
|
||||||
title="Cards (tap for game state):",
|
|
||||||
description=', '.join([repr(card) for card in
|
|
||||||
list_subtract(player.cards, playable)]),
|
|
||||||
input_message_content=InputTextMessageContent(
|
|
||||||
"Current player: " + display_name(game.current_player.user) +
|
|
||||||
"\n" +
|
|
||||||
"Last card: " + repr(game.last_card) + "\n" +
|
|
||||||
"Players: " + " -> ".join(players))
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def player_list(game):
|
|
||||||
players = list()
|
|
||||||
for player in game.players:
|
|
||||||
add_player(player, players)
|
|
||||||
return players
|
|
||||||
|
|
||||||
|
|
||||||
def add_no_game(results):
|
|
||||||
results.append(
|
|
||||||
InlineQueryResultArticle(
|
|
||||||
"nogame",
|
|
||||||
title="You are not playing",
|
|
||||||
input_message_content=
|
|
||||||
InputTextMessageContent('Not playing right now. Use /new to start '
|
|
||||||
'a game or /join to join the current game '
|
|
||||||
'in this group')
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def add_not_started(results):
|
|
||||||
results.append(
|
|
||||||
InlineQueryResultArticle(
|
|
||||||
"nogame",
|
|
||||||
title="The game wasn't started yet",
|
|
||||||
input_message_content=
|
|
||||||
InputTextMessageContent('Start the game with /start')
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def add_draw(player, results):
|
|
||||||
results.append(
|
|
||||||
Sticker(
|
|
||||||
"draw", sticker_file_id=c.STICKERS['option_draw'],
|
|
||||||
input_message_content=
|
|
||||||
InputTextMessageContent('Drawing %d card(s)'
|
|
||||||
% (player.game.draw_counter or 1))
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def add_gameinfo(game, results):
|
|
||||||
players = player_list(game)
|
|
||||||
|
|
||||||
results.append(
|
|
||||||
Sticker(
|
|
||||||
"gameinfo",
|
|
||||||
sticker_file_id=c.STICKERS['option_info'],
|
|
||||||
input_message_content=InputTextMessageContent(
|
|
||||||
"Current player: " + display_name(game.current_player.user) +
|
|
||||||
"\n" +
|
|
||||||
"Last card: " + repr(game.last_card) + "\n" +
|
|
||||||
"Players: " + " -> ".join(players))
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def add_pass(results):
|
|
||||||
results.append(
|
|
||||||
Sticker(
|
|
||||||
"pass", sticker_file_id=c.STICKERS['option_pass'],
|
|
||||||
input_message_content=InputTextMessageContent('Pass')
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def add_call_bluff(results):
|
|
||||||
results.append(
|
|
||||||
Sticker(
|
|
||||||
"call_bluff",
|
|
||||||
sticker_file_id=c.STICKERS['option_bluff'],
|
|
||||||
input_message_content=
|
|
||||||
InputTextMessageContent("I'm calling your bluff!")
|
|
||||||
)
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def add_play_card(game, card, results, can_play):
|
|
||||||
players = player_list(game)
|
|
||||||
|
|
||||||
if can_play:
|
|
||||||
results.append(
|
|
||||||
Sticker(str(card), sticker_file_id=c.STICKERS[str(card)])
|
|
||||||
)
|
|
||||||
else:
|
|
||||||
results.append(
|
|
||||||
Sticker(str(uuid4()), sticker_file_id=c.STICKERS_GREY[str(card)],
|
|
||||||
input_message_content=InputTextMessageContent(
|
|
||||||
"Current player: " + display_name(
|
|
||||||
game.current_player.user) +
|
|
||||||
"\n" +
|
|
||||||
"Last card: " + repr(game.last_card) + "\n" +
|
|
||||||
"Players: " + " -> ".join(players)))
|
|
||||||
)
|
|
||||||
|
|
||||||
|
|
||||||
def add_player(itplayer, players):
|
|
||||||
players.append(itplayer.user.first_name + " (%d cards)"
|
|
||||||
% len(itplayer.cards))
|
|
||||||
|
|
||||||
|
|
||||||
def process_result(bot, update):
|
def process_result(bot, update):
|
||||||
""" Check the players actions and act accordingly """
|
""" Check the players actions and act accordingly """
|
||||||
try:
|
try:
|
||||||
|
|
|
||||||
144
results.py
Normal file
144
results.py
Normal file
|
|
@ -0,0 +1,144 @@
|
||||||
|
from uuid import uuid4
|
||||||
|
|
||||||
|
from telegram import InlineQueryResultArticle, InputTextMessageContent, \
|
||||||
|
InlineQueryResultCachedSticker as Sticker
|
||||||
|
|
||||||
|
import card as c
|
||||||
|
from utils import *
|
||||||
|
|
||||||
|
|
||||||
|
def add_choose_color(results):
|
||||||
|
for color in c.COLORS:
|
||||||
|
results.append(
|
||||||
|
InlineQueryResultArticle(
|
||||||
|
id=color,
|
||||||
|
title="Choose Color",
|
||||||
|
description=display_color(color),
|
||||||
|
input_message_content=
|
||||||
|
InputTextMessageContent(display_color(color))
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def add_other_cards(playable, player, results, game):
|
||||||
|
if not playable:
|
||||||
|
playable = list()
|
||||||
|
|
||||||
|
players = player_list(game)
|
||||||
|
|
||||||
|
results.append(
|
||||||
|
InlineQueryResultArticle(
|
||||||
|
"hand",
|
||||||
|
title="Cards (tap for game state):",
|
||||||
|
description=', '.join([repr(card) for card in
|
||||||
|
list_subtract(player.cards, playable)]),
|
||||||
|
input_message_content=InputTextMessageContent(
|
||||||
|
"Current player: " + display_name(game.current_player.user) +
|
||||||
|
"\n" +
|
||||||
|
"Last card: " + repr(game.last_card) + "\n" +
|
||||||
|
"Players: " + " -> ".join(players))
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def player_list(game):
|
||||||
|
players = list()
|
||||||
|
for player in game.players:
|
||||||
|
add_player(player, players)
|
||||||
|
return players
|
||||||
|
|
||||||
|
|
||||||
|
def add_no_game(results):
|
||||||
|
results.append(
|
||||||
|
InlineQueryResultArticle(
|
||||||
|
"nogame",
|
||||||
|
title="You are not playing",
|
||||||
|
input_message_content=
|
||||||
|
InputTextMessageContent('Not playing right now. Use /new to start '
|
||||||
|
'a game or /join to join the current game '
|
||||||
|
'in this group')
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def add_not_started(results):
|
||||||
|
results.append(
|
||||||
|
InlineQueryResultArticle(
|
||||||
|
"nogame",
|
||||||
|
title="The game wasn't started yet",
|
||||||
|
input_message_content=
|
||||||
|
InputTextMessageContent('Start the game with /start')
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def add_draw(player, results):
|
||||||
|
results.append(
|
||||||
|
Sticker(
|
||||||
|
"draw", sticker_file_id=c.STICKERS['option_draw'],
|
||||||
|
input_message_content=
|
||||||
|
InputTextMessageContent('Drawing %d card(s)'
|
||||||
|
% (player.game.draw_counter or 1))
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def add_gameinfo(game, results):
|
||||||
|
players = player_list(game)
|
||||||
|
|
||||||
|
results.append(
|
||||||
|
Sticker(
|
||||||
|
"gameinfo",
|
||||||
|
sticker_file_id=c.STICKERS['option_info'],
|
||||||
|
input_message_content=InputTextMessageContent(
|
||||||
|
"Current player: " + display_name(game.current_player.user) +
|
||||||
|
"\n" +
|
||||||
|
"Last card: " + repr(game.last_card) + "\n" +
|
||||||
|
"Players: " + " -> ".join(players))
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def add_pass(results):
|
||||||
|
results.append(
|
||||||
|
Sticker(
|
||||||
|
"pass", sticker_file_id=c.STICKERS['option_pass'],
|
||||||
|
input_message_content=InputTextMessageContent('Pass')
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def add_call_bluff(results):
|
||||||
|
results.append(
|
||||||
|
Sticker(
|
||||||
|
"call_bluff",
|
||||||
|
sticker_file_id=c.STICKERS['option_bluff'],
|
||||||
|
input_message_content=
|
||||||
|
InputTextMessageContent("I'm calling your bluff!")
|
||||||
|
)
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def add_play_card(game, card, results, can_play):
|
||||||
|
players = player_list(game)
|
||||||
|
|
||||||
|
if can_play:
|
||||||
|
results.append(
|
||||||
|
Sticker(str(card), sticker_file_id=c.STICKERS[str(card)])
|
||||||
|
)
|
||||||
|
else:
|
||||||
|
results.append(
|
||||||
|
Sticker(str(uuid4()), sticker_file_id=c.STICKERS_GREY[str(card)],
|
||||||
|
input_message_content=InputTextMessageContent(
|
||||||
|
"Current player: " + display_name(
|
||||||
|
game.current_player.user) +
|
||||||
|
"\n" +
|
||||||
|
"Last card: " + repr(game.last_card) + "\n" +
|
||||||
|
"Players: " + " -> ".join(players)))
|
||||||
|
)
|
||||||
|
|
||||||
|
|
||||||
|
def add_player(itplayer, players):
|
||||||
|
players.append(itplayer.user.first_name + " (%d cards)"
|
||||||
|
% len(itplayer.cards))
|
||||||
|
|
||||||
31
utils.py
Normal file
31
utils.py
Normal file
|
|
@ -0,0 +1,31 @@
|
||||||
|
from telegram import Emoji
|
||||||
|
|
||||||
|
|
||||||
|
def list_subtract(list1, list2):
|
||||||
|
""" Helper function to subtract two lists and return the sorted result """
|
||||||
|
list1 = list1.copy()
|
||||||
|
|
||||||
|
for x in list2:
|
||||||
|
list1.remove(x)
|
||||||
|
|
||||||
|
return list(sorted(list1))
|
||||||
|
|
||||||
|
|
||||||
|
def display_name(user):
|
||||||
|
""" Get the current players name including their username, if possible """
|
||||||
|
user_name = user.first_name
|
||||||
|
if user.username:
|
||||||
|
user_name += ' (@' + user.username + ')'
|
||||||
|
return user_name
|
||||||
|
|
||||||
|
|
||||||
|
def display_color(color):
|
||||||
|
""" Convert a color code to actual color name """
|
||||||
|
if color == "r":
|
||||||
|
return Emoji.HEAVY_BLACK_HEART + " Red"
|
||||||
|
if color == "b":
|
||||||
|
return Emoji.BLUE_HEART + " Blue"
|
||||||
|
if color == "g":
|
||||||
|
return Emoji.GREEN_HEART + " Green"
|
||||||
|
if color == "y":
|
||||||
|
return Emoji.YELLOW_HEART + " Yellow"
|
||||||
Loading…
Reference in a new issue