blob: 502873eca2f5cc3a24f980949f5a0743ad0d1f5b [file] [log] [blame]
Guido van Rossumf06ee5f1996-11-27 19:52:01 +00001#! /usr/bin/env python
Guido van Rossum2ba9f301992-03-02 16:20:32 +00002
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
Guido van Rossume7b88e71992-03-30 11:13:59 +000013import sys, os
Guido van Rossum2ba9f301992-03-02 16:20:32 +000014
15LINK = '.LINK' # Name of special symlink at the top.
16
17debug = 0
18
19def main():
Tim Peters70c43782001-01-17 08:48:39 +000020 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
Guido van Rossum2ba9f301992-03-02 16:20:32 +000049
50def linknames(old, new, link):
Tim Peters70c43782001-01-17 08:48:39 +000051 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)
Guido van Rossum2ba9f301992-03-02 16:20:32 +000078
79sys.exit(main())