File size: 658 Bytes
2c966e2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
from starlette.middleware.base import BaseHTTPMiddleware
from starlette.requests import Request
from helpers.file_utils import delete_folders, delete_videos
class CleanupMiddleware(BaseHTTPMiddleware):
async def dispatch(self, request: Request, call_next):
response = await call_next(request)
output_dir_for_extracted_frames = "extracted_frames"
output_dir_for_sampled_frames = "sampled_frames"
video_files_to_delete = ["processed*.mp4"]
delete_folders(output_dir_for_extracted_frames, output_dir_for_sampled_frames)
delete_videos(video_files_to_delete)
return response
|