def message(cls, bot, update): cmd_chat_history = chat_history.history[update.message.from_user.id][0] if len(cmd_chat_history) == 1: if cmd_chat_history[-1] == "Search": reply_markup = telegram.ReplyKeyboardRemove(selective=True) bot.send_message( chat_id=update.message.chat_id, text="Search a code or an actress. e.g. ABP-231 or 桃乃木かな", reply_markup=reply_markup, reply_to_message_id=update.message.message_id) elif cmd_chat_history[-1] == "New": bot.send_message(chat_id=update.message.chat_id, text="How many results? [Integer]") elif cmd_chat_history[-1] == "Random": cls.random(bot, update) chat_history.clear_history(update.message.from_user.id) elif cmd_chat_history[-1] == "Brief": bot.send_message(chat_id=update.message.chat_id, text="Search a code. e.g. ABP-231") elif cmd_chat_history[-1] == "Magnet": bot.send_message(chat_id=update.message.chat_id, text="Search a code. e.g. ABP-231") elif len(cmd_chat_history) == 2: if cmd_chat_history[-2] == "Search": if re.search(r"\d", update.message.text): res = Functions.search_by_code(update.message.text) send_av(bot, update, res) chat_history.clear_history(update.message.from_user.id) else: if cmd_chat_history[-2] == "Search": bot.send_message(chat_id=update.message.chat_id, text="How many results? [Integer]") elif cmd_chat_history[-2] == "New": send_brief( bot, update, Functions.get_newly_released(int(cmd_chat_history[-1]), False)) chat_history.clear_history(update.message.from_user.id) elif cmd_chat_history[-2] == "Brief": send_brief(bot, update, Functions.get_brief(cmd_chat_history[-1])) chat_history.clear_history(update.message.from_user.id) elif cmd_chat_history[-2] == "Magnet": send_magnet(bot, update, Functions.get_magnet(cmd_chat_history[-1])) chat_history.clear_history(update.message.from_user.id) elif len(cmd_chat_history) == 3: if cmd_chat_history[-3] == "Search": res, _ = Functions.search_by_actress(cmd_chat_history[-2], int(cmd_chat_history[-1])) send_brief(bot, update, res) chat_history.clear_history(update.message.from_user.id) return
def search_magnet_by_code(): params = json.loads(request.data.decode("utf-8")) print(params) res = [] if params["code"]: res = Functions.get_magnet(params["code"]) if res: res = [x.to_dict() for x in res] rsp = jsonify(res) rsp.headers["Access-Control-Allow-Origin"] = "*" return rsp
def get_magnet(bot, update, args): send_magnet(bot, update, Functions.get_magnet(args[0]))