Example shell script

example_shell_script.py

#!/usr/bin/env python

# The MIT License (MIT)
#
# Copyright (c) 2015 David Woodruff, Michael-Keith Bernard
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in all
# copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
# SOFTWARE.

import csv
import yaml
import shlex
import argparse
import subprocess

__author__ = "David Woodruff, Michael-Keith Bernard"

CMD = """echo mutt -s "{subject}" "{recipient}" """
SUBJECT = "Foobar is pretty great"
RECIPIENT = "[email protected]"

def load_file(filename):
    with open(filename) as f:
        reader = csv.DictReader(f)
        return list(reader)
    
def run_cmd(cmd, input=None):
    parts = shlex.split(cmd)
    proc = subprocess.Popen(parts, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
    return proc.communicate(input)

def format_line(line):
    return yaml.safe_dump(line, default_flow_style=False)

def main(args=None):
    parser = argparse.ArgumentParser(description="David's cool thing")
    parser.add_argument("csvfile", help="Path to csv file")
    parser.add_argument("-s", "--subject", help="Subject for email", default=SUBJECT)
    parser.add_argument("-r", "--recipient", help="Recipient of email", default=RECIPIENT)
    
    args = parser.parse_args(args)
    
    lines = load_file(args.csvfile)
    print "Starting send..."
    for line in lines:
        cmd = CMD.format(subject=args.subject, recipient=args.recipient)
        stdin = format_line(line)
        print run_cmd(cmd, stdin)[0]
        

if __name__ == "__main__":
    main()