mirror of
https://github.com/kevingruesser/bootstrap-vz.git
synced 2025-08-22 09:50:37 +00:00
d34aeab6e0
File_copy supports creation of dirs and copying of files from the host system into the image. bug 168
65 lines
2 KiB
Python
65 lines
2 KiB
Python
from bootstrapvz.base import Task
|
|
from bootstrapvz.common import phases
|
|
|
|
import os
|
|
import shutil
|
|
|
|
|
|
class ValidateSourcePaths(Task):
|
|
description = 'Check whether the files to be copied exist'
|
|
phase = phases.preparation
|
|
|
|
@classmethod
|
|
def run(cls, info):
|
|
from bootstrapvz.common.exceptions import TaskError
|
|
for file_entry in info.manifest.plugins['file_copy']['files']:
|
|
srcfile = file_entry['src']
|
|
if not os.path.isfile(srcfile):
|
|
msg = 'The source file %s does not exist.' % srcfile
|
|
raise TaskError(msg)
|
|
|
|
|
|
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 = 'copy files into the image'
|
|
phase = phases.system_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 = 'copy files into the image'
|
|
phase = phases.system_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)
|