使用Python解析Fortigate配置文件

时间:2012-08-30 21:24:34

标签: python acl

我基本上是在尝试继续解决有关强制配置文件的未回答的问题。

Reading a fortigate configuration file with Python

根本问题是此配置包含许多这样的记录,

edit 1
    set srcintf "port26"
    set dstintf "port25"
        set srcaddr "all"             
        set dstaddr "all"             
    set action accept
    set utm-status enable
    set schedule "always"
        set service "ANY"             
    set av-profile "default"
    set nat enable
    set central-nat enable
next

我想在一行中获取每个acl的输出,以便我可以将它们导入CSV。问题是每个记录可以有可变数量的行,缩进显示前一行的子部分。另一篇文章确实得到了一些,但它没有处理缩进。我想出了一些用任意字符替换空格的变通方法,但我不知道是否有方法来读取数字制表符/空格并使用它来表示定位。

由于

1 个答案:

答案 0 :(得分:1)

所以我设法读取你的文本并将其变成python中的字典。这很简单。你基本上必须做一些事情:

conFigFile=open('./config.txt')

data=dict()
record=0

for line in conFigFile:
    if line.find('edit')>=0:
        record=int(line.replace('edit',''))
        data[record]={}
    if line.find('set')>=0:
        line=line.replace('set','')
        line=line.strip()
        print line
        key,val=line.split(' ')
        data[record][key]=val
conFigFile.close()

这将生成一个字典,然后允许您拨打电话,如:

>>> data[1]['nat']
'enable'
>>> data[1].keys()
['nat', 'service', 'schedule', 'central-nat', 'srcaddr', 'av-profile', 'dstintf', 'srcintf', 'action', 'dstaddr', 'utm-status']

所以现在可以生成一个csv文件:

csvFile=open('./data.csv','w')
records=data.keys()
for record in records:
    values=data[record].keys()
    valList=['Record',str(record)]
    for val in values:
        valList.append(val)
        valList.append(data[record][val])
    csvFile.write(",".join(valList))
csvFile.close()

生成csv文件:

Record,1,nat,enable,service,"ANY",schedule,"always",central-nat,enable,srcaddr,"all",av-profile,"default",dstintf,"port25",srcintf,"port26",action,accept,dstaddr,"all",utm-status,enable

如果您真的想要计算行前的空格,可以执行以下操作:

>>> a='     test: one     '
>>> a.count(' ') #count all spaces
11
>>> (len(a) - len(a.lstrip())) #count leading spaces
5
相关问题