| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566 |
- #! /usr/bin/env python3
- # mkreal
- #
- # turn a symlink to a directory into a real directory
- import sys
- import os
- from stat import *
- join = os.path.join
- error = 'mkreal error'
- BUFSIZE = 32*1024
- def mkrealfile(name):
- st = os.stat(name) # Get the mode
- mode = S_IMODE(st[ST_MODE])
- linkto = os.readlink(name) # Make sure again it's a symlink
- f_in = open(name, 'r') # This ensures it's a file
- os.unlink(name)
- f_out = open(name, 'w')
- while 1:
- buf = f_in.read(BUFSIZE)
- if not buf: break
- f_out.write(buf)
- del f_out # Flush data to disk before changing mode
- os.chmod(name, mode)
- def mkrealdir(name):
- st = os.stat(name) # Get the mode
- mode = S_IMODE(st[ST_MODE])
- linkto = os.readlink(name)
- files = os.listdir(name)
- os.unlink(name)
- os.mkdir(name, mode)
- os.chmod(name, mode)
- linkto = join(os.pardir, linkto)
- #
- for filename in files:
- if filename not in (os.curdir, os.pardir):
- os.symlink(join(linkto, filename), join(name, filename))
- def main():
- sys.stdout = sys.stderr
- progname = os.path.basename(sys.argv[0])
- if progname == '-c': progname = 'mkreal'
- args = sys.argv[1:]
- if not args:
- print('usage:', progname, 'path ...')
- sys.exit(2)
- status = 0
- for name in args:
- if not os.path.islink(name):
- print(progname+':', name+':', 'not a symlink')
- status = 1
- else:
- if os.path.isdir(name):
- mkrealdir(name)
- else:
- mkrealfile(name)
- sys.exit(status)
- if __name__ == '__main__':
- main()
|