python学习笔记～INI、REG文件读取函数（自动修复）

引入configparser，直接read整个INI文件，再调用get即可。但需要注意的是，如果INI文件本身不太规范，就会报各种错，而这又常常不可避免的。本文自定义函数通过try...except..来自动纠正再重读。

此外，注册表导出文件大概齐就是INI文件格式，但取初一行的声明也会被认定为没有SECTION头而报错。本文也进行了自动纠正。

极大概率是早有人造过更好的轮子，我写在这里权当作自己学习Python的笔记。功能、特色如下：

1. getINIValue函数三个参数，iniFile, section, option应该不用特别说明；

2. 若iniFile本身不存在或未知异常，返回None，同时输出报错信息；
3. INI文件编码自动尝试，目前顺序为gb2312 utf-8 utf-16 gbk gb18030；
4. INI文件内容不规范（section或option重复异常即configparser.DuplicateSectionError, configparser.DuplicateOptionError）自动纠正（注释后面的，以前面的为准）；
5. 注册表导出Reg文件不符合INI规范（缺少SECTION头异常即configparser.MissingSectionHeaderError）自动纠正（注释最初行）；
6. 找不到Option（考虑到一种情况：上位指定.reg的Option时没加"，自动追加）；
7. 自动纠正INI文件时产生的备份文件自动删除。

Python源码如下，欢迎讨论指正....

 import configparser
 import os

 bakpostfix = '.ibk'

 ## [修正不规则INI文件]
 def __iniFileFix(errorline, iniFile, encoding):
     # 备份INI文件名
     newIniFile = iniFile + bakpostfix

     try:
         fr = open(iniFile, 'r', encoding=encoding)
         fw = open(newIniFile, 'w', encoding=encoding)
         # 逐行读取
         line = fr.readline()
         lineno = 1
         while line != '':
             # 若到了问题行，则注释它
             if lineno == int(errorline):
                 line = '; ' + line
             # 逐行写入（修正后）
             fw.writelines(line)
             line = fr.readline()
             lineno += 1
         fr.close()
         fw.close()
     except Exception as e:
         # 异常时返回报错信息
         error = 'Error:[{0}]'.format(e)
         print(error)
         return error
     # 正常时返回新文件名
     return newIniFile

 ## [读取INI文件]
 def getINIValue(iniFile, section, option):
     config = configparser.ConfigParser()
     value = ''

     # 备份文件
     bBakFile = False

     # 异常：文件不存在
     if (os.path.isfile(iniFile) == False):
         print('Error: file "{0}" not exists...'.format(iniFile))
         return None

     bException = True
     # 尝试编码的次数
     counter = 1
     # 多次循环尝试修正INI为正确的格式，直到正常或遇到无法处理的异常
     while bException:
         try:
             encoding = ''
             # 依次进行如下编码打开尝试（后续根据需要添加）
             if counter == 1: encoding = 'gb2312'
             if counter == 2: encoding = 'utf-8'
             if counter == 3: encoding = 'utf-16'
             if counter == 4: encoding = 'gbk'
             if counter == 5: encoding = 'gb18030'
             if counter == 6:
                 # 暂无法处理的编码
                 print('Error: encoding unknown...')
                 return None
             config.read(iniFile, encoding=encoding)
             bException = False
         except UnicodeDecodeError as e:
             # 编码异常
             print('Error:[{0}]'.format(e))
             counter += 1
         except (configparser.DuplicateSectionError, configparser.DuplicateOptionError) as e:
             # SECTION重复异常
             # Option重复异常
             print('Error:[{0}]'.format(e))
             excep = '{0}'.format(e)
             errorline = excep[excep.find(' [line ') + len(' [line '):excep.find(']', excep.find(' [line '))]
             iniFile = __iniFileFix(errorline, iniFile, encoding)
             if iniFile.find('Error:[') == 0:
                 return None
             bBakFile = True
         except configparser.MissingSectionHeaderError as e:
             # 缺少SECTION头异常
             print('Error:[{0}]'.format(e))
             excep = '{0}'.format(e)
             errorline = excep[excep.find(', line: ') + len(', line: '):excep.find('\n', excep.find(', line: '))]
             iniFile = __iniFileFix(errorline, iniFile, encoding)
             if iniFile.find('Error:[') == 0:
                 return None
             bBakFile = True
         except Exception as e:
             # 未知新异常（后续根据需要追加）
             print('Error:[{0}]'.format(e))
             return None

     # 删除INI备份文件
     if bBakFile == True:
         # 可能存在多个INI备份，循环删除
         bMoreBakFile = True
         while (bMoreBakFile):
             os.remove(iniFile)
             iniFile = iniFile[0:iniFile.rfind(bakpostfix)]
             if iniFile.rfind(bakpostfix) == -1 :
                 bMoreBakFile = False

     # 多次循环尝试修正INI为正确的格式，直到正常或遇到无法处理的异常
     bException = True
     while bException:
         try:
             value = config.get(section, option)
             bException = False
         except configparser.NoSectionError as e:
             # 找不到SECTION
             print('Error:[{0}]'.format(e))
             return None
         except configparser.NoOptionError as e:
             # 找不到Option（考虑到一种情况：上位指定.reg的Option时没加"，自动追加）
             print('Error:[{0}]'.format(e))
             if option[0] != '"' and option[len(option)-1] != '"':
                 option = '"' + option + '"'
             else :
                 return None
         except Exception as e:
             # 未知新异常（后续根据需要追加）
             print('Error:[{0}]'.format(e))
             return None

     return value