ai_qback/core/api_get_article.py

# -*- coding: utf-8 -*-
from fastapi import FastAPI, Form, HTTPException, Request,status,APIRouter,Query,Path
from tools.loglog import logger
from gpt.get_article import GetArticle
from gpt.query_oss_file import query_file_content
from core.respone_format import *

from pydantic import BaseModel, ValidationError, conint
from typing import List, Optional


router = APIRouter()
get_article = GetArticle()


class ArticleRequest(BaseModel):
    meaning_ids: List[conint(ge=1)] 
    callback_url: Optional[str] = None 
    demo_name: Optional[str] = "无" 
    student_stage: Optional[int] = 1 
    vocabulary: Optional[int] = 500 
    class_id :Optional[int]

@router.post("/article")
def post_article(json_data:ArticleRequest,request:Request):
    real_ip = request.headers.get("X-Real-IP","localhost")
    words_meaning_ids: list = json_data.meaning_ids
    callback_url = json_data.callback_url
    demo_name = json_data.demo_name
    student_stage = json_data.student_stage
    vocabulary = json_data.vocabulary
    class_id = json_data.class_id 

    try:
        if not words_meaning_ids:
            return resp_404(message="没有词义id")

        r = get_article.submit_task(words_meaning_ids=words_meaning_ids,callback_url=callback_url,
                                    real_ip=real_ip,demo_name=demo_name,
                                    student_stage=student_stage,vocabulary=vocabulary,class_id=class_id)
        return r if not isinstance(r,str) else resp_500(message=r)

    except Exception as e:
        logger.error(f"{type(e).__name__},{e}")
        return resp_500(message=f"{type(e).__name__},{e}")


@router.post("/query_oss_file")
def query_oss_file(json_data:dict,request:Request):
    oss_key = json_data.get("key")

    if not oss_key:
        return resp_500(message="请提供key值")
    j = query_file_content(key=oss_key)
    if j == 0:
        return resp_500(message="错误：没有这个文件")
    return JSONResponse(j)