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

import struct

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")

    parser.add_option("-d", "--decode",

                        action="store_true", dest="decode", default=False,

                        help="decode files under target and create manifest")

    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, opts.decode)

def main():

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

    files = json.load(input_file)

    if decode:

        manifest = open(target + '/manifest', 'w')

    count = 0

    for f in files:

        count += 1

        owner = f['owner'] if 'owner' in f else "root"

        group = f['group'] if 'group' in f else "root"

        mode = f['mode'] if 'mode' in f else 288 # 440 in oct = 288 in dec

        filepath = f['path'] if not decode else str(count)

        filepath = target + "/" + filepath

        if os.path.lexists(filepath):

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

            os.rename(filepath, backup_file)

        parentdir = os.path.dirname(filepath)

        if not os.path.exists(parentdir):

            os.makedirs(parentdir)

        newfile = open(filepath, 'w')

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

        newfile.close()

        if decode:

            manifest.write(str(count))

            manifest.write(struct.pack('B', 0))

            manifest.write(owner)

            manifest.write(struct.pack('B', 0))

            manifest.write(group)

            manifest.write(struct.pack('B', 0))

            manifest.write("%o" % mode)

            manifest.write(struct.pack('B', 0))

            manifest.write(f['path'])

            manifest.write(struct.pack('B', 0))

    sys.stderr.write('Files were injected successfully\n')

    if decode:

        manifest.close()

    input_file.close()

    return 0

if __name__ == "__main__":

    sys.exit(main())

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