def get_output(locations: [str], output_generators: [str]) -> [[str]]: output_gen = OutputGenerators(locations) output = [] for output_generator_type in output_generators: output_generator = output_gen.get_ouput_generator(output_generator_type) if output_generator != None: output.append(output_generator.get_output()) return output
def get_output_generators(num_generators: int) -> [str]: output_generators = [] for i in range(num_generators): user_input = input().strip() if not OutputGenerators.is_valid_output_generator(user_input): _print_error_msg('Invalid output type: {}'.format(user_input if user_input != '' else 'undefined')) return None output_generators.append(user_input) return output_generators