diff options
Diffstat (limited to 'src/view')
| -rw-r--r-- | src/view/app.py | 140 | ||||
| -rw-r--r-- | src/view/download_type_chooser.py | 33 | ||||
| -rw-r--r-- | src/view/file_chooser.py | 51 |
3 files changed, 224 insertions, 0 deletions
diff --git a/src/view/app.py b/src/view/app.py new file mode 100644 index 0000000..203cc56 --- /dev/null +++ b/src/view/app.py @@ -0,0 +1,140 @@ +#! /usr/bin/python3 + +from kivy.app import App +from kivy.uix.boxlayout import BoxLayout +from kivy.uix.button import Button +from kivy.uix.floatlayout import FloatLayout +from kivy.uix.textinput import TextInput +from kivy.uix.popup import Popup +from kivy.uix.label import Label + +import threading +from pathlib import Path +import os + +from model.youload_playlist import YouLoadPlayList +from view.file_chooser import YouloadFileChooser +from view.download_type_chooser import YouloadDownloadTypeChooser + +class YouloadApp(App): + def build(self): + layout = BoxLayout(orientation='vertical') + + # Data members + self.url = "https://youtube.com/playlist?list=PLXbRgJ9vx_KuA5CljV3BpiV58qx6znmRk&si=JFaVyspkeoMWUDNL" + self.is_downloading = False + self.should_stop_download = False + + # File chooser. + self.file_chooser = YouloadFileChooser() + self.file_chooser.set_app(self) + + # Download type chooser. + self.download_type_chooser = YouloadDownloadTypeChooser() + self.download_type_chooser.set_app(self) + + # Url input. + url_input = TextInput(text=self.url, multiline=False, size_hint=(0.8, 1.0)) + url_input.bind(text=self.uid_url_input) + + # Submit button. + self.submit = Button(text="Download", size_hint=(0.2, 1.0)) + self.submit.bind(on_press=self.submit_cb) + + # Url and submit layout. + url_and_submit = BoxLayout(size_hint=(1.0, 0.1)) + url_and_submit.add_widget(url_input) + url_and_submit.add_widget(self.submit) + + # Download status. + self.download_status = Label(text="Nothing downloading", size_hint=(0.3, 1.0)) + + # Download type chooser button. + self.download_type_button = Button(text=self.download_type_chooser.current_type, size_hint=(0.1, 1.0)) + self.download_type_button.bind(on_press=self.download_type_chooser.open) + + # Folder display. + self.folder_display = Label(text=os.getcwd(), size_hint=(0.4, 1.0)) + + # choose folder button. + self.choose_folder_button = Button(text="Folder", size_hint=(0.2, 1.0)) + self.choose_folder_button.bind(on_press=self.choose_folder_cb) + + # Info line. + info_line = BoxLayout(size_hint=(1.0, 0.1)) + info_line.add_widget(self.download_status) + info_line.add_widget(self.download_type_button) + info_line.add_widget(self.folder_display) + info_line.add_widget(self.choose_folder_button) + + # Info display. + self.downloads_display = Label(text="", size_hint=(1.0, 0.8)) + + # Everything else (: + layout.add_widget(url_and_submit) + layout.add_widget(info_line) + layout.add_widget(self.downloads_display) + + return layout + + def uid_url_input(self, instance, value): + self.url = value + + # Download the videos in a different thread so the ui still works. + def download_playlist_thread(self): + try: + self.download_status.text = "Fetching playlist" + self.submit.text = "Stop" + self.is_downloading = True + + # Get playlist. + playlist = YouLoadPlayList(self.url) + + # Set directory and other stuff. + playlist.set_download_directory(self.folder_display.text) + playlist.download_type = self.download_type_chooser.current_type + playlist.prepare_for_download() + self.downloads_display.text = f"Downloading to {playlist.folder_name}\n" + + # Download each video + for i in range(playlist.video_count): + # Stop this mother fucker + if self.should_stop_download: + break + + self.download_status.text = f"Downloading {i+1}/{playlist.video_count}" + self.downloads_display.text += playlist.download_video(i) + "\n" + + # Complete download. + self.download_status.text = "Download complete" + except FileExistsError: + self.download_status.text = "Folder already exists" + except KeyError: + self.download_status.text = "Error getting playlist" + except AttributeError: + self.download_status.text = "Download type not supported" + + self.submit.text = "Download" + self.is_downloading = False + + def stop_download(self): + self.download_status.text = "Stopping download" + self.should_stop_download = True + + def submit_cb(self, instance): + # Is already downloading something. + if self.is_downloading: + self.stop_download() + return + + # Start download thread. + download_thread = threading.Thread(target=self.download_playlist_thread) + self.should_stop_download = False + download_thread.start() + + def choose_folder_cb(self, instance): + self.file_chooser.open() + + def on_stop(self): + self.stop_download() + diff --git a/src/view/download_type_chooser.py b/src/view/download_type_chooser.py new file mode 100644 index 0000000..927d7a4 --- /dev/null +++ b/src/view/download_type_chooser.py @@ -0,0 +1,33 @@ +from kivy.uix.label import Label +from kivy.uix.button import Button +from kivy.uix.popup import Popup +from kivy.uix.boxlayout import BoxLayout + +from model.youload_playlist import YouLoadPlayList + +class YouloadDownloadTypeChooser(Popup): + + def __init__(self, **kwargs): + super(YouloadDownloadTypeChooser, self).__init__(**kwargs) + + self.title = "Download types" + + layout = BoxLayout(orientation="vertical") + self.current_type = YouLoadPlayList.DEFAULT_DOWNLOAD_TYPE + + # Add option buttons. + for download_type in YouLoadPlayList.DOWNLOAD_TYPES: + item = Button(text=download_type) + item.bind(on_press=self.option_button_cb) + layout.add_widget(item) + + self.content = layout + + def set_app(self, app): + self.app = app + + def option_button_cb(self, instance): + self.current_type = instance.text + self.app.download_type_button.text = self.current_type + self.dismiss() + diff --git a/src/view/file_chooser.py b/src/view/file_chooser.py new file mode 100644 index 0000000..dfba199 --- /dev/null +++ b/src/view/file_chooser.py @@ -0,0 +1,51 @@ +from kivy.uix.popup import Popup +from kivy.uix.button import Button +from kivy.uix.boxlayout import BoxLayout +from kivy.uix.filechooser import FileChooserListView +from kivy.uix.textinput import TextInput +from pathlib import Path +import os + +class YouloadFileChooser(Popup): + + def __init__(self, **kwargs): + super(YouloadFileChooser, self).__init__(**kwargs) + + layout = BoxLayout(orientation='vertical') + + self.title = "Find folder" + + # Path input. + self.path_input = TextInput(text=str(Path.home()), multiline=False, size_hint=(0.9, 1.0)) + self.path_input.bind(text=self.uid_path_input) + + # Close button. + close_button = Button(text="Close", size_hint=(0.1, 1.0)) + close_button.bind(on_press=self.close_button_cb) + + top_bar_layout = BoxLayout(size_hint=(1.0, 0.1)) + top_bar_layout.add_widget(self.path_input) + top_bar_layout.add_widget(close_button) + + # File chooser. + self.file_chooser = FileChooserListView(size_hint=(1.0, 0.9)) + self.file_chooser.path = str(Path.home()) + + layout.add_widget(top_bar_layout) + layout.add_widget(self.file_chooser) + self.content = layout + + def set_app(self, app): + self.app = app + + def close_button_cb(self, instance): + self.app.folder_display.text = self.get_folder() + self.dismiss() + + def uid_path_input(self, instance, value): + if os.path.exists(value): + self.file_chooser.path = value + + def get_folder(self): + return self.file_chooser.path + |