SmartSolTech/Touchh
1
0
Fork 0
Code Issues Pull Requests 2 Actions Packages Projects Releases Wiki Activity

plugins devel

Browse Source
This commit is contained in:
trevor
2024-12-23 21:23:45 +09:00
parent d850f50e25
commit 77bf0b8381
5 changed files with 225 additions and 203 deletions
Download Patch File Download Diff File Expand all files Collapse all files

67
bot/operations/statistics.py
View File

@@ -16,6 +16,7 @@ from touchh.utils.log import CustomLogger
logger = CustomLogger(__name__).get_logger()
async def statistics(update: Update, context: ContextTypes.DEFAULT_TYPE):
"""Вывод списка отелей для статистики."""
@@ -37,7 +38,7 @@ async def statistics(update: Update, context: ContextTypes.DEFAULT_TYPE):
# Формируем кнопки для выбора отеля
keyboard = [
[InlineKeyboardButton(hotel.hotel.name, callback_data=f"stats_hotel_{hotel.hotel.id}")]
[InlineKeyboardButton(f'🏨 {hotel.hotel.name}', callback_data=f"stats_hotel_{hotel.hotel.id}")]
for hotel in user_hotels
]
keyboard.append([InlineKeyboardButton("🏠 Главная", callback_data="main_menu")])
@@ -91,53 +92,8 @@ def ensure_datetime(value):
logging.warning(f"Получено значение неизвестного типа для преобразования в datetime: {value}")
return None
# async def generate_statistics(update: Update, context: ContextTypes.DEFAULT_TYPE):
# """Генерация и отправка статистики."""
# query = update.callback_query
# await query.answer()
# try:
# hotel_id = context.user_data.get("selected_hotel")
# if not hotel_id:
# raise ValueError(f"ID отеля не найден в user_data: {context.user_data}")
# period = query.data.split("_")[2]
# now = ensure_datetime(datetime.utcnow())
# # Получаем диапазон дат
# start_date, end_date = get_period_dates(period, now)
# reservations = await sync_to_async(list)(
# Reservation.objects.filter(
# hotel_id=hotel_id,
# check_in__gte=start_date,
# check_in__lte=end_date
# ).select_related('hotel')
# )
# if not reservations:
# await query.edit_message_text("Нет данных для статистики за выбранный период.")
# return
# hotel = await sync_to_async(Hotel.objects.get)(id=hotel_id)
# file_path = await generate_pdf_report(hotel.name, reservations, start_date, end_date)
# with open(file_path, "rb") as file:
# await query.message.reply_document(document=file, filename=f"{hotel.name}_report.pdf")
# if os.path.exists(file_path):
# os.remove(file_path)
# except Exception as e:
# logging.error(f"Ошибка в generate_statistics: {str(e)}", exc_info=True)
# logging.error(f'start_date_type: {type(start_date)}, \n end_date_type: {type(end_date)}\n')
# await query.edit_message_text(f"Произошла ошибка: {str(e)}")
async def generate_statistics(update: Update, context: ContextTypes.DEFAULT_TYPE):
"""Генерация и отправка статистики."""
logger = CustomLogger(__name__).get_logger()
query = update.callback_query
await query.answer()
@@ -166,27 +122,17 @@ async def generate_statistics(update: Update, context: ContextTypes.DEFAULT_TYPE
hotel = await sync_to_async(Hotel.objects.get)(id=hotel_id)
print(f"[DEBUG] start_date: {start_date}, type: {type(start_date)}")
print(f"[DEBUG] end_date: {end_date}, type: {type(end_date)}")
file_path = await generate_pdf_report(hotel.name, reservations, start_date, end_date)
# Генерация PDF-отчета
file_path = await generate_pdf_report(
hotel.name,
reservations,
start_date=start_date.strftime('%Y-%m-%d %H:%M:%S'),
end_date=end_date.strftime('%Y-%m-%d %H:%M:%S')
)
# Отправка PDF-файла
with open(file_path, "rb") as file:
await query.message.reply_document(document=file, filename=f"{hotel.name}_report.pdf")
# Удаление временного файла
if os.path.exists(file_path):
os.remove(file_path)
except Exception as e:
logger.error(f"Ошибка в generate_statistics: {str(e)}", exc_info=True)
logging.error(f"Ошибка в generate_statistics: {str(e)}", exc_info=True)
logging.error(f'start_date_type: {type(start_date)}, \n end_date_type: {type(end_date)}\n')
await query.edit_message_text(f"Произошла ошибка: {str(e)}")
@@ -235,3 +181,6 @@ async def stats_back(update: Update, context):
keyboard.append([InlineKeyboardButton("🏠 Главная", callback_data="main_menu")])
reply_markup = InlineKeyboardMarkup(keyboard)
await query.edit_message_text("Выберите отель:", reply_markup=reply_markup)