def ask_for_options() -> bool: while True: want = input(Color.get_yes_no()) if want in ['y', 'Y']: return True if want in ['n', 'N']: return False Print.wrong_option() continue
def download_srt(url): """Downloads and extracts .srt file from zip url""" my_zip = Helper.__get_zip_file(url) storage_path = Helper.__get_downloads_folder() Print.bold_string( Constants.download_zip_text.format(Color.RED, storage_path, url)) for file in my_zip.namelist(): if my_zip.getinfo(file).filename.endswith('.srt'): my_zip.extract( file, storage_path ) # extract the file to current folder if it is a text file
def create_query() -> SearchQuery: Print.long_hash() s = input(Color.get_bold_string(Constants.search_string_text)) while not s: s = input(Color.get_bold_string(Constants.search_string_text)) q = 'all' g = Helper.take_optional_input('genre') o = Helper.take_optional_input('order') return SearchQuery(s, q, g, 0, o, 0, 'all')
def take_int_input(no_of_options) -> int: index = None while True: try: index = int( input(Color.get_bold_string(Constants.choose_option_text))) if 1 <= index <= no_of_options: break else: Print.wrong_option() continue except ValueError: Print.wrong_option() continue return index
def take_optional_input(input_type): Print.long_hash() if input_type not in Constants.input_types: Print.bold_string('Wrong input type: {0}'.format(input_type)) exit(1) selection_text = Constants.selection_text[input_type] specific_final_option = Constants.specific_final_option[input_type] special_final_option = Constants.special_final_option[input_type] Print.bold_string(selection_text) want = Helper.ask_for_options() options = Constants.options[input_type] if not want: Print.colored_note(special_final_option) return options[0] final_option = Helper.take_input(input_type, options) Print.colored_note(specific_final_option.format(final_option)) return final_option
def create_query() -> SearchQuery: Print.long_hash() s = input(Color.get_bold_string(Constants.search_string_text)) q = 'all' """ print('Please enter any specific quality of your torrent: ') quality_options = Constants.quality for i in range(len(quality_options)): print('{0}: {1}'.format(i, quality_options[i])) q = int(input()) while q > 4 or q < 0: print('Wrong option, Try again') q = quality_options[q] """ g = Helper.take_optional_input('genre') o = Helper.take_optional_input('order') return SearchQuery(s, q, g, 0, o, 0, 'all')
def take_input(input_type, options): if input_type not in Constants.input_types: Print.bold_string('Wrong input type: {0}'.format(input_type)) exit(1) specific_text = Constants.specific_text[input_type] no_of_options = len(options) Print.bold_string(specific_text) for i in range(1, len(options)): Print.option(i, options[i]) while True: index = int(input()) if 1 <= index <= no_of_options: break else: Print.wrong_option() continue return options[index - 1]
def take_input(input_type, options) -> int: if input_type not in Constants.input_types: Print.bold_string('Wrong input type: {0}'.format(input_type)) exit(1) specific_text = Constants.specific_text[input_type] no_of_options = len(options) Print.bold_string(specific_text) for i in range(1, len(options)): Print.option(i, options[i]) index = Helper.take_int_input(no_of_options) return options[index]
def show_movies(self, movies: MoviesList): print('Movies List: ') for ind, movie in enumerate(movies): print('{0}{1}: {2} ({3}){4}'.format(Color.PURPLE, ind + 1, movie.name, movie.year, Color.END)) while True: mid = int( input(Color.get_bold_string(Constants.movie_download_text))) if mid > len(movies) or mid < 1: Print.wrong_option() continue else: break movie_selected = movies[mid - 1] Print.bold_string(Constants.available_torrents_text) available_torrents = self.get_available_torrents( movie_selected.torrents) if len(available_torrents.values()) == 0: print('{0}{1}{2}'.format(Color.RED, Constants.no_torrent_text, Color.END)) else: ati = 1 for torrent_format in list(available_torrents): print('{0}{1}: {2}{3}'.format(Color.YELLOW, ati, torrent_format, Color.END)) ati += 1 if len(available_torrents) == 1: op = input( 'Press 1 to Download, Press any other key to exit\n') if op == '1': torrent_link = list(available_torrents.values())[0] Print.bold_string('{0}{1}{2}{3}'.format( Constants.click_link_text, Color.RED, torrent_link, Color.END)) else: Print.bold_string(Constants.movie_quality_text) qu = int(input()) torrent_link = list(available_torrents.values())[qu - 1] Helper.open_magnet_link(torrent_link) Print.bold_string('{0}{1}{2}{3}'.format( Constants.click_link_text, Color.RED, torrent_link, Color.END)) Print.long_hash() if movie_selected.subtitle_url and movie_selected.subtitle_url != '': Print.bold_string(Constants.selection_text['subtitle']) download_subtitle = Helper.ask_for_options() if download_subtitle: subtitle = Subtitle() subtitle.search_subtitle(movie_selected.subtitle_url) Print.long_hash() print( Constants.another_movies_text.format( Color.RED, Color.get_bold_string(self.search_query.search_term))) reshow_movies = input(Color.get_yes_no()) if reshow_movies == 'y' or reshow_movies == 'Y': self.show_movies(movies)
def sigint_handler(signum, frame): Print.thanks() exit(1)