blob: bc4dd1300a1231d2682a1e81d057854e106f8509 [file] [log] [blame]
Guido van Rossumd5c57e11993-04-01 20:46:40 +00001#! /usr/local/bin/python
Guido van Rossumec758ea1991-06-04 20:36:54 +00002
3# mkreal
4#
5# turn a symlink to a directory into a real directory
6
7import sys
Guido van Rossume7b88e71992-03-30 11:13:59 +00008import os
Guido van Rossumec758ea1991-06-04 20:36:54 +00009from stat import *
10
Guido van Rossume7b88e71992-03-30 11:13:59 +000011join = os.path.join
Guido van Rossumec758ea1991-06-04 20:36:54 +000012
13error = 'mkreal error'
14
15BUFSIZE = 32*1024
16
17def mkrealfile(name):
Guido van Rossume7b88e71992-03-30 11:13:59 +000018 st = os.stat(name) # Get the mode
Guido van Rossumec758ea1991-06-04 20:36:54 +000019 mode = S_IMODE(st[ST_MODE])
Guido van Rossume7b88e71992-03-30 11:13:59 +000020 linkto = os.readlink(name) # Make sure again it's a symlink
Guido van Rossumec758ea1991-06-04 20:36:54 +000021 f_in = open(name, 'r') # This ensures it's a file
Guido van Rossume7b88e71992-03-30 11:13:59 +000022 os.unlink(name)
Guido van Rossumec758ea1991-06-04 20:36:54 +000023 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
Guido van Rossume7b88e71992-03-30 11:13:59 +000029 os.chmod(name, mode)
Guido van Rossumec758ea1991-06-04 20:36:54 +000030
31def mkrealdir(name):
Guido van Rossume7b88e71992-03-30 11:13:59 +000032 st = os.stat(name) # Get the mode
Guido van Rossumec758ea1991-06-04 20:36:54 +000033 mode = S_IMODE(st[ST_MODE])
Guido van Rossume7b88e71992-03-30 11:13:59 +000034 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)
Guido van Rossumec758ea1991-06-04 20:36:54 +000040 #
41 for file in files:
Guido van Rossume7b88e71992-03-30 11:13:59 +000042 if file not in (os.curdir, os.pardir):
43 os.symlink(join(linkto, file), join(name, file))
Guido van Rossumec758ea1991-06-04 20:36:54 +000044
45def main():
46 sys.stdout = sys.stderr
Guido van Rossume7b88e71992-03-30 11:13:59 +000047 progname = os.path.basename(sys.argv[0])
Guido van Rossumd5c57e11993-04-01 20:46:40 +000048 if progname == '-c': progname = 'mkreal'
Guido van Rossumec758ea1991-06-04 20:36:54 +000049 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:
Guido van Rossume7b88e71992-03-30 11:13:59 +000055 if not os.path.islink(name):
Guido van Rossumec758ea1991-06-04 20:36:54 +000056 print progname+':', name+':', 'not a symlink'
57 status = 1
58 else:
Guido van Rossume7b88e71992-03-30 11:13:59 +000059 if os.path.isdir(name):
Guido van Rossumec758ea1991-06-04 20:36:54 +000060 mkrealdir(name)
61 else:
62 mkrealfile(name)
63 sys.exit(status)
64
65main()