#!/usr/bin/env python3
"""
Создание Excel дашборда с аудитом отелей
Лист 1: Дашборд с графиками и статистикой
Лист 2: Детальная таблица аудита по отелям
"""

import psycopg2
from psycopg2.extras import RealDictCursor
from urllib.parse import unquote
import pandas as pd
import openpyxl
from openpyxl import Workbook
from openpyxl.styles import Font, Alignment, PatternFill, Border, Side, NamedStyle
from openpyxl.chart import BarChart, PieChart, LineChart, Reference
from openpyxl.chart.label import DataLabelList
from openpyxl.utils.dataframe import dataframe_to_rows
from openpyxl.drawing.image import Image
from datetime import datetime
import json

DB_CONFIG = {
    'host': '147.45.189.234',
    'port': 5432,
    'database': 'default_db',
    'user': 'gen_user',
    'password': unquote('2~~9_%5EkVsU%3F2%5CS')
}

def get_audit_data():
    """Получить данные аудита из БД"""
    conn = psycopg2.connect(**DB_CONFIG, cursor_factory=RealDictCursor)
    cur = conn.cursor()
    
    # Получаем данные аудита с информацией об отелях
    cur.execute("""
        SELECT 
            har.hotel_id,
            har.hotel_name,
            har.region_name,
            har.website,
            har.has_website,
            har.total_score,
            har.max_score,
            har.score_percentage,
            har.audit_date,
            har.audit_version,
            har.criteria_results,
            hm.full_name,
            hm.website_address,
            hm.owner_inn,
            hm.owner_ogrn,
            hm.addresses,
            hm.phone,
            hm.email
        FROM hotel_audit_results har
        LEFT JOIN hotel_main hm ON hm.id = har.hotel_id
        ORDER BY har.audit_date DESC
    """)
    
    audit_data = cur.fetchall()
    
    # Статистика по регионам
    cur.execute("""
        SELECT 
            region_name,
            COUNT(*) as total_hotels,
            AVG(score_percentage) as avg_score,
            MAX(audit_date) as last_audit,
            COUNT(CASE WHEN has_website = true THEN 1 END) as with_website,
            COUNT(CASE WHEN score_percentage >= 50 THEN 1 END) as compliant_hotels
        FROM hotel_audit_results 
        WHERE region_name IS NOT NULL
        GROUP BY region_name 
        ORDER BY total_hotels DESC
    """)
    
    region_stats = cur.fetchall()
    
    # Статистика по критериям (пропускаем пока - сложная структура JSONB)
    criteria_stats = []
    
    cur.close()
    conn.close()
    
    return audit_data, region_stats, criteria_stats

