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