Tools/scripts/linktree.py

   1 #! /usr/bin/env 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 \
  64                not os.path.islink(oldname):
  65                 try:
  66                     os.mkdir(newname, 0777)
  67                     ok = 1
  68                 except:
  69                     print newname + \
  70                           ': warning: cannot mkdir:', msg
  71                     ok = 0
  72                 if ok:
  73                     linkname = os.path.join(os.pardir,
  74                                             linkname)
  75                     linknames(oldname, newname, linkname)
  76             else:
  77                 os.symlink(linkname, newname)
  78
  79 if __name__ == '__main__':
  80     sys.exit(main())