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]))