#! /usr/bin/env python3 | |
# mkreal | |
# | |
# turn a symlink to a directory into a real directory | |
import sys | |
import os | |
from stat import * | |
join = os.path.join | |
error = 'mkreal error' | |
BUFSIZE = 32*1024 | |
def mkrealfile(name): | |
st = os.stat(name) # Get the mode | |
mode = S_IMODE(st[ST_MODE]) | |
linkto = os.readlink(name) # Make sure again it's a symlink | |
with open(name, 'rb') as f_in: # This ensures it's a file | |
os.unlink(name) | |
with open(name, 'wb') as f_out: | |
while 1: | |
buf = f_in.read(BUFSIZE) | |
if not buf: break | |
f_out.write(buf) | |
os.chmod(name, mode) | |
def mkrealdir(name): | |
st = os.stat(name) # Get the mode | |
mode = S_IMODE(st[ST_MODE]) | |
linkto = os.readlink(name) | |
files = os.listdir(name) | |
os.unlink(name) | |
os.mkdir(name, mode) | |
os.chmod(name, mode) | |
linkto = join(os.pardir, linkto) | |
# | |
for filename in files: | |
if filename not in (os.curdir, os.pardir): | |
os.symlink(join(linkto, filename), join(name, filename)) | |
def main(): | |
sys.stdout = sys.stderr | |
progname = os.path.basename(sys.argv[0]) | |
if progname == '-c': progname = 'mkreal' | |
args = sys.argv[1:] | |
if not args: | |
print('usage:', progname, 'path ...') | |
sys.exit(2) | |
status = 0 | |
for name in args: | |
if not os.path.islink(name): | |
print(progname+':', name+':', 'not a symlink') | |
status = 1 | |
else: | |
if os.path.isdir(name): | |
mkrealdir(name) | |
else: | |
mkrealfile(name) | |
sys.exit(status) | |
if __name__ == '__main__': | |
main() | |