def find_all_ini(self): """ This function is used for finding all _fix files in the user defined directory """ for dir_name in os.listdir(self.dirname): if dir_name.endswith(".ini"): self.lists.append(os.path.join(self.dirname, dir_name)) for file_name in self.lists: if FileHelper.check_file(file_name, "r") is False: continue try: config = configparser.ConfigParser() filehander = codecs.open(file_name, 'r', encoding=settings.defenc) config.readfp(filehander) fields = {} if config.has_section('premigrate'): section = 'premigrate' else: section = 'preupgrade' for option in config.options(section): fields[option] = config.get(section, option) self.loaded[file_name] = [fields] except configparser.MissingSectionHeaderError: MessageHelper.print_error_msg(title="Missing section header") except configparser.NoSectionError: MessageHelper.print_error_msg(title="Missing section header") except configparser.ParsingError: MessageHelper.print_error_msg(title="Incorrect INI file\n", msg=file_name) os.sys.exit(1)
def find_all_ini(self): """ Find all ini files in the self.dirname path. Then read each ini file and save all options in 'preupgrade' section with their values to self.loded dict: self.loaded[ini_file_path] = {option1: value, option2: value, ...} """ for dir_name in os.listdir(self.dirname): if dir_name.endswith(".ini"): self.lists.append(os.path.join(self.dirname, dir_name)) for file_name in self.lists: if FileHelper.check_file(file_name, "r") is False: continue filehander = codecs.open(file_name, 'r', encoding=settings.defenc) config = configparser.ConfigParser() config.readfp(filehander) fields = {} section = 'preupgrade' for option in config.options(section): fields[option] = config.get(section, option) self.loaded[file_name] = fields
def check_files_are_readable(files): for file in files: if not FileHelper.check_file(file, "r"): sys.exit("Error: Can't read '{0}'.".format(file))