selfhostyourtech/reflector
2
0
Fork 0
mirror of https://github.com/Monadical-SAS/reflector.git synced 2025-12-20 12:19:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

server: ensure upload is removed

Browse Source
This commit is contained in:
Mathieu Virbel
2023-12-13 12:30:59 +01:00
parent 5dac1dd121
commit b664b5fff6
1 changed files with 16 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
server/reflector/pipelines/main_live_pipeline.py
View File

@@ -474,6 +474,15 @@ class PipelineMainWaveform(PipelineMainFromTopics):
] ]
@get_transcript
async def pipeline_remove_upload(transcript: Transcript, logger: Logger):
logger.info("Starting remove upload")
uploads = transcript.data_path.glob("upload.*")
for upload in uploads:
upload.unlink(missing_ok=True)
logger.info("Remove upload done")
@get_transcript @get_transcript
async def pipeline_waveform(transcript: Transcript, logger: Logger): async def pipeline_waveform(transcript: Transcript, logger: Logger):
logger.info("Starting waveform") logger.info("Starting waveform")
@@ -560,6 +569,12 @@ async def pipeline_summaries(transcript: Transcript, logger: Logger):
# =================================================================== # ===================================================================
@shared_task
@asynctask
async def task_pipeline_remove_upload(*, transcript_id: str):
await pipeline_remove_upload(transcript_id=transcript_id)
@shared_task @shared_task
@asynctask @asynctask
async def task_pipeline_waveform(*, transcript_id: str): async def task_pipeline_waveform(*, transcript_id: str):
@@ -604,6 +619,7 @@ def pipeline_post(*, transcript_id: str):
task_pipeline_waveform.si(transcript_id=transcript_id) task_pipeline_waveform.si(transcript_id=transcript_id)
| task_pipeline_convert_to_mp3.si(transcript_id=transcript_id) | task_pipeline_convert_to_mp3.si(transcript_id=transcript_id)
| task_pipeline_upload_mp3.si(transcript_id=transcript_id) | task_pipeline_upload_mp3.si(transcript_id=transcript_id)
| task_pipeline_remove_upload.si(transcript_id=transcript_id)
| task_pipeline_diarization.si(transcript_id=transcript_id) | task_pipeline_diarization.si(transcript_id=transcript_id)
) )
chain_title_preview = task_pipeline_title_and_short_summary.si( chain_title_preview = task_pipeline_title_and_short_summary.si(