diff --git a/tubearchivist/home/src/download/yt_dlp_handler.py b/tubearchivist/home/src/download/yt_dlp_handler.py index 700eee51..0ea813ae 100644 --- a/tubearchivist/home/src/download/yt_dlp_handler.py +++ b/tubearchivist/home/src/download/yt_dlp_handler.py @@ -20,7 +20,7 @@ from home.src.index.playlist import YoutubePlaylist from home.src.index.video import YoutubeVideo, index_new_video from home.src.index.video_constants import VideoTypeEnum from home.src.ta.config import AppConfig -from home.src.ta.helper import clean_string, ignore_filelist +from home.src.ta.helper import ignore_filelist class DownloadPostProcess: @@ -361,23 +361,16 @@ class VideoDownloader: videos = self.config["application"]["videos"] host_uid = self.config["application"]["HOST_UID"] host_gid = self.config["application"]["HOST_GID"] - channel_name = clean_string(vid_dict["channel"]["channel_name"]) - if len(channel_name) <= 3: - # fall back to channel id - channel_name = vid_dict["channel"]["channel_id"] - # make archive folder with correct permissions - new_folder = os.path.join(videos, channel_name) - if not os.path.exists(new_folder): - os.makedirs(new_folder) - if host_uid and host_gid: - os.chown(new_folder, host_uid, host_gid) - # find real filename + # make folder + folder = os.path.join(videos, vid_dict["channel"]["channel_id"]) + if not os.path.exists(folder): + os.makedirs(folder) + if host_uid and host_gid: + os.chown(folder, host_uid, host_gid) + # move media file + media_file = vid_dict["youtube_id"] + ".mp4" cache_dir = self.config["application"]["cache_dir"] - all_cached = ignore_filelist(os.listdir(cache_dir + "/download/")) - for file_str in all_cached: - if vid_dict["youtube_id"] in file_str: - old_file = file_str - old_path = os.path.join(cache_dir, "download", old_file) + old_path = os.path.join(cache_dir, "download", media_file) new_path = os.path.join(videos, vid_dict["media_url"]) # move media file and fix permission shutil.move(old_path, new_path, copy_function=shutil.copyfile) diff --git a/tubearchivist/home/src/index/video.py b/tubearchivist/home/src/index/video.py index b4daae2d..7b2da2d6 100644 --- a/tubearchivist/home/src/index/video.py +++ b/tubearchivist/home/src/index/video.py @@ -20,7 +20,7 @@ from home.src.index.video_streams import ( DurationConverter, MediaStreamExtractor, ) -from home.src.ta.helper import clean_string, randomizor +from home.src.ta.helper import randomizor from home.src.ta.ta_redis import RedisArchivist from ryd_client import ryd_client @@ -292,19 +292,10 @@ class YoutubeVideo(YouTubeItem, YoutubeSubtitle): def add_file_path(self): """build media_url for where file will be located""" - channel_name = self.json_data["channel"]["channel_name"] - clean_channel_name = clean_string(channel_name) - if len(clean_channel_name) <= 3: - # fall back to channel id - clean_channel_name = self.json_data["channel"]["channel_id"] - - timestamp = self.json_data["published"].replace("-", "") - youtube_id = self.json_data["youtube_id"] - title = self.json_data["title"] - clean_title = clean_string(title) - filename = f"{timestamp}_{youtube_id}_{clean_title}.mp4" - media_url = os.path.join(clean_channel_name, filename) - self.json_data["media_url"] = media_url + self.json_data["media_url"] = os.path.join( + self.json_data["channel"]["channel_id"], + self.json_data["youtube_id"] + ".mp4", + ) def delete_media_file(self): """delete video file, meta data"""