python - 从文本文件创建嵌套字典

Question

/tmp/bond0:

Ethernet Channel Bonding Driver: v3.7.1 (April 27, 2011)

Bonding Mode: IEEE 802.3ad Dynamic link aggregation
Transmit Hash Policy: layer2+3 (2)
MII Status: up
MII Polling Interval (ms): 100
Up Delay (ms): 0
Down Delay (ms): 0

802.3ad info
LACP rate: slow
Min links: 0
Aggregator selection policy (ad_select): stable
Active Aggregator Info:
Aggregator ID: 2
Number of ports: 2
Actor Key: 11
Partner Key: 705
Partner Mac Address: 02:1c:73:9c:3c:fe

Slave Interface: p1p1
MII Status: up
Speed: 10000 Mbps
Duplex: full
Link Failure Count: 0
Permanent HW addr: 9c:dc:71:45:eb:80
Aggregator ID: 2
Slave queue ID: 0

Slave Interface: p4p1
MII Status: up
Speed: 10000 Mbps
Duplex: full
Link Failure Count: 0
Permanent HW addr: 9c:dc:71:4d:80:20
Aggregator ID: 2
Slave queue ID: 0

我有上面的文本输出，我想创建一个如下所示的嵌套字典:在上面的文本中可能有两个以上的从接口(interface) block

bond0 : {
 'MII Status:' : 'up',
 'Aggregator ID:' : '2',
 'Slave Interfaces' : { 'p1p1' : { 'MII Status' : 'up',
 'Permanent HW addr' : '9c:dc:71:45:eb:80',
 'MII Status' : up },
 'p4p1' : { ''MII Status' : 'up',
 'Permanent HW addr' : '9c:dc:71:4d:80:20',
 'MII Status' : up },
 },

我开始做一些编码，如下所示，但仍然没有实现: #/usr/bin/python

从 future 导入 print_function 导入打印件 导入操作系统 进口重新 导入子流程

class BndClass(dict):
   def __init__(self, Bnd=None):
      self['Name'] = Bnd
      self.uPdateInfo()
      super(BndClass, self).__init__()

   def uPdateInfo(self):
      OutBnd = subprocess.Popen(['cat', '/tmp/'\
    + self['Name']],shell=False,stdout=subprocess.PIPE,stderr=subprocess.PIPE)
      for line in OutBnd.stdout:
          match = re.search(r'^Cur.*?:\s+(.*?)$', line)
          if match:
              self['act_int'] = match.group(1)

          match = re.search(r'^\s*?Aggregator ID:\s+(\d)$', line)
          if match:
              self['agid'] = match.group(1)

          match = re.search(r'^Slave\sInterface:\s(.*?)$', line)
          if match:
              self.setdefault('slvs', []).append(match.group(1))


  if __name__ == '__main__':
      Y = BndClass('bond0')

  Y:

  {'Name': 'bond0', 'agid': '2', 'slvs': ['p1p1', 'p4p1']}

<小时/>

我可能有更多的“债券”文件，即 bond1、2、3、4 等。所以我认为开设类(class)更有意义，因此我转换为类(class)形式。然而它失败了。有什么想法吗？

#!/usr/bin/python
from __future__ import print_function
from collections import defaultdict
import pprint
import os
import re
import subprocess



class BndClass(dict):
    def __init__(self, Bnd=None):
        self['Name'] = Bnd
        self.uPdateInfo()
        super(BndClass, self).__init__()

    def uPdateInfo(self):
        with open(self['Name'], "r") as f:
            for line in f:
                line = line.strip()  # clean that up a bit :)
                if line.strip() == "": continue
                match = re.search(r'^\s*?(Aggregator ID):\s+(\d)$', line)
                if match:
                    self[match.group(1)] = match.group(2)
                    continue

                match = re.search(r'^(Slave\sInterface):\s(.*?)$', line)
                if match:
                    self[match.group(1)] = match.group(2)
                    while True:
                        try:
                            line = next(f).strip()
                        except:
                            break
                        if line == "":
                            break
                        slave_match = re.search(r'^(MII\sStatus):\s+(\w+)$', line)
                        if slave_match:
                            self.setdefault(match.group(1), {}).setdefault(match.group(2), {})[slave_match.group(1)] = slave_match.group(2)
                            continue
                        slave_match = re.search(r'^(Permanent\sHW\saddr):\s+(.+)$', line)
                        if slave_match:
                            self.setdefault(match.group(1), {}).setdefault(match.group(2), {})[slave_match.group(1)] = slave_match.group(2)
                            continue

if __name__ == '__main__':
   B = BndClass('bond0')

Traceback (most recent call last):
File "./bc6.py", line 47, in <module>
    B = BndClass('bond0')
 File "./bc6.py", line 14, in __init__
   self.uPdateInfo()
 File "./bc6.py", line 39, in uPdateInfo
   self.setdefault(match.group(1), {}).setdefault(match.group(2), {}) 
 [slave_match.group(1)] = slave_match.group(2)
 AttributeError: 'str' object has no attribute 'setdefault'

Answer 1

抱歉，我开始了，然后做了其他事情然后忘记了......

这是一个解决方案，它不是最性感的，但它仍然有效。如果您的文件具有非常严格的格式(正如它看起来的那样)，您可以使用 break 和 continue 语句更有效地避免无用的正则表达式搜索。

import re
from collections import defaultdict

final_dict = defaultdict(lambda: defaultdict(str))

with open("bound0_file.txt", "r") as f:
    for line in f:
        line = line.strip()  # clean that up a bit :)
        if line.strip() == "": continue
        match = re.search(r'^\s*?(Aggregator ID):\s+(\d)$', line)
        if match:
            final_dict[match.group(1)] = match.group(2)
            continue

        match = re.search(r'^(Slave\sInterface):\s(.*?)$', line)
        if match:
            final_dict[match.group(1)][match.group(2)] = {}
            while True:
                try:
                    line = next(f).strip()
                except:
                    break
                if line == "":
                    break
                slave_match = re.search(r'^(MII\sStatus):\s+(\w+)$', line)
                if slave_match:
                    final_dict[match.group(1)][match.group(2)][slave_match.group(1)] = slave_match.group(2)
                    continue
                slave_match = re.search(r'^(Permanent\sHW\saddr):\s+(.+)$', line)
                if slave_match:
                    final_dict[match.group(1)][match.group(2)][slave_match.group(1)] = slave_match.group(2)
                    continue

    print(final_dict)

<小时/>

from collections import defaultdict
final_dict = defaultdict(lambda: defaultdict(str))

这里我们使用 defaultdict ，它允许我们“根据请求创建 key ”，基本上，如果您搜索不存在的 key ，defaultdict 将创建它引发错误。

我嵌套了其中两个，因为我真正想要的是第二个，而且最多有 2 层。

...
match = re.search(r'^\s*?(Aggregator ID):\s+(\d)$', line)
if match:
        final_dict[match.group(1)] = match.group(2)
        continue

这里非常有解释性，如果我的行是聚合器ID，我只是将其放入我的final_dict中。 注意围绕“聚合器 id) 添加的组。然后，因为我知道我已经完成了这一行，所以我使用 continue 语句跳过循环的其余部分并继续到下一行。

