def add_pet_photo_state(message, user, is_entry=False):
if is_entry:
bot.send_message(message.chat.id,
DICTIONARY[user.language]['add_pet_photo_msg'],
reply_markup=get_back_menu_keyboard(user.language))
else:
if message.text == DICTIONARY[user.language]['back_menu_btn']:
pet = Pet.objects(pet_id=user.current_pet).first()
if pet:
pet.delete()
return True, 'main_menu_state'
else:
if message.photo:
pet = Pet.objects(pet_id=user.current_pet).first()
pet.photo_link = message.photo[-1].file_id
pet.save()
bot.send_message(message.chat.id,
DICTIONARY[user.language]['ok_msg'])
return True, 'add_pet_confirmation_state'
else:
bot.send_message(
message.chat.id,
DICTIONARY[user.language]['photo_not_valid_msg'])
return True, 'add_pet_photo_state'
return False, ''
def add_pet_sex_state(message, user, is_entry=False):
if is_entry:
bot.send_message(message.chat.id,
DICTIONARY[user.language]['add_pet_sex_msg'],
reply_markup=get_add_pet_sex_keyboard(user.language))
else:
if message.text == DICTIONARY[user.language]['male_pet_btn']:
pet = Pet.objects(pet_id=user.current_pet).first()
pet.sex = False
pet.save()
bot.send_message(message.chat.id,
DICTIONARY[user.language]['ok_msg'])
return True, 'add_pet_age_state'
elif message.text == DICTIONARY[user.language]['female_pet_btn']:
pet = Pet.objects(pet_id=user.current_pet).first()
pet.sex = True
pet.save()
bot.send_message(message.chat.id,
DICTIONARY[user.language]['ok_msg'])
return True, 'add_pet_age_state'
elif message.text == DICTIONARY[user.language]['back_menu_btn']:
pet = Pet.objects(pet_id=user.current_pet).first()
if pet:
pet.delete()
return True, 'main_menu_state'
else:
bot.send_message(message.chat.id,
DICTIONARY[user.language]['use_keyboard_msg'],
reply_markup=get_add_pet_sex_keyboard(
user.language))
return False, ''
def add_pet_confirmation_state(message, user, is_entry=False):
if is_entry:
pet = Pet.objects(pet_id=user.current_pet).first()
message_answer = DICTIONARY[user.language]['print_info_msg'].format(
pet.kind, (pet.name if pet.name is not None else
DICTIONARY[user.language]['add_pet_no_name_btn']),
(DICTIONARY[user.language]['female_pet_btn']
if pet.sex else DICTIONARY[user.language]['male_pet_btn']),
pet.breed, pet.age, pet.description, pet.user_id)
bot.send_photo(message.chat.id,
photo=pet.photo_link,
caption=message_answer,
parse_mode='HTML',
reply_markup=get_add_pet_confirmation_keyboard(
user.language))
else:
if message.text == DICTIONARY[user.language]['yes_btn']:
pet = Pet.objects(pet_id=user.current_pet).first()
# pet.view = True
pet.save()
bot.send_photo(
ADMIN_CHAT_ID,
photo=pet.photo_link,
caption=DICTIONARY[user.language]['print_info_msg'].format(
pet.kind,
(pet.name if pet.name is not None else
DICTIONARY[user.language]['add_pet_no_name_btn']),
(DICTIONARY[user.language]['female_pet_btn'] if pet.sex
else DICTIONARY[user.language]['male_pet_btn']),
pet.breed, pet.age, pet.description, pet.user_id),
reply_markup=get_moder_keyboard(language=user.language,
callback_data=pet.pet_id),
parse_mode='html')
bot.send_message(message.chat.id,
DICTIONARY[user.language]['add_ok_pet_msg'])
return True, 'main_menu_state'
elif message.text == DICTIONARY[user.language]['no_btn']:
pet = Pet.objects(pet_id=user.current_pet).first()
if pet:
pet.delete()
bot.send_message(message.chat.id,
DICTIONARY[user.language]['add_no_pet_msg'])
return True, 'main_menu_state'
elif message.text == DICTIONARY[user.language]['back_menu_btn']:
pet = Pet.objects(pet_id=user.current_pet).first()
if pet:
pet.delete()
return True, 'main_menu_state'
else:
bot.send_message(message.chat.id,
DICTIONARY[user.language]['use_keyboard_msg'],
reply_markup=get_add_pet_confirmation_keyboard(
user.language))
return False, ''
def callback_handler(call, user):
pet = Pet.objects(pet_id=re.search('[0-9]+', call.data).group()).first()
if pet:
if 'apply' in call.data:
pet.view = True
pet.save()
user = User.objects(user_id=pet.user_id).first()
if user:
bot.send_message(user.user_id,
DICTIONARY[user.language]['applied_msg'])
else:
user = User.objects(user_id=pet.user_id).first()
pet.delete()
if user:
bot.send_message(user.user_id,
DICTIONARY[user.language]['denied_msg'])
bot.edit_message_reply_markup(chat_id=call.message.chat.id,
message_id=call.message.message_id)
else:
bot.edit_message_text(chat_id=call.message.chat.id,
message_id=call.message.message_id,
text='X')
bot.edit_message_reply_markup(chat_id=call.message.chat.id,
message_id=call.message.message_id,
reply_markup=types.ReplyKeyboardRemove())
bot.answer_callback_query(call.id)
def add_pet_description_state(message, user, is_entry=False):
if is_entry:
pet = Pet.objects(pet_id=user.current_pet).first()
symbols_left = 1024 - len(
DICTIONARY[user.language]['print_info_msg'].format(
pet.kind, (pet.name if pet.name is not None else
DICTIONARY[user.language]['add_pet_no_name_btn']),
(DICTIONARY[user.language]['female_pet_btn']
if pet.sex else DICTIONARY[user.language]['male_pet_btn']),
pet.breed, pet.age, '', pet.user_id))
bot.send_message(message.chat.id,
DICTIONARY[user.language]
['add_pet_description_msg'].format(symbols_left),
reply_markup=get_back_menu_keyboard(user.language))
else:
if message.text == DICTIONARY[user.language]['back_menu_btn']:
pet = Pet.objects(pet_id=user.current_pet).first()
if pet:
pet.delete()
return True, 'main_menu_state'
else:
pet = Pet.objects(pet_id=user.current_pet).first()
symbols_left = 1024 - len(
DICTIONARY[user.language]['print_info_msg'].format(
pet.kind,
(pet.name if pet.name is not None else
DICTIONARY[user.language]['add_pet_no_name_btn']),
(DICTIONARY[user.language]['female_pet_btn'] if pet.sex
else DICTIONARY[user.language]['male_pet_btn']),
pet.breed, pet.age, '', pet.user_id))
if len(message.text) <= symbols_left:
pet = Pet.objects(pet_id=user.current_pet).first()
pet.description = message.text
pet.save()
bot.send_message(message.chat.id,
DICTIONARY[user.language]['ok_msg'])
return True, 'add_pet_photo_state'
else:
bot.send_message(
message.chat.id,
DICTIONARY[user.language]['add_pet_description_bad_msg'])
return True, 'add_pet_description_state'
return False, ''
def want_take_pet_state(message, user, is_entry=False):
if is_entry:
bot.send_message(message.chat.id,
DICTIONARY[user.language]['want_take_pet_msg'],
reply_markup=get_want_take_pet_keyboard(
user.language))
else:
if message.text == DICTIONARY[user.language]['dog_btn']:
pets = Pet.objects(
Q(kind=DICTIONARY['ua']['dog'])
| Q(kind=DICTIONARY['ru']['dog']))
if pets:
for pet in pets:
if pet.view:
message_answer = DICTIONARY[
user.language]['print_info_msg'].format(
pet.kind,
(pet.name if pet.name is not None else
DICTIONARY[user.language]['no_name']),
(DICTIONARY[user.language]['female_pet_btn']
if pet.sex else
DICTIONARY[user.language]['male_pet_btn']),
pet.breed, pet.age, pet.description,
pet.user_id)
bot.send_photo(message.chat.id,
photo=pet.photo_link,
caption=message_answer,
parse_mode='HTML')
else:
bot.send_message(
message.chat.id,
DICTIONARY[user.language]['not_found_pet_msg'],
parse_mode='HTML')
elif message.text == DICTIONARY[user.language]['cat_btn']:
pets = Pet.objects(
Q(kind=DICTIONARY['ua']['cat'])
| Q(kind=DICTIONARY['ru']['cat']))
if pets:
for pet in pets:
if pet.view:
message_answer = DICTIONARY[
user.language]['print_info_msg'].format(
pet.kind,
(pet.name if pet.name is not None else
DICTIONARY[user.language]['no_name']),
(DICTIONARY[user.language]['female_pet_btn']
if pet.sex else
DICTIONARY[user.language]['male_pet_btn']),
pet.breed, pet.age, pet.description,
pet.user_id)
bot.send_photo(message.chat.id,
photo=pet.photo_link,
caption=message_answer,
parse_mode='HTML')
else:
bot.send_message(
message.chat.id,
DICTIONARY[user.language]['not_found_pet_msg'],
parse_mode='HTML')
elif message.text == DICTIONARY[user.language]['different_btn']:
pets = Pet.objects()
message_answer = ''
if pets:
for pet in pets:
if pet.view and pet.kind != DICTIONARY['ru']['dog'] and pet.kind != DICTIONARY['ru']['cat'] \
and pet.kind != DICTIONARY['ua']['dog'] and pet.kind != DICTIONARY['ua']['cat']:
message_answer = DICTIONARY[
user.language]['print_info_msg'].format(
pet.kind,
(pet.name if pet.name is not None else
DICTIONARY[user.language]['no_name']),
(DICTIONARY[user.language]['female_pet_btn']
if pet.sex else
DICTIONARY[user.language]['male_pet_btn']),
pet.breed, pet.age, pet.description,
pet.user_id)
bot.send_photo(message.chat.id,
photo=pet.photo_link,
caption=message_answer,
parse_mode='HTML')
if message_answer == '':
if message_answer == '':
bot.send_message(
message.chat.id,
DICTIONARY[user.language]['not_found_pet_msg'],
parse_mode='HTML')
else:
bot.send_message(
message.chat.id,
DICTIONARY[user.language]['not_found_pet_msg'],
parse_mode='HTML')
elif message.text == DICTIONARY[user.language]['show_all_btn']:
pets = Pet.objects()
if pets:
for pet in Pet.objects():
if pet.view:
message_answer = DICTIONARY[
user.language]['print_info_msg'].format(
pet.kind,
(pet.name if pet.name is not None else
DICTIONARY[user.language]['no_name']),
(DICTIONARY[user.language]['female_pet_btn']
if pet.sex else
DICTIONARY[user.language]['male_pet_btn']),
pet.breed, pet.age, pet.description,
pet.user_id)
bot.send_photo(message.chat.id,
photo=pet.photo_link,
caption=message_answer,
parse_mode='HTML')
else:
bot.send_message(
message.chat.id,
DICTIONARY[user.language]['not_found_pet_msg'],
parse_mode='HTML')
elif message.text == DICTIONARY[user.language]['back_menu_btn']:
return True, 'main_menu_state'
else:
bot.send_message(message.chat.id,
DICTIONARY[user.language]['use_keyboard_msg'],
reply_markup=get_want_take_pet_keyboard(
user.language))
return False, ''
def add_pet_kind_state(message, user, is_entry=False):
if is_entry:
bot.send_message(message.chat.id,
DICTIONARY[user.language]['add_pet_kind_msg'],
reply_markup=get_add_pet_kind_state_keyboard(
user.language))
else:
if message.text == DICTIONARY[user.language]['dog_btn']:
pet = Pet(user_id=user.user_id,
kind=DICTIONARY[user.language]['dog'])
pet.save()
user.current_pet = pet.pet_id
user.save()
bot.send_message(message.chat.id,
DICTIONARY[user.language]['ok_msg'])
return True, 'add_pet_name_state'
elif message.text == DICTIONARY[user.language]['cat_btn']:
pet = Pet(user_id=user.user_id,
kind=DICTIONARY[user.language]['cat'])
pet.save()
user.current_pet = pet.pet_id
user.save()
bot.send_message(message.chat.id,
DICTIONARY[user.language]['ok_msg'])
return True, 'add_pet_name_state'
elif message.text == DICTIONARY[user.language]['back_menu_btn']:
pet = Pet.objects(pet_id=user.current_pet).first()
if pet:
pet.delete()
return True, 'main_menu_state'
else:
pet = Pet(user_id=user.user_id, kind=message.text)
pet.save()
user.current_pet = pet.pet_id
user.save()
bot.send_message(message.chat.id,
DICTIONARY[user.language]['ok_msg'])
return True, 'add_pet_name_state'
return False, ''