Tools/scripts/mkreal.py

   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()