Synnefo » snf-image

#!/usr/bin/env python

"""Inject files into a directory

This program injects files into a target directory.

The files are passed to the program through a JSON string either read from a

file or from standard input.

"""

import sys

import os

import json

import datetime

import base64

from optparse import OptionParser

def timestamp():

    now = datetime.datetime.now()

    current_time = now.strftime("%Y%m%d.%H%M%S")

    return current_time

def parse_arguments(input_args):

    usage = "Usage: %prog [options] <target>"

    parser = OptionParser(usage=usage)

    parser.add_option("-i", "--input",

                        action="store",type='string', dest="input_file",

                        help="get input from FILE instead of stdin",

                        metavar="FILE")

    opts, args = parser.parse_args(input_args)

    if len(args) != 1:

        parser.error('target is missing')

    target = args[0]

    if not os.path.isdir(target):

        parser.error('target is not a directory')

    input_file = opts.input_file

    if input_file is None:

        input_file = sys.stdin

    else:

        if not os.path.isfile(input_file):

            parser.error('input file does not exist')

        input_file = open(input_file,'r')

    return (input_file, target)

def main():

    (input_file, target) = parse_arguments(sys.argv[1:])

    files = json.load(input_file)

    for f in files:

        real_path = target + '/' + f['path']

        if os.path.lexists(real_path):

            backup_file = real_path + '.bak.' + timestamp()

            os.rename(real_path, backup_file)

        parentdir = os.path.dirname(real_path)

        if not os.path.exists(parentdir):

            os.makedirs(parentdir)

        newfile = open(real_path, 'w')

        newfile.write(base64.b64decode(f['contents']))

        newfile.close()

        os.chmod(real_path, 0440)

    sys.stderr.write('Successful personalization of Image\n')

    input_file.close()

    return 0

if __name__ == "__main__":

    sys.exit(main())

# vim: set sta sts=4 shiftwidth=4 sw=4 et ai :