暂无图片
暂无图片
暂无图片
暂无图片
暂无图片

快速整理文件——使用Python遍历文件

一只大鸽子 2022-11-24
507

案例

鸽子忘记自己的mp3文件存储在哪些地方了,想将它们都找出来并整理。
利用os.walk
方法遍历目录(D盘),然后将它们都保存到一个文件夹下:

import os 
import shutil
import pathlib
from tqdm import tqdm

search_path = "D:/" # 查找D盘 
store_path = "D:/音乐"

#如果sotre_path不存在则创建目录。
pathlib.Path(store_path).mkdir(parents=True, exist_ok=True

for dirpath, dirnames, filenames in tqdm(os.walk(search_path)):
    if dirpath == store_path:
        print(f"跳过{store_path}")
        continue
    for filename in filenames:
        if filename.endswith(".mp3"):
            f = os.path.join(dirpath, filename)
            file_size = os.path.getsize(f)
            if file_size > 2**20# 文件大小超过1MB(2**20B)
                print(f, file_size)
                dst = os.path.join(store_path,filename)
                shutil.copyfile(f, dst) # 复制到目标文件夹下

注:tqdm用来显示进度,如果没有安装tqdm,可以先安装tqdm:
python -m pip install tqdm

下面简单介绍一下用到的库:

os 库

标准库,提供与操作系统相关的功能。例如 文件名、环境变量、进程相关操作。

https://docs.python.org/zh-cn/3/library/os.html

通常很少使用os库,就像文档开头说的:如果你只是想读写一个文件,请参阅 open();如果你想操作文件路径,请参阅 os.path
 模块;如果你想读取通过命令行给出的所有文件中的所有行,请参阅 fileinput
 模块。为了创建临时文件和目录,请参阅 tempfile
 模块;对于高级文件和目录处理,请参阅 shutil
 模块。

os.walk

这里介绍一下 os.walk

os.walk(top, topdown=True, onerror=None, followlinks=False)
 

https://docs.python.org/zh-cn/3/library/os.html#os.walk 
官方文档说得比较繁琐,可以直接看下面的例子:

例子

import os

search_path = r'D:\资料'
for dirpath, dirnames, filenames in os.walk(search_path):
    print(f"dirpath={dirpath}, dirnames={dirnames}, filenames={filenames}")

目录如下:

运行代码输出:

dirpath=D:\资料, dirnames=['电子书''表格'], filenames=['资料说明.txt']
dirpath=D:\资料\电子书, dirnames=['精选电子书'], filenames=['a.txt''b.txt']
dirpath=D:\资料\电子书\精选电子书, dirnames=[], filenames=['精品.txt']
dirpath=D:\资料\表格, dirnames=[], filenames=['表格1.xlsx''表格2.xlsx']

os.walk
 是递归地向下遍历(深度优先遍历),访问所有的文件夹。每次遍历返回一个三元组dirpath, dirnames, filenames
。 
dirpath
是当前访问的路径, dirnames
是路径下的文件夹(列表), filename
是路径下的文件(列表)。

在本例中,访问的路径依次是: 

-->资料 
-->资料->电子书 
-->资料->电子书->精选电子书 
-->资料->表格

shutil库

标准库,与文件操作相关。
https://docs.python.org/zh-cn/3/library/shutil.html

shutil.copyfile(src, dst, *, follow_symlinks=True)
 将名为 src
 的文件的内容(不包括元数据(元数据是文件大小、修改时间等信息))拷贝到名为 dst
 的文件并以尽可能高效的方式返回 dst
。 
src
 和 dst
 均为路径类对象或以字符串形式给出的路径名。

如果要拷贝目录,可能要用到copytree
: 
shutil.copytree(src, dst, symlinks=False, ignore=None, copy_function=copy2, ignore_dangling_symlinks=False, dirs_exist_ok=False)
 
将以 src
 为根起点的整个目录树拷贝到名为 dst
 的目录并返回目标目录。 dirs_exist_ok
 表示是否允许 dst
 已存在或父目录缺失。

shutil.move(src, dst, copy_function=copy2)
 
递归地将一个文件或目录 (src
) 移至另一位置 (dst
) 并返回目标位置。


小结

使用os.walk可以遍历文件夹,结合其它的一些库(shutil),可以很方便地操作文件。如果需要对文件内容进行判断,还可以进一步处理。

文章转载自一只大鸽子,如果涉嫌侵权,请发送邮件至:contact@modb.pro进行举报,并提供相关证据,一经查实,墨天轮将立刻删除相关内容。

评论