From 1e3a2f3efa6ffdcc26af6a3eb158fe8e165ebf4c Mon Sep 17 00:00:00 2001 From: maksim <kashin20031984@mail.ru> Date: Sun, 2 Jun 2024 17:33:50 +0400 Subject: [PATCH] =?UTF-8?q?=D0=A0=D0=B0=D0=B7=D0=B4=D1=80=D0=BE=D0=B1?= =?UTF-8?q?=D0=B8=D0=BB=20=D1=80=D0=BE=D1=83=D1=82=D0=B5=D1=80=20=D0=B0?= =?UTF-8?q?=D0=B4=D0=B5=D0=BA=D0=B2=D0=B0=D1=82=D0=BD=D0=BE?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- main.py | 4 +++- router_class.py | 25 +++++++++++++++++++++++++ router.py => router_questions.py | 13 +------------ 3 files changed, 29 insertions(+), 13 deletions(-) create mode 100644 router_class.py rename router.py => router_questions.py (66%) diff --git a/main.py b/main.py index 6d12b31..1f134a5 100644 --- a/main.py +++ b/main.py @@ -3,7 +3,8 @@ import logging from contextlib import asynccontextmanager from database import create_tables, delete_tables -from router import router as questions_router +from router_questions import router as questions_router +from router_class import router as class_router # Настройка логирования logging.basicConfig(level=logging.INFO) @@ -20,3 +21,4 @@ async def lifespan(app: FastAPI): app = FastAPI(lifespan=lifespan) app.include_router(questions_router) +app.include_router(class_router) diff --git a/router_class.py b/router_class.py new file mode 100644 index 0000000..9e9c1c5 --- /dev/null +++ b/router_class.py @@ -0,0 +1,25 @@ +from typing import Annotated +from fastapi import APIRouter, Depends +from typing import List + +from enums import TypeMood, TypeModel +from repository import QuestionRepository +from schemas import SQuestionAdd, SQuestion, SQuestionId + +router = APIRouter( + prefix="/class", + tags=["Class"], +) + +@router.get("/negative") +async def get_class_names() -> List[str]: + with open(".//neural_network/classification/class_names_negative.txt", "r", encoding="utf-8") as file: + class_names = [line.strip() for line in file.readlines()] + return class_names + +@router.get("/positive") +async def get_class_names() -> List[str]: + with open(".//neural_network/classification/class_names_positive.txt", "r", encoding="utf-8") as file: + class_names = [line.strip() for line in file.readlines()] + return class_names + diff --git a/router.py b/router_questions.py similarity index 66% rename from router.py rename to router_questions.py index 632878d..705f9f3 100644 --- a/router.py +++ b/router_questions.py @@ -11,18 +11,6 @@ router = APIRouter( tags=["Questions"], ) -@router.get("/class_negative") -async def get_class_names() -> List[str]: - with open(".//neural_network/classification/class_names_negative.txt", "r", encoding="utf-8") as file: - class_names = [line.strip() for line in file.readlines()] - return class_names - -@router.get("/class_positive") -async def get_class_names() -> List[str]: - with open(".//neural_network/classification/class_names_positive.txt", "r", encoding="utf-8") as file: - class_names = [line.strip() for line in file.readlines()] - return class_names - @router.post("") async def add_question( question: Annotated[SQuestionAdd, Depends()], @@ -42,3 +30,4 @@ async def get_questions() -> list[SQuestion]: async def get_questions_by_email(email_user: str) -> list[SQuestion]: questions = await QuestionRepository.find_by_email(email_user) return questions +