def generate_ticket_handle(text, context):
return gt.make_ticket(fio=context['name'],
from_=context['departure_city'],
to=context['destination_city'],
date=context['departure_date'],
time=context['time_flight'],
landing_time=context['landing_time'])
def check_phone_number(update, context):
"""
Проверка номера телефона пользователя и занесение данных в БД
"""
if update.message.text == "да":
log.info(f"получен верный номер телефона")
update.message.reply_text(f"Спасибо, {context.user_data['name']}! Ожидайте звонка. Ваш билет.")
image_ticket = make_ticket(
fio=context.user_data['name'], from_=context.user_data["city_of_departure"],
to=context.user_data["arrival_city"], date=context.user_data["date_of_departure"],
person=context.user_data["name_avatar"]
)
context.bot.send_photo(chat_id=update.message.chat.id, photo=open(image_ticket, 'rb'))
Registration(
name=context.user_data['name'],
date_of_departure=context.user_data["date_of_departure"],
city_of_departure=context.user_data["city_of_departure"],
arrival_city=context.user_data["arrival_city"],
flight=context.user_data["flight"],
number_of_seats=context.user_data["number_of_seats"],
comment=context.user_data["comment"],
phone_number=context.user_data["phone_number"],
name_avatar=context.user_data["name_avatar"]
)
return ConversationHandler.END
else:
log.info(f"получен неверный номер телефона")
update.message.reply_text("Введите номер телефона.")
return PHONE_NUMBER
def test_image_generation(self):
ticket_file = make_ticket(fio='Клементьев Данила Сергеевич',
from_='Moscow',
to='London',
date='27/02/2020')
with open('files/test_ticket.png', 'rb') as expected_file:
expected_bytes = expected_file.read()
assert ticket_file.read() == expected_bytes
def generate_ticket(text, context):
data = context['done_steps']
return make_ticket(fio=data['name'], from_=data['dep_city'], to=data['dest_city'], date=data['fly_date'])