Merge branch 'main' of https://github.com/dm1sh/porridger_tmp

back/main.py

@@ -196,12 +196,15 @@ def add_points(user_id: int):
 # Отправляем стихи
 @app.get("/api/user/poem") # пока не работает
 def poems_to_front(): # db: Annotated[Session, Depends(utils.get_db)] 
-    kolvo_stixov = 109  # пока количество стихотворений = 101
+    kolvo_stixov = database.query(models.Poems).count()  # пока количество стихотворений = 101
-    rand_id = random.randint(1, kolvo_stixov)  # номер стихотворения
+    if kolvo_stixov > 1:
-    poem_json = dict()
+        rand_id = random.randint(1, kolvo_stixov)  # номер стихотворения
-    poem = database.query(models.Poems).filter(models.Poems.id == rand_id).first()
+        poem_json = dict()
-    poem_json = {"title": poem.title, "text": poem.text, "author":poem.author}
+        poem = database.query(models.Poems).filter(models.Poems.id == rand_id).first()
-    return poem_json
+        poem_json = {"title": poem.title, "text": poem.text, "author": poem.author}
+        return poem_json
+    else:
+        raise HTTPException(status_code=404, detail="Poems not found")
 
 @app.get("/api/trashbox", response_model=List[schemas.TrashboxResponse])
 def get_trashboxes(lat:float, lng:float):#крутая функция для работы с api

back/service.py

@@ -31,7 +31,7 @@ def add_poems_to_db(db: Session):
                       author += str1
               if(str1 != f"стих {a+1}\n"):
                   stixi+=str1#удаление /n и заключение в список
-        poem = Poems(title=name, text=stixi, author=author)
+        poem = models.Poems(title=name, text=stixi, author=author)
         # В конце каждой итерации добавляем в базу данных
         db.add(poem)
         db.commit()