tropicoo/yt-dlp-bot
1
1
Fork 0
mirror of https://github.com/tropicoo/yt-dlp-bot.git synced 2024-09-20 06:46:08 +08:00
Code Issues Packages Projects Releases Wiki Activity

Maintenance

Browse source
This commit is contained in:
Taras Terletskyi 2023-04-04 22:14:41 +03:00
parent 91e53ade24
commit 9ddc433174
2 changed files with 6 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

6
app_worker/worker/core/media_service.py
View file

@ -183,8 +183,7 @@ class MediaService:
dst = os.path.join(settings.STORAGE_PATH, file.filename)
self._log.info('Copying "%s" to storage "%s"', file.filepath, dst)
await asyncio.to_thread(shutil.copy2, file.filepath, dst)
file.saved_to_storage = True
file.storage_path = dst
file.mark_as_saved_to_storage(storage_path=dst)
def _err_file_cleanup(self, video: DownMedia) -> None:
"""Cleanup any downloaded/created data if post-processing failed."""
@ -194,5 +193,4 @@ class MediaService:
async def _handle_download_exception(
self, err: Exception, task: Task, db: AsyncSession
) -> None:
exception_msg = str(err)
await self._repository.save_as_failed(db, task, exception_msg)
await self._repository.save_as_failed(db=db, task=task, error_message=str(err))

4
yt_shared/yt_shared/schemas/media.py
View file

@ -48,6 +48,10 @@ class BaseMedia(RealBaseModel):
def file_size_human(self) -> str:
return format_bytes(num=self.file_size)
def mark_as_saved_to_storage(self, storage_path: str) -> None:
self.storage_path = storage_path
self.saved_to_storage = True
class Audio(BaseMedia):
"""Model representing downloaded audio file."""