def from_directory(self, directory): for file in os.listdir(directory): if file.endswith('.yml'): config = os.path.join(directory, file) answer = Message.input( 'Is %s the open api config file? [Y/n] ' % config) if answer is 'Y': Message.debug( 'The %s file was selected as open api config file' % config) return self.from_file(config) else: Message.debug('Looking for other *.yml files') continue raise Exception('Unable not find *.yml OpenAPI config file in %s' % directory)
def select_directory_input(): return PoliciesTester.directory_exists(Message.input('Project directory: '))