def referral(message):
user = session.query(User).get(message.chat.id)
referral_user = session.query(User).filter_by(username=message.text.lower().replace(' ', '')).first()
if referral_user is not None:
referral_user.referral_invited.append(user)
referral_user.commit_to_db()
user.state = State.menu
markup = Markup.menu(lang=user.lang)
bot.send_message(message.chat.id, dictionary.register_new_user['hello'][user.lang].format(message.chat.first_name), reply_markup=markup)
def apply(self):
if self.accepted:
return
user = session.query(User).get(self.order_id)
user.balance += self.amount
self.accepted = True
session.add(self)
session.add(user)
session.commit()
def choose_language(message):
user = session.query(User).get(message.chat.id)
if message.text == Markup.choose_language(button=Lang.ru):
user.lang = Lang.ru
elif message.text == Markup.choose_language(button=Lang.en):
user.lang = Lang.en
else:
return
user.state = State.referral
user.commit_to_db()
bot.send_message(message.chat.id, dictionary.register_new_user['referral'][user.lang])
def webhook_for_donate():
user_id = request.args.get('id', type=int)
if user_id is None:
abort(400)
amount = request.args.get('amount', default=config.MAX_IN_SUM, type=float)
user = session.query(User).get(user_id)
if amount > float(config.MAX_IN_SUM) or amount < float(
config.MIN_IN_SUM) or user is None:
abort(400)
return render_template('donate.html',
user=user,
amount=amount,
merchant=config.TRANSACTION_MERCHANT,
payment_url=config.TRANSACTION_PAYMENT_URL[str(
user.lang)])
def menu(message):
user = session.query(User).get(message.chat.id)
state = Markup.menu(lang=user.lang, text=message.text)
bot.send_message(message.chat.id, str(state))
def reset_DEBUG(message):
user = session.query(User).get(message.chat.id)
session.delete(user)
session.commit()
from CupGamesBot.bot import bot, dictionary
from CupGamesBot.bot.markups import Markup
from CupGamesBot.database import session
from CupGamesBot.database.user import User
from CupGamesBot.database.state import State
from CupGamesBot.database.lang import Lang
@bot.message_handler(commands=['reset'])
def reset_DEBUG(message):
user = session.query(User).get(message.chat.id)
session.delete(user)
session.commit()
@bot.message_handler(func=lambda message: session.query(User).get(message.chat.id) is None)
def register_new_user(message):
# Register user
user = User(id=message.chat.id, username=message.chat.username.lower(), state=State.lang)
user.commit_to_db()
# Select lang
markup = Markup.choose_language()
bot.send_message(message.chat.id, dictionary.register_new_user['lang'], reply_markup=markup)
@bot.message_handler(func=lambda message: session.query(User).get(message.chat.id).state is State.lang)
def choose_language(message):
user = session.query(User).get(message.chat.id)
if message.text == Markup.choose_language(button=Lang.ru):
user.lang = Lang.ru
elif message.text == Markup.choose_language(button=Lang.en):