hoelee/tubesync
1
0
Fork 0
mirror of https://github.com/meeb/tubesync.git synced 2025-06-24 05:56:37 +00:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Store Media.downloaded_filesize for imported files

Browse Source 
Fixes #69
This commit is contained in:
tcely 2025-02-25 04:13:02 -05:00 committed by GitHub
parent ff91cbf6ba
commit 533c1631b7
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tubesync/sync/management/commands/import-existing-media.py
View File

@ -16,7 +16,7 @@ class Command(BaseCommand):
log.info('Building directory to Source map...') log.info('Building directory to Source map...')
dirmap = {} dirmap = {}
for s in Source.objects.all(): for s in Source.objects.all():
dirmap[s.directory_path] = s dirmap[str(s.directory_path)] = s
log.info(f'Scanning sources...') log.info(f'Scanning sources...')
file_extensions = list(FileExtension.values) + self.extra_extensions file_extensions = list(FileExtension.values) + self.extra_extensions
for sourceroot, source in dirmap.items(): for sourceroot, source in dirmap.items():
@ -38,7 +38,8 @@ class Command(BaseCommand):
ext = ext.strip().lower() ext = ext.strip().lower()
if ext not in file_extensions: if ext not in file_extensions:
continue continue
on_disk.append(str(rootpath / filename)) filepath = Path(rootpath / filename).resolve(strict=True)
on_disk.append(str(filepath))
filemap = {} filemap = {}
for item in media: for item in media:
for filepath in on_disk: for filepath in on_disk:
@ -50,7 +51,8 @@ class Command(BaseCommand):
for filepath, item in filemap.items(): for filepath, item in filemap.items():
log.info(f'Matched on-disk file: {filepath} ' log.info(f'Matched on-disk file: {filepath} '
f'to media item: {item.source} / {item}') f'to media item: {item.source} / {item}')
item.media_file.name = filepath item.media_file.name = str(Path(filepath).relative_to(item.media_file.storage.location))
item.downloaded = True item.downloaded = True
item.downloaded_filesize = Path(filepath).stat().st_size
item.save() item.save()
log.info('Done') log.info('Done')