posta2csv.py

posta2csv.py

#!/usr/bin/env python3
#
# Convertitore delle mail del Raspberry Pi di Ugone
# Scritto da Frafra (2017)

import collections
import csv
import glob

messages = glob.glob('messaggi/*.txt')
output = 'output.csv'

keys = [
    "Data",
    "Humidity",
    "Pressure",
    "Temperature",
]

with open(output, 'w', newline='') as csvfile:
    csvwriter = csv.writer(csvfile)
    csvwriter.writerow(keys)
    for message in messages:
        with open(message) as mail:
            result = collections.defaultdict(str)
            for line in mail.readlines():
                while ': ' in line:
                    key, value = line.rsplit(': ', 1)
                    key, value = key.strip(), value.strip()
                    if ' ' in key:
                        line, key = key.rsplit(' ', 1)
                    else:
                        line = ''
                    result[key] = value
        csvwriter.writerow([result[key] for key in keys])