请选择 进入手机版 | 继续访问电脑版
设为首页收藏本站

Crossin的编程教室

 找回密码
 立即加入
查看: 3103|回复: 0

还你系统空间的 Python 小程序

[复制链接]

169

主题

1

好友

733

积分

版主

Rank: 7Rank: 7Rank: 7

发表于 2018-7-11 17:09:33 |显示全部楼层

Windows 系统用久了,磁盘中就会产生大量的“垃圾”文件。这些文件有的是程序使用过的临时文件,有的是操作记录和日志信息等。因为往往不能被有效地清理干净,越积越多,导致用户的可用空间越来越小。同时也会因为碎片文件过多,使得系统的运行速度受到一定影响。

像我这种强迫症用户显然受不了这种状况,定期清理垃圾文件很有必要。

对于 Windows 系统,网上有一些现成的垃圾文件清理脚本。不过作为一个 Python 学习者,当然要自己动手来实现一个才合理嘛。

Python 的 os 模块,提供了较为丰富的处理系统文件与路径的函数。下面我们就靠它,来实现一个自己的磁盘清理小程序。

动手前的友情提示:

本文附带代码运行环境为 WIN7 + Python 2.7。

某些缓存文件可以提高程序的执行速度,比如缓存 cookie、使用记录 recent、预读取 prefetch 等。所以清理临时文件并不代表系统运行就会变快,有时也可能变慢。

