forked from lzc978/digraph
-
Notifications
You must be signed in to change notification settings - Fork 0
/
get_config.py
61 lines (48 loc) · 2.13 KB
/
get_config.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
import configparser
from utils import logger
class ReadConfigFile(object):
def __init__(self, ini_path):
self.ini_path = ini_path
self.cfg_parser = None
self.cfg = {}
def read_ini(self):
self.cfg_parser = configparser.ConfigParser()
try:
self.cfg_parser.read(self.ini_path, encoding="utf8")
self._digraph()
self._inotify()
self._tcpsocket()
self._mysql()
return self.cfg
except configparser.Error as e:
self.cfg_parser = None
logger.error('errcode:10004, 读取配置文件失败,error:{err}'.format(err=e))
return False
def _digraph(self):
cfg_digraph = {}
section_name = 'Digraph'
cfg_digraph['_data_path'] = self.cfg_parser.get(section_name, '_data_path')
cfg_digraph['_sentence_path'] = self.cfg_parser.get(section_name, '_sentence_path')
self.cfg['digraph'] = cfg_digraph; return
def _inotify(self):
cfg_inotify = {}
section_name = 'Inotify'
cfg_inotify['WATCH_PATH'] = self.cfg_parser.get(section_name, 'WATCH_PATH')
cfg_inotify['PICKLE_NAME'] = self.cfg_parser.get(section_name, 'PICKLE_NAME')
cfg_inotify['FILE_NAME'] = self.cfg_parser.get(section_name, 'FILE_NAME')
self.cfg['inotify'] = cfg_inotify; return
def _tcpsocket(self):
cfg_tcpsocket = {}
section_name = 'TCPSocket'
cfg_tcpsocket['host'] = self.cfg_parser.get(section_name, 'host')
cfg_tcpsocket['port'] = int(self.cfg_parser.get(section_name, 'port'))
self.cfg['tcpsocket'] = cfg_tcpsocket; return
def _mysql(self):
cfg_mysql = {}
section_name = 'Mysql'
cfg_mysql['host'] = self.cfg_parser.get(section_name, 'host')
cfg_mysql['port'] = int(self.cfg_parser.get(section_name, 'port'))
cfg_mysql['user'] = self.cfg_parser.get(section_name, 'user')
cfg_mysql['password'] = self.cfg_parser.get(section_name, 'password')
cfg_mysql['database'] = self.cfg_parser.get(section_name, 'database')
self.cfg['mysql'] = cfg_mysql; return