async def callback_underground_line(query: types.CallbackQuery):
if query.data == "Любая станция":
session = Session(bind=db.models.engine)
user = session.query(db.models.User).filter(db.models.User.telegram_id == query.from_user.id).one()
current_query = [q for q in user.queries if q.id == user.settings.editing_query][0]
current_query.underground = None
await bot.bot_instance.send_message(
text=messages.search_terms_answer(current_query.region,
current_query.deal,
current_query.underground,
current_query.rooms,
current_query.apartment_type,
current_query.price),
chat_id=query.from_user.id,
parse_mode='Markdown'
)
await asyncio.sleep(1)
await bot.bot_instance.send_message(
text=messages.SEARCH_TERMS,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.SEARCH_TERMS)
)
else:
metro_stations = filters.underground_voc[query.data].items()
metro_stations = [(s, f"underground_station {id}") for s, id in metro_stations]
await bot.bot_instance.send_message(
text=messages.GET_UNDERGROUND2,
chat_id=query.from_user.id,
disable_web_page_preview=True,
reply_markup=buttons.build_inlinekeyboard(buttons.Buttons(items=metro_stations)))
await query.answer()
async def get_price_from_msg(message: types.Message):
re_parser = re.search(r"(\d+)\s*-\s*(\d+)", message.text)
try:
min_price = float(re_parser[1].replace(',', '.'))
max_price = float(re_parser[2].replace(',', '.'))
except (IndexError, ValueError):
await message.answer(messages.GET_PRICE_FROM_MSG_ALERT)
return
session = Session(bind=db.models.engine)
user = session.query(db.models.User).filter(db.models.User.telegram_id == message.from_user.id).one()
current_query = [q for q in user.queries if q.id == user.settings.editing_query][0]
current_query.price = [min_price, max_price]
await bot.bot_instance.send_message(
text=messages.search_terms_answer(current_query.region,
current_query.deal,
current_query.underground,
current_query.rooms,
current_query.apartment_type,
current_query.price),
chat_id=message.from_user.id,
parse_mode='Markdown'
)
await asyncio.sleep(1)
await bot.bot_instance.send_message(
text=messages.SEARCH_TERMS,
chat_id=message.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.SEARCH_TERMS)
)
db.utils.safe_commit(session)
async def callback_get_underground_station(query: types.CallbackQuery):
session = Session(bind=db.models.engine)
user = session.query(db.models.User).filter(db.models.User.telegram_id == query.from_user.id).one()
current_query = [q for q in user.queries if q.id == user.settings.editing_query][0]
station = int(query.data.split(' ')[1])
current_query.underground = station
current_query.pages = None
await bot.bot_instance.send_message(
text=messages.search_terms_answer(current_query.region,
current_query.deal,
current_query.underground,
current_query.rooms,
current_query.apartment_type,
current_query.price),
chat_id=query.from_user.id,
parse_mode='Markdown'
)
await asyncio.sleep(1)
await bot.bot_instance.send_message(
text=messages.SEARCH_TERMS,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.SEARCH_TERMS)
)
db.utils.safe_commit(session)
await query.answer()
async def start(message: types.Message):
if not db.models.User.is_user(telegram_id=message.from_user.id):
print(message.from_user.first_name, message.from_user.last_name, 'connected')
db. models.User.insert_user(telegram_id=message.from_user.id,
first_name=message.from_user.first_name,
last_name=message.from_user.last_name)
await message.reply(messages.START_MESSAGE,
reply_markup=buttons.build_inlinekeyboard(['Задать критерии поиска']))
async def search_terms(message: types.Message):
try:
db.models.UserSettings.update_current_edit_msg(telegram_id=message.from_user.id,
message_id=message.message_id)
except Exception:
print(traceback.print_exc())
raise NotImplementedError
await message.reply(messages.SEARCH_TERMS,
reply_markup=buttons.build_inlinekeyboard(buttons.SEARCH_TERMS))
async def callback_search_terms(query: types.CallbackQuery):
if query.data == 'city':
await bot.bot_instance.send_message(
text=messages.GET_CITY,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.CITY)
)
elif query.data == 'deal':
await bot.bot_instance.send_message(
text=messages.GET_DEAL_TYPE,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.DEAL_TYPE)
)
elif query.data == 'underground':
await bot.bot_instance.send_message(
text=messages.GET_UNDERGROUND1,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.UNDERGROUND_LINE)
)
elif query.data == 'rooms':
await bot.bot_instance.send_message(
text=messages.GET_ROOMS,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.ROOMS)
)
elif query.data == 'apart_type':
session = Session(bind=db.models.engine)
user = session.query(db.models.User).filter(db.models.User.telegram_id == query.from_user.id).one()
current_query = [q for q in user.queries if q.id == user.settings.editing_query][0]
if current_query.deal == 'flatsale':
await bot.bot_instance.send_message(
text=messages.GET_APARTMENT_TYPE,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.APARTMENT_TYPE)
)
else:
await query.answer(messages.APART_TYPE_ALERT, show_alert=True)
elif query.data == 'price':
session = Session(bind=db.models.engine)
user = session.query(db.models.User).filter(db.models.User.telegram_id == query.from_user.id).one()
current_query = [q for q in user.queries if q.id == user.settings.editing_query][0]
if current_query.deal != 'flatsale':
items = buttons.PRICE['rent']
else:
items = buttons.PRICE['sale']
await bot.bot_instance.send_message(
text=messages.GET_PRICE,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(items)
)
else:
raise TypeError("Invalid query for app.handlers.callback_search_terms")
await query.answer()
async def callback_get_rooms(query: types.CallbackQuery):
session = Session(bind=db.models.engine)
user = session.query(db.models.User).filter(db.models.User.telegram_id == query.from_user.id).one()
current_query = [q for q in user.queries if q.id == user.settings.editing_query][0]
current_query.pages = None
if query.data != 'Готово!':
selected_rooms_type = filters.rooms_filter(query.data, mode='to_int')
if not current_query.rooms:
result = {*selected_rooms_type}
else:
result = set(current_query.rooms)
for room in selected_rooms_type:
if room not in current_query.rooms:
result.add(room)
else:
result.remove(room)
current_query.rooms = list(result)
db.utils.safe_commit(session)
await query.answer("Квартиры: {}".format(filters.rooms_filter(list(result), mode='array_to_str')))
else:
session = Session(bind=db.models.engine)
user = session.query(db.models.User).filter(db.models.User.telegram_id == query.from_user.id).one()
current_query = [q for q in user.queries if q.id == user.settings.editing_query][0]
await bot.bot_instance.send_message(
text=messages.search_terms_answer(current_query.region,
current_query.deal,
current_query.underground,
current_query.rooms,
current_query.apartment_type,
current_query.price),
chat_id=query.from_user.id,
parse_mode='Markdown'
)
await asyncio.sleep(1)
await bot.bot_instance.send_message(
text=messages.SEARCH_TERMS,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.SEARCH_TERMS)
)
session.close()
await query.answer()
async def callback_search_terms_main(query: types.CallbackQuery):
if query.data == 'Задать критерии поиска':
await bot.bot_instance.send_message(
text=messages.SETTINGS_FROM_START,
chat_id=query.from_user.id,
reply_markup=buttons.build_replykeyboard(buttons.MAIN))
await asyncio.sleep(2)
await bot.bot_instance.send_message(
text=messages.SEARCH_TERMS,
chat_id=query.from_user.id,
reply_markup=buttons.build_inlinekeyboard(buttons.SEARCH_TERMS)
)
try:
db.models.UserSettings.update_current_edit_msg(telegram_id=query.from_user.id,
message_id=query.message.message_id)
except Exception:
print(traceback.print_exc())
raise NotImplementedError
await query.answer()
async def notifications_settings(message: types.Message):
await message.answer(text=messages.NOTIFICATIONS_MAIN,
reply_markup=buttons.build_inlinekeyboard(buttons.NOTIFICATIONS))