Python读取写入获取文件简单例子

Python读取写入获取文件简单例子

# 对个获取指定目录的所有文件
def get_all_files(dir):
    fileList = []
    """遍历获取指定文件夹下面所有文件"""
    if os.path.isdir(dir):
        filelist = os.listdir(dir)
        for ret in filelist:
            filename = dir + "\\" + ret
            if os.path.isfile(filename):
                fileList.append(filename)
    return fileList


# 对个获取指定目录的所有文件
def get_file_list_by_walk(dir):
    fileList = []
    """使用listdir循环遍历"""
    if not os.path.isdir(dir):
        return fileList
    dirlist = os.walk(dir)
    for root, dirs, files in dirlist:
        for fi in files:
            fileList.append(os.path.join(root, fi))
    return fileList

#指定文件路径获取文件最后文件的路径包含文件
#如：D:\test\file.txt 返回的结果为：D:\test\
def get_file_path(path):
    #获取文件名
    #return os.path.split(path)[1] 
    # 获取文件路径
    return os.path.split(path)[0]

#指定文件路径获取文件最后文件的路径包含文件
#如：D:\test\file.txt 返回的结果为：file.txt
def get_file_name(path):
    return os.path.basename(path)
    
# 创建目录
def mkdir(path):
    # 去除尾部 \ 符号
    pathx = path.strip().rstrip("\\")
    # 判断路径是否存在
    # 存在     True
    # 不存在   False
    isExists = os.path.exists(pathx)
    # 判断结果
    if not isExists:
        # 如果不存在则创建目录创建目录操作函数
        os.makedirs(path)
        print(path + ' 创建成功')
        return True
    else:
        # 如果目录存在则不创建，并提示目录已存在
        print(path + ' 目录已存在')
        return False


# 创建一个txt文件,并向文件写入msg
"""
@file_dir参数 代表文件目录 如：D:\test
@file_name参数 代表文件名称 如：file.txt
@msg参数 代表要写入文件的内容信息
"""
def writer_to_file(file_dir, file_name, msg):
    # 先创建目录
    mkdir(file_dir)
    # 再打开文件
    full_path = file_dir + "\\" + file_name
    fi = open(full_path, 'w')
    # 写入文件
    fi.write(msg) 
    fi.close()

"""
通过指定文件路径文件进行读取内容
如：D:\test\file.txt
"""
def reader_file(path):
    #解决乱码问题
    fi = open(path,'r',encoding='gbk',errors = 'ignore')
    strs=[]
    #splitlines解决不换行\n输出
    for line in fi.read().splitlines():
        if(len(line)>0):
            strs.append(line)
    return strs



   
#测试__main__方法
if __name__ == '__main__':
    path = r"D:\test\test.txt"
    #读取内容
    fileStr = reader_file(path)
    
    # 最终统计的信息写入文件
    writer_to_file(path, "file.txt", fileStr)
    
    print('do finish')