Update outlook_relatorio.py

outlook_relatorio.py

@@ -1,12 +1,15 @@
-
 # -*- coding: utf-8 -*-
-import streamlit as st
-import pandas as pd
-from datetime import datetime, timedelta, date
+import os
+import platform
 import io
 import re
 import unicodedata
-import pythoncom  # ✅ COM init/finalize para evitar 'CoInitialize não foi chamado'
+from datetime import datetime, timedelta, date
+from typing import Optional
+
+import streamlit as st
+import pandas as pd
+import plotly.express as px
 
 # (opcional) auditoria, se existir no seu projeto
 try:
@@ -15,6 +18,10 @@ try:
 except Exception:
     _HAS_AUDIT = False
 
+# Ambiente / Flags
+_IS_WINDOWS = platform.system().lower() == "windows"
+_DISABLE_OUTLOOK = os.getenv("DISABLE_OUTLOOK", "0") == "1"  # permite desativar mesmo no Windows
+
 # ==============================
 # 🎨 Estilos (UX)
 # ==============================
@@ -73,30 +80,33 @@ def _build_downloads(df: pd.DataFrame, base_name: str):
 
     # Excel (com autoajuste de larguras)
     xlsx_buf = io.BytesIO()
-    with pd.ExcelWriter(xlsx_buf, engine="openpyxl") as writer:
-        df.to_excel(writer, index=False, sheet_name="Relatorio")
-        ws = writer.sheets["Relatorio"]
-
-        # 🔎 Ajuste automático de largura das colunas
-        from openpyxl.utils import get_column_letter
-        for col_idx, col_cells in enumerate(ws.columns, start=1):
-            max_len = 0
-            for cell in col_cells:
-                try:
-                    v = "" if cell.value is None else str(cell.value)
-                    max_len = max(max_len, len(v))
-                except Exception:
-                    pass
-            ws.column_dimensions[get_column_letter(col_idx)].width = min(max_len + 2, 60)
+    try:
+        with pd.ExcelWriter(xlsx_buf, engine="openpyxl") as writer:
+            df.to_excel(writer, index=False, sheet_name="Relatorio")
+            ws = writer.sheets["Relatorio"]
+
+            # 🔎 Ajuste automático de largura das colunas
+            from openpyxl.utils import get_column_letter
+            for col_idx, col_cells in enumerate(ws.columns, start=1):
+                max_len = 0
+                for cell in col_cells:
+                    try:
+                        v = "" if cell.value is None else str(cell.value)
+                        max_len = max(max_len, len(v))
+                    except Exception:
+                        pass
+                ws.column_dimensions[get_column_letter(col_idx)].width = min(max_len + 2, 60)
 
-    xlsx_buf.seek(0)
-    st.download_button(
-        "⬇️ Baixar Excel",
-        data=xlsx_buf,
-        file_name=f"{base_name}.xlsx",
-        mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
-        key=f"dl_xlsx_{base_name}"
-    )
+        xlsx_buf.seek(0)
+        st.download_button(
+            "⬇️ Baixar Excel",
+            data=xlsx_buf,
+            file_name=f"{base_name}.xlsx",
+            mime="application/vnd.openxmlformats-officedocument.spreadsheetml.sheet",
+            key=f"dl_xlsx_{base_name}"
+        )
+    except Exception as e:
+        st.info(f"Excel: não foi possível gerar o arquivo (openpyxl). Detalhe: {e}")
 
     # PDF (resumo até 100 linhas)
     try:
@@ -174,7 +184,7 @@ def _ensure_client_exploded(df: pd.DataFrame) -> pd.DataFrame:
     return df2
 
 # ==============================
-# ✅ NOVO: visão da Tabela (cliente único + Data/Hora + ícones)
+# ✅ Visão da Tabela (cliente único + Data/Hora + ícones)
 # ==============================
 def _build_table_view_unique_client(df: pd.DataFrame) -> pd.DataFrame:
     """
@@ -250,8 +260,6 @@ def _build_table_view_unique_client(df: pd.DataFrame) -> pd.DataFrame:
 # ==============================
 # Indicadores (visual + profissional)
 # ==============================
-import plotly.express as px
-
 def _render_indicators_custom(df: pd.DataFrame, dt_col_name: str, cols_for_topn: list[str], topn_default: int = 10):
     """
     Indicadores com:
@@ -623,7 +631,7 @@ def _format_plate(p: str) -> str:
         return f"{p0[:3]}-{p0[3:]}"
     return p0
 
-def _extract_plate_from_text(text: str) -> str | None:
+def _extract_plate_from_text(text: str) -> Optional[str]:
     """Extrai placa do texto com robustez."""
     if not text:
         return None
@@ -662,21 +670,21 @@ def _extract_plate_from_text(text: str) -> str | None:
     return None
 
 # ====== Sanitização de CLIENTE ======
