Zimbra zmprov отформатировал файл в csv и ldif - PullRequest
Новая
       85

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

0 голосов
/ 19 февраля 2019

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

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

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

Это входные данные zmp_file,для преобразования в csv и ldif 

ca user1@domain.com.br      ''
ma user1@domain.com.br cn   'User One'
ma user1@domain.com.br cpf  ''
ma user1@domain.com.br l    'Porto Alegre'

ca user2@domain.com.br      ''
ma user2@domain.com.br cn   'User Two'
ma user2@domain.com.br cpf  '0123456789'
ma user2@domain.com.br l    ''

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

mail,cn,cpf,l
user1@domain.com.br,"User One",,"Porto Alegre"
user2@domain.com.br,"User Two",0123456789,

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

dn:   'uid=user1@domain.com.br'
cn:   'User One'
l:    'Porto Alegre'
mail: 'user1@domain.com.br'

dn:   'uid=user2@domain.com.br'
cn:   'User Two'
cpf:  '0123456789'
mail: 'user2@domain.com.br'

Как далеко я мог бы пройти: 

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

Ответы [ 2 ]

0 голосов
/ 19 февраля 2019

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

Я знаю, что это не 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)
0 голосов
/ 19 февраля 2019

Вы можете преобразовать 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
user1@domain.com.br,'User One','','Porto Alegre'
user2@domain.com.br,'User Two','0123456789',''

Выводfile.ldif: 

dn:   'uid=user1@domain.com.br'
cn:   'User One'
l:    'Porto Alegre'
mail: 'user1@domain.com.br'

dn:   'uid=user2@domain.com.br'
cn:   'User Two'
cpf:  '0123456789'
mail: 'user2@domain.com.br'
...