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