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

fix legacy import alias for yt-dlp

Browse Source
This commit is contained in:
simon 2022-01-02 15:38:45 +07:00
parent 6d16d75ef9
commit fd334bcafe
No known key found for this signature in database
GPG Key ID: 2C15AA5E89985DD4
3 changed files with 22 additions and 26 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tubearchivist/home/src/download.py
View File

@ -13,7 +13,7 @@ from datetime import datetime
from time import sleep
import requests
import yt_dlp as youtube_dl
import yt_dlp
from home.src.config import AppConfig
from home.src.helper import (
DurationConverter,
@ -155,8 +155,8 @@ class PendingList:
"simulate": True,
}
try:
vid = youtube_dl.YoutubeDL(obs).extract_info(youtube_id)
except youtube_dl.utils.DownloadError:
vid = yt_dlp.YoutubeDL(obs).extract_info(youtube_id)
except yt_dlp.utils.DownloadError:
print("failed to extract info for: " + youtube_id)
return False
# stop if video is streaming live now
@ -309,7 +309,7 @@ class ChannelSubscription:
}
if limit:
obs["playlistend"] = self.channel_size
chan = youtube_dl.YoutubeDL(obs).extract_info(url, download=False)
chan = yt_dlp.YoutubeDL(obs).extract_info(url, download=False)
last_videos = [(i["id"], i["title"]) for i in chan["entries"]]
return last_videos
@ -551,7 +551,7 @@ class VideoDownloader:
try:
self.dl_single_vid(youtube_id)
except youtube_dl.utils.DownloadError:
except yt_dlp.utils.DownloadError:
print("failed to download " + youtube_id)
continue
vid_dict = index_new_video(youtube_id)
@ -593,7 +593,7 @@ class VideoDownloader:
@staticmethod
def progress_hook(response):
"""process the progress_hooks from youtube_dl"""
"""process the progress_hooks from yt_dlp"""
# title
path = os.path.split(response["filename"])[-1][12:]
filename = os.path.splitext(os.path.splitext(path)[0])[0]
@ -683,10 +683,10 @@ class VideoDownloader:
for file_name in all_cached:
if youtube_id in file_name:
obs["outtmpl"] = os.path.join(dl_cache, file_name)
with youtube_dl.YoutubeDL(obs) as ydl:
with yt_dlp.YoutubeDL(obs) as ydl:
try:
ydl.download([youtube_id])
except youtube_dl.utils.DownloadError:
except yt_dlp.utils.DownloadError:
print("retry failed download: " + youtube_id)
sleep(10)
ydl.download([youtube_id])

4
tubearchivist/home/src/helper.py
View File

@ -13,7 +13,7 @@ from urllib.parse import parse_qs, urlparse
import redis
import requests
import yt_dlp as youtube_dl
import yt_dlp
def get_total_hits(index, es_url, es_auth, match_field):
@ -139,7 +139,7 @@ class UrlListParser:
"extract_flat": True,
"playlistend": 0,
}
url_info = youtube_dl.YoutubeDL(obs).extract_info(url, download=False)
url_info = yt_dlp.YoutubeDL(obs).extract_info(url, download=False)
try:
channel_id = url_info["channel_id"]
except KeyError as error:

28
tubearchivist/home/src/index.py
View File

@ -12,7 +12,7 @@ from datetime import datetime
from time import sleep
import requests
import yt_dlp as youtube_dl
import yt_dlp
from bs4 import BeautifulSoup
from home.src.config import AppConfig
from home.src.helper import DurationConverter, UrlListParser, clean_string
@ -257,7 +257,7 @@ class YoutubeChannel:
"skip_download": True,
"extract_flat": True,
}
playlists = youtube_dl.YoutubeDL(obs).extract_info(url)
playlists = yt_dlp.YoutubeDL(obs).extract_info(url)
all_entries = [(i["id"], i["title"]) for i in playlists["entries"]]
return all_entries
@ -289,7 +289,7 @@ class YoutubeVideo:
self.vid_dict = None
def get_vid_dict(self):
"""wrapper to loop around youtube_dl to retry on failure"""
"""wrapper to loop around yt_dlp to retry on failure"""
print(f"get video data for {self.youtube_id}")
vid_dict = False
for i in range(3):
@ -315,10 +315,10 @@ class YoutubeVideo:
"noplaylist": True,
}
try:
vid = youtube_dl.YoutubeDL(obs).extract_info(youtube_id)
vid = yt_dlp.YoutubeDL(obs).extract_info(youtube_id)
except (
youtube_dl.utils.ExtractorError,
youtube_dl.utils.DownloadError,
yt_dlp.utils.ExtractorError,
yt_dlp.utils.DownloadError,
):
print("failed to get info for " + youtube_id)
return False
@ -489,12 +489,10 @@ class YoutubePlaylist:
"playlistend": 0,
}
try:
playlist = youtube_dl.YoutubeDL(obs).extract_info(
url, download=False
)
playlist = yt_dlp.YoutubeDL(obs).extract_info(url, download=False)
except (
youtube_dl.utils.ExtractorError,
youtube_dl.utils.DownloadError,
yt_dlp.utils.ExtractorError,
yt_dlp.utils.DownloadError,
):
print("failed to get info for " + self.playlist_id)
return False
@ -535,12 +533,10 @@ class YoutubePlaylist:
obs["playlistend"] = playlistend
try:
playlist = youtube_dl.YoutubeDL(obs).extract_info(
url, download=False
)
playlist = yt_dlp.YoutubeDL(obs).extract_info(url, download=False)
except (
youtube_dl.utils.ExtractorError,
youtube_dl.utils.DownloadError,
yt_dlp.utils.ExtractorError,
yt_dlp.utils.DownloadError,
):
print("failed to get plealist entries for " + self.playlist_id)
return False