大家好,又到了 Python 办公自动化专题
本文跟大家分享一个文件整理脚本的实现过程。具体的功能很简单,给定一个打算整理的文件夹目录,这个脚本可以将该目录下的所有文件都揪出来,并且根据后缀名归类到不同的文件夹里,就像下图一样,简直是整理爱好者的福音
通过本例还可以学到:os
、glob
、shutil
三个标准库的综合运用,现在让我们一起来捋一捋实现的逻辑。
首先是导入需要的库
import os
import shutil
import glob
os
库可以完成操作系统层面大量操作,例如文件夹的创建、移动、重命名、删除等,有些功能实现的不是很完美,就用到shutil
库跟其互补了,例如文件的复制、移动等。glob
库可以利用通配符进行文件的搜索获取,非常强大
# 设置建立分类总文件夹的路径,这里按自己的实际路径修改
mkdir_path = r'C:\Users\chenx\文件夹分类'
# 设置需要遍历整理的文件夹路径,可以依据自己的实际需求修改
goal_dir = r'C:\xxxxxxxx)'
if not os.path.exists(mkdir_path):
os.mkdir(mkdir_path)
os.mkdir
可以在指定路径创建文件夹,但如果文件夹已经存在则会报错,因此谨慎一点可以利用os.path.exists
先对文件夹的存在与否进行判断,接下来是代码核心循环,为了方便理解先简化成如下形式:
for file in glob.glob(f'{goal_dir}/**/*', recursive=True):
if os.path.isfile(file):
print(file)
glob.glob(f'{goal_dir}/**/*', recursive=True)
中**/*
是通配符的重要用法,*
可以代表任意个字符,包括 0 个字符,recursive
参数的设置确保遍历。由于需要找出所有的文件而非文件夹,这里用os.path.isfile
进行判断。最后可以输出文件的绝对路径先看看代码有没出现错误,让我们继续往下写
for file in glob.glob(f'{goal_dir}/**/*', recursive=True):
if os.path.isfile(file):
filename = os.path.basename(file)
if '.' in filename:
suffix = filename.split('.')[-1]
else:
suffix = 'others'
这里发生了什么呢?确认遍历到的是文件后,先用os.path.basename
获取绝对路径中的文件名,接下来就是获取后缀名了。可以简单用split
根据.
将字符串 “劈开”,然后取最后一个元素就是后缀名了,但注意这里必须要考虑一个特殊情况:有些文件没有后缀名 (文件类型就叫 文件),且名字中也没有.
,这时用字符串方法split
就会报错。如下图:
因此需要先判断文件中有没有.
。由于我们是利用后缀名建立文件夹,所以索性将文件名中没有.
的 文件 类型统一分类到others
文件夹了,(这个实现逻辑大致上没有问题,但是依然忽略了一种极端情况:有些文件没有后缀名,且文件名中有个.
,哈哈哈哈这种就会被上面的实现逻辑拆解出错误的后缀名了。更好的方法是有个函数可以直接获取文件的后缀名,利用这个对文件进行分类,感兴趣的读者可以自己尝试),接下来就可以根据后缀名产生文件夹了,这里依然要注意先判断文件夹是否已经产生
for file in glob.glob(f'{goal_dir}/**/*', recursive=True):
if os.path.isfile(file):
filename = os.path.basename(file)
if '.' in filename:
suffix = filename.split('.')[-1]
else:
suffix = 'others'
if not os.path.exists(f'{mkdir_path}/{suffix}'):
os.mkdir(f'{mkdir_path}/{suffix}')
shutil.copy(file, f'{mkdir_path}/{suffix}')
为了避免移动文件夹而造成的异常,尤其是系统盘,因此这里用的是复制。按照需要也可以换成shutil.move
最后我们可以加上分类文件夹和所有文件的计数并输出。完整代码如下,拿走就能用!
import os
import shutil
import glob
mkdir_path = r'C:\Users\chenx\文件夹分类'
goal_dir = r'C:\xxxxxxxx'
if not os.path.exists(mkdir_path):
os.mkdir(mkdir_path)
file_num = 0
dir_num = 0
for file in glob.glob(f'{goal_dir}/**/*', recursive=True):
if os.path.isfile(file):
filename = os.path.basename(file)
if '.' in filename:
suffix = filename.split('.')[-1]
else:
suffix = 'others'
if not os.path.exists(f'{mkdir_path}/{suffix}'):
os.mkdir(f'{mkdir_path}/{suffix}')
dir_num += 1
shutil.copy(file, f'{mkdir_path}/{suffix}')
file_num += 1
print(f'整理完成,有{file_num}个文件分类到了{dir_num}个文件夹中')
这里给大家分享一份Python全套学习资料,包括学习路线、软件、源码、视频、面试题等等,都是我自己学习时整理的,希望可以对正在学习或者想要学习Python的朋友有帮助!
CSDN大礼包:全网最全《全套Python学习资料》免费分享🎁
😝有需要的小伙伴,可以点击下方链接免费领取或者V扫描下方二维码免费领取🆓
1️⃣零基础入门
① 学习路线
对于从来没有接触过Python的同学,我们帮你准备了详细的学习成长路线图。可以说是最科学最系统的学习路线,你可以按照上面的知识点去找对应的学习资源,保证自己学得较为全面。
② 路线对应学习视频
还有很多适合0基础入门的学习视频,有了这些视频,轻轻松松上手Python~
③练习题
每节视频课后,都有对应的练习题哦,可以检验学习成果哈哈!
因篇幅有限,仅展示部分资料
2️⃣国内外Python书籍、文档
① 文档和书籍资料
3️⃣Python工具包+项目源码合集
①Python工具包
学习Python常用的开发软件都在这里了!每个都有详细的安装教程,保证你可以安装成功哦!
②Python实战案例
光学理论是没用的,要学会跟着一起敲代码,动手实操,才能将自己的所学运用到实际当中去,这时候可以搞点实战案例来学习。100+实战案例源码等你来拿!
③Python小游戏源码
如果觉得上面的实战案例有点枯燥,可以试试自己用Python编写小游戏,让你的学习过程中增添一点趣味!
4️⃣Python面试题
我们学会了Python之后,有了技能就可以出去找工作啦!下面这些面试题是都来自阿里、腾讯、字节等一线互联网大厂,并且有阿里大佬给出了权威的解答,刷完这一套面试资料相信大家都能找到满意的工作。
5️⃣Python兼职渠道
而且学会Python以后,还可以在各大兼职平台接单赚钱,各种兼职渠道+兼职注意事项+如何和客户沟通,我都整理成文档了。
上述所有资料 ⚡️ ,朋友们如果有需要 📦《全套Python学习资料》的,可以扫描下方二维码免费领取 🆓
😝有需要的小伙伴,可以点击下方链接免费领取或者V扫描下方二维码免费领取🆓