changed get epub method

TartuNLP · Jan 11, 2023 · 4a672a5 · 4a672a5
1 parent f65451c
commit 4a672a5
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/app/api/routers.py b/app/api/routers.py
@@ -153,10 +153,9 @@ async def get_audiobook(job_id: str, session: AsyncSession = Depends(database.ge
 async def get_epub(job_id: str, _: str = Depends(get_username),
                     session: AsyncSession = Depends(database.get_session)):
     job_info = await database.read_job(session, job_id)
-    if job_info.state in [State.QUEUED, State.IN_PROGRESS]:
-        if job_info.state == State.QUEUED:
-            await database.update_job(session, job_id, State.IN_PROGRESS)
-        return FileResponse(os.path.join(api_settings.storage_path, f"{job_id}.epub"), filename=f"{job_id}.epub")
+    if job_info.state == State.QUEUED:
+        await database.update_job(session, job_id, State.IN_PROGRESS)
+    return FileResponse(os.path.join(api_settings.storage_path, f"{job_id}.epub"), filename=f"{job_id}.epub")
 
 
 @router.post('/{job_id}/failed', response_model=JobInfo, response_model_exclude_none=True,