88 lines
2.9 KiB
Python
88 lines
2.9 KiB
Python
"""
|
||
Update existing job to use predefined task and add new event sync job
|
||
"""
|
||
import asyncio
|
||
import sys
|
||
from pathlib import Path
|
||
|
||
# Add backend to path
|
||
sys.path.insert(0, str(Path(__file__).parent.parent))
|
||
|
||
from sqlalchemy import text, update
|
||
from app.database import engine
|
||
from app.models.db.scheduled_job import ScheduledJob, JobType
|
||
|
||
|
||
async def update_jobs():
|
||
"""Update existing job and add new event sync job"""
|
||
async with engine.begin() as conn:
|
||
print("🔄 Updating scheduled jobs...")
|
||
|
||
# 1. Update existing job to use predefined task
|
||
result = await conn.execute(text("""
|
||
UPDATE scheduled_jobs
|
||
SET
|
||
job_type = 'predefined',
|
||
predefined_function = 'sync_solar_system_positions',
|
||
function_params = '{"days": 7, "source": "nasa_horizons_cron"}'::jsonb,
|
||
description = '每日同步太阳系天体位置数据（使用内置任务）'
|
||
WHERE id = 1
|
||
RETURNING id, name
|
||
"""))
|
||
updated = result.fetchone()
|
||
if updated:
|
||
print(f"✅ Updated job ID {updated[0]}: {updated[1]} -> predefined task")
|
||
|
||
# 2. Add new celestial events sync job (disabled)
|
||
result = await conn.execute(text("""
|
||
INSERT INTO scheduled_jobs (
|
||
name,
|
||
job_type,
|
||
predefined_function,
|
||
function_params,
|
||
cron_expression,
|
||
description,
|
||
is_active
|
||
)
|
||
VALUES (
|
||
'天体事件同步',
|
||
'predefined',
|
||
'sync_celestial_events',
|
||
'{"days_ahead": 30}'::jsonb,
|
||
'0 3 * * *',
|
||
'每日凌晨3点同步未来30天的天体事件（预留功能，暂未实现）',
|
||
false
|
||
)
|
||
ON CONFLICT DO NOTHING
|
||
RETURNING id, name
|
||
"""))
|
||
new_job = result.fetchone()
|
||
if new_job:
|
||
print(f"✅ Added new job ID {new_job[0]}: {new_job[1]} (disabled)")
|
||
else:
|
||
print("ℹ️ Event sync job already exists")
|
||
|
||
# 3. Show all jobs
|
||
print("\n📋 Current scheduled jobs:")
|
||
result = await conn.execute(text("""
|
||
SELECT
|
||
id,
|
||
name,
|
||
job_type,
|
||
predefined_function,
|
||
is_active,
|
||
cron_expression
|
||
FROM scheduled_jobs
|
||
ORDER BY id
|
||
"""))
|
||
|
||
for row in result.fetchall():
|
||
status = "🟢 启用" if row[4] else "🔴 禁用"
|
||
job_type_display = "内置任务" if row[2] == 'predefined' else "自定义代码"
|
||
print(f" {status} ID {row[0]}: {row[1]}")
|
||
print(f" 类型: {job_type_display} | 函数: {row[3]} | CRON: {row[5]}")
|
||
|
||
|
||
if __name__ == "__main__":
|
||
asyncio.run(update_jobs())
|