查看原文
其他

10 行 Python 代码自动清理电脑重复文件,解放双手!

程序人生 2020-12-18

The following article is from 早起Python Author 陈熹

作者 | 陈熹来源 | 早起Python(ID:zaoqi-python)


前言


大家好,又到了 Python 办公自动化系列。
今天分享一个系统层面的自动化案例:
「给定一个文件夹,使用 Python 检查给定文件夹下有无文件重复,若存在重复则删除」
主要涉及的知识点有:
  • os 模块综合应用
  • glob 模块综合应用
  • 利用 filecmp 模块比较两个文件


步骤分析


该程序实现的逻辑可以具化为:
遍历获取给定文件夹下的所有文件,然后通过嵌套循环两两比较文件是否相同,如果相同则删除后者。
实现问题的关键就变成了👇如何判断两个文件是否相同?在这里我们可以使用 filecmp 模块,来看看官方的介绍文档:
  • filecmp.cmp(f1, f2, shallow=True)

  • 比较名为 f1 和 f2 的文件,如果它们似乎相等则返回 True,否则返回False

  • 如果 shallow 为真,那么具有相同 os.stat() 签名的文件将会被认为是相等的。否则,将比较文件的内容。

所以可以这样使用👇

# 假设x和y两个文件是相同的
print(filecmp.cmp(x, y))
# True

解决了这个问题,我们就可以开始写代码了!

Python实现

导入需要的库并设置目标文件夹路径:

import os
import glob
import filecmp

dir_path = r'C:\\xxxx'

接着遍历获取所有文件的绝对路径,我们可以利用 glob 模块的通配符结合 recursive 参数即可完成,框架如下:

for file in glob.glob(path + '/**/*', recursive=True):
    pass

由于遍历获取每一个文件或者文件夹后,需要判断是否是文件,如果是文件则可能将绝对路径存放到列表中,这里需要再完成两个事情:
  1. 首先创建一个空列表,后面用 list.append(i) 添加文件路径。

  2. 接着利用 os.path.isfile(i) 判断是否是文件,返回 True 则执行添加元素的操作。

具体代码如下:

file_lst = []

for i in glob.glob(dir_path + '/**/*', recursive=True):
    if os.path.isfile(i):
        file_lst.append(i)

上一步我们获取了目标文件夹下的所有文件路径,接下来就可以嵌套遍历这个路径列表,其中 filecmp.cmp 进行文件判断,os.remove 进行文件删除。

for x in file_lst:
    for y in file_lst:
        if x != y:
            if filecmp.cmp(x, y):
                os.remove(y)

这里的代码已经实现了大致逻辑,但有一个细节需要考虑到:有可能循环到文件已经被前面的判断删除了,导致 os.remove(file) 由于文件不存在而报错。因此,可以用 os.path.exists 对文件存在进行判断,如下所示:

for x in file_lst:
    for y in file_lst:
        if x != y and os.path.exists(x) and os.path.exists(y):
            if filecmp.cmp(x, y):
                os.remove(y)

这样,一个简易的文件去重小程序就完成了,完整代码如下:

import os
import glob
import filecmp

dir_path = r'C:\xxxx'

file_lst = []

for i in glob.glob(dir_path + '/**/*', recursive=True):
    if os.path.isfile(i):
        file_lst.append(i)

for x in file_lst:
    for y in file_lst:
        if x != y and os.path.exists(x) and os.path.exists(y):
            if filecmp.cmp(x, y):
                os.remove(y)


写在最后

通过本文的 Python 自动化脚本制作过程,我们可以再次体会 Python 办公自动化的强大,同时本次实现的功能可以和之前自动整理文件的脚本结合使用!
最后还是希望大家能够理解 Python 办公自动化的一个核心就是批量操作-解放双手,让复杂的工作自动化!



更多精彩推荐
12306 背后的技术大牛:我不跟人拼智商,我就跟他们拼狠!
融资 2000 万美元后,他竟将核心代码全开源,这……能行吗?
鸿蒙加海思,麒麟加龙芯,组合拳能否渡劫“生态”危机
用 Python 详解《英雄联盟》游戏取胜的重要因素!
2020 年最厉害的 10 门编程语言
“善”用区块链:善是初心,链是保障
点分享点点赞点在看

    您可能也对以下帖子感兴趣

    文章有问题?点此查看未经处理的缓存