比较后在TXT文件中在特定行之前获取特定行

Question

我正在进行2个TXT文件之间的比较，并且我想在差线之前添加界面的名称，其中包含此行的名称如下：。

第一个文件包含：

!CfgFileCrc:f4fcebea
!Software Version V800R021C00SPC100
!Last configuration was updated at 2022-04-07 10:16:05 UTC 
!Last configuration was saved at 2022-04-09 21:00:41 UTC 
!MKHash 
info-center filter-id bymodule-alias system hwSecurityRisk
info-center loghost source LoopBack1
set service-mode forwarding-mode enhance
interface GigabitEthernet4/0/14
 shutdown
interface GigabitEthernet4/0/15
 negotiation auto
 undo shutdown
interface GigabitEthernet4/0/16
 negotiation auto
 undo shutdown

第二个文件包含：

!CfgFileCrc:f4fcebea
!Software Version V800R021C00SPC100
!Last configuration was updated at 2022-04-07 10:16:05 UTC
!Last configuration was saved at 2022-04-09 21:00:41 UTC 
!MKHash 
info-center filter-id bymodule-alias system hwSecurityRisk
info-center loghost source LoopBack1
set service-mode forwarding-mode enhance
interface GigabitEthernet4/0/14
 shutdown
interface GigabitEthernet4/0/15
 negotiation auto
 description CEM-Smart-Care-PS
 undo shutdown
interface GigabitEthernet4/0/16
 negotiation auto
 description CEM-Smart-Care-PS
 undo shutdown

代码如下：

def files(Devices):
 doc = open('C:/Users/Ahmed Shouaib/Desktop/Difference/test/10-4-2022/'+Devices+'.txt', 'r')
 dox = open('C:/Users/Ahmed Shouaib/Desktop/Difference/test/7-4-2022/'+Devices+'.txt', 'r')
 f1 = [x for x in doc.readlines()]
 f2 = [x for x in dox.readlines()]
 with open('C:/Users/Ahmed Shouaib/Desktop/Difference/test/result/' + Devices + '.txt', 'w') as new:
  GGG = ['CfgFileCrc', 'Last configuration was','MKHash' , 'username ftpuser password']
  for line in f1:
      if line not in f2 and not any(x in line for x in GGG):
          new.write("+"+line + '\n')

  GGG = ['CfgFileCrc','Last configuration was','MKHash','username ftpuser password']
  XX=1
  for line in f2:
      if line not in f1 and not any(x in line for x in GGG):
          if ( XX == 1):
           new.write('-' * 80 + '\n'+ '\n')
           XX = 0
          new.write("-"+line + '\n')
  doc.close()
  dox.close()

files('10.0.130.71')

代码结果如下：

+ description CEM-Smart-Care-PS

+ description CEM-Smart-Care-PS

我需要的代码结果：

interface GigabitEthernet4/0/15
+ description CEM-Smart-Care-PS

interface GigabitEthernet4/0/16
+ description CEM-Smart-Care-PS

Answer 1

不需要重新实现算法来查找差异，您可以使用 difflib.ndiff()，只需要一些准备。我们只需要向 ndiff() 传递属于接口的行。

我们可以使用返回接口名称的生成器函数，并且所有行都属于该接口：

def iter_interface(f):
    interface = ""
    lines = []
    for line in f:
        if interface:
            if line.startswith(" "):
                lines.append(line.strip())
            else:
                yield interface, lines
                interface = ""
                lines = []
        if line.startswith("interface"):
            interface = line.rstrip()
    if interface:
        yield interface, lines

然后我们只需打开两个文件并使用生成器函数迭代它们。如果两个文件中的接口名称相同，我们只需将行传递给 ndiff() 并仅打印差异：

from difflib import ndiff

...

with open("file1.txt") as f1, open("file2.txt") as f2:
    for (interface_f1, lines_f1), (interface_f2, lines_f2) in \
            zip(iter_interface(f1), iter_interface(f2)):
        if interface_f1 == interface_f2:
            interface_printed = False
            for diff_line in ndiff(lines_f1, lines_f2):
                if diff_line[0] != " ":
                    if not interface_printed:
                        print(interface_f1)
                        interface_printed = True
                    print(diff_line)
            if interface_printed:
                print()

使用您提供的两个文件示例，您将获得下一个输出：

interface GigabitEthernet4/0/15
+ description CEM-Smart-Care-PS

interface GigabitEthernet4/0/16
+ description CEM-Smart-Care-PS

PS 使用生成器函数允许您不将整个文件内容保留在内存中，因此此方法对于大文件应该可以正常工作。