-
Notifications
You must be signed in to change notification settings - Fork 0
/
cfg_parse.py
37 lines (25 loc) · 845 Bytes
/
cfg_parse.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
def parse_cfg(cfgfile):
"""
Takes a config file ,parses it line by line
It creates blocks from the config file and appends it to list of modules
creating the layers for the network.
"""
block = {}
blocks = []
files = open(cfgfile,'r')
lines = files.read().split("\n")
lines = [x for x in lines if len(x) > 0]
lines = [x for x in lines if x[0] !="#"]
lines = [x.rstrip().lstrip() for x in lines]
for line in lines:
if line[0] == "[":
if len(block) !=0:
#print(block)
blocks.append(block)
block = {}
block['type'] = line[1:-1].rstrip()
else:
key,value = line.split("=")
block[key.rstrip()] = value.lstrip()
blocks.append(block)
return blocks