"""
Report generation module (HTML + PDF)
"""
import os
import logging
from datetime import datetime
from pathlib import Path
from typing import Dict, List
from weasyprint import HTML, CSS
from weasyprint.text.fonts import FontConfiguration
from .config import settings
logger = logging.getLogger(__name__)
class ReportGenerator:
"""Generator for HTML and PDF reports."""
def __init__(self):
"""Initialize report generator."""
self.reports_dir = Path(settings.reports_dir)
self.reports_dir.mkdir(parents=True, exist_ok=True)
def generate_html_content(
self,
sentiments: Dict,
total_items: int,
negative_items: List[Dict]
) -> str:
"""
Generate HTML report content.
Args:
sentiments: Dictionary with neg/pos/neu counts
total_items: Total number of items analyzed
negative_items: List of negative items
Returns:
HTML content as string
"""
percent_neg = (sentiments['neg'] / total_items * 100) if total_items > 0 else 0
percent_pos = (sentiments['pos'] / total_items * 100) if total_items > 0 else 0
percent_neu = (sentiments['neu'] / total_items * 100) if total_items > 0 else 0
# Crisis alert
crisis_level = "CRÍTICO" if percent_neg > 60 else "MODERADO" if percent_neg > 40 else "BAIXO"
crisis_color = "#d32f2f" if percent_neg > 60 else "#f57c00" if percent_neg > 40 else "#388e3c"
html = f"""
Relatório de Crise: {settings.search_query}
📊 RELATÓRIO DE CRISE: {settings.search_query}
Período: Últimos {settings.num_emails} emails de Google Alerts Total de alertas: {total_items} notícias analisadas Gerado em: {datetime.now().strftime('%d/%m/%Y às %H:%M:%S')} Nível de Crise:{crisis_level}
{sentiments['neg']}
Negativos
{percent_neg:.1f}%
{sentiments['pos']}
Positivos
{percent_pos:.1f}%
{sentiments['neu']}
Neutros
{percent_neu:.1f}%
{percent_neg:.0f}%
DOS ALERTAS SÃO NEGATIVOS → Nível de crise: {crisis_level}. {'Ação imediata necessária!' if percent_neg > 60 else 'Monitoramento recomendado.'}
🎯 PLANO DE AÇÃO URGENTE
1. Resolver Reclame Aqui (hoje): Acesse o painel administrativo → responda todas as reclamações pendentes → solicite desativação de posts resolvidos.
2. Desindexar do Google (24h): Google Search Console → "Remover URL" → envie links das páginas negativas (especialmente Reclame Aqui).
3. Monitoramento diário: Configure novo alerta: "{settings.search_query}" -site:reclameaqui.com.br
4. Contramedidas de PR: Publique nota oficial + invista R$ 50k em assessoria de imprensa positiva (prazo: 3 dias).
5. SEO Defensivo: Crie 5 artigos positivos em sites de autoridade para empurrar resultados negativos para 2ª página.
⚠️ TOP 10 ALERTAS NEGATIVOS
"""
if not negative_items:
html += '
Nenhum alerta negativo identificado.
'
for idx, item in enumerate(negative_items[:10], 1):
date_display = item['date'].split(',')[0] if ',' in item['date'] else item['date'][:20]
reclameaqui_badge = 'RECLAME AQUI' if item.get('is_reclameaqui') else ''
html += f"""
#{idx} - {item['title'][:150]}{'...' if len(item['title']) > 150 else ''}{reclameaqui_badge}
NEGATIVO
📅 {date_display} |
🌐 {item['source']}
🔗 {item['link'][:100]}{'...' if len(item['link']) > 100 else ''}
{f'
💡 {item.get("reason", "")}
' if item.get("reason") else ''}
"""
html += f"""
"""
return html
def save_html_report(
self,
sentiments: Dict,
total_items: int,
negative_items: List[Dict],
filename: str = None
) -> Path:
"""
Save HTML report to file.
Args:
sentiments: Sentiment counts
total_items: Total items
negative_items: Negative items list
filename: Optional custom filename
Returns:
Path to saved HTML file
"""
html_content = self.generate_html_content(sentiments, total_items, negative_items)
if not filename:
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
filename = f"relatorio_{timestamp}.html"
filepath = self.reports_dir / filename
with open(filepath, 'w', encoding='utf-8') as f:
f.write(html_content)
logger.info(f"HTML report saved to {filepath}")
return filepath
def generate_pdf_report(
self,
sentiments: Dict,
total_items: int,
negative_items: List[Dict],
filename: str = None
) -> Path:
"""
Generate PDF report from HTML.
Args:
sentiments: Sentiment counts
total_items: Total items
negative_items: Negative items list
filename: Optional custom filename
Returns:
Path to saved PDF file
"""
html_content = self.generate_html_content(sentiments, total_items, negative_items)
if not filename:
timestamp = datetime.now().strftime('%Y%m%d_%H%M%S')
filename = f"relatorio_{timestamp}.pdf"
filepath = self.reports_dir / filename
# Font configuration for WeasyPrint
font_config = FontConfiguration()
# Generate PDF
HTML(string=html_content).write_pdf(
filepath,
font_config=font_config
)
logger.info(f"PDF report saved to {filepath}")
return filepath
def cleanup_old_reports(self):
"""Remove old reports keeping only the most recent ones."""
reports = sorted(
self.reports_dir.glob('relatorio_*'),
key=lambda p: p.stat().st_mtime,
reverse=True
)
if len(reports) > settings.max_reports_stored:
for old_report in reports[settings.max_reports_stored:]:
old_report.unlink()
logger.info(f"Deleted old report: {old_report.name}")