v5.1: add auto-scheduler trigger endpoint + route protection, fix short gen

ai_runtime_patch_fast.py

@@ -1,9 +1,10 @@
-"""Final patch v5: SYNCHRONOUS short video generation, no silent errors, log everything."""
+"""Final patch v5: SYNCHRONOUS short video generation, no silent errors, log everything. + Auto scheduler trigger."""
 import re, threading, time, json, os, asyncio, hashlib, subprocess, requests, sys, logging
 import ai_runtime_final6 as f6
 from ai_runtime_final6 import app, rt, f5, HTMLResponse, JSONResponse, FileResponse, Request, Query
 import html as html_lib
 from urllib.parse import urlparse
+from datetime import datetime, timezone, timedelta
 
 _log = logging.getLogger("patch_fast")
 _log.setLevel(logging.INFO)
@@ -234,6 +235,49 @@ def _gen_short_sync(post) -> str:
         _log.error("No part files generated!")
     return ''
 
+# ===== Auto Scheduler Trigger =====
+_AUTO_SCHEDULER_RUNNING = False
+_AUTO_SCHEDULER_LAST = 0
+_AUTO_SCHEDULER_RESULTS = []
+
+def _run_scheduler_now():
+    """Trigger auto_scheduler._run_scheduled_posting() safely."""
+    global _AUTO_SCHEDULER_RUNNING, _AUTO_SCHEDULER_LAST, _AUTO_SCHEDULER_RESULTS
+    if _AUTO_SCHEDULER_RUNNING:
+        _log.warning("Scheduler already running!")
+        return {"status": "running", "message": "Scheduler đang chạy, vui lòng đợi."}
+    _AUTO_SCHEDULER_RUNNING = True
+    try:
+        import auto_scheduler as _as
+        _as.LOG.info("Triggered manually via API")
+        _as._run_scheduled_posting()
+        _AUTO_SCHEDULER_LAST = time.time()
+        _AUTO_SCHEDULER_RESULTS = [{'time': datetime.now().strftime('%H:%M %d/%m/%Y'), 'status': 'done'}]
+        _log.info("Auto scheduler completed successfully")
+        return {"status": "done", "message": "Đã đăng 3 bài tự động thành công!"}
+    except Exception as e:
+        _log.error(f"Auto scheduler failed: {e}", exc_info=True)
+        return {"status": "error", "message": f"Lỗi: {e}"}
+    finally:
+        _AUTO_SCHEDULER_RUNNING = False
+
+@app.post('/api/auto/schedule')
+async def _api_auto_schedule():
+    """Trigger auto-scheduler to post 3 articles NOW."""
+    _log.info("POST /api/auto/schedule triggered")
+    result = await asyncio.get_event_loop().run_in_executor(None, _run_scheduler_now)
+    return JSONResponse(result)
+
+@app.get('/api/auto/status')
+async def _api_auto_status():
+    """Check auto-scheduler status."""
+    return JSONResponse({
+        'running': _AUTO_SCHEDULER_RUNNING,
+        'last_run': _AUTO_SCHEDULER_LAST,
+        'last_run_str': datetime.fromtimestamp(_AUTO_SCHEDULER_LAST).strftime('%H:%M %d/%m/%Y') if _AUTO_SCHEDULER_LAST else 'never',
+        'results': _AUTO_SCHEDULER_RESULTS
+    })
+
 _bg_home={"t":0,"d":[]};_bg_shorts={"t":0,"d":[]};_bg_lock=False
 def _bg():
     global _bg_lock
@@ -250,7 +294,7 @@ def _bg():
 @app.on_event("startup")
 async def _s():threading.Thread(target=_bg,daemon=True).start()
 threading.Thread(target=lambda:[time.sleep(600) or _bg() for _ in iter(int,1)],daemon=True).start()
-app.router.routes=[r for r in app.router.routes if not (getattr(r,'path',None) in ('/api/homepage','/api/shorts','/api/ai_wall','/api/topic_post','/api/article/ask','/api/topic/rewrite','/api/rewrite_share','/api/url_wall','/api/short/comments','/api/short/comment','/api/storage_status','/api/ai/short/{post_id}','/api/ai/short-file/{file_id}','/') and any(m in getattr(r,'methods',set()) for m in ('GET','POST')))]
+app.router.routes=[r for r in app.router.routes if not (getattr(r,'path',None) in ('/api/homepage','/api/shorts','/api/ai_wall','/api/topic_post','/api/article/ask','/api/topic/rewrite','/api/rewrite_share','/api/url_wall','/api/short/comments','/api/short/comment','/api/storage_status','/api/ai/short/{post_id}','/api/ai/short-file/{file_id}','/api/auto/schedule','/api/auto/status','/') and any(m in getattr(r,'methods',set()) for m in ('GET','POST')))]
 @app.get('/api/homepage')
 def _h():
     n=time.time()