Python学习笔记6——练习mkreal

关于软链接的几点注意：
1、该程序区别软链接和硬链接，针对的是软链接
2、http://blog.csdn.net/jackbai1990/article/details/7432577 关于软链接中的相对路径与绝对路径，例如，ln -s /home/jack  /home/john. 这个软链接中链接的是绝对地址
3、文件软链接与文件夹软链接的区别对待：文件软链接则创建文件，文件夹软链接则创建文件夹，新文件夹中每个文件都是到target的一个软链接.

#!   /usr/bin/env python3
#    -*- coding:utf8 -*-
#    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

/*
*  1、保存链接文件的mode值
*  2、先打开链接文件，然后断开链接，再打开name文件
*  3、进行文件A-->B的复制，最后del关闭文件时进行flush动作.
*  4、修改文件mode属性为先前保存的值
*/
def mkrealfile(name):
    st = os.stat(name)                         #Get the mode，这也是链接文件的mode
    mode = S_IMODE(st[ST_MODE])
    linkto = os.readlink(name)             # Make sure again it's a symlink, linkto 是一个字符串，相对路径。至少做实验的时候是这样的.
    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)

/*
* 1、保存文件夹的mode属性
* 2、保存链接文件夹下所有文件的名称，断开链接，创建新的文件夹
* 3、为链接文件夹下的每一个文件，创建一个链接
*/
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) #是不是有点重复啊?
    print('linkto : ', linkto)
    linkto = join(os.pardir, linkto)       #竟然是..，首先要注意join的参数中存在绝对地址的情况，其次要分析如何重组相对地址到链接文件夹的。
    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])      #跟basename命令的效果一样
    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()