-def _sanitize_cliente(s: str | None) -> str | None:
+def _sanitize_cliente(s: Optional[str]) -> Optional[str]:
     if not s:
         return s
     s2 = re.sub(r"\s*:\s*", " ", s)
     s2 = re.sub(r"\s+", " ", s2).strip(" -:|")
     return s2.strip()
 
-def _split_semicolon(s: str | None) -> list[str] | None:
+def _split_semicolon(s: Optional[str]) -> Optional[list[str]]:
     if not s:
         return None
     parts = [p.strip(" -:|").strip() for p in s.split(";")]
     parts = [p for p in parts if p]
     return parts or None
 
-def _cliente_to_list(cliente_raw: str | None) -> list[str] | None:
+def _cliente_to_list(cliente_raw: Optional[str]) -> Optional[list[str]]:
     s = _sanitize_cliente(cliente_raw)
     return _split_semicolon(s)
 
@@ -1007,8 +1015,16 @@ def gerar_relatorio_outlook_desktop_multi(
     filtro_remetente: str = "",
     extrair_campos: bool = True
 ) -> pd.DataFrame:
-    """Inicializa COM, conecta ao Outlook, lê múltiplas pastas (todas as stores) e finaliza COM."""
+    """
+    Inicializa COM, conecta ao Outlook, lê múltiplas pastas (todas as stores) e finaliza COM.
+    ⚠️ Disponível apenas em Windows (Outlook Desktop via pywin32).
+    """
+    if not _IS_WINDOWS or _DISABLE_OUTLOOK:
+        st.info("📌 Leitura via Outlook Desktop está indisponível neste ambiente.")
+        return pd.DataFrame()
+
     try:
+        import pythoncom
         import win32com.client
         pythoncom.CoInitialize()
         ns = win32com.client.Dispatch("Outlook.Application").GetNamespace("MAPI")
@@ -1391,28 +1407,31 @@ def main():
             st.success("Cache limpo. Gere novamente os dados.")
 
         if submit_test:
-            try:
-                import win32com.client
-                pythoncom.CoInitialize()
-                ns = win32com.client.Dispatch("Outlook.Application").GetNamespace("MAPI")
-                stores = ns.Folders.Count
-                inbox_def = ns.GetDefaultFolder(6)
-                sent_def = ns.GetDefaultFolder(5)
-                st.info(f"Stores detectadas: **{stores}** | Inbox padrão: **{inbox_def.Name}** | Sent: **{sent_def.Name}**")
-                if pastas_escolhidas:
-                    for p in pastas_escolhidas:
-                        try:
-                            f = _get_folder_by_path_any_store(ns, p)
-                            st.write(f"📁 {p} → '{f.Name}' itens: {f.Items.Count}")
-                        except Exception as e:
-                            st.warning(f"[Teste] Falha ao acessar '{p}': {e}")
-            except Exception as e:
-                st.error(f"[Teste] Erro ao conectar: {e}")
-            finally:
+            if not _IS_WINDOWS or _DISABLE_OUTLOOK:
+                st.info("Teste indisponível neste ambiente (Linux/Spaces).")
+            else:
                 try:
-                    pythoncom.CoUninitialize()
-                except Exception:
-                    pass
+                    import pythoncom, win32com.client
+                    pythoncom.CoInitialize()
+                    ns = win32com.client.Dispatch("Outlook.Application").GetNamespace("MAPI")
+                    stores = ns.Folders.Count
+                    inbox_def = ns.GetDefaultFolder(6)
+                    sent_def = ns.GetDefaultFolder(5)
+                    st.info(f"Stores detectadas: **{stores}** | Inbox padrão: **{inbox_def.Name}** | Sent: **{sent_def.Name}**")
+                    if pastas_escolhidas:
+                        for p in pastas_escolhidas:
+                            try:
+                                f = _get_folder_by_path_any_store(ns, p)
+                                st.write(f"📁 {p} → '{f.Name}' itens: {f.Items.Count}")
+                            except Exception as e:
+                                st.warning(f"[Teste] Falha ao acessar '{p}': {e}")
+                except Exception as e:
+                    st.error(f"[Teste] Erro ao conectar: {e}")
+                finally:
+                    try:
+                        pythoncom.CoUninitialize()
+                    except Exception:
+                        pass
 
         if submit_fresh:
             if not pastas_escolhidas:
@@ -1618,7 +1637,4 @@ def main():
                 pass
 
     else:
-        st.info("👉 Selecione a caixa (Entrada/Saída), opcionalmente uma subpasta, e clique em **🔍 Gerar (com cache)** ou **⚡ Atualizar agora (sem cache)**. Use **🧪 Teste de conexão** se vier vazio.")
-
-# if __name__ == "__main__":
-#     main()
+        st.info("👉 Selecione a caixa (Entrada/Saída), opcionalmente uma subpasta, e clique em **🔍 Gerar (com cache)** ou **⚡ Atualizar agora (sem cache)**. Use **🧪 Teste de conexão** se vier vazio.")