mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 19:44:20 +00:00
Start something
This commit is contained in:
parent
463c33650b
commit
a98b6c79fc
2 changed files with 132 additions and 128 deletions
|
|
@ -1,24 +1,19 @@
|
||||||
import typing
|
from .ytdlinfo import YtdlInfo
|
||||||
import logging as _logging
|
|
||||||
import discord
|
|
||||||
import os
|
|
||||||
import dateparser
|
|
||||||
import datetime
|
|
||||||
from youtube_dl import YoutubeDL
|
|
||||||
from ..utils import ytdldateformat
|
|
||||||
|
|
||||||
log = _logging.getLogger(__name__)
|
|
||||||
|
|
||||||
|
|
||||||
class DownloaderError(Exception):
|
|
||||||
pass
|
|
||||||
|
|
||||||
|
|
||||||
class InterruptDownload(DownloaderError):
|
|
||||||
"""Raised from a progress_hook to interrupt the video download."""
|
|
||||||
|
|
||||||
|
|
||||||
class YtdlFile:
|
class YtdlFile:
|
||||||
|
"""Information about a youtube-dl downloaded file."""
|
||||||
|
|
||||||
|
def __init__(self,
|
||||||
|
url: str,
|
||||||
|
info: YtdlInfo = None,
|
||||||
|
filename: str):
|
||||||
|
...
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
class OldYtdlFile:
|
||||||
"""A wrapper around a youtube_dl downloaded file."""
|
"""A wrapper around a youtube_dl downloaded file."""
|
||||||
|
|
||||||
ytdl_args = {
|
ytdl_args = {
|
||||||
|
|
@ -74,113 +69,3 @@ class YtdlFile:
|
||||||
No checks are done when deleting, so it may try to delete a non-existing file and raise an exception or do some other weird stuff with weird filenames."""
|
No checks are done when deleting, so it may try to delete a non-existing file and raise an exception or do some other weird stuff with weird filenames."""
|
||||||
os.remove(self.video_filename)
|
os.remove(self.video_filename)
|
||||||
|
|
||||||
|
|
||||||
class YtdlInfo:
|
|
||||||
"""A wrapper around youtube_dl extracted info."""
|
|
||||||
|
|
||||||
def __init__(self, info: typing.Dict[str, typing.Any]):
|
|
||||||
"""Create a YtdlInfo from the dict returned by the :py:func:`youtube_dl.YoutubeDL.extract_info` function.
|
|
||||||
|
|
||||||
Warning:
|
|
||||||
Does not download the info, for that use :py:func:`royalnet.audio.YtdlInfo.create_from_url`."""
|
|
||||||
self.id: typing.Optional[str] = info.get("id")
|
|
||||||
self.uploader: typing.Optional[str] = info.get("uploader")
|
|
||||||
self.uploader_id: typing.Optional[str] = info.get("uploader_id")
|
|
||||||
self.uploader_url: typing.Optional[str] = info.get("uploader_url")
|
|
||||||
self.channel_id: typing.Optional[str] = info.get("channel_id")
|
|
||||||
self.channel_url: typing.Optional[str] = info.get("channel_url")
|
|
||||||
self.upload_date: typing.Optional[datetime.datetime] = dateparser.parse(ytdldateformat(info.get("upload_date")))
|
|
||||||
self.license: typing.Optional[str] = info.get("license")
|
|
||||||
self.creator: typing.Optional[...] = info.get("creator")
|
|
||||||
self.title: typing.Optional[str] = info.get("title")
|
|
||||||
self.alt_title: typing.Optional[...] = info.get("alt_title")
|
|
||||||
self.thumbnail: typing.Optional[str] = info.get("thumbnail")
|
|
||||||
self.description: typing.Optional[str] = info.get("description")
|
|
||||||
self.categories: typing.Optional[typing.List[str]] = info.get("categories")
|
|
||||||
self.tags: typing.Optional[typing.List[str]] = info.get("tags")
|
|
||||||
self.subtitles: typing.Optional[typing.Dict[str, typing.List[typing.Dict[str, str]]]] = info.get("subtitles")
|
|
||||||
self.automatic_captions: typing.Optional[dict] = info.get("automatic_captions")
|
|
||||||
self.duration: typing.Optional[datetime.timedelta] = datetime.timedelta(seconds=info.get("duration", 0))
|
|
||||||
self.age_limit: typing.Optional[int] = info.get("age_limit")
|
|
||||||
self.annotations: typing.Optional[...] = info.get("annotations")
|
|
||||||
self.chapters: typing.Optional[...] = info.get("chapters")
|
|
||||||
self.webpage_url: typing.Optional[str] = info.get("webpage_url")
|
|
||||||
self.view_count: typing.Optional[int] = info.get("view_count")
|
|
||||||
self.like_count: typing.Optional[int] = info.get("like_count")
|
|
||||||
self.dislike_count: typing.Optional[int] = info.get("dislike_count")
|
|
||||||
self.average_rating: typing.Optional[...] = info.get("average_rating")
|
|
||||||
self.formats: typing.Optional[list] = info.get("formats")
|
|
||||||
self.is_live: typing.Optional[bool] = info.get("is_live")
|
|
||||||
self.start_time: typing.Optional[float] = info.get("start_time")
|
|
||||||
self.end_time: typing.Optional[float] = info.get("end_time")
|
|
||||||
self.series: typing.Optional[str] = info.get("series")
|
|
||||||
self.season_number: typing.Optional[int] = info.get("season_number")
|
|
||||||
self.episode_number: typing.Optional[int] = info.get("episode_number")
|
|
||||||
self.track: typing.Optional[...] = info.get("track")
|
|
||||||
self.artist: typing.Optional[...] = info.get("artist")
|
|
||||||
self.extractor: typing.Optional[str] = info.get("extractor")
|
|
||||||
self.webpage_url_basename: typing.Optional[str] = info.get("webpage_url_basename")
|
|
||||||
self.extractor_key: typing.Optional[str] = info.get("extractor_key")
|
|
||||||
self.playlist: typing.Optional[str] = info.get("playlist")
|
|
||||||
self.playlist_index: typing.Optional[int] = info.get("playlist_index")
|
|
||||||
self.thumbnails: typing.Optional[typing.List[typing.Dict[str, str]]] = info.get("thumbnails")
|
|
||||||
self.display_id: typing.Optional[str] = info.get("display_id")
|
|
||||||
self.requested_subtitles: typing.Optional[...] = info.get("requested_subtitles")
|
|
||||||
self.requested_formats: typing.Optional[tuple] = info.get("requested_formats")
|
|
||||||
self.format: typing.Optional[str] = info.get("format")
|
|
||||||
self.format_id: typing.Optional[str] = info.get("format_id")
|
|
||||||
self.width: typing.Optional[int] = info.get("width")
|
|
||||||
self.height: typing.Optional[int] = info.get("height")
|
|
||||||
self.resolution: typing.Optional[...] = info.get("resolution")
|
|
||||||
self.fps: typing.Optional[int] = info.get("fps")
|
|
||||||
self.vcodec: typing.Optional[str] = info.get("vcodec")
|
|
||||||
self.vbr: typing.Optional[int] = info.get("vbr")
|
|
||||||
self.stretched_ratio: typing.Optional[...] = info.get("stretched_ratio")
|
|
||||||
self.acodec: typing.Optional[str] = info.get("acodec")
|
|
||||||
self.abr: typing.Optional[int] = info.get("abr")
|
|
||||||
self.ext: typing.Optional[str] = info.get("ext")
|
|
||||||
|
|
||||||
@staticmethod
|
|
||||||
def create_from_url(url, **ytdl_args) -> typing.List["YtdlInfo"]:
|
|
||||||
# So many redundant options!
|
|
||||||
ytdl = YoutubeDL({
|
|
||||||
"logger": log, # Log messages to a logging.Logger instance.
|
|
||||||
"quiet": True, # Do not print messages to stdout.
|
|
||||||
"noplaylist": True, # Download single video instead of a playlist if in doubt.
|
|
||||||
"no_warnings": True, # Do not print out anything for warnings.
|
|
||||||
**ytdl_args
|
|
||||||
})
|
|
||||||
first_info = ytdl.extract_info(url=url, download=False)
|
|
||||||
# If it is a playlist, create multiple videos!
|
|
||||||
if "entries" in first_info:
|
|
||||||
return [YtdlInfo(second_info) for second_info in first_info["entries"]]
|
|
||||||
return [YtdlInfo(first_info)]
|
|
||||||
|
|
||||||
def download(self, outtmpl="%(title)s-%(id)s.%(ext)s", **ytdl_args) -> YtdlFile:
|
|
||||||
return YtdlFile(self, outtmpl, **ytdl_args)
|
|
||||||
|
|
||||||
def to_discord_embed(self) -> discord.Embed:
|
|
||||||
embed = discord.Embed(title=self.title,
|
|
||||||
colour=discord.Colour(0xcc0000),
|
|
||||||
url=self.webpage_url)
|
|
||||||
embed.set_thumbnail(
|
|
||||||
url=self.thumbnail)
|
|
||||||
embed.set_author(name=self.uploader, url=self.uploader_url)
|
|
||||||
embed.set_footer(text="Source: youtube-dl", icon_url="https://i.imgur.com/TSvSRYn.png")
|
|
||||||
embed.add_field(name="Duration", value=str(self.duration), inline=True)
|
|
||||||
embed.add_field(name="Published on", value=self.upload_date.strftime("%d %b %Y"), inline=True)
|
|
||||||
return embed
|
|
||||||
|
|
||||||
def __repr__(self):
|
|
||||||
if self.title:
|
|
||||||
return f"<YtdlInfo of {self.title}>"
|
|
||||||
if self.webpage_url:
|
|
||||||
return f"<YtdlInfo for {self.webpage_url}>"
|
|
||||||
return f"<YtdlInfo id={self.id} ...>"
|
|
||||||
|
|
||||||
def __str__(self):
|
|
||||||
if self.title:
|
|
||||||
return self.title
|
|
||||||
if self.webpage_url:
|
|
||||||
return self.webpage_url
|
|
||||||
return self.id
|
|
||||||
|
|
|
||||||
119
royalnet/audio/ytdlinfo.py
Normal file
119
royalnet/audio/ytdlinfo.py
Normal file
|
|
@ -0,0 +1,119 @@
|
||||||
|
import typing
|
||||||
|
import datetime
|
||||||
|
import dateparser
|
||||||
|
import youtube_dl
|
||||||
|
import discord
|
||||||
|
import royalnet.utils as u
|
||||||
|
|
||||||
|
|
||||||
|
class YtdlInfo:
|
||||||
|
"""A wrapper around youtube_dl extracted info."""
|
||||||
|
|
||||||
|
def __init__(self, info: typing.Dict[str, typing.Any]):
|
||||||
|
"""Create a YtdlInfo from the dict returned by the :py:func:`youtube_dl.YoutubeDL.extract_info` function.
|
||||||
|
|
||||||
|
Warning:
|
||||||
|
Does not download the info, for that use :py:func:`royalnet.audio.YtdlInfo.retrieve_for_url`."""
|
||||||
|
self.id: typing.Optional[str] = info.get("id")
|
||||||
|
self.uploader: typing.Optional[str] = info.get("uploader")
|
||||||
|
self.uploader_id: typing.Optional[str] = info.get("uploader_id")
|
||||||
|
self.uploader_url: typing.Optional[str] = info.get("uploader_url")
|
||||||
|
self.channel_id: typing.Optional[str] = info.get("channel_id")
|
||||||
|
self.channel_url: typing.Optional[str] = info.get("channel_url")
|
||||||
|
self.upload_date: typing.Optional[datetime.datetime] = dateparser.parse(u.ytdldateformat(info.get("upload_date")))
|
||||||
|
self.license: typing.Optional[str] = info.get("license")
|
||||||
|
self.creator: typing.Optional[...] = info.get("creator")
|
||||||
|
self.title: typing.Optional[str] = info.get("title")
|
||||||
|
self.alt_title: typing.Optional[...] = info.get("alt_title")
|
||||||
|
self.thumbnail: typing.Optional[str] = info.get("thumbnail")
|
||||||
|
self.description: typing.Optional[str] = info.get("description")
|
||||||
|
self.categories: typing.Optional[typing.List[str]] = info.get("categories")
|
||||||
|
self.tags: typing.Optional[typing.List[str]] = info.get("tags")
|
||||||
|
self.subtitles: typing.Optional[typing.Dict[str, typing.List[typing.Dict[str, str]]]] = info.get("subtitles")
|
||||||
|
self.automatic_captions: typing.Optional[dict] = info.get("automatic_captions")
|
||||||
|
self.duration: typing.Optional[datetime.timedelta] = datetime.timedelta(seconds=info.get("duration", 0))
|
||||||
|
self.age_limit: typing.Optional[int] = info.get("age_limit")
|
||||||
|
self.annotations: typing.Optional[...] = info.get("annotations")
|
||||||
|
self.chapters: typing.Optional[...] = info.get("chapters")
|
||||||
|
self.webpage_url: typing.Optional[str] = info.get("webpage_url")
|
||||||
|
self.view_count: typing.Optional[int] = info.get("view_count")
|
||||||
|
self.like_count: typing.Optional[int] = info.get("like_count")
|
||||||
|
self.dislike_count: typing.Optional[int] = info.get("dislike_count")
|
||||||
|
self.average_rating: typing.Optional[...] = info.get("average_rating")
|
||||||
|
self.formats: typing.Optional[list] = info.get("formats")
|
||||||
|
self.is_live: typing.Optional[bool] = info.get("is_live")
|
||||||
|
self.start_time: typing.Optional[float] = info.get("start_time")
|
||||||
|
self.end_time: typing.Optional[float] = info.get("end_time")
|
||||||
|
self.series: typing.Optional[str] = info.get("series")
|
||||||
|
self.season_number: typing.Optional[int] = info.get("season_number")
|
||||||
|
self.episode_number: typing.Optional[int] = info.get("episode_number")
|
||||||
|
self.track: typing.Optional[...] = info.get("track")
|
||||||
|
self.artist: typing.Optional[...] = info.get("artist")
|
||||||
|
self.extractor: typing.Optional[str] = info.get("extractor")
|
||||||
|
self.webpage_url_basename: typing.Optional[str] = info.get("webpage_url_basename")
|
||||||
|
self.extractor_key: typing.Optional[str] = info.get("extractor_key")
|
||||||
|
self.playlist: typing.Optional[str] = info.get("playlist")
|
||||||
|
self.playlist_index: typing.Optional[int] = info.get("playlist_index")
|
||||||
|
self.thumbnails: typing.Optional[typing.List[typing.Dict[str, str]]] = info.get("thumbnails")
|
||||||
|
self.display_id: typing.Optional[str] = info.get("display_id")
|
||||||
|
self.requested_subtitles: typing.Optional[...] = info.get("requested_subtitles")
|
||||||
|
self.requested_formats: typing.Optional[tuple] = info.get("requested_formats")
|
||||||
|
self.format: typing.Optional[str] = info.get("format")
|
||||||
|
self.format_id: typing.Optional[str] = info.get("format_id")
|
||||||
|
self.width: typing.Optional[int] = info.get("width")
|
||||||
|
self.height: typing.Optional[int] = info.get("height")
|
||||||
|
self.resolution: typing.Optional[...] = info.get("resolution")
|
||||||
|
self.fps: typing.Optional[int] = info.get("fps")
|
||||||
|
self.vcodec: typing.Optional[str] = info.get("vcodec")
|
||||||
|
self.vbr: typing.Optional[int] = info.get("vbr")
|
||||||
|
self.stretched_ratio: typing.Optional[...] = info.get("stretched_ratio")
|
||||||
|
self.acodec: typing.Optional[str] = info.get("acodec")
|
||||||
|
self.abr: typing.Optional[int] = info.get("abr")
|
||||||
|
self.ext: typing.Optional[str] = info.get("ext")
|
||||||
|
|
||||||
|
@staticmethod
|
||||||
|
def retrieve_for_url(url, **ytdl_args) -> typing.List["YtdlInfo"]:
|
||||||
|
"""Fetch the info for an url through YoutubeDL.
|
||||||
|
|
||||||
|
Returns:
|
||||||
|
A :py:class:`list` containing the infos for the requested videos."""
|
||||||
|
# So many redundant options!
|
||||||
|
ytdl = youtube_dl.YoutubeDL({
|
||||||
|
"quiet": True, # Do not print messages to stdout.
|
||||||
|
"noplaylist": True, # Download single video instead of a playlist if in doubt.
|
||||||
|
"no_warnings": True, # Do not print out anything for warnings.
|
||||||
|
**ytdl_args
|
||||||
|
})
|
||||||
|
first_info = ytdl.extract_info(url=url, download=False)
|
||||||
|
# If it is a playlist, create multiple videos!
|
||||||
|
if "entries" in first_info:
|
||||||
|
return [YtdlInfo(second_info) for second_info in first_info["entries"]]
|
||||||
|
return [YtdlInfo(first_info)]
|
||||||
|
|
||||||
|
def to_discord_embed(self) -> discord.Embed:
|
||||||
|
"""Return this info as a :py:class:`discord.Embed`."""
|
||||||
|
embed = discord.Embed(title=self.title,
|
||||||
|
colour=discord.Colour(0xcc0000),
|
||||||
|
url=self.webpage_url)
|
||||||
|
embed.set_thumbnail(
|
||||||
|
url=self.thumbnail)
|
||||||
|
embed.set_author(name=self.uploader, url=self.uploader_url)
|
||||||
|
embed.set_footer(text="Source: youtube-dl", icon_url="https://i.imgur.com/TSvSRYn.png")
|
||||||
|
embed.add_field(name="Duration", value=str(self.duration), inline=True)
|
||||||
|
embed.add_field(name="Published on", value=self.upload_date.strftime("%d %b %Y"), inline=True)
|
||||||
|
return embed
|
||||||
|
|
||||||
|
def __repr__(self):
|
||||||
|
if self.title:
|
||||||
|
return f"<YtdlInfo of {self.title}>"
|
||||||
|
if self.webpage_url:
|
||||||
|
return f"<YtdlInfo for {self.webpage_url}>"
|
||||||
|
return f"<YtdlInfo id={self.id} ...>"
|
||||||
|
|
||||||
|
def __str__(self):
|
||||||
|
"""Return the video name."""
|
||||||
|
if self.title:
|
||||||
|
return self.title
|
||||||
|
if self.webpage_url:
|
||||||
|
return self.webpage_url
|
||||||
|
return self.id
|
||||||
Loading…
Reference in a new issue