home *** CD-ROM | disk | FTP | other *** search
/ OS/2 Shareware BBS: 10 Tools / 10-Tools.zip / pyth_os2.zip / python-1.0.2 / Demo / scripts / linktree.py < prev    next >
Text File  |  1992-10-02  |  2KB  |  77 lines
  1. #! /usr/local/bin/python
  2.  
  3. # linktree
  4. #
  5. # Make a copy of a directory tree with symbolic links to all files in the
  6. # original tree.
  7. # All symbolic links go to a special symbolic link at the top, so you
  8. # can easily fix things if the original source tree moves.
  9. # See also "mkreal".
  10. #
  11. # usage: mklinks oldtree newtree
  12.  
  13. import sys, os
  14.  
  15. LINK = '.LINK' # Name of special symlink at the top.
  16.  
  17. debug = 0
  18.  
  19. def main():
  20.     if not 3 <= len(sys.argv) <= 4:
  21.         print 'usage:', sys.argv[0], 'oldtree newtree [linkto]'
  22.         return 2
  23.     oldtree, newtree = sys.argv[1], sys.argv[2]
  24.     if len(sys.argv) > 3:
  25.         link = sys.argv[3]
  26.         link_may_fail = 1
  27.     else:
  28.         link = LINK
  29.         link_may_fail = 0
  30.     if not os.path.isdir(oldtree):
  31.         print oldtree + ': not a directory'
  32.         return 1
  33.     try:
  34.         os.mkdir(newtree, 0777)
  35.     except os.error, msg:
  36.         print newtree + ': cannot mkdir:', msg
  37.         return 1
  38.     linkname = os.path.join(newtree, link)
  39.     try:
  40.         os.symlink(os.path.join(os.pardir, oldtree), linkname)
  41.     except os.error, msg:
  42.         if not link_may_fail:
  43.             print linkname + ': cannot symlink:', msg
  44.             return 1
  45.         else:
  46.             print linkname + ': warning: cannot symlink:', msg
  47.     linknames(oldtree, newtree, link)
  48.     return 0
  49.  
  50. def linknames(old, new, link):
  51.     if debug: print 'linknames', (old, new, link)
  52.     try:
  53.         names = os.listdir(old)
  54.     except os.error, msg:
  55.         print old + ': warning: cannot listdir:', msg
  56.         return
  57.     for name in names:
  58.         if name not in (os.curdir, os.pardir):
  59.         oldname = os.path.join(old, name)
  60.         linkname = os.path.join(link, name)
  61.         newname = os.path.join(new, name)
  62.         if debug > 1: print oldname, newname, linkname
  63.         if os.path.isdir(oldname) and not os.path.islink(oldname):
  64.             try:
  65.                 os.mkdir(newname, 0777)
  66.                 ok = 1
  67.             except:
  68.                 print newname + ': warning: cannot mkdir:', msg
  69.                 ok = 0
  70.             if ok:
  71.                 linkname = os.path.join(os.pardir, linkname)
  72.                 linknames(oldname, newname, linkname)
  73.         else:
  74.             os.symlink(linkname, newname)
  75.  
  76. sys.exit(main())
  77.