tubearchivist/tubearchivist-frontend
1
0
Fork 0
mirror of https://github.com/tubearchivist/tubearchivist-frontend.git synced 2024-11-22 20:00:15 +00:00
Code Issues Projects Releases Wiki Activity

add thumbnail download for manual import

Browse Source
This commit is contained in:
simon 2021-10-16 17:03:39 +07:00
parent 0c3c1b650a
commit 08338f3270
2 changed files with 9 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tubearchivist/home/src/reindex.py
View File

@ -413,6 +413,8 @@ class ManualImport:
def process_import(self): def process_import(self):
"""go through identified media files""" """go through identified media files"""
all_videos_added = []
for media_file in self.identified: for media_file in self.identified:
json_file = media_file["json_file"] json_file = media_file["json_file"]
video_file = media_file["video_file"] video_file = media_file["video_file"]
@ -425,6 +427,9 @@ class ManualImport:
# identify and archive # identify and archive
vid_dict = index_new_video(youtube_id) vid_dict = index_new_video(youtube_id)
VideoDownloader([youtube_id]).move_to_archive(vid_dict) VideoDownloader([youtube_id]).move_to_archive(vid_dict)
youtube_id = vid_dict["youtube_id"]
thumb_url = vid_dict["vid_thumb_url"]
all_videos_added.append((youtube_id, thumb_url))
# cleanup # cleanup
if os.path.exists(video_path): if os.path.exists(video_path):
@ -433,6 +438,8 @@ class ManualImport:
json_path = os.path.join(self.CACHE_DIR, "import", json_file) json_path = os.path.join(self.CACHE_DIR, "import", json_file)
os.remove(json_path) os.remove(json_path)
return all_videos_added
def move_to_cache(self, video_path, youtube_id): def move_to_cache(self, video_path, youtube_id):
"""move identified video file to cache, convert to mp4""" """move identified video file to cache, convert to mp4"""
file_name = os.path.split(video_path)[-1] file_name = os.path.split(video_path)[-1]

3
tubearchivist/home/tasks.py
View File

@ -114,7 +114,8 @@ def run_manual_import():
if have_lock: if have_lock:
import_handler = ManualImport() import_handler = ManualImport()
if import_handler.identified: if import_handler.identified:
import_handler.process_import() all_videos_added = import_handler.process_import()
ThumbManager().download_vid(all_videos_added)
else: else:
print("Did not acquire lock form import.") print("Did not acquire lock form import.")