Zimbra zmprov отформатировал файл в csv и ldif

Question

Я изучаю Python, и мое первое задание - преобразовать Zimbra zmprov отформатированный файл в csv а также ldif,

Так как я не знаю встроенных функций Python для выполнения этой задачи, я беру долгий путь и перебираю строки и печать.

Я был бы очень признателен, если бы вы, ребята, смогли показать мне, как это сделать правильно.

Это входной файл zmp_file, который нужно преобразовать в csv и ldif

ca [e-mail]      ''
ma [e-mail] cn   'User One'
ma [e-mail] cpf  ''
ma [e-mail] l    'Porto Alegre'

ca [e-mail]      ''
ma [e-mail] cn   'User Two'
ma [e-mail] cpf  '0123456789'
ma [e-mail] l    ''

Желаемый вывод.csv (порядок полей не важен)

mail,cn,cpf,l
[e-mail],"User One",,"Porto Alegre"
[e-mail],"User Two",0123456789,

И желаемый вывод.ldif (порядок полей не важен)

dn:   'uid=[e-mail]'
cn:   'User One'
l:    'Porto Alegre'
mail: '[e-mail]'

dn:   'uid=[e-mail]'
cn:   'User Two'
cpf:  '0123456789'
mail: '[e-mail]'

Как далеко я мог бы получить:

with zmp_file as input_file
    for line in input_file:
        if line.startswith('ca'):
            mail = line.split()[1]
            print "dn: uid={0}".format(mail)
            print "mail: {0}".format(mail)
        elif line.startswith('ma'):
            words = shlex.split(line)[-2:]
            print "{0}: {1}".format(words[0], words[1])
        else:
            print

Answer 1

Хорошо. Понял.

Я знаю, что это не codereview.stackexchange.com, но если у кого-то есть комментарии, я здесь, чтобы узнать.

#!/usr/bin/env python

import csv
import os
import shlex
import sys
from ldif import LDIFParser, LDIFWriter

def zmp_to_csv_and_ldif(zmp_file):

    all_attrs = set()
    data      = {}
    records   = {}

    with zmp_file as input_file:
        for line in input_file:
            if line.startswith('ca'):
                cmd, mail, pwd       = line.split()
                data['mail']         = mail
                data['userpassword'] = pwd
                records[mail]        = data
                all_attrs.update(['mail','userpassword'])
            elif line.startswith('ma'):
                cmd, mail, attr, value = shlex.split(line)
                data[attr]             = value
                records[mail]          = data
                all_attrs.add(attr)
            else:
                data = {}

    with open('/tmp/rag-parsed.csv', 'w') as output_file:
        csv_writer = csv.DictWriter(output_file, fieldnames=all_attrs, extrasaction='ignore', lineterminator='\n')
        csv_writer.writeheader()
        for mail, data in sorted(records.items()):
            csv_writer.writerow(data)

    with open('/tmp/rag-parsed.ldif', 'w') as output_file:
        b64_attrs   = map(str.lower, ['jpegPhoto', 'userPassword'])
        ldif_writer = LDIFWriter(output_file, base64_attrs=b64_attrs, cols=999)
        for mail, data in sorted(records.items()):
            dn = "uid={0}".format(mail)
            data_in_ldap_fmt = dict([k, v.split('\n')] for k, v in data.items() if v)
            ldif_writer.unparse(dn, data_in_ldap_fmt)

Answer 2

Вы можете конвертировать zmprov в csv а также ldif с помощью regex а также pandas:

import re
import pandas as pd

with open('file.txt', 'r') as f:
    data = f.read()

mail = re.findall('ca (\S+)', data)
cn = re.findall('cn +(.*)', data)
cpf = re.findall('cpf + (.*)', data)
l = re.findall('l + (.*)', data)

df = pd.DataFrame({'mail': mail, 'cn': cn, 'cpf': cpf, 'l': l})
df.to_csv('file.csv', sep=',', encoding='utf-8', index=False)

with open('file.ldif', 'w') as f:
    for m_, cn_, cpf_, l_ in zip(mail, cn, cpf, l):
        if re.findall('\w+', l_):
            f.write("""dn:   'uid={m}'\ncn:   {cn}\nl:    {l}\nmail: '{m}'\n\n""".format(m=m_, cn=cn_, l=l_))
        elif cpf_:
            f.write("""dn:   'uid={m}'\ncn:   {cn}\ncpf:  {cpf}\nmail: '{m}'\n\n""".format(m=m_, cn=cn_, cpf=cpf_))

Выход file.csv:

mail,cn,cpf,l
[e-mail],'User One','','Porto Alegre'
[e-mail],'User Two','0123456789',''

Выход file.ldif:

dn:   'uid=[e-mail]'
cn:   'User One'
l:    'Porto Alegre'
mail: '[e-mail]'

dn:   'uid=[e-mail]'
cn:   'User Two'
cpf:  '0123456789'
mail: '[e-mail]'