From fb3763b9104b18cfd0a787bf59c40a990cfcfd18 Mon Sep 17 00:00:00 2001 From: DmitryGantimurov Date: Fri, 18 Aug 2023 18:49:40 +0300 Subject: [PATCH] Check of poems in database added in function /api/user/poem --- back/api.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/back/api.py b/back/api.py index 9d01112..00415df 100644 --- a/back/api.py +++ b/back/api.py @@ -34,8 +34,7 @@ if not os.path.exists("./uploads"): app.mount("/uploads", StaticFiles(directory = "./uploads")) # Записываем стихи в базу данных, если их еще нет (запускать только если стихов в базе нет). -# from .db import database -# service.add_poems_to_db(database) +from .db import database @app.get("/api/announcements", response_model=List[schemas.Announcement])#адрес объявлений @@ -193,7 +192,7 @@ def add_points(user_id: int, db: Annotated[Session, Depends(utils.get_db)]): def poems_to_front(db: Annotated[Session, Depends(utils.get_db)]): # db: Annotated[Session, Depends(utils.get_db)] num_of_poems = db.query(models.Poems).count() # определяем кол-во стихов в бд if num_of_poems < 1: - raise HTTPException(status_code=404, detail="Poems not found") + service.add_poems_to_db(database) # добавляем поэмы в базу данных rand_id = random.randint(1, num_of_poems) # генерируем номер стихотворения poem = db.query(models.Poems).filter(models.Poems.id == rand_id).first() # находим стих в бд if not poem: