Feat: [main] baekjoon-bot-v1
This commit is contained in:
sm4640
187
app.py
Normal file
187
app.py
Normal file
@@ -0,0 +1,187 @@
|
||||
from typing import Optional
|
||||
|
||||
from fastapi import FastAPI, HTTPException, Query, Depends
|
||||
from fastapi.responses import JSONResponse
|
||||
from dotenv import load_dotenv
|
||||
from sqlalchemy import text
|
||||
from sqlalchemy.ext.asyncio import AsyncSession
|
||||
|
||||
from utils import env, resolve_difficulty, resolve_tags, build_query, get_problem
|
||||
from db import get_db
|
||||
from workbook_picker import pick_from_workbook
|
||||
from workbook_importer import import_workbook
|
||||
from workbook_enricher import enrich_workbook
|
||||
|
||||
|
||||
load_dotenv()
|
||||
|
||||
app = FastAPI()
|
||||
|
||||
|
||||
@app.get("/")
|
||||
def root():
|
||||
return {"status": "ok"}
|
||||
|
||||
|
||||
@app.post("/admin/workbooks/{workbook_id}/enrich")
|
||||
async def admin_enrich_workbook(
|
||||
workbook_id: int,
|
||||
only_missing: bool = Query(True, description="True면 NULL만 채움 / False면 덮어씀"),
|
||||
commit_every: int = Query(50, ge=1, le=500, description="몇 개마다 commit 할지"),
|
||||
sleep_sec: float = Query(0.12, ge=0.0, le=2.0, description="solved.ac 호출 사이 sleep"),
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
result = await enrich_workbook(
|
||||
db,
|
||||
workbook_id=workbook_id,
|
||||
only_missing=only_missing,
|
||||
commit_every=commit_every,
|
||||
sleep_sec=sleep_sec,
|
||||
)
|
||||
return {"status": "ok", "result": result}
|
||||
|
||||
|
||||
@app.delete("/admin/workbooks/{workbook_id}/reset")
|
||||
async def reset_workbook_progress(
|
||||
workbook_id: int,
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
try:
|
||||
res = await db.execute(
|
||||
text("DELETE FROM workbook_sends WHERE workbook_id = :wid"),
|
||||
{"wid": workbook_id},
|
||||
)
|
||||
await db.commit()
|
||||
|
||||
# res.rowcount: 삭제된 행 수(=초기화된 문제 수)
|
||||
return {
|
||||
"status": "ok",
|
||||
"workbook_id": workbook_id,
|
||||
"deleted_sends": int(res.rowcount or 0),
|
||||
"message": "workbook progress reset (problems can be picked again)",
|
||||
}
|
||||
except Exception as e:
|
||||
raise HTTPException(status_code=500, detail=str(e))
|
||||
|
||||
|
||||
@app.get("/today")
|
||||
async def today(
|
||||
source_mode: str = Query(env("SOURCE_MODE_DEFAULT", "search"), description="search|workbook"),
|
||||
workbook_id: Optional[int] = Query(None, description="문제집 모드일 때 workbook id"),
|
||||
workbook_pick: str = Query("level_asc", description="random|level_asc"),
|
||||
|
||||
difficulty_mode: str = Query(env("DIFFICULTY_MODE_DEFAULT", "easy"), description="easy|hard|all"),
|
||||
tag_mode: str = Query(env("TAG_MODE_DEFAULT", "easy"), description="easy|hard|all"),
|
||||
difficulty: Optional[str] = Query(None, description="예: 6..10 (주면 mode보다 우선)"),
|
||||
tags: Optional[str] = Query(None, description="예: dp,graphs (주면 mode보다 우선)"),
|
||||
lang: str = Query(env("LANG_DEFAULT", "all"), description="ko | en | ko,en | all"),
|
||||
|
||||
db: AsyncSession = Depends(get_db),
|
||||
):
|
||||
sm = (source_mode or "").lower().strip()
|
||||
if sm == "workbook":
|
||||
wid = workbook_id or (int(env("WORKBOOK_ID_DEFAULT")) if env("WORKBOOK_ID_DEFAULT") else None)
|
||||
if not wid:
|
||||
return JSONResponse(status_code=400, content={"error": "workbook_id is required for workbook mode"})
|
||||
|
||||
pid, title, level = await pick_from_workbook(db, wid, pick=workbook_pick)
|
||||
if not pid:
|
||||
return JSONResponse(status_code=409, content={"error": "no_more_problems_in_workbook", "workbook_id": wid})
|
||||
|
||||
problem_url = f"https://www.acmicpc.net/problem/{pid}"
|
||||
solved_url = f"https://solved.ac/problems/id/{pid}"
|
||||
level_text = f"Lv. {level}" if level is not None else "Lv. ?"
|
||||
|
||||
discord_payload = {
|
||||
"embeds": [{
|
||||
"title": "🔔 오늘의 백준 추천 문제 (문제집)",
|
||||
"description": (
|
||||
f"**{pid}번: {title}**\n"
|
||||
f"난이도: **{level_text}**\n"
|
||||
f"source_mode: `workbook` / workbook_id: `{wid}`"
|
||||
),
|
||||
"fields": [
|
||||
{"name": "문제 링크", "value": f"[바로가기]({problem_url})", "inline": True},
|
||||
{"name": "해설/정보", "value": f"[Solved.ac]({solved_url})", "inline": True},
|
||||
],
|
||||
"footer": {"text": "매일 오전 10시 정기 알림 (n8n)"}
|
||||
}]
|
||||
}
|
||||
|
||||
return {
|
||||
"source_mode": "workbook",
|
||||
"workbook_id": wid,
|
||||
"problemId": pid,
|
||||
"title": title,
|
||||
"level": level,
|
||||
"problemUrl": problem_url,
|
||||
"solvedUrl": solved_url,
|
||||
"discordPayload": discord_payload,
|
||||
}
|
||||
|
||||
# 2) 기존 search 모드(네가 쓰던 그대로)
|
||||
dm = (difficulty_mode or "").lower()
|
||||
tm = (tag_mode or "").lower()
|
||||
|
||||
if dm not in ("easy", "hard", "all"):
|
||||
return JSONResponse(status_code=400, content={"error": "difficulty_mode must be easy|hard|all"})
|
||||
if tm not in ("easy", "hard", "all"):
|
||||
return JSONResponse(status_code=400, content={"error": "tag_mode must be easy|hard|all"})
|
||||
|
||||
chosen_difficulty = resolve_difficulty(difficulty, dm)
|
||||
chosen_tags = resolve_tags(tags if tags is not None else None, tm)
|
||||
query = build_query(chosen_difficulty, chosen_tags, lang)
|
||||
|
||||
pid, title, level = get_problem(query=query)
|
||||
if not pid:
|
||||
return JSONResponse(status_code=503, content={"error": "failed_to_fetch_problem", "query": query})
|
||||
|
||||
problem_url = f"https://www.acmicpc.net/problem/{pid}"
|
||||
solved_url = f"https://solved.ac/problems/id/{pid}"
|
||||
level_text = f"Lv. {level}" if level is not None else "Lv. ?"
|
||||
|
||||
discord_payload = {
|
||||
"embeds": [{
|
||||
"title": "🔔 오늘의 백준 추천 문제",
|
||||
"description": (
|
||||
f"**{pid}번: {title}**\n"
|
||||
f"난이도: **{level_text}**\n"
|
||||
f"difficulty_mode: `{dm}` / tag_mode: `{tm}` / lang: `{lang}`\n"
|
||||
f"filter: `{chosen_difficulty}` / tags: `{', '.join(chosen_tags) if chosen_tags else 'none'}`"
|
||||
),
|
||||
"fields": [
|
||||
{"name": "문제 링크", "value": f"[바로가기]({problem_url})", "inline": True},
|
||||
{"name": "해설/정보", "value": f"[Solved.ac]({solved_url})", "inline": True},
|
||||
],
|
||||
"footer": {"text": "매일 오전 10시 정기 알림 (n8n)"}
|
||||
}]
|
||||
}
|
||||
|
||||
return {
|
||||
"source_mode": "search",
|
||||
"difficulty_mode": dm,
|
||||
"tag_mode": tm,
|
||||
"lang": lang,
|
||||
"difficulty": chosen_difficulty,
|
||||
"tags": chosen_tags,
|
||||
"query": query,
|
||||
"problemId": pid,
|
||||
"title": title,
|
||||
"level": level,
|
||||
"problemUrl": problem_url,
|
||||
"solvedUrl": solved_url,
|
||||
"discordPayload": discord_payload,
|
||||
}
|
||||
|
||||
|
||||
# @app.post("/admin/workbooks/{workbook_id}/import")
|
||||
# async def admin_import_workbook(
|
||||
# workbook_id: int,
|
||||
# title: Optional[str] = Query(None, description="문제집 제목(옵션)"),
|
||||
# db: AsyncSession = Depends(get_db),
|
||||
# ):
|
||||
# try:
|
||||
# result = await import_workbook(db, workbook_id=workbook_id, title=title)
|
||||
# return {"status": "ok", "result": result}
|
||||
# except Exception as e:
|
||||
# return JSONResponse(status_code=500, content={"error": str(e)})
|
||||
Reference in New Issue
Block a user