port25 PowerMTA bounce accounting with send to MongoDB

bounce_acct.conf

[DEFAULT]

[headers]
headers = type,timeLogged,timeQueued,orig,rcpt,orcpt,dsnAction,dsnStatus,dsnDiag,dsnMta,bounceCat,srcType,srcMta,dlvType,dlvSourceIp,dlvDestinationIp,dlvEsmtpAvailable,dlvSize,vmta,jobId,envId,queue,vmtaPool

[mongo]
host = 127.0.0.1
port = 27017
queueName = pmta-events

bounce_acct.py

import sys
import datetime
import configparser
import pymongo
import csv
import io

config = configparser.ConfigParser()
config.read("./bounce.conf")

# read accounting headers
headers = config.get("headers", "headers")
keys = ("_t," + headers).split(",")

# read mongo config
mongoSection = config["mongo"]
host = mongoSection.get("host")
port = mongoSection.getint("port")
queueName = mongoSection.get("queueName")

# connect to mongo
try:
  c = pymongo.MongoClient(host, port)
except ConnectionFailure, e:
  sys.stderr.write("Could not connect to MongoDB: %s" % e)
  sys.exit(1)

db = c.Scheduler
queue = db.Queue

# read first line - unused
line = sys.stdin.readline()

while line:
  line = sys.stdin.readline()
  if line:
    # write accounting data to queue
    values = list(csv.reader(io.StringIO('EmailStatusReceived,' + line)))[0]
    message = dict(zip(keys, values))
    queue.insert({
      "QueueName": queueName,
      "Message": message,
      "NextVisibleTime": datetime.datetime.utcnow(),
	    "DequeueCount": 0
    })