pyparsing을 사용한 bacula 설정 파서

import pprint
from pyparsing import Word, Literal, Forward, Group, ZeroOrMore
from pyparsing import alphas, printables
 
class Parser:
    def __init__(self, conf_file):
        self.conf_file = conf_file
 
    def get_str_conf(self):
        conf_lines = []
        str_conf = open(self.conf_file, "r").read()
        for line in str_conf.strip().split("\n"):
            index = line.find("#")
            if index > -1:
                line = line[:index].strip()
            if line.find(";") > -1:
                conf_lines += line.split(";")
            else:
                conf_lines.append(line)
        conf_lines.insert(0, "conf {")
        conf_lines.append("}")
        return "\n".join(conf_lines)
 
    def syntax(self):
        lbr = Literal( '{' ).suppress()
        rbr = Literal( '}' ).suppress()
        equals = Literal("=").suppress()
        nonequals = "".join( [ c for c in printables if c != "=" ] ) + " \t"
        topic = Word( alphas )
        key = Word( alphas + " \t")
        value = Word ( nonequals )
        expr = Forward()
        opt = Group(expr) | key + equals + value
        opts = ZeroOrMore(opt)
        expr << topic + Group(lbr + opts + rbr)
        return expr
 
    def parse(self):
        conf = {
            'jobs': [],
            'pools': [],
            'clients': [],
            'schedules': [],
            'filesets': []
        }
        str_conf = self.get_str_conf()
        expr = self.syntax()
        result = expr.parseString(str_conf).asList()
        for i in range(len(result[1])):
            conf_name = result[1][i][0].strip().lower().replace(" ", "_")
            if conf_name in ["job", "pool", "client"]:
                keys = [x.strip().lower().replace(" ", "_") for x in result[1][i][1][0::2]]
                values = result[1][i][1][1::2]
                data = dict(zip(keys, values))
                if conf_name == "job":
                    conf['jobs'].append(data)
                elif conf_name == "pool":
                    conf['pools'].append(data)
                elif conf_name == "client":
                    conf['clients'].append(data)
            elif conf_name == "schedule":
                keys = [x.strip().lower().replace(" ", "_") for x in result[1][i][1][0::2]]
                values = result[1][i][1][1::2]
                run_ids = []
                index = 0
                for key in keys:
                    if key == "run":
                        run_ids.append(index)
                    index += 1
                run_ids.reverse()
                runs = []
                for id in run_ids:
                    keys.pop(id)
                    runs.append(values.pop(id))
                data = dict(zip(keys, values))
                data['runs'] = runs
                conf['schedules'].append(data)
            elif conf_name == "fileset":
                data = {
                    'name': "",
                    'include': {
                        'option': {},
                        'files': []
                    },
                    'exclude': {
                        'files': []
                    }
                }
                for j in range(len(result[1][i][1])):
                    if type(result[1][i][1][j]) == str:
                        i_key = result[1][i][1][j].strip().lower().replace(" ", "_")
                        if i_key == "name":
                            data['name'] = result[1][i][1][j+1]
                    elif type(result[1][i][1][j]) == list:
                        j_key = result[1][i][1][j][0].strip().lower().replace(" ", "_")
                        if j_key == "include":
                            option_data = {}
                            temp = []
                            for k in range(len(result[1][i][1][j][1])):
                                if type(result[1][i][1][j][1][k]) == list:
                                    k_key = result[1][i][1][j][1][k][0].strip().lower().replace(" ", "_")
                                    if k_key == "options":
                                        keys = [x.strip().lower().replace(" ", "_") for x in result[1][i][1][j][1][k][1][0::2]]
                                        values = result[1][i][1][j][1][k][1][1::2]
                                        option_data = dict(zip(keys, values))
                                else:
                                    temp.append(result[1][i][1][j][1][k])
                            keys = [x.strip().lower().replace(" ", "_") for x in temp[0::2]]
                            values = temp[1::2]
 
                            file_ids = []
                            index = 0
                            for key in keys:
                                if key == "file":
                                    file_ids.append(index)
                                index += 1
                            file_ids.reverse()
                            files = []
                            for id in file_ids:
                                keys.pop(id)
                                files.append(values.pop(id))
                            include_data = dict(zip(keys, values))
                            include_data['files'] = files
                            include_data['option'] = option_data
                            data['include'] = include_data
                        if j_key == "exclude":
                            keys = [x.strip().lower().replace(" ", "_") for x in result[1][i][1][j][1][0::2]]
                            values = result[1][i][1][j][1][1::2]
 
                            file_ids = []
                            index = 0
                            for key in keys:
                                if key == "file":
                                    file_ids.append(index)
                                index += 1
                            file_ids.reverse()
                            files = []
                            for id in file_ids:
                                keys.pop(id)
                                files.append(values.pop(id))
                            exclude_data = dict(zip(keys, values))
                            exclude_data['files'] = files
                            data['exclude'] = exclude_data
                conf['filesets'].append(data)
        return conf
 
if __name__ == "__main__":
    parser = Parser("bacula-test.conf")
    pprint.pprint(parser.parse())