From 1a51c580d4709d1922f646b675943b2b56664bf7 Mon Sep 17 00:00:00 2001 From: dm1sh Date: Tue, 15 Aug 2023 23:42:04 +0300 Subject: [PATCH] fixed error on empty poems table --- back/api.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/back/api.py b/back/api.py index 6cf0d11..fb7488b 100644 --- a/back/api.py +++ b/back/api.py @@ -191,6 +191,8 @@ def add_points(user_id: int, db: Annotated[Session, Depends(utils.get_db)]): @app.get("/api/user/poem", response_model=schemas.Poem) # пока не работает 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") rand_id = random.randint(1, num_of_poems) # генерируем номер стихотворения poem = db.query(models.Poems).filter(models.Poems.id == rand_id).first() # находим стих в бд if not poem: