Check of poems in database added in function /api/user/poem

This commit is contained in:
DmitryGantimurov 2023-08-18 18:49:40 +03:00
parent f345ed6587
commit fb3763b910

View File

@ -34,8 +34,7 @@ if not os.path.exists("./uploads"):
app.mount("/uploads", StaticFiles(directory = "./uploads")) app.mount("/uploads", StaticFiles(directory = "./uploads"))
# Записываем стихи в базу данных, если их еще нет (запускать только если стихов в базе нет). # Записываем стихи в базу данных, если их еще нет (запускать только если стихов в базе нет).
# from .db import database from .db import database
# service.add_poems_to_db(database)
@app.get("/api/announcements", response_model=List[schemas.Announcement])#адрес объявлений @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)] 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() # определяем кол-во стихов в бд num_of_poems = db.query(models.Poems).count() # определяем кол-во стихов в бд
if num_of_poems < 1: 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) # генерируем номер стихотворения rand_id = random.randint(1, num_of_poems) # генерируем номер стихотворения
poem = db.query(models.Poems).filter(models.Poems.id == rand_id).first() # находим стих в бд poem = db.query(models.Poems).filter(models.Poems.id == rand_id).first() # находим стих в бд
if not poem: if not poem: