Python ldif注入和csv读取脚本

Question

有人能帮我找出我做错了什么吗？我正在编写一个python shell脚本，它接受一个ldif文件和一个csv文件，然后将csv文件中的内容附加到ldif中每个记录的末尾。比如：

示例CSV：

    "KEY","VALUE"
    "abc","def"
    "foo","bar"
    "qwop","flop"

样品本地设计院（ldif）：

    dn: Aziz
    cn: Aziz_09

    dn: Carl
    cn: Carl_04

在pythonmyscript.py"sample.ldif""sample.csv"之后

    dn: Aziz
    cn: Aziz_09
    KEY: VALUE
    abc: def
    foo: bar
    qwop: flop

    dn: Carl
    cn: Carl_04
    KEY: VALUE
    abc: def
    foo: bar
    qwop: flop

到目前为止，我的代码可以编译，但不能正确修改文件。我正在创建一个对象，该对象在创建时采用csv文件路径名字符串，然后将键存储到列表字段中，并将值存储到列表字段中。然后打开ldif文件，解析记录之间的转义字符，并在每条记录的末尾插入列表字段（键和值）：

    import sys, csv

    #  Make new object that can open a csv and set csv data in its arrays
    class Container(object):
      def __init__(self, filename=None, keys=None, values=None):
        self.filename = filename
        self.keys = []
        self.values = []

      #  Opens self.filename and puts 0th and 1st rows into keys and values respectively  
      def csv_to_list():
        with open(self.filename, 'rb') as f:
          reader = csv.reader(f)
          for row in reader:
            self.keys = row[0]
            self.values = row[1]

    haruhi = Container("./content/test_pairs.txt")
    haruhi.csv_to_list

    # open first argument of the command line call to ldif_record_a.py for read/writing
    with open(sys.argv[1],'r+') as f1:
      lines=[x.strip() for x in f1]  # Create list with each line as an element
      f1.truncate(0)
      f1.seek(0)
      count = 0
      for x in lines:
         if x:
           f1.write(x+'\n')
         else:
           f1.write("{0}: {1}\n\n".format(haruhi.keys[count] , haruhi.values[count]))
           count = count + 1
      f1.write("{0}: {1}\n\n".format(haruhi.keys[count] , haruhi.values[count]))

我是Python的新手！任何帮助，建议和/或资源方向都将不胜感激！谢谢你SO

Answer 1

好吧，这是我临时安排的，所以它需要改进，但这里是：

import csv
import re

csv_data = list(csv.reader(open('/home/jon/tmp/data.csv'))) # (1)
csv_text = '\n' + '\n'.join('{0} : {1}'.format(*row) for row in csv_data) # (2)

with open('/home/jon/tmp/other.ldif') as f:
    contents = f.read() # (3)
    print re.sub(r'(\n\n)|(\n$)', csv_text + '\n\n', contents) # (4)

>

（1）将CSV文件读入列表中

csv_data == [['键'，'值']，['abc'，'def']，['foo'，'bar']，['qwop'，'flp']]

（2）创建要附加到每个ldif的文本表示

键：值abc:def foo:bar qwop:flop

（3） 打开并将全部内容读入内存（请注意，效率不高）

（4） 使用正则表达式查找ldif后面的“下一位”，并输入文本

印刷品：

dn: Aziz
cn: Aziz_09
KEY : VALUE
abc : def
foo : bar
qwop : flop

dn: Carl
cn: Carl_04
KEY : VALUE
abc : def
foo : bar
qwop : flop

您需要对其进行调整，以将数据写回或任何您想要的内容。。。，但这是一个可能的起点——但强烈建议您在Python手册的基础上使用它。请随时要求澄清。