mirror of
https://github.com/python/cpython.git
synced 2024-11-25 02:44:06 +08:00
66 lines
1.6 KiB
Python
Executable File
66 lines
1.6 KiB
Python
Executable File
#! /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
|
|
with open(name, 'rb') as f_in: # This ensures it's a file
|
|
os.unlink(name)
|
|
with open(name, 'wb') as f_out:
|
|
while 1:
|
|
buf = f_in.read(BUFSIZE)
|
|
if not buf: break
|
|
f_out.write(buf)
|
|
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()
|