Initial commit

pages/mailer.py

@@ -0,0 +1,118 @@
+import streamlit as st
+import pandas as pd
+from datetime import date, datetime
+from zoneinfo import ZoneInfo
+from app_core.services.mailer_service import MailerService
+
+
+def render_page():
+    if st.session_state.get("auth_user") is None:
+        st.warning("Please login to continue.")
+        st.stop()
+
+    st.markdown("## Mailer")
+    st.caption("Automated daily email sending logs and status.")
+
+    service = MailerService()
+    ist = ZoneInfo("Asia/Kolkata")
+
+    st.markdown("### Scheduler Status")
+    
+    # Check if scheduler container is running by checking if daily report was sent today
+    today_logs = [log for log in service.recent_logs(limit=1000) if log.get('date_for') == str(date.today())]
+    daily_report_sent_today = any('Daily Report' in str(log.get('subject', '')) for log in today_logs)
+    
+    if daily_report_sent_today:
+        st.success("Scheduler is running - Daily report already sent today")
+    else:
+        # Check if it's past 8 PM IST today
+        now_ist = datetime.now(ist)
+        eight_pm_today = now_ist.replace(hour=20, minute=0, second=0, microsecond=0)
+        
+        if now_ist >= eight_pm_today:
+            st.warning("Scheduler is running - Waiting for next scheduled run (8:00 PM IST daily)")
+        else:
+            next_run_ist = eight_pm_today
+            st.success(f"Scheduler is running - Next report will be sent at {next_run_ist.strftime('%B %d, %Y at %I:%M %p IST')}")
+    
+    st.markdown("---")
+
+    # Show system status
+    st.markdown("### System Status")
+    col1, col2, col3 = st.columns(3)
+    
+    with col1:
+        st.metric("Total Emails Sent", len([log for log in service.recent_logs(limit=1000) if log.get('status') == 'sent']))
+    
+    with col2:
+        failed_count = len([log for log in service.recent_logs(limit=1000) if log.get('status') == 'failed'])
+        st.metric("Failed Sends", failed_count, delta=f"-{failed_count}" if failed_count > 0 else None)
+    
+    with col3:
+        today_logs = [log for log in service.recent_logs(limit=1000) if log.get('date_for') == str(date.today())]
+        st.metric("Today's Sends", len(today_logs))
+    
+    st.markdown("---")
+    
+    # Manual trigger section
+    st.markdown("### Manual Controls")
+    mcol1, mcol2, mcol3 = st.columns([2, 2, 3])
+    
+    with mcol1:
+        target_date = st.date_input("Report Date", value=date.today())
+        
+    with mcol2:
+        st.write("") # Spacer
+        force_resend = st.checkbox("Force Resend", value=True, help="Send the report even if it was already sent for this date.")
+        
+    with mcol3:
+        st.write("") # Spacer
+        if st.button("Send Report Now", type="primary", use_container_width=True):
+            with st.spinner(f"Sending report for {target_date}..."):
+                try:
+                    from app_core.services.daily_report import main as run_daily_report
+                    # Pass the selected date and force flag
+                    result = run_daily_report(for_date=str(target_date), force=force_resend)
+                    if result == 0:
+                        st.success(f"Report for {target_date} sent successfully!")
+                        st.rerun()
+                    else:
+                        st.error(f"Failed to send report (exit code: {result})")
+                except Exception as e:
+                    st.error(f"Error: {str(e)}")
+    
+    st.caption("Select a date to manually trigger or re-trigger the daily report email. This is useful for reconciliations.")
+    
+    st.markdown("---")
+    
+    # Show email logs table
+    st.markdown("### Email Logs")
+    logs = service.recent_logs(limit=100)
+    if not logs:
+        st.info("No email logs yet. Automated emails will appear here once sent.")
+    else:
+        df_logs = pd.DataFrame(logs)
+        col_map = {
+            "id": "ID",
+            "sent_at": "Sent At",
+            "recipients": "Recipients",
+            "subject": "Subject",
+            "status": "Status",
+            "error": "Error",
+            "date_for": "Report Date",
+        }
+        df_logs = df_logs[["id", "sent_at", "date_for", "recipients", "subject", "status", "error"]]
+        df_logs = df_logs.rename(columns=col_map)
+        
+        # Add status styling
+        def style_status(val):
+            if val == 'sent':
+                return 'background-color: #D1FAE5; color: #065F46; font-weight: 600;'
+            elif val == 'failed':
+                return 'background-color: #FEE2E2; color: #991B1B; font-weight: 600;'
+            return ''
+        
+        styled_logs = df_logs.style.map(style_status, subset=['Status'])
+        st.dataframe(styled_logs, use_container_width=True, height=400)
+
+# trigger reload