home *** CD-ROM | disk | FTP | other *** search
/ PC World 2001 April / PCWorld_2001-04_cd.bin / Software / TemaCD / webclean / !!!python!!! / BeOpen-Python-2.0.exe / MKREAL.PY < prev    next >
Encoding:
Python Source  |  2000-09-28  |  1.4 KB  |  66 lines
  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 file in files:
  42.         if file not in (os.curdir, os.pardir):
  43.             os.symlink(join(linkto, file), join(name, file))
  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. main()
  66.