...
match = re.search(r'^(Slave\sInterface):\s(.*?)$', line)
if match:
    final_dict[match.group(1)][match.group(2)] = {}

这里开始棘手的部分。如果前一个匹配失败(又名行不是聚合器id)，那么我们尝试这个，如果它不是从属接口(interface)，我们只是循环下一行。

但是，如果是这一行，则意味着我们输入一个从属接口(interface) block ，该 block 将以空行结尾(稍后见)。

此 final_dict 行是我必须使用 defaultdict 的原因，因为我将创建嵌套字典 Slave interface: { 'p1p1': {} } 立即。

        ...
        while True:
            try:
                line = next(f).strip()
            except:
                break

我们输入一个“子循环”，我用它来遍历从属接口(interface) block 来查找您想要的条目(MII 状态 和 永久硬件地址)。我们将在这个子循环中做一些事情(如下)，但是当我们找到空行时，这意味着我们已经完成了当前 block 。 (这里的 try-expect 语句是为了在到达文件末尾时中断)。

            ...
            if slave_match:
                final_dict[match.group(1)][match.group(2)][slave_match.group(1)] = slave_match.group(2)
                continue
            slave_match = re.search(r'^(Permanent\sHW\saddr):\s+(.+)$', line)
            if slave_match:
                final_dict[match.group(1)][match.group(2)][slave_match.group(1)] = slave_match.group(2)
                continue

这与第一个匹配完全相同，我们查找正确的行并将其添加到final_dict中。但是，我们需要使用另一个变量，因为我们需要 match 变量来访问字典中的正确位置。

<小时/>

正如你所看到的，你真的很亲近。但这种方法可能不是最好的方法。