109 lines
3.1 KiB
Python
109 lines
3.1 KiB
Python
import shutil
|
|
|
|
def join(*paths):
|
|
"""Like os.path.join but doesn't treat absolute RHS specially"""
|
|
import os.path
|
|
return os.path.normpath("/".join(paths))
|
|
|
|
def relative(src, dest):
|
|
""" Return a relative path from src to dest.
|
|
|
|
>>> relative("/usr/bin", "/tmp/foo/bar")
|
|
../../tmp/foo/bar
|
|
|
|
>>> relative("/usr/bin", "/usr/lib")
|
|
../lib
|
|
|
|
>>> relative("/tmp", "/tmp/foo/bar")
|
|
foo/bar
|
|
"""
|
|
import os.path
|
|
|
|
if hasattr(os.path, "relpath"):
|
|
return os.path.relpath(dest, src)
|
|
else:
|
|
destlist = os.path.normpath(dest).split(os.path.sep)
|
|
srclist = os.path.normpath(src).split(os.path.sep)
|
|
|
|
# Find common section of the path
|
|
common = os.path.commonprefix([destlist, srclist])
|
|
commonlen = len(common)
|
|
|
|
# Climb back to the point where they differentiate
|
|
relpath = [ os.path.pardir ] * (len(srclist) - commonlen)
|
|
if commonlen < len(destlist):
|
|
# Add remaining portion
|
|
relpath += destlist[commonlen:]
|
|
|
|
return os.path.sep.join(relpath)
|
|
|
|
def format_display(path, metadata):
|
|
""" Prepare a path for display to the user. """
|
|
rel = relative(metadata.getVar("TOPDIR", 1), path)
|
|
if len(rel) > len(path):
|
|
return path
|
|
else:
|
|
return rel
|
|
|
|
|
|
class Error(EnvironmentError):
|
|
pass
|
|
|
|
# Based on shutil.copytree but with features removed and
|
|
# No fatal error is dst already exists
|
|
# Handle symlinks that already exist
|
|
def copytree(src, dst):
|
|
names = os.listdir(src)
|
|
|
|
bb.mkdirhier(dst)
|
|
|
|
errors = []
|
|
for name in names:
|
|
srcname = os.path.join(src, name)
|
|
dstname = os.path.join(dst, name)
|
|
try:
|
|
if os.path.islink(srcname):
|
|
linkto = os.readlink(srcname)
|
|
if os.path.lexists(dstname):
|
|
os.unlink(dstname)
|
|
os.symlink(linkto, dstname)
|
|
elif os.path.isdir(srcname):
|
|
copytree(srcname, dstname)
|
|
else:
|
|
bb.utils.copyfile(srcname, dstname)
|
|
except (IOError, os.error), why:
|
|
errors.append((srcname, dstname, str(why)))
|
|
# catch the Error from the recursive copytree so that we can
|
|
# continue with other files
|
|
except Error, err:
|
|
errors.extend(err.args[0])
|
|
try:
|
|
shutil.copystat(src, dst)
|
|
except OSError, why:
|
|
errors.extend((src, dst, str(why)))
|
|
if errors:
|
|
raise Error, errors
|
|
|
|
def remove(path):
|
|
"""Equivalent to rm -f or rm -rf"""
|
|
import os, errno, shutil, glob
|
|
for name in glob.glob(path):
|
|
try:
|
|
os.unlink(name)
|
|
except OSError, exc:
|
|
if exc.errno == errno.EISDIR:
|
|
shutil.rmtree(path)
|
|
elif exc.errno != errno.ENOENT:
|
|
raise
|
|
|
|
def symlink(source, destination, force=False):
|
|
"""Create a symbolic link"""
|
|
import os, errno
|
|
try:
|
|
if force:
|
|
remove(destination)
|
|
os.symlink(source, destination)
|
|
except OSError, e:
|
|
if e.errno != errno.EEXIST or os.readlink(destination) != source:
|
|
raise
|