xtaodada/ytdlbot
1
0
Fork 0
mirror of https://github.com/omg-xtao/ytdlbot.git synced 2024-11-16 03:45:23 +00:00
Code Issues Packages Projects Releases Wiki Activity

better direct audio

Browse Source
This commit is contained in:
BennyThink 2022-03-13 13:12:18 +08:00
parent 87d6b0ee51
commit 97dbacad04
No known key found for this signature in database
GPG Key ID: 6CD0DBDA5235D481
3 changed files with 26 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
ytdlbot/config.py
View File

@ -51,7 +51,7 @@ MYSQL_HOST = os.getenv("MYSQL_HOST")
MYSQL_USER = os.getenv("MYSQL_USER", "root")
MYSQL_PASS = os.getenv("MYSQL_PASS", "root")
AUDIO_FORMAT = os.getenv("AUDIO_FORMAT", "m4a")
AUDIO_FORMAT = os.getenv("AUDIO_FORMAT")
ARCHIVE_ID = os.getenv("ARCHIVE_ID")
IPv6 = os.getenv("IPv6", False)

27
ytdlbot/downloader.py
View File

@ -257,11 +257,32 @@ def ytdl_download(url, tempdir, bm) -> dict:
def convert_audio_format(resp: "dict", bm):
# 1. file is audio, default format
# 2. file is video, default format
# 3. non default format
if resp["status"]:
# all_converted = []
path: pathlib.PosixPath
path: "pathlib.Path"
for path in resp["filepath"]:
if path.suffix != f".{AUDIO_FORMAT}":
streams = ffmpeg.probe(path)["streams"]
if (AUDIO_FORMAT is None and
len(streams) == 1 and
streams[0]["codec_type"] == "audio"):
logging.info("%s is audio, default format, no need to convert", path)
elif AUDIO_FORMAT is None and len(streams) >= 2:
logging.info("%s is video, default format, need to extract audio", path)
audio_stream = {"codec_name": "m4a"}
for stream in streams:
if stream["codec_type"] == "audio":
audio_stream = stream
break
ext = audio_stream["codec_name"]
new_path = path.with_suffix(f".{ext}")
run_ffmpeg(["ffmpeg", "-y", "-i", path, "-vn", "-acodec", "copy", new_path], bm)
path.unlink()
index = resp["filepath"].index(path)
resp["filepath"][index] = new_path
else:
logging.info("Not default format, converting %s to %s", path, AUDIO_FORMAT)
new_path = path.with_suffix(f".{AUDIO_FORMAT}")
run_ffmpeg(["ffmpeg", "-y", "-i", path, new_path], bm)
path.unlink()

2
ytdlbot/tasks.py
View File

@ -362,7 +362,7 @@ def gen_video_markup():
[
[ # First row
InlineKeyboardButton( # Generates a callback query when pressed
f"convert to audio({AUDIO_FORMAT})",
"convert to audio",
callback_data="convert"
)
]