diff --git a/app/ytdl.py b/app/ytdl.py index 16a4fff..8b05857 100644 --- a/app/ytdl.py +++ b/app/ytdl.py @@ -39,6 +39,7 @@ class DownloadInfo: self.custom_name_prefix = custom_name_prefix self.msg = self.percent = self.speed = self.eta = None self.status = "pending" + self.size = None self.timestamp = time.time_ns() self.error = error @@ -138,7 +139,9 @@ class Download: return self.tmpfilename = status.get('tmpfilename') if 'filename' in status: - self.info.filename = os.path.relpath(status.get('filename'), self.download_dir) + fileName = status['filename'] + self.info.filename = os.path.relpath(fileName, self.download_dir) + self.info.size = os.path.getsize(fileName) if os.path.exists(fileName) else None # Set correct file extension for thumbnails if(self.info.format == 'thumbnail'):