fix post vk
This commit is contained in:
+58
-20
@@ -3,28 +3,61 @@ from apscheduler.triggers.cron import CronTrigger
|
||||
|
||||
from db import PostScheduler
|
||||
|
||||
# =========================
|
||||
# Flask app (будет установлен из app.py)
|
||||
# =========================
|
||||
flask_app = None
|
||||
|
||||
# ===== Scheduler =====
|
||||
# =========================
|
||||
# Scheduler
|
||||
# =========================
|
||||
scheduler: BackgroundScheduler | None = None
|
||||
JOB_ID = "vk_publish_job"
|
||||
|
||||
|
||||
def start_scheduler():
|
||||
global scheduler
|
||||
def init_scheduler(app):
|
||||
"""
|
||||
Инициализация планировщика с Flask-приложением
|
||||
Вызывать ОДИН раз при старте приложения
|
||||
"""
|
||||
global flask_app, scheduler
|
||||
|
||||
flask_app = app
|
||||
|
||||
if scheduler is None:
|
||||
scheduler = BackgroundScheduler()
|
||||
scheduler.start()
|
||||
|
||||
|
||||
# ===== Добавление задачи =====
|
||||
# =========================
|
||||
# JOB wrapper
|
||||
# =========================
|
||||
def vk_publish_job():
|
||||
"""
|
||||
Обёртка для APScheduler
|
||||
"""
|
||||
if flask_app is None:
|
||||
raise RuntimeError("Scheduler is not initialized with Flask app")
|
||||
|
||||
from vk_handler import handle_vk_post
|
||||
|
||||
with flask_app.app_context():
|
||||
handle_vk_post()
|
||||
|
||||
|
||||
# =========================
|
||||
# Enable job
|
||||
# =========================
|
||||
def enable_publish_job():
|
||||
"""
|
||||
Включает выполнение handle_vk_post
|
||||
Включает выполнение публикации постов
|
||||
"""
|
||||
from vk_handler import handle_vk_post
|
||||
if not scheduler:
|
||||
return
|
||||
|
||||
scheduleData = PostScheduler.query.first()
|
||||
if not scheduleData or not scheduleData.enabled:
|
||||
disable_publish_job()
|
||||
return
|
||||
|
||||
start_hour = scheduleData.start_hour
|
||||
@@ -32,34 +65,39 @@ def enable_publish_job():
|
||||
interval_minutes = scheduleData.interval_minutes
|
||||
|
||||
trigger = CronTrigger(
|
||||
hour=f"{start_hour}-{end_hour - 1}", minute=f"*/{interval_minutes}"
|
||||
hour=f"{start_hour}-{end_hour - 1}",
|
||||
minute=f"*/{interval_minutes}",
|
||||
)
|
||||
|
||||
scheduler.add_job(handle_vk_post, trigger=trigger, id=JOB_ID, replace_existing=True)
|
||||
scheduler.add_job(
|
||||
vk_publish_job,
|
||||
trigger=trigger,
|
||||
id=JOB_ID,
|
||||
replace_existing=True,
|
||||
)
|
||||
|
||||
|
||||
# ===== Отключение задачи =====
|
||||
# =========================
|
||||
# Disable job
|
||||
# =========================
|
||||
def disable_publish_job():
|
||||
if scheduler and scheduler.get_job(JOB_ID):
|
||||
scheduler.remove_job(JOB_ID)
|
||||
|
||||
|
||||
# =========================
|
||||
# Status
|
||||
# =========================
|
||||
def get_scheduler_status() -> dict:
|
||||
scheduler_running = bool(scheduler and scheduler.running)
|
||||
|
||||
job = scheduler.get_job(JOB_ID) if scheduler_running else None
|
||||
|
||||
status = {
|
||||
return {
|
||||
"scheduler": scheduler_running,
|
||||
"vk_publish_job": job is not None,
|
||||
"next_run_time": None,
|
||||
}
|
||||
|
||||
if job:
|
||||
status["next_run_time"] = (
|
||||
"next_run_time": (
|
||||
job.next_run_time.strftime("%Y-%m-%d %H:%M:%S")
|
||||
if job.next_run_time
|
||||
if job and job.next_run_time
|
||||
else None
|
||||
)
|
||||
|
||||
return status
|
||||
),
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user