使用 Python 的 LDIF Parser 跳过一些属性并写入新的 LDIF 文件

问题描述

我正在使用 python LDIF 解析器 (link) 编辑一些属性并生成文件的修改版本。

我的 LDIF 文件是

# entry-id: 1
dn: cn=Directory Administrators,dc=organization,dc=corp
nsUniqueId: 2947f1b3-1dd211b2-80b89250-3a51c428
objectClass: top
objectClass: groupofuniquenames
cn: Directory Administrators
creatorsName: cn=directory manager
modifiersName: cn=directory manager
createTimestamp: 20151110180921Z
modifyTimestamp: 20151110180921Z

# entry-id: 2
dn: cn=214-All-Matrix-100342-ALXD,ou=groups,dc=corp
modifyTimestamp: 20190905182416Z
modifiersName: cn=directory manager
owner: uid=hwuebker,ou=people,dc=corp
nsUniqueId: 4350c83d-1dd211b2-80a59250-3a51c428
uniqueMember: uid=anthonys,dc=corp
GroupType: Core
Application: AppUID
adminGroupAdmin: cn=IDM System Managers,dc=corp

# entry-id: 3
dn: uid=twalsh,dc=corp
nsUniqueId: 3df58701-1dd211b2-80489250-3a51c428
modifyTimestamp: 20180606194655Z
modifiersName: cn=directory manager
initials: 1
Document: 0M8R4KGxGuEAAAAAAAAAAAAAAAAAAAAAPgADAP7/CQAGAAAAAAAAAAAAAAAC
 AAAAAQAAAAAAAAAAEAAAAgAAAAIAAAD+////AAAAAAAAAACAAAAA////////////////////////
 /////////////////////////////////////////////////////////AAAAAA==
Document: RE  E-mail details .msg

我目前用于 LDIF 解析器的 Python 代码是

parser = MyLDIF(open(fileName,'rb'),sys.stdout)
parser.parse()

#LDIF Parser go through records
for dn,entry in parser.parse():
    
        Doc = entry['Document']
        uidList= entry['uid']  
        uid = uidList[0]
        DocValue = Doc[0]
        DocSource = Doc[1]
    
        writer = LDIFWriter(open("data.ldif","ab"))
        writer.unparse(dn,{
            "nsUniqueId": entry['nsUniqueId'],"Docs": entry['Document'],})

由于第一个和第二个条目没有 DOCUMENT 属性，这会引发错误。

Doc = entry['Document']
  KeyError: 'Document'  

如何在不做任何更改的情况下复制第 1 个和第 2 个条目的同时操作第 3 个条目？

解决方法

您可以使用内置方法 get(key[,default]) 获取字典中 key 的值，或者如果未设置 key，则获取 默认 值。

我建议你在你的类中使用 handle 方法来实现这一点，并直接写入输出文件（而不是写入 stdout 并为输出文件创建另一个写入器，你仍然可以从句柄方法打印数据）：

class MyLDIF(LDIFParser):
   def __init__(self,input,output):
      LDIFParser.__init__(self,input)
      self.writer = LDIFWriter(output)

   def handle(self,dn,entry):
      if entry.get('Document',False):
          # modify entry if 'Document' is set
          entry['attribute1'] = 'value1'
          entry['attribute2'] = 'value2'
          # ...
      self.writer.unparse(dn,entry)

parser = MyLDIF(open('data.ldif','r'),open('output.ldif','w'))
parser.parse()