def create_dashboard_sheet(workbook, region_stats, criteria_stats):
    """Создать лист дашборда"""
    ws = workbook.active
    ws.title = "📊 Дашборд"
    
    # Стили
    header_font = Font(name='Arial', size=14, bold=True, color='FFFFFF')
    header_fill = PatternFill(start_color='366092', end_color='366092', fill_type='solid')
    subheader_font = Font(name='Arial', size=12, bold=True)
    normal_font = Font(name='Arial', size=10)
    
    # Заголовок
    ws['A1'] = "🏨 ДАШБОРД АУДИТА ОТЕЛЕЙ"
    ws['A1'].font = Font(name='Arial', size=16, bold=True, color='366092')
    ws['A1'].alignment = Alignment(horizontal='center')
    ws.merge_cells('A1:H1')
    
    # Общая статистика
    ws['A3'] = "📈 ОБЩАЯ СТАТИСТИКА"
    ws['A3'].font = subheader_font
    ws['A3'].fill = PatternFill(start_color='E7E6E6', end_color='E7E6E6', fill_type='solid')
    
    # Подсчитываем общую статистику
    total_hotels = sum(row['total_hotels'] for row in region_stats)
    total_with_website = sum(row['with_website'] for row in region_stats)
    total_compliant = sum(row['compliant_hotels'] for row in region_stats)
    avg_score = sum(row['avg_score'] * row['total_hotels'] for row in region_stats) / total_hotels if total_hotels > 0 else 0
    
    ws['A4'] = f"Всего отелей проаудировано: {total_hotels}"
    ws['A5'] = f"Отелей с сайтами: {total_with_website} ({total_with_website/total_hotels*100:.1f}%)"
    ws['A6'] = f"Соответствующих требованиям (≥50%): {total_compliant} ({total_compliant/total_hotels*100:.1f}%)"
    ws['A7'] = f"Средний балл соответствия: {avg_score:.1f}%"
    
    for cell in ['A4', 'A5', 'A6', 'A7']:
        ws[cell].font = normal_font
    
    # Статистика по регионам
    ws['A9'] = "📍 СТАТИСТИКА ПО РЕГИОНАМ"
    ws['A9'].font = subheader_font
    ws['A9'].fill = PatternFill(start_color='E7E6E6', end_color='E7E6E6', fill_type='solid')
    
    # Заголовки таблицы регионов
    headers = ['Регион', 'Отелей', 'С сайтами', 'Соответствуют', 'Средний балл', 'Последний аудит']
    for i, header in enumerate(headers, 1):
        cell = ws.cell(row=10, column=i, value=header)
        cell.font = header_font
        cell.fill = header_fill
        cell.alignment = Alignment(horizontal='center')
    
    # Данные по регионам
    for i, region in enumerate(region_stats, 11):
        ws.cell(row=i, column=1, value=region['region_name'])
        ws.cell(row=i, column=2, value=region['total_hotels'])
        ws.cell(row=i, column=3, value=region['with_website'])
        ws.cell(row=i, column=4, value=region['compliant_hotels'])
        ws.cell(row=i, column=5, value=f"{region['avg_score']:.1f}%")
        ws.cell(row=i, column=6, value=region['last_audit'].strftime('%d.%m.%Y') if region['last_audit'] else '')
        
        # Форматирование
        for col in range(1, 7):
            ws.cell(row=i, column=col).font = normal_font
            ws.cell(row=i, column=col).alignment = Alignment(horizontal='center')
    
    # График по регионам
    chart1 = BarChart()
    chart1.title = "Количество отелей по регионам"
    chart1.x_axis.title = "Регионы"
    chart1.y_axis.title = "Количество отелей"
    
    data = Reference(ws, min_col=2, min_row=10, max_row=10+len(region_stats), max_col=2)
    cats = Reference(ws, min_col=1, min_row=11, max_row=10+len(region_stats))
    chart1.add_data(data, titles_from_data=False)
    chart1.set_categories(cats)
    chart1.height = 10
    chart1.width = 15
    
    ws.add_chart(chart1, "A15")
    
    # Информация о дате создания
    ws['A30'] = f"📅 Отчет создан: {datetime.now().strftime('%d.%m.%Y %H:%M')}"
    ws['A30'].font = Font(name='Arial', size=10, italic=True, color='666666')
    
    # Настройка ширины колонок
    column_widths = [25, 10, 12, 15, 12, 15]
    for i, width in enumerate(column_widths, 1):
        ws.column_dimensions[openpyxl.utils.get_column_letter(i)].width = width

