【Python】读取MySQL 配置文件的脚本

记录一个读取my.cnf配置的脚本,可以基于该函数做一些mysql 后端运维工作。各位可以基于自己的需求进行修改。
脚本名称 getcnf.py

1. import sys
   
2. import os
   
3. 
   
4. def read_cnf(cnf_path):
   
5.         assert cnf_path is not None and os.path.exists(cnf_path)
   
6.         cnf_dict = {}
   
7.         cur_section = None
   
8.         with open(cnf_path) as cnf_reader:
   
9.             for line in cnf_reader.readlines():
   
10.                 line = ''.join(line.split())
    
11.                 if len(line) <= 0 or '#' == line[0]:
    
12.                     continue
    
13.                 if '[' == line[0] and ']' == line[-1]:
    
14.                     cur_section = line[len('['):len(line) - 1]
    
15.                     if cur_section not in cnf_dict:
    
16.                         cnf_dict[cur_section] = {}
    
17.                 elif '=' in line and line.count('=') == 1:
    
18.                     if cur_section is None:
    
19.                         LOGGER.warning('cur_section is None')
    
20.                         continue
    
21.                     tokens = line.split('=')
    
22.                     key = tokens[0].replace('"', '').replace("'", '')
    
23.                     value = tokens[1].replace('"', '').replace("'", "")
    
24.                     cnf_dict[cur_section][key] = value
    
25.         return cnf_dict
    
26. 
    
27. 
    
28. def main():
    
29.     cnf_path="/u01/my3353/my.cnf"
    
30.     mycnf = read_cnf(cnf_path)
    
31.     print mycnf['mysqld']['tmpdir']
    
32. 
    
33. if __name__ == '__main__':
    
34.     main()

运行效果图