Python获取文件路径的父级目录、文件名、扩展名

import os


def split_file_path(text_file, file_type):
    '''
    文件路径拆分
    :param text_file:
    :return:
    '''
    try:
        res_text = None         
        
        # 提取父目录
        if file_type == 'dir':
            res_text = os.path.dirname(text_file)
            
        # 提取完整文件名
        elif file_type == 'file_ext':
            res_text = os.path.basename(text_file)
        
        # 提取文件名
        elif file_type == 'file':
            file_full_name = os.path.basename(text_file)
            res_text = os.path.splitext(file_full_name)[0]
        
        # 提取扩展名
        elif file_type == 'ext':
            file_full_name = os.path.basename(text_file)
            res_text = os.path.splitext(file_full_name)[1][1:]
        
        return res_text if res_text else None
    
    except Exception as e:
        print(e)
        return None



if __name__ == '__main__':

    text_file = r'F:\text\gold.txt'

    # 提取父目录
    print(split_file_path(text_file, 'dir'))
    # F:\text

    # 提取完整文件名
    print(split_file_path(text_file, 'file_ext'))
    # gold.txt

    # 提取文件名
    print(split_file_path(text_file, 'file'))
    # gold

    # 提取扩展名
    print(split_file_path(text_file, 'ext'))
    # txt


猜你喜欢

转载自blog.csdn.net/weixin_43721000/article/details/131411328