python常用文件处理

#!python3
# fileUtil.py 文件处理工具类
# 包含功能：文件重命名、txt文件合并、获取文件列表

import os
import shutil


def changeName(path, oldname, newname):
    """
    批量变更文件名:包含子目录
    :param path: 文件路径
    :param oldname: 要替换的字符串
    :param newname: 替换后的字符串
    :return: 处理结果：-1:文件或目录不存在,数字：重命名文件和目录的总数
    """
    if not os.path.exists(path):
        return -1

    cnt = 0
    for folderName, subFolders, fileNames in os.walk(path):
        """
        注意：要先重命名文件再重命名目录才不会报错
        """
        # 第一步：重命名当前目录中的文件
        for filename in fileNames:
            if oldname in filename:
                #  如果包含要替换的字符串则重命名文件
                shutil.move(os.path.join(folderName, filename),
                            os.path.join(folderName, filename.replace(oldname, newname)))
                cnt += 1

        #  第二步：重命名当前目录
        if oldname in folderName:
            #  如果包含要替换的字符串则重命名目录
            shutil.move(folderName, folderName.replace(oldname, newname))
            cnt += 1
    return cnt


def mergeTxt(path, resfile):
    """
    合并目录下的所有txt文件到一个txt文件中
    copy *.txt 合并.txt
    :param path: 文件所在的目录
    :param resfile: 合并后的文件，不能与源文件同一个目录
    :return:
    """
    if not os.path.exists(path):
        return -1

    resfile = open(resfile, 'w')

    for folderName, subFolders, fileNames in os.walk(path):
        for filename in fileNames:
            if filename.endswith(".txt"):
                txtFile = open(os.path.join(folderName, filename))
                lines = txtFile.readlines()
                resfile.writelines(lines)
                print(os.path.join(folderName, filename))
    return 1


def getFileList(path):
    """
    获取当前目录及子目录下的文件列表
    :param path:
    :return:
    """
    lst = []
    for folderName, subfolders, filenames in os.walk(path):
        for filename in filenames:
            # print(folderName + "\\" + filename)
            lst.append(os.path.join(folderName, filename))
    return lst