better

diff --git a/dependencies.txt b/dependencies.txt
index cd5e770101fe9ca395444cd52a1ef6a89f85367e..2fd322a8358480d94fe4be585803b53227a001b1 100644 (file)
--- a/dependencies.txt
+++ b/dependencies.txt
@@ -1 +1,2 @@
 pytube
+ffmpeg-python
\ No newline at end of file

diff --git a/links.txt b/links.txt
new file mode 100644 (file)
index 0000000..2ee388a
--- /dev/null
+++ b/links.txt
@@ -0,0 +1 @@
+https://www.youtube.com/watch?v=dQw4w9WgXcQ
\ No newline at end of file

diff --git a/links_file.txt b/links_file.txt
deleted file mode 100644 (file)
index e69de29..0000000

diff --git a/main.py b/main.py
index fa0371b6dec18b263af2f40b87cf92e71647cdf8..3c4c10e6066afbeacaed56dc685d52b2fa2befe3 100644 (file)
--- a/main.py
+++ b/main.py
@@ -1,31 +1,63 @@
 from pytube import YouTube
+import ffmpeg
 
-SAVE_PATH = "D:/Youtube/"
 
-with open('links_file.txt', 'r') as f:
-    links = f.read().split('\n')
-    if links[-1] == "":
-        links = links[:-1]
+RES = ["1440p", "1080p", "720p", "480p", "360p", "240p", "144p"]
+ABR = ["160kbps", "128kbps", "70kbps", "50kbps", "48kbps"]
+target_res = 0
+target_abr=0
 
-VIDEO = False
+if __name__ == "__main__":
+
+    # get list of links from file
+    links = []
+    with open('links.txt', 'r') as f:
+        links = f.read().split('\n')
+        if links[-1] == "":
+            links = links[:-1]
+
+    # download links one by one
+    for link in links:
+        success = True
+        yt = YouTube(link)
+        video_streams = []
+        while len(video_streams) == 0:
+            video_streams = yt.streams.filter(file_extension='mp4', res=RES[target_res]) # find available streams
+            if target_res + 1 < len(RES):
+                target_res = target_res + 1
+            else:
+                success = False
+                break
+        if not success:
+            print(f"Unable to find video stream for {yt.title}")
+            break
+        vstream = video_streams[0]
+
+
+        # audio
+        audio_streams = []
+        while len(audio_streams) == 0:
+            audio_streams = yt.streams.filter(only_audio=True, abr=ABR[target_abr]) # find available streams
+
+            if target_abr + 1 < len(RES):
+                target_abr = target_abr + 1
+            else:
+                success = False
+                break
+        if not success:
+            print(f"Unable to find audio stream for {yt.title}")
+            break
+        astream = audio_streams[0]
+
+
+        vstream.download(output_path="downloaded/video_only")
+        astream.download(output_path="downloaded/audio_only")
+
+        input_video = ffmpeg.input(f"downloaded/video_only/{vstream.default_filename}")
+        input_audio = ffmpeg.input(f"downloaded/audio_only/{astream.default_filename}")
+
+        ffmpeg.concat(input_video, input_audio, v=1, a=1).output(f'downloaded/{yt.title}.mp4').run()
 
-def download_video(link):
-    yt = YouTube(link)
-    mp4_files = yt.streams.filter(file_extension="mp4")
-    mp4_files = mp4_files.get_highest_resolution()
-    mp4_files.download(output_path=SAVE_PATH)
-    print("Download is completed successfully")
-def download_audio(link):
-    yt = YouTube(link)
-    audio = yt.streams.filter(only_audio=True).get_audio_only()
-    audio.download(output_path=SAVE_PATH)
-    print("Download is completed successfully")
 
 
-for i in links:
-    if VIDEO:
-        download_video(i)
-    else:
-        download_audio(i)
 
-print('Items Downloaded!')