9280be7e52
- RSS 수집 시 media:thumbnail, media:content, enclosure, <img> 태그에서 이미지 추출 - source_image를 topic → article → publisher로 전달 - 발행 시 우선순위: 원본 소스 이미지 → Pexels → Unsplash Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>
1374 lines
54 KiB
Python
1374 lines
54 KiB
Python
"""
|
|
스케줄러 (scheduler.py)
|
|
역할: 모든 봇의 실행 시간 관리 + Telegram 수동 명령 리스너
|
|
라이브러리: APScheduler + python-telegram-bot
|
|
"""
|
|
import asyncio
|
|
import json
|
|
import logging
|
|
import os
|
|
import sys
|
|
from datetime import datetime
|
|
from logging.handlers import RotatingFileHandler
|
|
from pathlib import Path
|
|
|
|
from runtime_guard import ensure_project_runtime
|
|
|
|
ensure_project_runtime(
|
|
"scheduler",
|
|
["apscheduler", "python-dotenv", "python-telegram-bot", "anthropic"],
|
|
)
|
|
|
|
from apscheduler.schedulers.asyncio import AsyncIOScheduler
|
|
from dotenv import load_dotenv
|
|
from telegram import Update, InlineKeyboardButton, InlineKeyboardMarkup
|
|
from telegram.ext import Application, CommandHandler, MessageHandler, CallbackQueryHandler, filters, ContextTypes
|
|
|
|
import anthropic
|
|
import re
|
|
|
|
load_dotenv()
|
|
|
|
BASE_DIR = Path(__file__).parent.parent
|
|
CONFIG_DIR = BASE_DIR / 'config'
|
|
DATA_DIR = BASE_DIR / 'data'
|
|
LOG_DIR = BASE_DIR / 'logs'
|
|
LOG_DIR.mkdir(exist_ok=True)
|
|
|
|
log_handler = RotatingFileHandler(
|
|
LOG_DIR / 'scheduler.log',
|
|
maxBytes=5 * 1024 * 1024,
|
|
backupCount=3,
|
|
encoding='utf-8',
|
|
)
|
|
logging.basicConfig(
|
|
level=logging.INFO,
|
|
format='%(asctime)s [%(levelname)s] %(message)s',
|
|
handlers=[log_handler, logging.StreamHandler()],
|
|
)
|
|
logger = logging.getLogger(__name__)
|
|
|
|
TELEGRAM_BOT_TOKEN = os.getenv('TELEGRAM_BOT_TOKEN', '')
|
|
TELEGRAM_CHAT_ID = os.getenv('TELEGRAM_CHAT_ID', '')
|
|
ANTHROPIC_API_KEY = os.getenv('ANTHROPIC_API_KEY', '')
|
|
|
|
_claude_client: anthropic.Anthropic | None = None
|
|
_conversation_history: dict[int, list] = {}
|
|
|
|
CLAUDE_SYSTEM_PROMPT = """당신은 "AI? 그게 뭔데?" 블로그의 운영 어시스턴트입니다.
|
|
블로그 운영자 eli가 Telegram으로 명령하면 도와주는 역할입니다.
|
|
슬로건: "어렵지 않아요, 그냥 읽어봐요"
|
|
블로그 주소: eli-ai.blogspot.com
|
|
|
|
이 시스템(v3)은 4계층 구조로 운영됩니다:
|
|
|
|
[LAYER 1] AI 콘텐츠 생성: Gemini 2.5-flash가 원본 마크다운 1개 생성
|
|
[LAYER 2] 변환 엔진: 원본 → 블로그HTML / 인스타카드 / X스레드 / 뉴스레터 자동 변환
|
|
[LAYER 3] 배포 엔진: Blogger / Instagram / X / TikTok / YouTube 순차 발행
|
|
[LAYER 4] 분석봇: 성과 수집 + 주간 리포트 + 피드백 루프
|
|
|
|
8개 카테고리: AI인사이트, 여행맛집, 스타트업, 제품리뷰, 생활꿀팁, 앱추천, 재테크절약, 팩트체크
|
|
|
|
사용 가능한 텔레그램 명령:
|
|
/status — 봇 상태
|
|
/topics — 오늘 수집된 글감
|
|
/collect — 글감 즉시 수집
|
|
/write [번호] [방향] — 특정 글감으로 글 작성
|
|
/pending — 검토 대기 글 목록
|
|
/approve [번호] — 글 승인 및 발행
|
|
/reject [번호] — 글 거부
|
|
/report — 주간 리포트
|
|
/images — 이미지 제작 현황
|
|
/convert — 수동 변환 실행
|
|
/novel_list — 연재 소설 목록
|
|
/novel_gen [novel_id] — 에피소드 즉시 생성
|
|
/novel_status — 소설 파이프라인 진행 현황
|
|
|
|
모든 글은 발행 전 운영자 승인이 필요합니다.
|
|
사용자의 자연어 요청을 이해하고 적절히 안내하거나 답변해주세요.
|
|
한국어로 간결하게 답변하세요."""
|
|
IMAGE_MODE = os.getenv('IMAGE_MODE', 'manual').lower()
|
|
# request 모드에서 이미지 대기 시 사용하는 상태 변수
|
|
# {chat_id: prompt_id} — 다음에 받은 이미지를 어느 프롬프트에 연결할지 기억
|
|
_awaiting_image: dict[int, str] = {}
|
|
|
|
_publish_enabled = True
|
|
|
|
|
|
def load_schedule() -> dict:
|
|
with open(CONFIG_DIR / 'schedule.json', 'r', encoding='utf-8') as f:
|
|
return json.load(f)
|
|
|
|
|
|
# ─── 스케줄 작업 ──────────────────────────────────────
|
|
|
|
def job_collector():
|
|
logger.info("[스케줄] 수집봇 시작")
|
|
try:
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import collector_bot
|
|
collector_bot.run()
|
|
except Exception as e:
|
|
logger.error(f"수집봇 오류: {e}")
|
|
|
|
|
|
def job_ai_writer():
|
|
logger.info("[스케줄] AI 글 작성 트리거")
|
|
if not _publish_enabled:
|
|
logger.info("발행 중단 상태 — 건너뜀")
|
|
return
|
|
try:
|
|
_trigger_openclaw_writer()
|
|
except Exception as e:
|
|
logger.error(f"AI 글 작성 트리거 오류: {e}")
|
|
|
|
|
|
def _trigger_openclaw_writer():
|
|
topics_dir = DATA_DIR / 'topics'
|
|
drafts_dir = DATA_DIR / 'drafts'
|
|
originals_dir = DATA_DIR / 'originals'
|
|
drafts_dir.mkdir(exist_ok=True)
|
|
originals_dir.mkdir(exist_ok=True)
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
topic_files = sorted(topics_dir.glob(f'{today}_*.json'))
|
|
if not topic_files:
|
|
logger.info("오늘 처리할 글감 없음")
|
|
return
|
|
for topic_file in topic_files[:3]:
|
|
draft_check = drafts_dir / topic_file.name
|
|
original_check = originals_dir / topic_file.name
|
|
if draft_check.exists() or original_check.exists():
|
|
continue
|
|
topic_data = json.loads(topic_file.read_text(encoding='utf-8'))
|
|
logger.info(f"글 작성 요청: {topic_data.get('topic', '')}")
|
|
_call_openclaw(topic_data, original_check)
|
|
|
|
|
|
def _safe_slug(text: str) -> str:
|
|
slug = re.sub(r'[^a-z0-9]+', '-', text.lower()).strip('-')
|
|
return slug or datetime.now().strftime('article-%Y%m%d-%H%M%S')
|
|
|
|
|
|
def _build_openclaw_prompt(topic_data: dict) -> tuple[str, str]:
|
|
topic = topic_data.get('topic', '').strip()
|
|
corner = topic_data.get('corner', 'AI인사이트').strip() or 'AI인사이트'
|
|
description = topic_data.get('description', '').strip()
|
|
source = topic_data.get('source_url') or topic_data.get('source') or ''
|
|
published_at = topic_data.get('published_at', '')
|
|
is_english = topic_data.get('is_english', False)
|
|
system = (
|
|
"당신은 'AI? 그게 뭔데?' 블로그의 편집자 eli다. "
|
|
"비전문가도 쉽게 읽을 수 있는 친근한 톤으로 글을 쓴다. "
|
|
"반드시 아래 섹션 헤더 형식만 사용해 완성된 Blogger-ready HTML 원고를 출력하라. "
|
|
"본문(BODY)은 HTML로 작성하고, KEY_POINTS는 3줄 이내로 작성한다."
|
|
)
|
|
if is_english:
|
|
system += (
|
|
" 영문 원문을 단순 번역하지 말고, 한국 독자 관점에서 재해석하여 작성하라. "
|
|
"한국 시장/사용자에게 어떤 의미인지, 국내 대안이나 비교 서비스가 있다면 함께 언급하라. "
|
|
"제목도 한국어로 매력적으로 새로 작성하라."
|
|
)
|
|
lang_note = "\n⚠️ 영문 원문입니다. 단순 번역이 아닌, 한국 독자 맥락으로 재작성해주세요." if is_english else ""
|
|
prompt = f"""다음 글감을 바탕으로 한국어 블로그 원고를 작성해줘.{lang_note}
|
|
|
|
주제: {topic}
|
|
코너: {corner}
|
|
설명: {description}
|
|
출처: {source}
|
|
발행시점 참고: {published_at}
|
|
|
|
출력 형식은 아래 섹션만 정확히 사용해.
|
|
|
|
---TITLE---
|
|
제목
|
|
|
|
---META---
|
|
검색 설명 150자 이내
|
|
|
|
---SLUG---
|
|
영문 소문자 slug
|
|
|
|
---TAGS---
|
|
태그1, 태그2, 태그3
|
|
|
|
---CORNER---
|
|
{corner}
|
|
|
|
---BODY---
|
|
<h2>...</h2> 형식의 Blogger-ready HTML 본문
|
|
|
|
---KEY_POINTS---
|
|
- 핵심포인트1
|
|
- 핵심포인트2
|
|
- 핵심포인트3
|
|
|
|
---COUPANG_KEYWORDS---
|
|
키워드1, 키워드2
|
|
|
|
---SOURCES---
|
|
{source} | 참고 출처 | {published_at}
|
|
|
|
---DISCLAIMER---
|
|
필요 시 짧은 면책문구
|
|
"""
|
|
return system, prompt
|
|
|
|
|
|
def _call_openclaw(topic_data: dict, output_path: Path, direction: str = ''):
|
|
logger.info(f"글 작성 요청: {topic_data.get('topic', '')}")
|
|
sys.path.insert(0, str(BASE_DIR))
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
|
|
from engine_loader import EngineLoader
|
|
from article_parser import parse_output
|
|
|
|
system, prompt = _build_openclaw_prompt(topic_data)
|
|
if direction:
|
|
prompt += f"\n\n운영자 지시사항: {direction}"
|
|
writer = EngineLoader().get_writer()
|
|
raw_output = writer.write(prompt, system=system).strip()
|
|
if not raw_output:
|
|
raise RuntimeError('Writer 응답이 비어 있습니다.')
|
|
|
|
article = parse_output(raw_output)
|
|
if not article:
|
|
raise RuntimeError('Writer 출력 파싱 실패')
|
|
|
|
article.setdefault('title', topic_data.get('topic', '').strip())
|
|
article['slug'] = article.get('slug') or _safe_slug(article['title'])
|
|
article['corner'] = article.get('corner') or topic_data.get('corner', 'AI인사이트')
|
|
article['topic'] = topic_data.get('topic', '')
|
|
article['description'] = topic_data.get('description', '')
|
|
article['quality_score'] = topic_data.get('quality_score', 0)
|
|
article['source'] = topic_data.get('source', '')
|
|
article['source_url'] = topic_data.get('source_url') or topic_data.get('source') or ''
|
|
article['published_at'] = topic_data.get('published_at', '')
|
|
article['source_image'] = topic_data.get('source_image', '')
|
|
article['created_at'] = datetime.now().isoformat()
|
|
|
|
output_path.parent.mkdir(parents=True, exist_ok=True)
|
|
output_path.write_text(
|
|
json.dumps(article, ensure_ascii=False, indent=2),
|
|
encoding='utf-8',
|
|
)
|
|
logger.info(f"원고 저장 완료: {output_path.name}")
|
|
|
|
|
|
def _publish_next():
|
|
"""originals/ → pending_review/ 이동 (안전장치 체크)"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import publisher_bot
|
|
originals_dir = DATA_DIR / 'originals'
|
|
pending_dir = DATA_DIR / 'pending_review'
|
|
pending_dir.mkdir(exist_ok=True)
|
|
safety_cfg = publisher_bot.load_config('safety_keywords.json')
|
|
for f in sorted(originals_dir.glob('*.json')):
|
|
try:
|
|
article = json.loads(f.read_text(encoding='utf-8'))
|
|
needs_review, reason = publisher_bot.check_safety(article, safety_cfg)
|
|
article['pending_reason'] = reason or '수동 승인 필요'
|
|
dest = pending_dir / f.name
|
|
dest.write_text(json.dumps(article, ensure_ascii=False, indent=2), encoding='utf-8')
|
|
f.unlink()
|
|
logger.info(f"검토 대기로 이동: {f.name} ({reason})")
|
|
except Exception as e:
|
|
logger.error(f"publish_next 오류 ({f.name}): {e}")
|
|
|
|
|
|
def job_convert():
|
|
"""08:30 — 변환 엔진: 원본 마크다운 → 5개 포맷 생성"""
|
|
if not _publish_enabled:
|
|
logger.info("[스케줄] 발행 중단 — 변환 건너뜀")
|
|
return
|
|
logger.info("[스케줄] 변환 엔진 시작")
|
|
try:
|
|
_run_conversion_pipeline()
|
|
except Exception as e:
|
|
logger.error(f"변환 엔진 오류: {e}")
|
|
|
|
|
|
def _run_conversion_pipeline():
|
|
"""originals/ 폴더의 미변환 원본을 5개 포맷으로 변환"""
|
|
originals_dir = DATA_DIR / 'originals'
|
|
originals_dir.mkdir(exist_ok=True)
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
|
|
converters_path = str(BASE_DIR / 'bots' / 'converters')
|
|
sys.path.insert(0, converters_path)
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
|
|
for orig_file in sorted(originals_dir.glob(f'{today}_*.json')):
|
|
converted_flag = orig_file.with_suffix('.converted')
|
|
if converted_flag.exists():
|
|
continue
|
|
try:
|
|
article = json.loads(orig_file.read_text(encoding='utf-8'))
|
|
slug = article.get('slug', 'article')
|
|
|
|
# 1. 블로그 HTML
|
|
import blog_converter
|
|
blog_converter.convert(article, save_file=True)
|
|
|
|
# 2. 인스타 카드
|
|
import card_converter
|
|
card_path = card_converter.convert(article, save_file=True)
|
|
if card_path:
|
|
article['_card_path'] = card_path
|
|
|
|
# 3. X 스레드
|
|
import thread_converter
|
|
thread_converter.convert(article, save_file=True)
|
|
|
|
# 4. 쇼츠 영상 (Phase 2 — card 생성 후 시도, 실패해도 계속)
|
|
if card_path:
|
|
try:
|
|
import shorts_converter
|
|
shorts_converter.convert(article, card_path=card_path, save_file=True)
|
|
except Exception as shorts_err:
|
|
logger.debug(f"쇼츠 변환 건너뜀 (Phase 2): {shorts_err}")
|
|
|
|
# 5. 뉴스레터 발췌 (주간 묶음용 — 개별 저장은 weekly_report에서)
|
|
# newsletter_converter는 주간 단위로 묶어서 처리
|
|
|
|
# 변환 완료 플래그
|
|
converted_flag.touch()
|
|
logger.info(f"변환 완료: {slug}")
|
|
|
|
# drafts에 복사 (발행봇이 읽도록)
|
|
drafts_dir = DATA_DIR / 'drafts'
|
|
drafts_dir.mkdir(exist_ok=True)
|
|
draft_path = drafts_dir / orig_file.name
|
|
if not draft_path.exists():
|
|
draft_path.write_text(
|
|
orig_file.read_text(encoding='utf-8'), encoding='utf-8'
|
|
)
|
|
except Exception as e:
|
|
logger.error(f"변환 오류 ({orig_file.name}): {e}")
|
|
|
|
|
|
def job_publish(slot: int):
|
|
"""09:00 — 블로그 발행 (슬롯별)"""
|
|
if not _publish_enabled:
|
|
logger.info(f"[스케줄] 발행 중단 — 슬롯 {slot} 건너뜀")
|
|
return
|
|
logger.info(f"[스케줄] 발행봇 (슬롯 {slot})")
|
|
try:
|
|
_publish_next()
|
|
except Exception as e:
|
|
logger.error(f"발행봇 오류: {e}")
|
|
|
|
|
|
def job_distribute_instagram():
|
|
"""10:00 — 인스타그램 카드 발행"""
|
|
if not _publish_enabled:
|
|
return
|
|
logger.info("[스케줄] 인스타그램 발행")
|
|
try:
|
|
_distribute_instagram()
|
|
except Exception as e:
|
|
logger.error(f"인스타그램 배포 오류: {e}")
|
|
|
|
|
|
def _distribute_instagram():
|
|
sys.path.insert(0, str(BASE_DIR / 'bots' / 'distributors'))
|
|
import instagram_bot
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
outputs_dir = DATA_DIR / 'outputs'
|
|
for card_file in sorted(outputs_dir.glob(f'{today}_*_card.png')):
|
|
ig_flag = card_file.with_suffix('.ig_done')
|
|
if ig_flag.exists():
|
|
continue
|
|
slug = card_file.stem.replace(f'{today}_', '').replace('_card', '')
|
|
article = _load_article_by_slug(today, slug)
|
|
if not article:
|
|
logger.warning(f"Instagram: 원본 article 없음 ({slug})")
|
|
continue
|
|
# image_host.py가 로컬 경로 → 공개 URL 변환 처리
|
|
success = instagram_bot.publish_card(article, str(card_file))
|
|
if success:
|
|
ig_flag.touch()
|
|
logger.info(f"Instagram 발행 완료: {card_file.name}")
|
|
|
|
|
|
def job_distribute_instagram_reels():
|
|
"""10:30 — Instagram Reels (쇼츠 MP4) 발행"""
|
|
if not _publish_enabled:
|
|
return
|
|
logger.info("[스케줄] Instagram Reels 발행")
|
|
try:
|
|
_distribute_instagram_reels()
|
|
except Exception as e:
|
|
logger.error(f"Instagram Reels 배포 오류: {e}")
|
|
|
|
|
|
def _distribute_instagram_reels():
|
|
sys.path.insert(0, str(BASE_DIR / 'bots' / 'distributors'))
|
|
import instagram_bot
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
outputs_dir = DATA_DIR / 'outputs'
|
|
for shorts_file in sorted(outputs_dir.glob(f'{today}_*_shorts.mp4')):
|
|
flag = shorts_file.with_suffix('.ig_reels_done')
|
|
if flag.exists():
|
|
continue
|
|
slug = shorts_file.stem.replace(f'{today}_', '').replace('_shorts', '')
|
|
article = _load_article_by_slug(today, slug)
|
|
if not article:
|
|
logger.warning(f"Instagram Reels: 원본 article 없음 ({slug})")
|
|
continue
|
|
success = instagram_bot.publish_reels(article, str(shorts_file))
|
|
if success:
|
|
flag.touch()
|
|
logger.info(f"Instagram Reels 발행 완료: {shorts_file.name}")
|
|
|
|
|
|
def job_distribute_x():
|
|
"""11:00 — X 스레드 게시"""
|
|
if not _publish_enabled:
|
|
return
|
|
logger.info("[스케줄] X 스레드 게시")
|
|
try:
|
|
_distribute_x()
|
|
except Exception as e:
|
|
logger.error(f"X 배포 오류: {e}")
|
|
|
|
|
|
def _distribute_x():
|
|
sys.path.insert(0, str(BASE_DIR / 'bots' / 'distributors'))
|
|
import x_bot
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
outputs_dir = DATA_DIR / 'outputs'
|
|
for thread_file in sorted(outputs_dir.glob(f'{today}_*_thread.json')):
|
|
x_flag = thread_file.with_suffix('.x_done')
|
|
if x_flag.exists():
|
|
continue
|
|
slug = thread_file.stem.replace(f'{today}_', '').replace('_thread', '')
|
|
article = _load_article_by_slug(today, slug)
|
|
if not article:
|
|
continue
|
|
thread_data = json.loads(thread_file.read_text(encoding='utf-8'))
|
|
success = x_bot.publish_thread(article, thread_data)
|
|
if success:
|
|
x_flag.touch()
|
|
|
|
|
|
def job_distribute_tiktok():
|
|
"""18:00 — TikTok 쇼츠 업로드"""
|
|
if not _publish_enabled:
|
|
return
|
|
logger.info("[스케줄] TikTok 쇼츠 업로드")
|
|
try:
|
|
_distribute_shorts('tiktok')
|
|
except Exception as e:
|
|
logger.error(f"TikTok 배포 오류: {e}")
|
|
|
|
|
|
def job_distribute_youtube():
|
|
"""20:00 — YouTube 쇼츠 업로드"""
|
|
if not _publish_enabled:
|
|
return
|
|
logger.info("[스케줄] YouTube 쇼츠 업로드")
|
|
try:
|
|
_distribute_shorts('youtube')
|
|
except Exception as e:
|
|
logger.error(f"YouTube 배포 오류: {e}")
|
|
|
|
|
|
def _distribute_shorts(platform: str):
|
|
"""틱톡/유튜브 쇼츠 MP4 배포 공통 로직"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots' / 'distributors'))
|
|
if platform == 'tiktok':
|
|
import tiktok_bot as dist_bot
|
|
else:
|
|
import youtube_bot as dist_bot
|
|
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
outputs_dir = DATA_DIR / 'outputs'
|
|
for shorts_file in sorted(outputs_dir.glob(f'{today}_*_shorts.mp4')):
|
|
done_flag = shorts_file.with_suffix(f'.{platform}_done')
|
|
if done_flag.exists():
|
|
continue
|
|
slug = shorts_file.stem.replace(f'{today}_', '').replace('_shorts', '')
|
|
article = _load_article_by_slug(today, slug)
|
|
if not article:
|
|
logger.warning(f"{platform}: 원본 article 없음 ({slug})")
|
|
continue
|
|
success = dist_bot.publish_shorts(article, str(shorts_file))
|
|
if success:
|
|
done_flag.touch()
|
|
|
|
|
|
def _load_article_by_slug(date_str: str, slug: str) -> dict:
|
|
"""날짜+slug로 원본 article 로드"""
|
|
originals_dir = DATA_DIR / 'originals'
|
|
for f in originals_dir.glob(f'{date_str}_*{slug}*.json'):
|
|
try:
|
|
return json.loads(f.read_text(encoding='utf-8'))
|
|
except Exception:
|
|
pass
|
|
return {}
|
|
|
|
|
|
def _publish_next():
|
|
drafts_dir = DATA_DIR / 'drafts'
|
|
drafts_dir.mkdir(exist_ok=True)
|
|
for draft_file in sorted(drafts_dir.glob('*.json')):
|
|
try:
|
|
article = json.loads(draft_file.read_text(encoding='utf-8'))
|
|
if article.get('_pending_openclaw'):
|
|
continue
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
sys.path.insert(0, str(BASE_DIR / 'bots' / 'converters'))
|
|
import publisher_bot
|
|
import blog_converter
|
|
# 변환봇으로 HTML 생성 (이미 변환된 경우 outputs에서 읽음)
|
|
html = blog_converter.convert(article, save_file=False)
|
|
article['_html_content'] = html
|
|
article['_body_is_html'] = True
|
|
publisher_bot.publish(article)
|
|
draft_file.unlink(missing_ok=True)
|
|
break
|
|
except Exception as e:
|
|
logger.error(f"드래프트 처리 오류 ({draft_file.name}): {e}")
|
|
|
|
|
|
def job_analytics_daily():
|
|
logger.info("[스케줄] 분석봇 일일 리포트")
|
|
try:
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import analytics_bot
|
|
analytics_bot.daily_report()
|
|
except Exception as e:
|
|
logger.error(f"분석봇 오류: {e}")
|
|
|
|
|
|
def job_analytics_weekly():
|
|
logger.info("[스케줄] 분석봇 주간 리포트")
|
|
try:
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import analytics_bot
|
|
analytics_bot.weekly_report()
|
|
except Exception as e:
|
|
logger.error(f"분석봇 주간 리포트 오류: {e}")
|
|
|
|
|
|
def job_image_prompt_batch():
|
|
"""request 모드 전용 — 매주 월요일 10:00 프롬프트 배치 전송"""
|
|
if IMAGE_MODE != 'request':
|
|
return
|
|
logger.info("[스케줄] 이미지 프롬프트 배치 전송")
|
|
try:
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import image_bot
|
|
image_bot.send_prompt_batch()
|
|
except Exception as e:
|
|
logger.error(f"이미지 배치 오류: {e}")
|
|
|
|
|
|
def job_novel_pipeline():
|
|
"""소설 파이프라인 — 월/목 09:00 활성 소설 에피소드 자동 생성"""
|
|
logger.info("[스케줄] 소설 파이프라인 시작")
|
|
try:
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
from novel.novel_manager import NovelManager
|
|
manager = NovelManager()
|
|
results = manager.run_all()
|
|
if results:
|
|
for r in results:
|
|
if r.get('error'):
|
|
logger.error(f"소설 파이프라인 오류 [{r['novel_id']}]: {r['error']}")
|
|
else:
|
|
logger.info(
|
|
f"소설 에피소드 완료 [{r['novel_id']}] "
|
|
f"제{r['episode_num']}화 blog={bool(r['blog_path'])} "
|
|
f"shorts={bool(r['shorts_path'])}"
|
|
)
|
|
else:
|
|
logger.info("[소설] 오늘 발행 예정 소설 없음")
|
|
except Exception as e:
|
|
logger.error(f"소설 파이프라인 오류: {e}")
|
|
|
|
|
|
# ─── Telegram 명령 핸들러 ────────────────────────────
|
|
|
|
async def cmd_status(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
status = "🟢 발행 활성" if _publish_enabled else "🔴 발행 중단"
|
|
mode_label = {'manual': '수동', 'request': '요청', 'auto': '자동'}.get(IMAGE_MODE, IMAGE_MODE)
|
|
await update.message.reply_text(
|
|
f"블로그 엔진 상태: {status}\n이미지 모드: {mode_label} ({IMAGE_MODE})"
|
|
)
|
|
|
|
|
|
async def cmd_stop_publish(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
global _publish_enabled
|
|
_publish_enabled = False
|
|
await update.message.reply_text("🔴 발행이 중단되었습니다.")
|
|
|
|
|
|
async def cmd_resume_publish(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
global _publish_enabled
|
|
_publish_enabled = True
|
|
await update.message.reply_text("🟢 발행이 재개되었습니다.")
|
|
|
|
|
|
async def cmd_collect(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
await update.message.reply_text("🔄 글감 수집을 시작합니다...")
|
|
loop = asyncio.get_event_loop()
|
|
try:
|
|
await loop.run_in_executor(None, job_collector)
|
|
topics_dir = DATA_DIR / 'topics'
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
files = sorted(topics_dir.glob(f'{today}_*.json'))
|
|
if not files:
|
|
await update.message.reply_text("✅ 수집 완료! 오늘 수집된 글감이 없습니다.")
|
|
return
|
|
# 텔레그램 메시지 4096자 제한 고려, 페이지 나눠 전송
|
|
total = len(files)
|
|
page_size = 30
|
|
for page_start in range(0, total, page_size):
|
|
page_files = files[page_start:page_start + page_size]
|
|
if page_start == 0:
|
|
lines = [f"✅ 수집 완료! 오늘 글감 {total}개:"]
|
|
else:
|
|
lines = [f"📋 계속 ({page_start + 1}~{page_start + len(page_files)}):"]
|
|
for i, f in enumerate(page_files, page_start + 1):
|
|
try:
|
|
data = json.loads(f.read_text(encoding='utf-8'))
|
|
lines.append(f" {i}. [{data.get('corner','')}] {data.get('topic','')[:40]}")
|
|
except Exception:
|
|
pass
|
|
if page_start + page_size >= total:
|
|
lines.append(f"\n✍️ /write [번호] 로 글 작성 (1~{total})")
|
|
await update.message.reply_text('\n'.join(lines))
|
|
except Exception as e:
|
|
await update.message.reply_text(f"❌ 수집 오류: {e}")
|
|
|
|
|
|
async def cmd_write(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
topics_dir = DATA_DIR / 'topics'
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
files = sorted(topics_dir.glob(f'{today}_*.json'))
|
|
if not files:
|
|
await update.message.reply_text("오늘 수집된 글감이 없습니다. /collect 먼저 실행하세요.")
|
|
return
|
|
args = context.args
|
|
if not args:
|
|
lines = ["📋 글감 목록 (번호를 선택하세요):"]
|
|
for i, f in enumerate(files[:10], 1):
|
|
try:
|
|
data = json.loads(f.read_text(encoding='utf-8'))
|
|
lines.append(f" {i}. [{data.get('corner','')}] {data.get('topic','')[:50]}")
|
|
except Exception:
|
|
pass
|
|
lines.append("\n사용법: /write [번호] [방향(선택)]")
|
|
await update.message.reply_text('\n'.join(lines))
|
|
return
|
|
try:
|
|
idx = int(args[0]) - 1
|
|
if idx < 0 or idx >= len(files):
|
|
await update.message.reply_text(f"❌ 1~{len(files)} 사이 번호를 입력하세요.")
|
|
return
|
|
except ValueError:
|
|
await update.message.reply_text("❌ 숫자를 입력하세요. 예: /write 1")
|
|
return
|
|
topic_file = files[idx]
|
|
topic_data = json.loads(topic_file.read_text(encoding='utf-8'))
|
|
direction = ' '.join(args[1:]) if len(args) > 1 else ''
|
|
draft_path = DATA_DIR / 'originals' / topic_file.name
|
|
(DATA_DIR / 'originals').mkdir(exist_ok=True)
|
|
await update.message.reply_text(
|
|
f"✍️ 글 작성 중...\n주제: {topic_data.get('topic','')[:50]}"
|
|
+ (f"\n방향: {direction}" if direction else "")
|
|
)
|
|
loop = asyncio.get_event_loop()
|
|
try:
|
|
await loop.run_in_executor(None, _call_openclaw, topic_data, draft_path, direction)
|
|
# 자동으로 pending_review로 이동
|
|
await loop.run_in_executor(None, _publish_next)
|
|
# pending_review에서 방금 작성된 글 찾기
|
|
pending_dir = DATA_DIR / 'pending_review'
|
|
pending_file = pending_dir / topic_file.name
|
|
if pending_file.exists():
|
|
article = json.loads(pending_file.read_text(encoding='utf-8'))
|
|
title = article.get('title', '')[:50]
|
|
corner = article.get('corner', '')
|
|
body_preview = article.get('body', '')[:200].replace('<', '<').replace('>', '>')
|
|
# 인라인 버튼으로 승인/거부
|
|
keyboard = InlineKeyboardMarkup([
|
|
[
|
|
InlineKeyboardButton("✅ 승인 발행", callback_data=f"approve:{topic_file.name}"),
|
|
InlineKeyboardButton("🗑 거부", callback_data=f"reject:{topic_file.name}"),
|
|
]
|
|
])
|
|
await update.message.reply_text(
|
|
f"📝 [수동 검토 필요]\n\n"
|
|
f"<b>{title}</b>\n"
|
|
f"코너: {corner}\n\n"
|
|
f"미리보기:\n{body_preview}...\n",
|
|
parse_mode='HTML',
|
|
reply_markup=keyboard,
|
|
)
|
|
else:
|
|
await update.message.reply_text("✅ 완료! /pending 으로 검토하세요.")
|
|
except Exception as e:
|
|
await update.message.reply_text(f"❌ 글 작성 오류: {e}")
|
|
|
|
|
|
async def cmd_show_topics(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
topics_dir = DATA_DIR / 'topics'
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
files = sorted(topics_dir.glob(f'{today}_*.json'))
|
|
if not files:
|
|
await update.message.reply_text("오늘 수집된 글감이 없습니다.")
|
|
return
|
|
total = len(files)
|
|
page_size = 30
|
|
for page_start in range(0, total, page_size):
|
|
page_files = files[page_start:page_start + page_size]
|
|
if page_start == 0:
|
|
lines = [f"📋 오늘 수집된 글감 ({total}개):"]
|
|
else:
|
|
lines = [f"📋 계속 ({page_start + 1}~{page_start + len(page_files)}):"]
|
|
for i, f in enumerate(page_files, page_start + 1):
|
|
try:
|
|
data = json.loads(f.read_text(encoding='utf-8'))
|
|
lines.append(f" {i}. [{data.get('quality_score',0)}점][{data.get('corner','')}] {data.get('topic','')[:40]}")
|
|
except Exception:
|
|
pass
|
|
if page_start + page_size >= total:
|
|
lines.append(f"\n✍️ /write [번호] 로 글 작성 (1~{total})")
|
|
await update.message.reply_text('\n'.join(lines))
|
|
|
|
|
|
async def cmd_pending(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import publisher_bot
|
|
pending = publisher_bot.get_pending_list()
|
|
if not pending:
|
|
await update.message.reply_text("수동 검토 대기 글이 없습니다.")
|
|
return
|
|
for i, item in enumerate(pending[:10], 1):
|
|
filepath = item.get('_filepath', '')
|
|
filename = Path(filepath).name if filepath else ''
|
|
title = item.get('title', '')[:50]
|
|
corner = item.get('corner', '')
|
|
reason = item.get('pending_reason', '')
|
|
body_preview = item.get('body', '')[:150].replace('<', '<').replace('>', '>')
|
|
keyboard = InlineKeyboardMarkup([
|
|
[
|
|
InlineKeyboardButton("✅ 승인 발행", callback_data=f"approve:{filename}"),
|
|
InlineKeyboardButton("🗑 거부", callback_data=f"reject:{filename}"),
|
|
]
|
|
])
|
|
await update.message.reply_text(
|
|
f"🔍 [{i}/{len(pending)}] 수동 검토 대기\n\n"
|
|
f"<b>{title}</b>\n"
|
|
f"코너: {corner}\n"
|
|
f"사유: {reason}\n\n"
|
|
f"{body_preview}...",
|
|
parse_mode='HTML',
|
|
reply_markup=keyboard,
|
|
)
|
|
|
|
|
|
async def cmd_approve(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import publisher_bot
|
|
pending = publisher_bot.get_pending_list()
|
|
if not pending:
|
|
await update.message.reply_text("대기 글이 없습니다.")
|
|
return
|
|
args = context.args
|
|
idx = int(args[0]) - 1 if args and args[0].isdigit() else 0
|
|
if not (0 <= idx < len(pending)):
|
|
await update.message.reply_text("잘못된 번호입니다.")
|
|
return
|
|
success = publisher_bot.approve_pending(pending[idx].get('_filepath', ''))
|
|
await update.message.reply_text(
|
|
f"✅ 승인 완료: {pending[idx].get('title','')}" if success else "❌ 발행 실패. 로그 확인."
|
|
)
|
|
|
|
|
|
async def cmd_reject(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import publisher_bot
|
|
pending = publisher_bot.get_pending_list()
|
|
if not pending:
|
|
await update.message.reply_text("대기 글이 없습니다.")
|
|
return
|
|
args = context.args
|
|
idx = int(args[0]) - 1 if args and args[0].isdigit() else 0
|
|
if not (0 <= idx < len(pending)):
|
|
await update.message.reply_text("잘못된 번호입니다.")
|
|
return
|
|
publisher_bot.reject_pending(pending[idx].get('_filepath', ''))
|
|
await update.message.reply_text(f"🗑 거부 완료: {pending[idx].get('title','')}")
|
|
|
|
|
|
async def callback_approve_reject(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""인라인 버튼 콜백: 승인/거부"""
|
|
query = update.callback_query
|
|
await query.answer()
|
|
data = query.data # "approve:filename.json" or "reject:filename.json"
|
|
action, filename = data.split(':', 1)
|
|
pending_dir = DATA_DIR / 'pending_review'
|
|
filepath = pending_dir / filename
|
|
|
|
if not filepath.exists():
|
|
await query.edit_message_text("⚠️ 해당 글을 찾을 수 없습니다. (이미 처리됨)")
|
|
return
|
|
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import publisher_bot
|
|
|
|
if action == 'approve':
|
|
success = publisher_bot.approve_pending(str(filepath))
|
|
if success:
|
|
await query.edit_message_text(f"✅ 발행 완료!\n\n{query.message.text_html or query.message.text}", parse_mode='HTML')
|
|
else:
|
|
await query.edit_message_text("❌ 발행 실패. 로그를 확인하세요.")
|
|
elif action == 'reject':
|
|
publisher_bot.reject_pending(str(filepath))
|
|
await query.edit_message_text(f"🗑 거부 완료\n\n{query.message.text_html or query.message.text}", parse_mode='HTML')
|
|
|
|
|
|
async def cmd_report(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
await update.message.reply_text("주간 리포트 생성 중...")
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import analytics_bot
|
|
analytics_bot.weekly_report()
|
|
|
|
|
|
async def cmd_convert(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""수동 변환 실행"""
|
|
await update.message.reply_text("변환 엔진 실행 중...")
|
|
try:
|
|
_run_conversion_pipeline()
|
|
outputs_dir = DATA_DIR / 'outputs'
|
|
today = datetime.now().strftime('%Y%m%d')
|
|
blogs = len(list(outputs_dir.glob(f'{today}_*_blog.html')))
|
|
cards = len(list(outputs_dir.glob(f'{today}_*_card.png')))
|
|
threads = len(list(outputs_dir.glob(f'{today}_*_thread.json')))
|
|
await update.message.reply_text(
|
|
f"변환 완료\n"
|
|
f"블로그 HTML: {blogs}개\n"
|
|
f"인스타 카드: {cards}개\n"
|
|
f"X 스레드: {threads}개"
|
|
)
|
|
except Exception as e:
|
|
await update.message.reply_text(f"변환 오류: {e}")
|
|
|
|
|
|
# ─── 이미지 관련 명령 (request 모드) ────────────────
|
|
|
|
async def cmd_images(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""대기 중인 이미지 프롬프트 목록 표시"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import image_bot
|
|
pending = image_bot.get_pending_prompts('pending')
|
|
selected = image_bot.get_pending_prompts('selected')
|
|
done = image_bot.get_pending_prompts('done')
|
|
|
|
if not pending and not selected:
|
|
await update.message.reply_text(
|
|
f"🎨 대기 중인 이미지 요청이 없습니다.\n"
|
|
f"완료된 이미지: {len(done)}개\n\n"
|
|
f"/imgbatch — 지금 바로 배치 전송 요청"
|
|
)
|
|
return
|
|
|
|
lines = [f"🎨 이미지 제작 현황\n"]
|
|
if pending:
|
|
lines.append(f"⏳ 대기 ({len(pending)}건):")
|
|
for p in pending:
|
|
lines.append(f" #{p['id']} {p['topic'][:40]}")
|
|
if selected:
|
|
lines.append(f"\n🔄 진행 중 ({len(selected)}건):")
|
|
for p in selected:
|
|
lines.append(f" #{p['id']} {p['topic'][:40]}")
|
|
lines.append(f"\n✅ 완료: {len(done)}건")
|
|
lines.append(
|
|
f"\n/imgpick [번호] — 프롬프트 받기\n"
|
|
f"/imgbatch — 전체 목록 재전송"
|
|
)
|
|
await update.message.reply_text('\n'.join(lines))
|
|
|
|
|
|
async def cmd_imgpick(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""특정 번호 프롬프트 선택 → 전체 프롬프트 전송 + 이미지 대기 상태 진입"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import image_bot
|
|
|
|
args = context.args
|
|
if not args or not args[0].isdigit():
|
|
await update.message.reply_text("사용법: /imgpick [번호]\n예) /imgpick 3")
|
|
return
|
|
|
|
prompt_id = args[0]
|
|
prompt = image_bot.get_prompt_by_id(prompt_id)
|
|
if not prompt:
|
|
await update.message.reply_text(f"#{prompt_id} 번 프롬프트를 찾을 수 없습니다.\n/images 로 목록 확인")
|
|
return
|
|
|
|
if prompt['status'] == 'done':
|
|
await update.message.reply_text(f"#{prompt_id} 는 이미 완료된 항목입니다.")
|
|
return
|
|
|
|
# 단일 프롬프트 전송 (Telegram 메시지 길이 제한 고려해 분리 전송)
|
|
image_bot.send_single_prompt(prompt_id)
|
|
|
|
# 이미지 대기 상태 등록
|
|
chat_id = update.message.chat_id
|
|
_awaiting_image[chat_id] = prompt_id
|
|
logger.info(f"이미지 대기 등록: chat={chat_id}, prompt=#{prompt_id}")
|
|
|
|
|
|
async def cmd_imgbatch(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""전체 대기 프롬프트 배치 전송 (수동 트리거)"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import image_bot
|
|
image_bot.send_prompt_batch()
|
|
await update.message.reply_text("📤 프롬프트 배치 전송 완료.")
|
|
|
|
|
|
async def cmd_novel_list(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""소설 목록 조회"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
from novel.novel_manager import handle_novel_command
|
|
await handle_novel_command(update, context, 'list', [])
|
|
|
|
|
|
async def cmd_novel_gen(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""소설 에피소드 즉시 생성: /novel_gen [novel_id]"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
from novel.novel_manager import handle_novel_command
|
|
await handle_novel_command(update, context, 'gen', context.args or [])
|
|
|
|
|
|
async def cmd_novel_status(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""소설 파이프라인 진행 현황"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
from novel.novel_manager import handle_novel_command
|
|
await handle_novel_command(update, context, 'status', [])
|
|
|
|
|
|
async def cmd_imgcancel(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""이미지 대기 상태 취소"""
|
|
chat_id = update.message.chat_id
|
|
if chat_id in _awaiting_image:
|
|
pid = _awaiting_image.pop(chat_id)
|
|
await update.message.reply_text(f"❌ #{pid} 이미지 대기 취소.")
|
|
else:
|
|
await update.message.reply_text("현재 대기 중인 이미지 요청이 없습니다.")
|
|
|
|
|
|
# ─── 이미지/파일 수신 핸들러 ─────────────────────────
|
|
|
|
async def _receive_image(update: Update, context: ContextTypes.DEFAULT_TYPE,
|
|
file_getter, caption: str):
|
|
"""공통 이미지 수신 처리 (photo / document)"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
import image_bot
|
|
|
|
chat_id = update.message.chat_id
|
|
|
|
# 프롬프트 ID 결정: 대기 상태 > 캡션 파싱 > 없음
|
|
prompt_id = _awaiting_image.get(chat_id)
|
|
if not prompt_id and caption:
|
|
# 캡션에 #번호 형식이 있으면 추출
|
|
m = __import__('re').search(r'#(\d+)', caption)
|
|
if m:
|
|
prompt_id = m.group(1)
|
|
|
|
if not prompt_id:
|
|
await update.message.reply_text(
|
|
"⚠ 어느 주제의 이미지인지 알 수 없습니다.\n\n"
|
|
"방법 1: /imgpick [번호] 로 먼저 선택 후 이미지 전송\n"
|
|
"방법 2: 이미지 캡션에 #번호 입력 (예: #3)\n\n"
|
|
"/images — 현재 대기 목록 확인"
|
|
)
|
|
return
|
|
|
|
# Telegram에서 파일 다운로드
|
|
try:
|
|
tg_file = await file_getter()
|
|
file_bytes = (await tg_file.download_as_bytearray())
|
|
except Exception as e:
|
|
await update.message.reply_text(f"❌ 파일 다운로드 실패: {e}")
|
|
return
|
|
|
|
# 저장 및 프롬프트 완료 처리
|
|
image_path = image_bot.save_image_from_telegram(bytes(file_bytes), prompt_id)
|
|
if not image_path:
|
|
await update.message.reply_text(f"❌ 저장 실패. #{prompt_id} 번이 존재하는지 확인하세요.")
|
|
return
|
|
|
|
# 대기 상태 해제
|
|
_awaiting_image.pop(chat_id, None)
|
|
|
|
prompt = image_bot.get_prompt_by_id(prompt_id)
|
|
topic = prompt['topic'] if prompt else ''
|
|
await update.message.reply_text(
|
|
f"✅ <b>이미지 저장 완료!</b>\n\n"
|
|
f"#{prompt_id} {topic}\n"
|
|
f"경로: <code>{image_path}</code>\n\n"
|
|
f"이 이미지는 해당 만평 글 발행 시 자동으로 사용됩니다.",
|
|
parse_mode='HTML',
|
|
)
|
|
logger.info(f"이미지 수령 완료: #{prompt_id} → {image_path}")
|
|
|
|
|
|
async def handle_photo(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""Telegram 사진 수신"""
|
|
caption = update.message.caption or ''
|
|
photo = update.message.photo[-1] # 가장 큰 해상도
|
|
await _receive_image(
|
|
update, context,
|
|
file_getter=lambda: context.bot.get_file(photo.file_id),
|
|
caption=caption,
|
|
)
|
|
|
|
|
|
async def handle_document(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""Telegram 파일(문서) 수신 — 고해상도 이미지 전송 시"""
|
|
doc = update.message.document
|
|
mime = doc.mime_type or ''
|
|
if not mime.startswith('image/'):
|
|
return # 이미지 파일만 처리
|
|
caption = update.message.caption or ''
|
|
await _receive_image(
|
|
update, context,
|
|
file_getter=lambda: context.bot.get_file(doc.file_id),
|
|
caption=caption,
|
|
)
|
|
|
|
|
|
# ─── 텍스트 명령 ─────────────────────────────────────
|
|
|
|
async def handle_text(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
text = update.message.text.strip()
|
|
chat_id = update.message.chat_id
|
|
|
|
cmd_map = {
|
|
'발행 중단': cmd_stop_publish,
|
|
'발행 재개': cmd_resume_publish,
|
|
'오늘 수집된 글감 보여줘': cmd_show_topics,
|
|
'이번 주 리포트': cmd_report,
|
|
'대기 중인 글 보여줘': cmd_pending,
|
|
'이미지 목록': cmd_images,
|
|
'변환 실행': cmd_convert,
|
|
'오늘 뭐 발행했어?': cmd_status,
|
|
}
|
|
if text in cmd_map:
|
|
await cmd_map[text](update, context)
|
|
return
|
|
|
|
# Claude API로 자연어 처리
|
|
if not ANTHROPIC_API_KEY:
|
|
await update.message.reply_text(
|
|
"Claude API 키가 없습니다. .env 파일에 ANTHROPIC_API_KEY를 입력하세요."
|
|
)
|
|
return
|
|
|
|
global _claude_client
|
|
if _claude_client is None:
|
|
_claude_client = anthropic.Anthropic(api_key=ANTHROPIC_API_KEY)
|
|
|
|
history = _conversation_history.setdefault(chat_id, [])
|
|
history.append({"role": "user", "content": text})
|
|
|
|
# 대화 기록이 너무 길면 최근 20개만 유지
|
|
if len(history) > 20:
|
|
history[:] = history[-20:]
|
|
|
|
try:
|
|
await context.bot.send_chat_action(chat_id=chat_id, action="typing")
|
|
response = _claude_client.messages.create(
|
|
model="claude-opus-4-6",
|
|
max_tokens=1024,
|
|
system=CLAUDE_SYSTEM_PROMPT,
|
|
messages=history,
|
|
)
|
|
reply = response.content[0].text
|
|
history.append({"role": "assistant", "content": reply})
|
|
await update.message.reply_text(reply)
|
|
except Exception as e:
|
|
logger.error(f"Claude API 오류: {e}")
|
|
await update.message.reply_text(f"오류가 발생했습니다: {e}")
|
|
|
|
|
|
# ─── Shorts Bot 잡 ─────────────────────────────────────
|
|
|
|
def job_shorts_produce():
|
|
"""쇼츠 생산 (shorts_bot.produce) — 블로그 글 → YouTube Shorts."""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
try:
|
|
import shorts_bot
|
|
cfg = shorts_bot._load_config()
|
|
if not cfg.get('enabled', True):
|
|
logger.info("Shorts bot disabled in config — 건너뜀")
|
|
return
|
|
article = shorts_bot.pick_article(cfg)
|
|
if not article:
|
|
logger.info("쇼츠 생산: eligible 글 없음")
|
|
return
|
|
result = shorts_bot.produce(article, dry_run=False, cfg=cfg)
|
|
if result.success:
|
|
msg = f"🎬 쇼츠 발행 완료: {result.youtube_url}"
|
|
else:
|
|
msg = f"⚠️ 쇼츠 생산 실패: {result.error}"
|
|
logger.info(msg)
|
|
_telegram_notify(msg)
|
|
except Exception as e:
|
|
logger.error(f"쇼츠 잡 오류: {e}")
|
|
_telegram_notify(f"⚠️ 쇼츠 잡 오류: {e}")
|
|
|
|
|
|
# ─── Shorts Telegram 명령 ─────────────────────────────
|
|
|
|
async def cmd_shorts(update: Update, context: ContextTypes.DEFAULT_TYPE):
|
|
"""
|
|
/shorts [subcommand] [args]
|
|
subcommands: status, mode, input, character, upload, skip
|
|
"""
|
|
sys.path.insert(0, str(BASE_DIR / 'bots'))
|
|
args = context.args or []
|
|
sub = args[0].lower() if args else 'status'
|
|
|
|
if sub == 'status':
|
|
import shorts_bot
|
|
cfg = shorts_bot._load_config()
|
|
mode = cfg.get('production_mode', 'auto')
|
|
enabled = cfg.get('enabled', True)
|
|
converted = len(shorts_bot._get_converted_ids())
|
|
rendered_dir = DATA_DIR / 'shorts' / 'rendered'
|
|
rendered = len(list(rendered_dir.glob('*.mp4'))) if rendered_dir.exists() else 0
|
|
text = (
|
|
f"🎬 Shorts 현황\n"
|
|
f"{'🟢 활성' if enabled else '🔴 비활성'} | 모드: {mode}\n"
|
|
f"변환 완료: {converted}개 | 렌더링 완료: {rendered}개"
|
|
)
|
|
await update.message.reply_text(text)
|
|
|
|
elif sub == 'mode' and len(args) >= 2:
|
|
new_mode = 'semi_auto' if args[1] in ('semi', 'semi_auto') else 'auto'
|
|
cfg_path = BASE_DIR / 'config' / 'shorts_config.json'
|
|
cfg = json.loads(cfg_path.read_text(encoding='utf-8'))
|
|
cfg['production_mode'] = new_mode
|
|
cfg_path.write_text(json.dumps(cfg, ensure_ascii=False, indent=2), encoding='utf-8')
|
|
await update.message.reply_text(f"✅ Shorts 모드 변경: {new_mode}")
|
|
|
|
elif sub == 'input':
|
|
input_dirs = ['images', 'videos', 'scripts', 'audio']
|
|
lines = ['📂 input/ 폴더 현황']
|
|
for d in input_dirs:
|
|
p = BASE_DIR / 'input' / d
|
|
files = list(p.glob('*.*')) if p.exists() else []
|
|
lines.append(f" {d}/: {len(files)}개")
|
|
await update.message.reply_text('\n'.join(lines))
|
|
|
|
elif sub == 'input' and len(args) >= 2 and args[1] == 'clear':
|
|
import shutil
|
|
for d in ['images', 'videos', 'scripts', 'audio']:
|
|
p = BASE_DIR / 'input' / d
|
|
if p.exists():
|
|
for f in p.glob('*.*'):
|
|
f.unlink(missing_ok=True)
|
|
await update.message.reply_text('✅ input/ 폴더 초기화 완료')
|
|
|
|
elif sub == 'character' and len(args) >= 2:
|
|
char = args[1].lower()
|
|
if char not in ('bao', 'zero'):
|
|
await update.message.reply_text('캐릭터: bao 또는 zero')
|
|
return
|
|
cfg_path = BASE_DIR / 'config' / 'shorts_config.json'
|
|
cfg = json.loads(cfg_path.read_text(encoding='utf-8'))
|
|
# 다음 영상에 강제 적용 — corner_character_map 전체를 지정 캐릭터로 덮어씀
|
|
char_type = 'fourth_path' if char == 'zero' else 'tech_blog'
|
|
for corner in cfg.get('assets', {}).get('corner_character_map', {}):
|
|
cfg['assets']['corner_character_map'][corner] = char_type
|
|
cfg_path.write_text(json.dumps(cfg, ensure_ascii=False, indent=2), encoding='utf-8')
|
|
await update.message.reply_text(f'✅ 다음 쇼츠 캐릭터: {char} ({"Ø" if char == "zero" else "바오"})')
|
|
|
|
elif sub == 'upload' and len(args) >= 2:
|
|
video_path = ' '.join(args[1:])
|
|
await update.message.reply_text(f'📤 업로드 중: {video_path}')
|
|
import shorts_bot
|
|
result = shorts_bot.upload_existing(video_path)
|
|
if result.success:
|
|
await update.message.reply_text(f'✅ 업로드 완료: {result.youtube_url}')
|
|
else:
|
|
await update.message.reply_text(f'❌ 업로드 실패: {result.error}')
|
|
|
|
elif sub == 'skip' and len(args) >= 2:
|
|
article_id = args[1]
|
|
skip_dir = DATA_DIR / 'shorts' / 'published'
|
|
skip_dir.mkdir(parents=True, exist_ok=True)
|
|
skip_path = skip_dir / f'skip_{article_id}.json'
|
|
skip_path.write_text(
|
|
json.dumps({'article_id': article_id, 'skipped': True,
|
|
'time': datetime.now().isoformat()}, ensure_ascii=False),
|
|
encoding='utf-8',
|
|
)
|
|
await update.message.reply_text(f'✅ 쇼츠 건너뜀 등록: {article_id}')
|
|
|
|
elif sub == 'run':
|
|
await update.message.reply_text('🎬 쇼츠 즉시 생산 시작...')
|
|
import asyncio as _asyncio
|
|
loop = _asyncio.get_event_loop()
|
|
loop.run_in_executor(None, job_shorts_produce)
|
|
|
|
else:
|
|
help_text = (
|
|
"🎬 /shorts 명령어\n"
|
|
"/shorts status — 현황\n"
|
|
"/shorts mode auto|semi — 모드 전환\n"
|
|
"/shorts input — input/ 폴더 현황\n"
|
|
"/shorts character bao|zero — 캐릭터 강제 지정\n"
|
|
"/shorts upload [경로] — 렌더링된 영상 업로드\n"
|
|
"/shorts skip [article_id] — 특정 글 쇼츠 제외\n"
|
|
"/shorts run — 즉시 실행"
|
|
)
|
|
await update.message.reply_text(help_text)
|
|
|
|
|
|
# ─── 스케줄러 설정 + 메인 ─────────────────────────────
|
|
|
|
def setup_scheduler() -> AsyncIOScheduler:
|
|
scheduler = AsyncIOScheduler(timezone='Asia/Seoul')
|
|
schedule_cfg = load_schedule()
|
|
|
|
# schedule.json 기반 동적 잡 (기존)
|
|
job_map = {
|
|
'collector': job_collector,
|
|
'ai_writer': job_ai_writer,
|
|
'publish_1': lambda: job_publish(1),
|
|
'publish_2': lambda: job_publish(2),
|
|
'publish_3': lambda: job_publish(3),
|
|
'analytics': job_analytics_daily,
|
|
}
|
|
for job in schedule_cfg.get('jobs', []):
|
|
fn = job_map.get(job['id'])
|
|
if fn:
|
|
scheduler.add_job(fn, 'cron', hour=job['hour'], minute=job['minute'], id=job['id'])
|
|
|
|
# v3 고정 스케줄: 시차 배포
|
|
# 07:00 수집봇 (schedule.json에서 관리)
|
|
# 08:00 AI 글 작성 (schedule.json에서 관리)
|
|
scheduler.add_job(job_convert, 'cron', hour=8, minute=30, id='convert') # 08:30 변환
|
|
scheduler.add_job(lambda: job_publish(1), 'cron',
|
|
hour=9, minute=0, id='blog_publish') # 09:00 블로그
|
|
scheduler.add_job(job_distribute_instagram, 'cron',
|
|
hour=10, minute=0, id='instagram_dist') # 10:00 인스타 카드
|
|
scheduler.add_job(job_distribute_instagram_reels, 'cron',
|
|
hour=10, minute=30, id='instagram_reels_dist') # 10:30 인스타 릴스
|
|
scheduler.add_job(job_distribute_x, 'cron',
|
|
hour=11, minute=0, id='x_dist') # 11:00 X
|
|
scheduler.add_job(job_distribute_tiktok, 'cron',
|
|
hour=18, minute=0, id='tiktok_dist') # 18:00 틱톡
|
|
scheduler.add_job(job_distribute_youtube, 'cron',
|
|
hour=20, minute=0, id='youtube_dist') # 20:00 유튜브
|
|
scheduler.add_job(job_analytics_daily, 'cron',
|
|
hour=22, minute=0, id='daily_report') # 22:00 분석
|
|
scheduler.add_job(job_analytics_weekly, 'cron',
|
|
day_of_week='sun', hour=22, minute=30, id='weekly_report') # 일요일 주간
|
|
|
|
# request 모드: 매주 월요일 10:00 이미지 프롬프트 배치 전송
|
|
if IMAGE_MODE == 'request':
|
|
scheduler.add_job(job_image_prompt_batch, 'cron',
|
|
day_of_week='mon', hour=10, minute=0, id='image_batch')
|
|
logger.info("이미지 request 모드: 매주 월요일 10:00 배치 전송 등록")
|
|
|
|
# 소설 파이프라인: 매주 월/목 09:00
|
|
scheduler.add_job(job_novel_pipeline, 'cron',
|
|
day_of_week='mon,thu', hour=9, minute=0, id='novel_pipeline')
|
|
logger.info("소설 파이프라인: 매주 월/목 09:00 등록")
|
|
|
|
# Shorts Bot: 10:35 (첫 번째), 16:00 (두 번째)
|
|
try:
|
|
import json as _json
|
|
shorts_cfg_path = CONFIG_DIR / 'shorts_config.json'
|
|
if shorts_cfg_path.exists():
|
|
_shorts_cfg = _json.loads(shorts_cfg_path.read_text(encoding='utf-8'))
|
|
if _shorts_cfg.get('enabled', True):
|
|
scheduler.add_job(job_shorts_produce, 'cron',
|
|
hour=10, minute=35, id='shorts_produce_1') # 10:35 첫 번째 쇼츠
|
|
scheduler.add_job(job_shorts_produce, 'cron',
|
|
hour=16, minute=0, id='shorts_produce_2') # 16:00 두 번째 쇼츠
|
|
logger.info("Shorts Bot: 10:35, 16:00 등록")
|
|
except Exception as _e:
|
|
logger.warning(f"Shorts 스케줄 등록 실패: {_e}")
|
|
|
|
logger.info("스케줄러 설정 완료 (v3 시차 배포 + 소설 파이프라인 + Shorts Bot)")
|
|
return scheduler
|
|
|
|
|
|
async def main():
|
|
logger.info("=== 블로그 엔진 스케줄러 시작 ===")
|
|
scheduler = setup_scheduler()
|
|
scheduler.start()
|
|
|
|
if TELEGRAM_BOT_TOKEN:
|
|
app = Application.builder().token(TELEGRAM_BOT_TOKEN).build()
|
|
|
|
# 발행 관련
|
|
app.add_handler(CommandHandler('status', cmd_status))
|
|
app.add_handler(CommandHandler('collect', cmd_collect))
|
|
app.add_handler(CommandHandler('write', cmd_write))
|
|
app.add_handler(CommandHandler('approve', cmd_approve))
|
|
app.add_handler(CommandHandler('reject', cmd_reject))
|
|
app.add_handler(CommandHandler('pending', cmd_pending))
|
|
app.add_handler(CallbackQueryHandler(callback_approve_reject, pattern=r'^(approve|reject):'))
|
|
app.add_handler(CommandHandler('report', cmd_report))
|
|
app.add_handler(CommandHandler('topics', cmd_show_topics))
|
|
app.add_handler(CommandHandler('convert', cmd_convert))
|
|
|
|
# 이미지 관련 (request / manual 공통 사용 가능)
|
|
app.add_handler(CommandHandler('images', cmd_images))
|
|
app.add_handler(CommandHandler('imgpick', cmd_imgpick))
|
|
app.add_handler(CommandHandler('imgbatch', cmd_imgbatch))
|
|
app.add_handler(CommandHandler('imgcancel', cmd_imgcancel))
|
|
|
|
# 소설 파이프라인
|
|
app.add_handler(CommandHandler('novel_list', cmd_novel_list))
|
|
app.add_handler(CommandHandler('novel_gen', cmd_novel_gen))
|
|
app.add_handler(CommandHandler('novel_status', cmd_novel_status))
|
|
|
|
# Shorts Bot
|
|
app.add_handler(CommandHandler('shorts', cmd_shorts))
|
|
|
|
# 이미지 파일 수신
|
|
app.add_handler(MessageHandler(filters.PHOTO, handle_photo))
|
|
app.add_handler(MessageHandler(filters.Document.IMAGE, handle_document))
|
|
|
|
# 텍스트 명령
|
|
app.add_handler(MessageHandler(filters.TEXT & ~filters.COMMAND, handle_text))
|
|
|
|
logger.info("Telegram 봇 시작")
|
|
await app.initialize()
|
|
await app.start()
|
|
await app.updater.start_polling(drop_pending_updates=True)
|
|
|
|
try:
|
|
while True:
|
|
await asyncio.sleep(3600)
|
|
except (KeyboardInterrupt, SystemExit):
|
|
logger.info("종료 신호 수신")
|
|
finally:
|
|
await app.updater.stop()
|
|
await app.stop()
|
|
await app.shutdown()
|
|
scheduler.shutdown()
|
|
else:
|
|
logger.warning("TELEGRAM_BOT_TOKEN 없음 — 스케줄러만 실행")
|
|
try:
|
|
while True:
|
|
await asyncio.sleep(3600)
|
|
except (KeyboardInterrupt, SystemExit):
|
|
scheduler.shutdown()
|
|
|
|
logger.info("=== 블로그 엔진 스케줄러 종료 ===")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
asyncio.run(main())
|