34a6c6d67e
- 使用 draft API 同步文章（适配个人订阅号） - 使用 material API 同步视频（含详情获取） - 自动建表（videos）、UPSERT 已有 articles 表 - 同步删除：微信端删除的素材自动从数据库移除 - APScheduler 定时调度，支持 --once 手动触发 - Docker + docker-compose 部署配置 Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
63 lines
1.6 KiB
Python
63 lines
1.6 KiB
Python
import argparse
|
|
import logging
|
|
import sys
|
|
|
|
from apscheduler.schedulers.blocking import BlockingScheduler
|
|
|
|
import config
|
|
from db import Database
|
|
from sync import SyncService
|
|
from wechat import WeChatClient
|
|
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format="%(asctime)s [%(levelname)s] %(name)s: %(message)s",
|
|
datefmt="%Y-%m-%d %H:%M:%S",
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
def main():
|
|
parser = argparse.ArgumentParser(description="UpdateDB - sync WeChat materials to database")
|
|
parser.add_argument("--once", action="store_true", help="Run sync once and exit")
|
|
args = parser.parse_args()
|
|
|
|
logger.info("UpdateDB starting (sync interval: %dh)", config.SYNC_INTERVAL_HOURS)
|
|
|
|
db = Database()
|
|
wechat = WeChatClient()
|
|
sync = SyncService(wechat, db)
|
|
|
|
try:
|
|
db.ensure_tables()
|
|
|
|
# Run once at startup
|
|
sync.run_sync()
|
|
|
|
if args.once:
|
|
logger.info("--once flag set, exiting after first sync")
|
|
return
|
|
|
|
scheduler = BlockingScheduler()
|
|
scheduler.add_job(
|
|
sync.run_sync,
|
|
"interval",
|
|
hours=config.SYNC_INTERVAL_HOURS,
|
|
id="sync_job",
|
|
max_instances=1,
|
|
)
|
|
logger.info("Scheduler started, next sync in %d hours", config.SYNC_INTERVAL_HOURS)
|
|
scheduler.start()
|
|
except KeyboardInterrupt:
|
|
logger.info("Shutting down")
|
|
except Exception as e:
|
|
logger.error("Fatal error: %s", e, exc_info=True)
|
|
sys.exit(1)
|
|
finally:
|
|
wechat.close()
|
|
db.close()
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|