import
os
import
urllib
.
request
import
json
from
http
.
client
import
IncompleteRead
from
pytube
import
YouTube
# Введите URL видео YouTube
url
=
input
(
"Введите URL видео YouTube: "
)
yt
=
YouTube
(
url
)
while
True
:
choice
=
input
(
"Что нужно скачать? (a - аудио, v - видео): "
)
if
choice
==
"a"
:
# почему то крашит код на 256 кбпс,но мне по***....
audio_quality
=
input
(
"Выберите качество аудио (256kbps, 128kbps, 70kbps): "
)
file_extension
=
"mp3"
stream
=
yt
.
streams
.
filter
(
only_audio
=
True
,
abr
=
audio_quality
)
.
first
(
)
break
elif
choice
==
"v"
:
video_quality
=
input
(
"Выберите качество видео (720p, 480p, 360p, 240p, 144p): "
)
file_extension
=
"mp4"
stream
=
yt
.
streams
.
filter
(
res
=
video_quality
,
mime_type
=
"video/mp4"
)
.
first
(
)
break
else
:
print
(
"Некорректный выбор. Попробуйте еще раз."
)
if
not
stream
:
print
(
f"Не удалось найти поток для выбранного качества. Попробуйте выбрать другое качество."
)
else
:
save_dir
=
input
(
"Введите директорию для сохранения файла: "
)
while
True
:
try
:
file_name
=
f"{yt.title}.{file_extension}"
file_path
=
os
.
path
.
join
(
save_dir
,
file_name
)
stream
.
download
(
save_dir
)
break
except
IncompleteRead
as
e
:
data
=
e
.
partial

    info_file
=
os
.
path
.
join
(
save_dir
,
f"{yt.title}.txt"
)
with
open
(
info_file
,
"w"
,
encoding
=
"utf-8"
)
as
f
:
f
.
write
(
f"Название:{yt.title}\n"
)
f
.
write
(
f"Описание:{yt.description}\n"
)
f
.
write
(
f"Автор:{yt.author}\n"
)
if
yt
.
keywords
:
f
.
write
(
f"Теги:{', '.join(yt.keywords)}\n"
)
print
(
f"Файл '{file_name}' успешно сохранен в директории '{save_dir}'."
)