由于牵涉到文件删除操作,请在动手前务必反复确认代码,万一导致什么重要文件被删,本人可概不负责啊。别问我为什么要这么说,写了好几天的代码在调试时被误删了,我再去哭一会儿……(;Д`)

言归正传,电脑中的垃圾文件及文件夹主要有下面几类:

系统盘 %system% 下文件类型:【临时文件(*.tmp)】【临时文件(*._mp)】【日志文件(*.log)】【临时帮助文件(*.gid)】【磁盘检查文件(*.chk)】【临时备份文件(*.old)】【Excel备份文件(*.xlk)】【临时备份文件(*.bak)】--------------------------------用户目录 %userprofile% 下文件夹【COOKIE】 cookies\*.*【文件使用记录】 recent\*.*【IE临时文件】 Temporary Internet Files\*.*【临时文件文件夹】 Temp\*.*--------------------------------Windows 目录 %windir% 下文件夹【预读取数据文件夹】 prefetch\*.*【临时文件】 temp\*.*

如果你了解自己使用的软件产生的其它垃圾文件,也可以添加至待删除的列表上。

与我们直接在 Windows 可视化窗口删除文件或文件夹类似,使用 Python 清理磁盘大致亦可分为 获取文件地址、判断垃圾文件、删除垃圾文件和文件夹 三步:
1. 获取文件地址
在可视化窗口中,我们点击每个文件夹的图标,打开不同文件夹,可以看见窗口的地址栏也随之切换到了对应的目录下,在 Python  os 模块中,同样是根据文件地址来查找文件,相应的函数如下:

获取当前目录路径
  1. >>> import os
  2. >>> os.getcwd()
  3. 'E:\\Software\\Python27'
复制代码
跳转至指定的文件目录
  1. >>> os.chdir('d://temporary')
  2. >>> os.getcwd()
  3. 'd:\\temporary'
复制代码
获取系统盘符
  1. >>> os.environ['systemdrive']
  2. 'C:'
复制代码
获取用户目录
  1. >>> os.environ['userprofile']
  2. 'C:\\Users\\Administrator'
复制代码
获取 Windows 目录
  1. >>> os.environ['windir']
  2. 'C:\\Windows'
复制代码
调用 os.environ 函数得到字典格式的系统环境映射表,通过关键字可以很方便地得到我们想要的地址。

使用 os.walk 函数,可生成并展开以指定目录为根目录的目录树。

参数 topdown 指定展开方式是否从顶层到底层
  1. all_info = os.walk('d:\\temporary\\', topdown=False)
复制代码
os.walk() 函数一般配合如下的 for 循环使用,遍历目录树中每一层的根目录 roots,子目录 dirs 以及文件 files,并将值储存在对应的变量中:
  1. for roots, dirs, files in all_info:  
  2.     ...
复制代码
例如,对于 D 盘下根目录 AAA 目录树展开如下:

用 walk 函数遍历并输出后,结果如下:

将两个给定的目录进行组合
  1. >>> os.path.join('d:\\', 'temporary')
  2. 'd:\\temporary'
复制代码
用 join 函数,配合 walk 函数得到的根目录与文件名,就能组合出我们想要的文件地址了。
2. 判断垃圾文件
我们现在已知垃圾文件的扩展名(以及垃圾文件夹名),通过 walk 函数与 join 函数也得到了完整的文件名,要判断文件是不是垃圾文件,可以用正则表达式进行匹配判断,正则表达式还不牢固的同学请移步 【Python 第55课】 正则表达式(1)

如果对正则表达式的语法深感头疼,os 模块也有提供其它解决方案:
  1. >>> os.path.splitext(r'aaa\bbb\ccc.ddd')
  2. ('aaa\\bbb\\ccc', '.ddd')
复制代码
os.path.splitext 函数可以将文件的文件名与扩展名进行分离,并返回一个包含文件名与扩展名的二元元组。

得到了文件的扩展名,我们可以用 ‘in’ 来 判断该扩展名是否在需要删除的扩展名列表中:
  1. extension = os.path.splitext(r'aaa\bbb\ccc.ddd')
  2. extension_to_del = ['.aaa', '.bbb', '.ccc', '.ddd']
  3. if extension[1] in extension_to_del:
  4.     # to be deleted
复制代码
这里可能会踩到的一个坑是字符“\”,它既是 Windows 路径的分隔符,又是 Python 字符串中的转义符。在处理路径时,需要额外注意。
3. 删除文件
使用 os 模块删除数据有三点需要注意:

其一,删除文件与删除文件夹调用的是不同的函数。

删除文件
  1. os.remove('d:temporary/test/test.txt')
复制代码
删除文件夹
  1. os.rmdir('d:temporary/test/empty')
复制代码
如果你在尝试调用这个代码删除文件/文件夹时无效,确认下文件名是否带有中文、空格和特殊符号。建议先在纯英文的简单路径下调试成功再说。

无法删除的还有一个常见原因是另一个要注意的地方:

其二,os.rmdir 只能删除空文件夹,如果文件夹非空,则会报错。

所以,如果要把匹配到的垃圾文件夹整个删除,可能我们不得不选择先把文件夹中的所有文件全部删除,再从最内层文件夹开始往外逐层删除。这实在是挺麻烦的,那有没有其它解决方案呢?

有,但要依靠 os 以外的帮助。
  1. import shutil
  2. shutil.rmtree('d:/temporary/test/aaa')
复制代码
上面的解决办法是引用另外一个模块 shutil 的函数来删除非空文件夹,其实,这个外部引用的函数还是用 os 模块的函数来写的,想挑战的话,也可以自己写一个试试。

当然,直接调用 cmd 命令也是极好的:
  1. os.system('rd/s/q "d:temporary/test"')
复制代码
注意,整个命令是一个字符串,其中目录地址要额外加引号。

在某些情况下,尤其是当前有较多其它程序在运行时,即使用 shutil.rmtree 函数仍然会报错,甚至连删除文件都会报错,这就是第三个注意事项了:

其三,文件正在运行或者受到保护、当前账户没有足够权限时,删除会报错。

这个情况下,对应的文件最后就不要强行删了,不然轻则即使删除了,也会再次自动生成、重则还可能导致正在运行的程序崩溃。

因此,我们在删除文件或文件夹时,加上一个 try except 逻辑结构,跳过那些无法删除的文件:
  1. try:
  2.     os.remove(r'd:\temporary\using.xxx')
  3. except WindowsError:
  4.     print 'nothing has being removed'
复制代码
按照上面三个步骤写下来,磁盘清理脚本基本功能已经完善。不过除此之外,我们还可以添加一些数据监控代码,多一些对用户友好的人机交互,比如:

显示文件夹(路径)大小,单位 bite
  1. >>> os.path.getsize('d://temporary/test')
  2. 4096
复制代码
显示文件大小
  1. >>> os.path.getsize('d://temporary/test/aaa.txt')
  2. 135
复制代码
其它功能诸如 定时清理、开关机清理、清理时间统计 等等,都可以尝试实现。欢迎在留言区和论坛上讨论。

在公众号(Crossin的编程教室)回复 清理,可获取本文演示代码。

其他文章及回答:

学习编程的过程中可能会走哪些弯路,有哪些经验可以参考?你是如何自学 Python 的?数据分析:当赵雷唱民谣时他唱些什么?我去扒了杜蕾斯的微博一行代码扫出“敬业福”今天,你抢到票了吗?爆款游戏《贪吃蛇大作战》的 Python 实现

Crossin的编程教室

微信公众号ID:crossincode网站:crossincode.comQQ群:453300655微信加群:crossin11

回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即加入

QQ|手机版|Archiver|Crossin的编程教室 ( 苏ICP备15063769号  

GMT+8, 2024-3-29 22:01 , Processed in 0.026078 second(s), 23 queries .

Powered by Discuz! X2.5

© 2001-2012 Comsenz Inc.

回顶部