python文件操作常用方法整理

一、删除文件夹下的所有文件

导入os库

通过os.listdir方法获取给定目录下的所有文件名，使用os.path.join方法循环拼接文件名和目录名，通过os.unlink方法删除指定的文件

import os
def deldirfile(download_dir):
    for filename in os.listdir(download_dir):
        source_file_path = os.path.join(download_dir, filename)
        os.unlink(source_file_path)

二、判断目录是否存在，不存在的创建目录

和上面的方法一样，导入os库，判断目录是否存在

通过os.mkdir方法创建目录

if not os.path.isdir(target_directory):
	os.mkdir(target_directory)

三、判断文件名是否是指定的扩展名

通过os.endswith()方法判断是否以指定的文件后缀结束，这里判断的是否为.docx文件，也就是word文件。

if filename.endswith(".docx"):
	target_file_path = os.path.join(target_directory, filename)

四、判断文件是否哦存在，存在的化删除文件

os.isfile()方法用来判断给定的文件路径是否存在

os.remove()删除给定的文件

if os.path.isfile(target_file_path):
	os.remove(target_file_path)

注：os库中remove和unlink都可以删除文件，他们之间有什么区别呢？

1、remove可以删除文件和目录，unlink只能删除文件

2、unlink可以删除快捷方式背后的文件，remove只删除快捷方式本身，不删除指向的文件。

五、移动文件到新目录中

这里使用的shutil库

move方法需要两个参数，需要移动的文件路径和目标路径。使用之前需要确定文件没有被打开或被占用。

import shutil

shutil.move(source_file_path, target_file_path)

六、生成压缩文件

导入zipfile,os两个库

zipfile库主要用来对zip压缩文件进行操作，包括读取zip包、创建压缩文件、追加文件等。都是通过zipfile.ZipFile()方法进行，第一个参数是要操作的压缩包文件名，第二个参数是操作类型，w为创建、r为读取、a是追加。zipfile.ZIP_DEFLATED表示压缩文件使用的压缩算法是DEFLATE。

in os.walk()方法用于获取文件的三元信息，包括指定目录的目录地址、目录下的子目录列表、目录下的文件列表。

os.path.relpath()用来获取相对路径，为了使压缩包中的文件路径和实际目录路径保持一致。第一个参数是用来对比的相对目录，第二个参数默认为当前目录，如果配置则指定目录和目标目录的相对路径。

zipf.write(file_path, relative_path)用来将一个文件添加到zip包中，第一个参数是要添加的文件，第二个是添加到zip包中的位置。

import zipfile
import os
def zip_directory(directory_path, zip_filename):
    # 创建一个 ZIP 文件
    with zipfile.ZipFile(zip_filename, 'w', zipfile.ZIP_DEFLATED) as zipf:
        # 遍历目录中的文件和子目录
        for root, dirs, files in os.walk(directory_path):
            for file in files:
                file_path = os.path.join(root, file)
                # 计算相对路径以在 ZIP 文件中保持目录结构
                relative_path = os.path.relpath(file_path, directory_path)
                zipf.write(file_path, relative_path)

七、获取脚本当前目录

os模块下的getcwd方法，用来获取脚本工作的当前文件夹路径。cwd 是 Current Working Directory 的首字母。

import os  
  
# 获取当前工作目录  
current_directory = os.getcwd()  
print("当前工作目录是:", current_directory)