From b136bdf99756a7555c2e4750fece507fc22b54a5 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Jannes=20H=C3=B6ke?= Date: Sat, 2 Jul 2016 20:37:35 +0200 Subject: [PATCH] formatting, exception handling, documentation --- bot.py | 28 +++++++++++++++++++--------- game_manager.py | 2 ++ locales/unobot.pot | 4 ++++ simple_commands.py | 3 ++- 4 files changed, 27 insertions(+), 10 deletions(-) diff --git a/bot.py b/bot.py index 4076247..e27d733 100644 --- a/bot.py +++ b/bot.py @@ -51,18 +51,21 @@ logging.basicConfig( level=logging.INFO) logger = logging.getLogger(__name__) -remind_dict = {} + @user_locale -def next_game(bot, update): - """handler for /next_game command , pm people for next game""" +def notify_me(bot, update): + """Handler for /notify_me command, pm people for next game""" chat_id = update.message.chat_id if update.message.chat.type == 'private': - send_async(bot, chat_id,text="call this in a group to be reminded ") + send_async(bot, + chat_id, + text=_("Send this command in a group to be notified " + "when a new game is started there.")) else: try: - remind_dict[chat_id].append(update.message.from_user.id) + gm.remind_dict[chat_id].append(update.message.from_user.id) except KeyError: - remind_dict[chat_id] = [update.message.from_user.id] + gm.remind_dict[chat_id] = [update.message.from_user.id] @user_locale @@ -74,9 +77,16 @@ def new_game(bot, update): help(bot, update) else: - for user in remind_dict[update.message.chat_id]: - send_async(bot , user , text = "A new game has been started in " + update.message.chat.title) - del remind_dict[update.message.chat_id] + + if update.message.chat_id in gm.remind_dict: + for user in gm.remind_dict[update.message.chat_id]: + send_async(bot, + user, + text="A new game has been started in " + + update.message.chat.title) + + del gm.remind_dict[update.message.chat_id] + game = gm.new_game(update.message.chat) game.owner = update.message.from_user send_async(bot, chat_id, diff --git a/game_manager.py b/game_manager.py index c068fb1..5c94a22 100644 --- a/game_manager.py +++ b/game_manager.py @@ -33,6 +33,8 @@ class GameManager(object): self.chatid_games = dict() self.userid_players = dict() self.userid_current = dict() + self.remind_dict = dict() + self.logger = logging.getLogger(__name__) def new_game(self, chat): diff --git a/locales/unobot.pot b/locales/unobot.pot index a002baf..b65293e 100644 --- a/locales/unobot.pot +++ b/locales/unobot.pot @@ -75,6 +75,10 @@ msgstr "" msgid "Created a new game! Join the game with /join and start the game with /start" msgstr "" +#: bot.py +msgid "Send this command in a group to be notified when a new game is started there." +msgstr "" + #: bot.py:152 msgid "The lobby is closed" msgstr "" diff --git a/simple_commands.py b/simple_commands.py index 88920b2..05bece5 100644 --- a/simple_commands.py +++ b/simple_commands.py @@ -40,7 +40,8 @@ help_text = ("Follow these steps:\n\n" "the selected action.\n" "Players can join the game at any time. To leave a game, " "use /leave. If a player takes more than 90 seconds to play, " - "you can use /skip to skip that player.\n\n" + "you can use /skip to skip that player. Use /notify_me to " + "receive a private message when a new game is started.\n\n" "Language and other settings: /settings\n" "Other commands (only game creator):\n" "/close - Close lobby\n"