python 操作文件目录

# !/usr/bin/python
# -*- coding:UTF-8 -*-

import os
import sys

special_file = ('init-shell.sh', 'report-files.xml', 'report-detail.json', '.report_new.html', 'report-files-type.json',
                'success.check')
special_dir = 'Personal_Analysis'
need_dir = ('Web_Report', 'Needed_Data')


def create_link(source_path, target_path):
    try:
        os.link(source_path, target_path)
    except IOError:
        print "link file is exception", source_path
        sys.exit(1)


def make_dir(target_path):
    try:
        if os.path.exists(target_path):
            return
        os.makedirs(target_path)
    except IOError:
        print "make dir is exception", target_path
        sys.exit(1)


def deal_target_path(source_path, target_path):
    if os.path.isdir(source_path):
        make_dir(target_path)
        list_file = os.listdir(source_path)
        for file_name in list_file:
            if file_name != special_dir:
                new_target_path = target_path + '/' + file_name
                deal_target_path(source_path + '/' + file_name, new_target_path.replace("//", "/"))
    else:
        file_name = os.path.basename(source_path)
        if file_name not in special_file:
            create_link(source_path, target_path)


'''
sourcePath: 原始路径
targetPath: 目标路径
pathType: APP（项目路径） or DATA（其他路径）

 命令使用格式： python LinkAppPath.py sourcePath targetPath type
'''
if __name__ == '__main__':
    args = sys.argv
    if len(args) != 4:
        print '''exception source or target file path is empty'''
        sys.exit(1)

    sourcePath = args[1]
    targetPath = args[2]
    pathType = args[3]

    if not os.path.exists(sourcePath):
        print sourcePath + ''' exception source_path is not exists'''
        sys.exit(1)

    if pathType == "APP":
        Web_Report = sourcePath + "/" + need_dir[0]
        Needed_Data = sourcePath + "/" + need_dir[1]
        target_Web_Report = targetPath + "/" + need_dir[0]
        target_Needed_Data = targetPath + "/" + need_dir[1]
        deal_target_path(Web_Report, target_Web_Report)
        deal_target_path(Needed_Data, target_Needed_Data)
    else:
        if os.path.isfile(sourcePath):
            line = targetPath
            r = line.rfind("/")
            make_dir(line[0:r])
        else:
            make_dir(targetPath)
        deal_target_path(sourcePath, targetPath)
    sys.exit(0)