mirror of
https://github.com/apache/cloudstack.git
synced 2025-10-26 08:42:29 +01:00
76 lines
2.4 KiB
Python
Executable File
76 lines
2.4 KiB
Python
Executable File
#!/usr/bin/python
|
|
# Licensed to the Apache Software Foundation (ASF) under one
|
|
# or more contributor license agreements. See the NOTICE file
|
|
# distributed with this work for additional information
|
|
# regarding copyright ownership. The ASF licenses this file
|
|
# to you under the Apache License, Version 2.0 (the
|
|
# "License"); you may not use this file except in compliance
|
|
# with the License. You may obtain a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing,
|
|
# software distributed under the License is distributed on an
|
|
# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
|
|
# KIND, either express or implied. See the License for the
|
|
# specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
import sys
|
|
import tempfile
|
|
import os.path
|
|
import os
|
|
|
|
kernel = None
|
|
initrd = None
|
|
copy_to = None
|
|
|
|
def cmd(cmdstr, err=True):
|
|
print cmdstr
|
|
if os.system(cmdstr) != 0 and err:
|
|
raise Exception("Failed to run shell command: %s" % cmdstr)
|
|
|
|
def prepare():
|
|
global kernel, initrd, copy_to
|
|
try:
|
|
k = os.path.join(copy_to, "vmlinuz")
|
|
i = os.path.join(copy_to, "initrd.img")
|
|
if os.path.exists(k) and os.path.exists(i):
|
|
print "Having template(%s) prepared already, skip copying" % copy_to
|
|
return 0
|
|
else:
|
|
if not os.path.exists(copy_to):
|
|
os.makedirs(copy_to)
|
|
|
|
|
|
def copy_from_nfs(src, dst):
|
|
mnt_path = tempfile.mkdtemp()
|
|
try:
|
|
nfs_path = os.path.dirname(src)
|
|
filename = os.path.basename(src)
|
|
t = os.path.join(mnt_path, filename)
|
|
mnt = "mount %s %s" % (nfs_path, mnt_path)
|
|
cmd(mnt)
|
|
cp = "cp -f %s %s" % (t, dst)
|
|
cmd(cp)
|
|
finally:
|
|
umnt = "umount %s" % mnt_path
|
|
cmd(umnt, False)
|
|
rm = "rm -r %s" % mnt_path
|
|
cmd(rm, False)
|
|
|
|
copy_from_nfs(kernel, copy_to)
|
|
copy_from_nfs(initrd, copy_to)
|
|
except Exception, e:
|
|
print e
|
|
return 1
|
|
|
|
if __name__ == "__main__":
|
|
if len(sys.argv) < 4:
|
|
print "Usage: prepare_kickstart_kerneal_initrd.py path_to_kernel path_to_initrd path_kernel_initrd_copy_to"
|
|
sys.exit(1)
|
|
|
|
(kernel, initrd, copy_to) = sys.argv[1:]
|
|
sys.exit(prepare())
|
|
|