mau_mau_bot/utils.py

148 lines
4 KiB
Python
Raw Normal View History

2016-05-08 20:37:25 +08:00
#!/usr/bin/env python3
2016-05-20 05:15:46 +08:00
# -*- coding: utf-8 -*-
2016-05-08 20:37:25 +08:00
#
# Telegram bot to play UNO in group chats
# Copyright (c) 2016 Jannes Höke <uno@jhoeke.de>
#
# This program is free software: you can redistribute it and/or modify
# it under the terms of the GNU Affero General Public License as
# published by the Free Software Foundation, either version 3 of the
# License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU Affero General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with this program. If not, see <http://www.gnu.org/licenses/>.
import logging
from telegram.ext.dispatcher import run_async
from internationalization import _, __
2017-08-19 06:08:55 +08:00
from mwt import MWT
from shared_vars import gm
2019-10-17 22:58:11 +08:00
import card as c
logger = logging.getLogger(__name__)
TIMEOUT = 2.5
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} Red").format(emoji='❤️')
if color == "b":
return _("{emoji} Blue").format(emoji='💙')
if color == "g":
return _("{emoji} Green").format(emoji='💚')
if color == "y":
return _("{emoji} Yellow").format(emoji='💛')
2016-05-23 01:21:51 +08:00
def display_color_group(color, game):
""" Convert a color code to actual color name """
if color == "r":
return __("{emoji} Red", game.translate).format(
emoji='❤️')
2016-05-23 01:21:51 +08:00
if color == "b":
return __("{emoji} Blue", game.translate).format(
emoji='💙')
2016-05-23 01:21:51 +08:00
if color == "g":
return __("{emoji} Green", game.translate).format(
emoji='💚')
2016-05-23 01:21:51 +08:00
if color == "y":
return __("{emoji} Yellow", game.translate).format(
emoji='💛')
2019-10-17 22:58:11 +08:00
def display_card_text(card):
if card.special == c.DRAW_FOUR:
return __("⬛️+4")
if card.special == c.CHOOSE:
return __("Color Chooser")
if card.value == c.DRAW_TWO:
if card.color == 'r':
return __("❤️+2")
if card.color == 'b':
return __("💙+2")
if card.color == 'g':
return __("💚+2")
if card.color == 'y':
return __("💛+2")
else:
return repr(card)
def error(bot, update, error):
"""Simple error handler"""
logger.exception(error)
@run_async
def send_async(bot, *args, **kwargs):
"""Send a message asynchronously"""
if 'timeout' not in kwargs:
kwargs['timeout'] = TIMEOUT
try:
bot.sendMessage(*args, **kwargs)
except Exception as e:
error(None, None, e)
@run_async
def answer_async(bot, *args, **kwargs):
"""Answer an inline query asynchronously"""
if 'timeout' not in kwargs:
kwargs['timeout'] = TIMEOUT
try:
bot.answerInlineQuery(*args, **kwargs)
except Exception as e:
error(None, None, e)
2017-08-19 06:08:55 +08:00
def game_is_running(game):
return game in gm.chatid_games.get(game.chat.id, list())
def user_is_creator(user, game):
return user.id in game.owner
def user_is_admin(user, bot, chat):
return user.id in get_admin_ids(bot, chat.id)
def user_is_creator_or_admin(user, game, bot, chat):
return user_is_creator(user, game) or user_is_admin(user, bot, chat)
2017-08-19 06:08:55 +08:00
@MWT(timeout=60*60)
def get_admin_ids(bot, chat_id):
"""Returns a list of admin IDs for a given chat. Results are cached for 1 hour."""
return [admin.user.id for admin in bot.get_chat_administrators(chat_id)]