441 lines
16 KiB
Python
441 lines
16 KiB
Python
import os
|
|
import uuid
|
|
import logging
|
|
from fastapi import FastAPI, HTTPException, BackgroundTasks
|
|
from fastapi.middleware.cors import CORSMiddleware
|
|
from pydantic import BaseModel
|
|
from typing import List, Optional
|
|
|
|
from .config import CORS_ALLOW_ORIGINS, NAVER_CLIENT_ID, ANTHROPIC_API_KEY
|
|
from .db import (
|
|
init_db,
|
|
get_keyword_analyses, get_keyword_analysis, delete_keyword_analysis,
|
|
add_keyword_analysis,
|
|
get_posts, get_post, add_post, update_post, delete_post,
|
|
get_commissions, add_commission, update_commission, delete_commission,
|
|
get_dashboard_stats,
|
|
get_task, create_task, update_task,
|
|
add_brand_link, get_brand_links, update_brand_link, delete_brand_link,
|
|
link_brand_links_to_post,
|
|
)
|
|
from .naver_search import analyze_keyword_with_crawling
|
|
from .content_generator import generate_trend_brief, generate_blog_post, regenerate_blog_post
|
|
from .quality_reviewer import review_post
|
|
from .marketer import enhance_for_conversion
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
app = FastAPI()
|
|
|
|
_cors_origins = CORS_ALLOW_ORIGINS.split(",")
|
|
app.add_middleware(
|
|
CORSMiddleware,
|
|
allow_origins=[o.strip() for o in _cors_origins],
|
|
allow_credentials=False,
|
|
allow_methods=["GET", "POST", "PUT", "DELETE", "OPTIONS"],
|
|
allow_headers=["Content-Type"],
|
|
)
|
|
|
|
|
|
@app.on_event("startup")
|
|
def on_startup():
|
|
init_db()
|
|
os.makedirs("/app/data", exist_ok=True)
|
|
|
|
|
|
@app.get("/health")
|
|
def health():
|
|
return {"ok": True}
|
|
|
|
|
|
@app.get("/api/blog-marketing/status")
|
|
def service_status():
|
|
"""서비스 상태 및 설정 현황."""
|
|
return {
|
|
"ok": True,
|
|
"naver_api": bool(NAVER_CLIENT_ID),
|
|
"claude_api": bool(ANTHROPIC_API_KEY),
|
|
}
|
|
|
|
|
|
# ── 키워드 분석 API ──────────────────────────────────────────────────────────
|
|
|
|
class ResearchRequest(BaseModel):
|
|
keyword: str
|
|
|
|
|
|
def _run_research(task_id: str, keyword: str):
|
|
"""BackgroundTask: 네이버 검색 → 키워드 분석 → DB 저장."""
|
|
try:
|
|
update_task(task_id, "processing", 30, "네이버 검색 중...")
|
|
result = analyze_keyword_with_crawling(keyword)
|
|
|
|
update_task(task_id, "processing", 80, "분석 결과 저장 중...")
|
|
saved = add_keyword_analysis(result)
|
|
|
|
update_task(task_id, "succeeded", 100, "분석 완료", result_id=saved["id"])
|
|
except Exception as e:
|
|
logger.exception("Research failed for keyword=%s", keyword)
|
|
update_task(task_id, "failed", 0, "", error=str(e))
|
|
|
|
|
|
@app.post("/api/blog-marketing/research")
|
|
def start_research(req: ResearchRequest, background_tasks: BackgroundTasks):
|
|
"""키워드 분석 시작 (BackgroundTask). task_id 즉시 반환."""
|
|
if not NAVER_CLIENT_ID:
|
|
raise HTTPException(status_code=400, detail="Naver API 키가 설정되지 않았습니다")
|
|
if not req.keyword.strip():
|
|
raise HTTPException(status_code=400, detail="키워드를 입력하세요")
|
|
|
|
task_id = str(uuid.uuid4())
|
|
create_task(task_id, "research", {"keyword": req.keyword.strip()})
|
|
background_tasks.add_task(_run_research, task_id, req.keyword.strip())
|
|
return {"task_id": task_id}
|
|
|
|
|
|
@app.get("/api/blog-marketing/research/history")
|
|
def list_research(limit: int = 30):
|
|
return {"analyses": get_keyword_analyses(limit)}
|
|
|
|
|
|
@app.get("/api/blog-marketing/research/{analysis_id}")
|
|
def get_research(analysis_id: int):
|
|
result = get_keyword_analysis(analysis_id)
|
|
if not result:
|
|
raise HTTPException(status_code=404, detail="Analysis not found")
|
|
return result
|
|
|
|
|
|
@app.delete("/api/blog-marketing/research/{analysis_id}")
|
|
def remove_research(analysis_id: int):
|
|
if not delete_keyword_analysis(analysis_id):
|
|
raise HTTPException(status_code=404, detail="Analysis not found")
|
|
return {"ok": True}
|
|
|
|
|
|
# ── 작업 상태 폴링 API ──────────────────────────────────────────────────────
|
|
|
|
@app.get("/api/blog-marketing/task/{task_id}")
|
|
def get_task_status(task_id: str):
|
|
task = get_task(task_id)
|
|
if not task:
|
|
raise HTTPException(status_code=404, detail="Task not found")
|
|
return task
|
|
|
|
|
|
# ── AI 글 생성 API ──────────────────────────────────────────────────────────
|
|
|
|
class GenerateRequest(BaseModel):
|
|
keyword_id: int # keyword_analyses.id
|
|
|
|
|
|
class LinkRequest(BaseModel):
|
|
url: str
|
|
product_name: str
|
|
keyword_id: Optional[int] = None
|
|
post_id: Optional[int] = None
|
|
description: str = ""
|
|
placement_hint: str = ""
|
|
|
|
|
|
def _run_generate(task_id: str, keyword_id: int):
|
|
"""BackgroundTask: 트렌드 브리프 → 블로그 글 생성 → DB 저장."""
|
|
try:
|
|
analysis = get_keyword_analysis(keyword_id)
|
|
if not analysis:
|
|
update_task(task_id, "failed", 0, "", error="키워드 분석 결과를 찾을 수 없습니다")
|
|
return
|
|
|
|
# 연결된 브랜드커넥트 링크 조회
|
|
brand_links = get_brand_links(keyword_id=keyword_id)
|
|
|
|
update_task(task_id, "processing", 20, "트렌드 브리프 생성 중...")
|
|
trend_brief = generate_trend_brief(analysis)
|
|
|
|
update_task(task_id, "processing", 60, "블로그 글 작성 중...")
|
|
post_data = generate_blog_post(analysis, trend_brief, brand_links=brand_links)
|
|
|
|
update_task(task_id, "processing", 90, "저장 중...")
|
|
saved = add_post({
|
|
"keyword_id": keyword_id,
|
|
"title": post_data["title"],
|
|
"body": post_data["body"],
|
|
"excerpt": post_data["excerpt"],
|
|
"tags": post_data["tags"],
|
|
"status": "draft",
|
|
"trend_brief": trend_brief,
|
|
})
|
|
|
|
# keyword_id에 연결된 링크를 post_id에도 연결
|
|
link_brand_links_to_post(keyword_id=keyword_id, post_id=saved["id"])
|
|
|
|
update_task(task_id, "succeeded", 100, "글 생성 완료", result_id=saved["id"])
|
|
except Exception as e:
|
|
logger.exception("Generate failed for keyword_id=%s", keyword_id)
|
|
update_task(task_id, "failed", 0, "", error=str(e))
|
|
|
|
|
|
@app.post("/api/blog-marketing/generate")
|
|
def start_generate(req: GenerateRequest, background_tasks: BackgroundTasks):
|
|
"""AI 블로그 글 생성 시작. task_id 즉시 반환."""
|
|
if not ANTHROPIC_API_KEY:
|
|
raise HTTPException(status_code=400, detail="Claude API 키가 설정되지 않았습니다")
|
|
analysis = get_keyword_analysis(req.keyword_id)
|
|
if not analysis:
|
|
raise HTTPException(status_code=404, detail="키워드 분석 결과를 찾을 수 없습니다")
|
|
|
|
task_id = str(uuid.uuid4())
|
|
create_task(task_id, "generate", {"keyword_id": req.keyword_id})
|
|
background_tasks.add_task(_run_generate, task_id, req.keyword_id)
|
|
return {"task_id": task_id}
|
|
|
|
|
|
# ── 품질 리뷰 API ───────────────────────────────────────────────────────────
|
|
|
|
def _run_review(task_id: str, post_id: int):
|
|
"""BackgroundTask: 블로그 글 품질 리뷰."""
|
|
try:
|
|
post = get_post(post_id)
|
|
if not post:
|
|
update_task(task_id, "failed", 0, "", error="포스트를 찾을 수 없습니다")
|
|
return
|
|
|
|
update_task(task_id, "processing", 50, "품질 리뷰 중...")
|
|
result = review_post(post["title"], post["body"])
|
|
|
|
update_post(post_id, {
|
|
"review_score": result["total"],
|
|
"review_detail": result,
|
|
"status": "reviewed" if result["pass"] else "draft",
|
|
})
|
|
|
|
update_task(task_id, "succeeded", 100, "리뷰 완료", result_id=post_id)
|
|
except Exception as e:
|
|
logger.exception("Review failed for post_id=%s", post_id)
|
|
update_task(task_id, "failed", 0, "", error=str(e))
|
|
|
|
|
|
@app.post("/api/blog-marketing/review/{post_id}")
|
|
def start_review(post_id: int, background_tasks: BackgroundTasks):
|
|
"""블로그 글 품질 리뷰 시작. task_id 즉시 반환."""
|
|
if not ANTHROPIC_API_KEY:
|
|
raise HTTPException(status_code=400, detail="Claude API 키가 설정되지 않았습니다")
|
|
post = get_post(post_id)
|
|
if not post:
|
|
raise HTTPException(status_code=404, detail="Post not found")
|
|
|
|
task_id = str(uuid.uuid4())
|
|
create_task(task_id, "review", {"post_id": post_id})
|
|
background_tasks.add_task(_run_review, task_id, post_id)
|
|
return {"task_id": task_id}
|
|
|
|
|
|
# ── 재생성 API ───────────────────────────────────────────────────────────────
|
|
|
|
def _run_regenerate(task_id: str, post_id: int):
|
|
"""BackgroundTask: 피드백 기반 블로그 글 재생성."""
|
|
try:
|
|
post = get_post(post_id)
|
|
if not post:
|
|
update_task(task_id, "failed", 0, "", error="포스트를 찾을 수 없습니다")
|
|
return
|
|
|
|
analysis = get_keyword_analysis(post["keyword_id"]) if post["keyword_id"] else {}
|
|
feedback = post.get("review_detail", {}).get("feedback", "개선이 필요합니다")
|
|
|
|
update_task(task_id, "processing", 50, "글 재생성 중...")
|
|
result = regenerate_blog_post(
|
|
analysis or {"keyword": ""},
|
|
post.get("trend_brief", ""),
|
|
post["body"],
|
|
feedback,
|
|
)
|
|
|
|
update_post(post_id, {
|
|
"title": result["title"],
|
|
"body": result["body"],
|
|
"excerpt": result["excerpt"],
|
|
"tags": result["tags"],
|
|
"status": "draft",
|
|
"review_score": None,
|
|
"review_detail": {},
|
|
})
|
|
|
|
update_task(task_id, "succeeded", 100, "재생성 완료", result_id=post_id)
|
|
except Exception as e:
|
|
logger.exception("Regenerate failed for post_id=%s", post_id)
|
|
update_task(task_id, "failed", 0, "", error=str(e))
|
|
|
|
|
|
@app.post("/api/blog-marketing/regenerate/{post_id}")
|
|
def start_regenerate(post_id: int, background_tasks: BackgroundTasks):
|
|
"""피드백 기반 블로그 글 재생성. task_id 즉시 반환."""
|
|
if not ANTHROPIC_API_KEY:
|
|
raise HTTPException(status_code=400, detail="Claude API 키가 설정되지 않았습니다")
|
|
post = get_post(post_id)
|
|
if not post:
|
|
raise HTTPException(status_code=404, detail="Post not found")
|
|
|
|
task_id = str(uuid.uuid4())
|
|
create_task(task_id, "regenerate", {"post_id": post_id})
|
|
background_tasks.add_task(_run_regenerate, task_id, post_id)
|
|
return {"task_id": task_id}
|
|
|
|
|
|
# ── 포스트 CRUD API ──────────────────────────────────────────────────────────
|
|
|
|
@app.get("/api/blog-marketing/posts")
|
|
def list_posts(status: str = None, limit: int = 50):
|
|
return {"posts": get_posts(status=status, limit=limit)}
|
|
|
|
|
|
@app.get("/api/blog-marketing/posts/{post_id}")
|
|
def get_post_detail(post_id: int):
|
|
post = get_post(post_id)
|
|
if not post:
|
|
raise HTTPException(status_code=404, detail="Post not found")
|
|
return post
|
|
|
|
|
|
@app.put("/api/blog-marketing/posts/{post_id}")
|
|
def edit_post(post_id: int, data: dict):
|
|
result = update_post(post_id, data)
|
|
if not result:
|
|
raise HTTPException(status_code=404, detail="Post not found")
|
|
return result
|
|
|
|
|
|
@app.delete("/api/blog-marketing/posts/{post_id}")
|
|
def remove_post(post_id: int):
|
|
if not delete_post(post_id):
|
|
raise HTTPException(status_code=404, detail="Post not found")
|
|
return {"ok": True}
|
|
|
|
|
|
@app.post("/api/blog-marketing/posts/{post_id}/publish")
|
|
def publish_post(post_id: int, data: dict = None):
|
|
"""네이버 URL 등록 + 상태를 published로 변경."""
|
|
naver_url = (data or {}).get("naver_url", "")
|
|
result = update_post(post_id, {"status": "published", "naver_url": naver_url})
|
|
if not result:
|
|
raise HTTPException(status_code=404, detail="Post not found")
|
|
return result
|
|
|
|
|
|
# ── 브랜드커넥트 링크 API ──────────────────────────────────────────────────
|
|
|
|
@app.post("/api/blog-marketing/links", status_code=201)
|
|
def create_link(req: LinkRequest):
|
|
return add_brand_link(req.model_dump())
|
|
|
|
|
|
@app.get("/api/blog-marketing/links")
|
|
def list_links(post_id: int = None, keyword_id: int = None):
|
|
return {"links": get_brand_links(post_id=post_id, keyword_id=keyword_id)}
|
|
|
|
|
|
@app.put("/api/blog-marketing/links/{link_id}")
|
|
def edit_link(link_id: int, data: dict):
|
|
result = update_brand_link(link_id, data)
|
|
if not result:
|
|
raise HTTPException(status_code=404, detail="Link not found")
|
|
return result
|
|
|
|
|
|
@app.delete("/api/blog-marketing/links/{link_id}")
|
|
def remove_link(link_id: int):
|
|
if not delete_brand_link(link_id):
|
|
raise HTTPException(status_code=404, detail="Link not found")
|
|
return {"ok": True}
|
|
|
|
|
|
# ── 마케터 API ──────────────────────────────────────────────────────────────
|
|
|
|
def _run_market(task_id: str, post_id: int):
|
|
"""BackgroundTask: 마케터 전환율 강화."""
|
|
try:
|
|
post = get_post(post_id)
|
|
if not post:
|
|
update_task(task_id, "failed", 0, "", error="포스트를 찾을 수 없습니다")
|
|
return
|
|
|
|
brand_links = get_brand_links(post_id=post_id)
|
|
if not brand_links and post.get("keyword_id"):
|
|
brand_links = get_brand_links(keyword_id=post["keyword_id"])
|
|
|
|
if not brand_links:
|
|
update_task(task_id, "failed", 0, "", error="브랜드커넥트 링크가 없습니다. 먼저 링크를 등록하세요.")
|
|
return
|
|
|
|
analysis = get_keyword_analysis(post["keyword_id"]) if post.get("keyword_id") else {}
|
|
keyword = (analysis or {}).get("keyword", "")
|
|
|
|
update_task(task_id, "processing", 50, "마케터가 전환율 강화 중...")
|
|
result = enhance_for_conversion(
|
|
post_body=post["body"],
|
|
post_title=post["title"],
|
|
brand_links=brand_links,
|
|
keyword=keyword,
|
|
)
|
|
|
|
update_post(post_id, {
|
|
"title": result["title"],
|
|
"body": result["body"],
|
|
"excerpt": result["excerpt"],
|
|
"status": "marketed",
|
|
})
|
|
|
|
update_task(task_id, "succeeded", 100, "마케팅 강화 완료", result_id=post_id)
|
|
except Exception as e:
|
|
logger.exception("Market failed for post_id=%s", post_id)
|
|
update_task(task_id, "failed", 0, "", error=str(e))
|
|
|
|
|
|
@app.post("/api/blog-marketing/market/{post_id}")
|
|
def start_market(post_id: int, background_tasks: BackgroundTasks):
|
|
"""마케터 단계 실행. task_id 즉시 반환."""
|
|
if not ANTHROPIC_API_KEY:
|
|
raise HTTPException(status_code=400, detail="Claude API 키가 설정되지 않았습니다")
|
|
post = get_post(post_id)
|
|
if not post:
|
|
raise HTTPException(status_code=404, detail="Post not found")
|
|
|
|
task_id = str(uuid.uuid4())
|
|
create_task(task_id, "market", {"post_id": post_id})
|
|
background_tasks.add_task(_run_market, task_id, post_id)
|
|
return {"task_id": task_id}
|
|
|
|
|
|
# ── 수익 추적 API ────────────────────────────────────────────────────────────
|
|
|
|
@app.get("/api/blog-marketing/commissions")
|
|
def list_commissions(post_id: int = None, limit: int = 100):
|
|
return {"commissions": get_commissions(post_id=post_id, limit=limit)}
|
|
|
|
|
|
@app.post("/api/blog-marketing/commissions", status_code=201)
|
|
def create_commission(data: dict):
|
|
return add_commission(data)
|
|
|
|
|
|
@app.put("/api/blog-marketing/commissions/{comm_id}")
|
|
def edit_commission(comm_id: int, data: dict):
|
|
result = update_commission(comm_id, data)
|
|
if not result:
|
|
raise HTTPException(status_code=404, detail="Commission not found")
|
|
return result
|
|
|
|
|
|
@app.delete("/api/blog-marketing/commissions/{comm_id}")
|
|
def remove_commission(comm_id: int):
|
|
if not delete_commission(comm_id):
|
|
raise HTTPException(status_code=404, detail="Commission not found")
|
|
return {"ok": True}
|
|
|
|
|
|
# ── 대시보드 API ─────────────────────────────────────────────────────────────
|
|
|
|
@app.get("/api/blog-marketing/dashboard")
|
|
def dashboard():
|
|
return get_dashboard_stats()
|