bootstrap-vz/bootstrapvz/plugins/file_copy/tasks.py

from bootstrapvz.base import Task
from bootstrapvz.common import phases

import os
import shutil


def modify_path(info, path, entry):
    from bootstrapvz.common.tools import log_check_call
    if 'permissions' in entry:
        # We wrap the permissions string in str() in case
        #  the user specified a numeric bitmask
        chmod_command = ['chroot', info.root, 'chmod', str(entry['permissions']), path]
        log_check_call(chmod_command)

    if 'owner' in entry:
        chown_command = ['chroot', info.root, 'chown', entry['owner'], path]
        log_check_call(chown_command)

    if 'group' in entry:
        chgrp_command = ['chroot', info.root, 'chgrp', entry['group'], path]
        log_check_call(chgrp_command)


class MkdirCommand(Task):
    description = 'Creating directories requested by user'
    phase = phases.user_modification

    @classmethod
    def run(cls, info):
        from bootstrapvz.common.tools import log_check_call

        for dir_entry in info.manifest.plugins['file_copy']['mkdirs']:
            mkdir_command = ['chroot', info.root, 'mkdir', '-p', dir_entry['dir']]
            log_check_call(mkdir_command)
            modify_path(info, dir_entry['dir'], dir_entry)


class FileCopyCommand(Task):
    description = 'Copying user specified files into the image'
    phase = phases.user_modification
    predecessors = [MkdirCommand]

    @classmethod
    def run(cls, info):
        for file_entry in info.manifest.plugins['file_copy']['files']:
            # note that we don't use os.path.join because it can't
            #  handle absolute paths, which 'dst' most likely is.
            final_destination = os.path.normpath("%s/%s" % (info.root, file_entry['dst']))
            shutil.copy(file_entry['src'], final_destination)
            modify_path(info, file_entry['dst'], file_entry)