#!/usr/bin/python2

# This script inserts the mailman user name into the crontab entries after

# the 5 time/date fields so that it can be installed under /etc/cron.d

#

# usage: mailman-crontab-edit [-s src_file] [-u mailman_user] [-d dst_file]

# src_file defaults to stdin

# mailman_user defaults to mailman

# dst_file defaults to stdout

import sys, re, getopt

srcFile = None

dstFile = None

mmUser = None

opts, args = getopt.getopt(sys.argv[1:], "s:d:u:")

for o, a in opts:

    if o == "-s":

        srcFile = a

    if o == "-d":

        dstFile = a

    if o == "-u":

        mmUser = a

if srcFile:

    inFD = open(srcFile)

else:

    inFD = sys.stdin

if dstFile:

    outFD = open(dstFile, mode='w')

else:

    outFD = sys.stdout

if not mmUser:

    mmUser = "mailman"

comment_re = re.compile(r'^\s*#')

time_date_re = re.compile(r'(^\s*(\S+\s+){5,5})')

for line in inFD:

    if not comment_re.search(line):

        match = time_date_re.search(line)

        if match:

            line = time_date_re.sub(r'\1 %s ' % mmUser, line)

    print >>outFD, line,