1 #! /usr/bin/env python 2 3 # mkreal 4 # 5 # turn a symlink to a directory into a real directory 6 7 import sys 8 import os 9 from stat import * 10 11 join = os.path.join 12 13 error = 'mkreal error' 14 15 BUFSIZE = 32*1024 16 17 def mkrealfile(name): 18 st = os.stat(name) # Get the mode 19 mode = S_IMODE(st[ST_MODE]) 20 linkto = os.readlink(name) # Make sure again it's a symlink 21 f_in = open(name, 'r') # This ensures it's a file 22 os.unlink(name) 23 f_out = open(name, 'w') 24 while 1: 25 buf = f_in.read(BUFSIZE) 26 if not buf: break 27 f_out.write(buf) 28 del f_out # Flush data to disk before changing mode 29 os.chmod(name, mode) 30 31 def mkrealdir(name): 32 st = os.stat(name) # Get the mode 33 mode = S_IMODE(st[ST_MODE]) 34 linkto = os.readlink(name) 35 files = os.listdir(name) 36 os.unlink(name) 37 os.mkdir(name, mode) 38 os.chmod(name, mode) 39 linkto = join(os.pardir, linkto) 40 # 41 for filename in files: 42 if filename not in (os.curdir, os.pardir): 43 os.symlink(join(linkto, filename), join(name, filename)) 44 45 def main(): 46 sys.stdout = sys.stderr 47 progname = os.path.basename(sys.argv[0]) 48 if progname == '-c': progname = 'mkreal' 49 args = sys.argv[1:] 50 if not args: 51 print 'usage:', progname, 'path ...' 52 sys.exit(2) 53 status = 0 54 for name in args: 55 if not os.path.islink(name): 56 print progname+':', name+':', 'not a symlink' 57 status = 1 58 else: 59 if os.path.isdir(name): 60 mkrealdir(name) 61 else: 62 mkrealfile(name) 63 sys.exit(status) 64 65 if __name__ == '__main__': 66 main() 67
