Updating Project

Author: JawadSher

Project 14 - Media Searcher and Downloader/__modules__/youtube_Media_Downloader.py

@@ -0,0 +1,59 @@
+import yt_dlp
+import os
+
+def get_default_dir():
+    return os.path.join(os.path.expanduser('~'), "Downloads")
+
+def download_media(url, save_to=None, resolution=None):
+    if not save_to:
+        save_to = get_default_dir()
+
+    if not os.path.exists(save_to):
+        os.mkdir(save_to)
+
+    ydl_opts = {
+        'outtmpl': os.path.join(save_to, '%(title)s.%(ext)s'),  # Output path and filename template
+    }
+
+    try:
+        with yt_dlp.YoutubeDL(ydl_opts) as ydl:
+            info_dict = ydl.extract_info(url, download=True)
+            formats = info_dict.get('formats', [])
+
+            selected_format = None
+
+            for format_item in formats:
+                if 'height' in format_item:
+                    if resolution and f'{resolution}p' in format_item.get('format_note', '').lower():
+                        selected_format = format_item
+                        break
+                    elif not resolution:
+                        selected_format = format_item
+
+            if selected_format:
+                print(f"Selected Resolution: {selected_format['format_note']}")
+                ydl_opts['format'] = selected_format['format_id']
+                ydl.download([url])
+                downloaded_file_path = os.path.join(save_to, f"{info_dict.get('title')}.mp4")
+                print(f"Download completed successfully. File saved to: {downloaded_file_path}")
+            else:
+                print("No suitable format found for the requested resolution.")
+
+            print('\n------------------------------Video Info------------------------------\n')
+            print(f"Title: {info_dict.get('title')}")
+            print(f"Author: {info_dict.get('uploader')}")
+            print(f"Views: {info_dict.get('view_count')}")
+            print(f"Length: {info_dict.get('duration')} seconds")
+            print(f"Description: {info_dict.get('description')}")
+            print('------------------------------------------------------------------------\n')
+
+    except Exception as e:
+        print(f"An error occurred: {e}")
+
+def downloadMedia():
+    url = input("Enter URL Here: ")
+    save_to = input("Save to (default is 'Downloads'): ").strip()
+    resolution = input("Enter desired resolution (e.g., 360p, 720p): ").strip()
+    download_media(url, save_to, resolution)
+
+downloadMedia()