def invalid_invitation(self, bot, update, trainer, database): """Handes Invalid Invidation""" bot.send_message( update.message.chat_id, text="Invalid Invitationcode :( Maybe... IT WAS A TRAP?!") Trainer.create_new(database, trainer.id, update.message.from_user.first_name, 800)
def start(self, bot: Bot, update: Update, trainer: Trainer, database: Database): """Start cmd""" id = update.message.from_user.id name = update.message.from_user.first_name data = update.message.text invcode = data[7:] if trainer.does_exist(database): if data.find("fancycode") != -1: self.daily_code(bot, update, trainer, database) else: self.already_registered(bot, update, trainer) else: if invcode != None and invcode != " " and invcode != "" and int( invcode) != id: inviter = Trainer(int(invcode)) if inviter.does_exist(database): self.invite_code(bot, update, trainer, inviter, database) else: self.invalid_invitation(bot, update, trainer, database) else: Trainer.create_new(database, id, name, 800) self.register_user(bot, update, trainer, database)
def invite_code(self, bot, update, invcode, trainer: Trainer, inviter: Trainer, database: Database): """Handles Invite Code""" inviter.load_values(database, "pokedollar") inviter.pokedollar = inviter.pokedollar + 400 inviter.update_values(database, "pokedollar") Trainer.create_new(database, id, update.message.from_user.first_name, 1200) bot.send_message( update.message.chat_id, text="Invited by {0}! You both got 400₱ as gift!".format( inviter.name)) logger.info("{0} with ID {1} was recuted by {2} with ID {3}".format( update.message.from_user.first_name, id, inviter.name, inviter.id))