def create_audit_sheet(workbook, audit_data):
    """Создать лист детального аудита"""
    ws = workbook.create_sheet("🏨 Аудит отелей")
    
    # Стили
    header_font = Font(name='Arial', size=12, bold=True, color='FFFFFF')
    header_fill = PatternFill(start_color='366092', end_color='366092', fill_type='solid')
    normal_font = Font(name='Arial', size=9)
    
    # Заголовки
    headers = [
        'Отель', 'Регион', 'Сайт', 'Есть сайт', 'Балл', 'Процент',
        'ИНН', 'ОГРН', 'Адрес', 'Телефон', 'Email', 'Дата аудита'
    ]
    
    for i, header in enumerate(headers, 1):
        cell = ws.cell(row=1, column=i, value=header)
        cell.font = header_font
        cell.fill = header_fill
        cell.alignment = Alignment(horizontal='center')
    
    # Данные
    for i, hotel in enumerate(audit_data, 2):
        ws.cell(row=i, column=1, value=hotel['hotel_name'] or hotel['full_name'])
        ws.cell(row=i, column=2, value=hotel['region_name'])
        ws.cell(row=i, column=3, value=hotel['website'] or hotel['website_address'])
        ws.cell(row=i, column=4, value='Да' if hotel['has_website'] else 'Нет')
        ws.cell(row=i, column=5, value=f"{hotel['total_score']}/{hotel['max_score']}")
        ws.cell(row=i, column=6, value=f"{hotel['score_percentage']:.1f}%")
        ws.cell(row=i, column=7, value=hotel['owner_inn'])
        ws.cell(row=i, column=8, value=hotel['owner_ogrn'])
        ws.cell(row=i, column=9, value=str(hotel['addresses']) if hotel['addresses'] else '')
        ws.cell(row=i, column=10, value=hotel['phone'])
        ws.cell(row=i, column=11, value=hotel['email'])
        ws.cell(row=i, column=12, value=hotel['audit_date'].strftime('%d.%m.%Y %H:%M') if hotel['audit_date'] else '')
        
        # Цветовое кодирование процента
        percentage = hotel['score_percentage'] or 0
        if percentage >= 70:
            fill_color = 'C6EFCE'  # Зеленый
        elif percentage >= 40:
            fill_color = 'FFEB9C'  # Желтый
        else:
            fill_color = 'FFC7CE'  # Красный
            
        ws.cell(row=i, column=6).fill = PatternFill(start_color=fill_color, end_color=fill_color, fill_type='solid')
        
        # Форматирование
        for col in range(1, 13):
            ws.cell(row=i, column=col).font = normal_font
            ws.cell(row=i, column=col).alignment = Alignment(horizontal='center')
    
    # Настройка ширины колонок
    column_widths = [30, 20, 25, 10, 10, 10, 15, 15, 30, 15, 20, 15]
    for i, width in enumerate(column_widths, 1):
        ws.column_dimensions[openpyxl.utils.get_column_letter(i)].width = width
    
    # Фильтры
    ws.auto_filter.ref = f"A1:{openpyxl.utils.get_column_letter(len(headers))}{len(audit_data)+1}"
    
    # Заморозка заголовков
    ws.freeze_panes = "A2"

def main():
    """Основная функция"""
    print("🚀 СОЗДАНИЕ EXCEL ДАШБОРДА")
    print("=" * 50)
    
    # Получаем данные
    print("📊 Загружаем данные из БД...")
    audit_data, region_stats, criteria_stats = get_audit_data()
    
    print(f"✅ Загружено:")
    print(f"  🏨 Отелей: {len(audit_data)}")
    print(f"  📍 Регионов: {len(region_stats)}")
    print(f"  🎯 Критериев: {len(criteria_stats)}")
    
    # Создаем Excel файл
    print("\n📝 Создаем Excel файл...")
    workbook = Workbook()
    
    # Лист дашборда
    print("📊 Создаем дашборд...")
    create_dashboard_sheet(workbook, region_stats, criteria_stats)
    
    # Лист аудита
    print("🏨 Создаем таблицу аудита...")
    create_audit_sheet(workbook, audit_data)
    
    # Сохраняем файл
    timestamp = datetime.now().strftime("%Y%m%d_%H%M%S")
    filename = f"hotel_audit_dashboard_{timestamp}.xlsx"
    
    workbook.save(filename)
    
    print(f"\n✅ Дашборд сохранен: {filename}")
    print(f"📊 Листы:")
    print(f"  📈 Дашборд - графики и статистика")
    print(f"  🏨 Аудит отелей - детальная таблица")

if __name__ == "__main__":
    main()