forked from test/mau_mau_bot
106 lines
3.6 KiB
Python
106 lines
3.6 KiB
Python
#!/usr/bin/env python3
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# 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 import ReplyKeyboardMarkup, Emoji
|
|
from telegram.ext import CommandHandler, RegexHandler
|
|
|
|
from utils import send_async
|
|
from user_setting import UserSetting
|
|
from utils import _, user_locale
|
|
from shared_vars import dispatcher
|
|
|
|
available_locales = [['en_US', 'de_DE'],
|
|
['it_IT', 'pt_BR']]
|
|
|
|
|
|
@user_locale
|
|
def show_settings(bot, update):
|
|
chat = update.message.chat
|
|
|
|
if update.message.chat.type != 'private':
|
|
send_async(bot, chat.id,
|
|
text=_("Please edit your settings in a private chat with "
|
|
"the bot."))
|
|
return
|
|
|
|
us = UserSetting.get(id=update.message.from_user.id)
|
|
|
|
if not us:
|
|
us = UserSetting(id=update.message.from_user.id)
|
|
|
|
if not us.stats:
|
|
stats = Emoji.BAR_CHART + ' ' + _("Enable statistics")
|
|
else:
|
|
stats = Emoji.CROSS_MARK + ' ' + _("Delete all statistics")
|
|
|
|
kb = [[stats], [Emoji.EARTH_GLOBE_EUROPE_AFRICA + ' ' + _("Language")]]
|
|
send_async(bot, chat.id, text=Emoji.WRENCH + ' ' + _("Settings"),
|
|
reply_markup=ReplyKeyboardMarkup(keyboard=kb,
|
|
one_time_keyboard=True))
|
|
|
|
|
|
@user_locale
|
|
def kb_select(bot, update, groups):
|
|
chat = update.message.chat
|
|
user = update.message.from_user
|
|
option = groups[0]
|
|
|
|
if option == Emoji.BAR_CHART:
|
|
us = UserSetting.get(id=user.id)
|
|
us.stats = True
|
|
send_async(bot, chat.id, text=_("Enabled statistics!"))
|
|
|
|
elif option == Emoji.EARTH_GLOBE_EUROPE_AFRICA:
|
|
send_async(bot, chat.id, text=_("Select locale"),
|
|
reply_markup=ReplyKeyboardMarkup(keyboard=available_locales,
|
|
one_time_keyboard=True))
|
|
|
|
elif option == Emoji.CROSS_MARK:
|
|
us = UserSetting.get(id=user.id)
|
|
us.stats = False
|
|
us.first_places = 0
|
|
us.games_played = 0
|
|
us.cards_played = 0
|
|
send_async(bot, chat.id, text=_("Deleted and disabled statistics!"))
|
|
|
|
|
|
@user_locale
|
|
def locale_select(bot, update, groups):
|
|
chat = update.message.chat
|
|
user = update.message.from_user
|
|
option = groups[0]
|
|
|
|
if option in [locale for row in available_locales for locale in row]:
|
|
us = UserSetting.get(id=user.id)
|
|
us.lang = option
|
|
_.push(option)
|
|
send_async(bot, chat.id, text=_("Set locale!"))
|
|
_.pop()
|
|
|
|
|
|
dispatcher.add_handler(CommandHandler('settings', show_settings))
|
|
dispatcher.add_handler(RegexHandler('^([' + Emoji.BAR_CHART +
|
|
Emoji.EARTH_GLOBE_EUROPE_AFRICA +
|
|
Emoji.CROSS_MARK + ']) .+$',
|
|
kb_select, pass_groups=True))
|
|
dispatcher.add_handler(RegexHandler(r'^(\w\w_\w\w)$',
|
|
locale_select, pass_groups=True))
|