使用Python怎么编写一个自动清理系统垃圾功能

发布时间：2021-01-18 16:21:58 阅读：295 作者：Leah 栏目：开发技术

Python开发者专用服务器限时活动，0元免费领，库存有限，领完即止！点击查看>>

这篇文章给大家介绍使用Python怎么编写一个自动清理系统垃圾功能，内容非常详细，感兴趣的小伙伴们可以参考借鉴，希望对大家能有所帮助。

一、清理目标

系统盘 %system% 下文件类型：

【临时文件（*.tmp）】
【临时文件（*._mp）】
【日志文件（*.log）】
【临时帮助文件（*.gid）】
【磁盘检查文件（*.chk）】
【临时备份文件（*.old）】
【Excel备份文件（*.xlk）】
【临时备份文件（*.bak）】

用户目录 %userprofile% 下文件夹

【COOKIE】 cookies\*.*
【文件使用记录】 recent\*.*
【IE临时文件】 Temporary Internet Files\*.*
【临时文件文件夹】 Temp\*.*.

Windows 目录 %windir% 下文件夹

【预读取数据文件夹】 prefetch\*.*
【临时文件】 temp\*.*

Python os

Python os 模块提供了非常丰富的方法用来处理文件和目录，它会自适应于不同的操作系统平台，根据不同的平台进行相应的操作，在python编程时，经常和文件、目录打交道，这时就离不了os模块。
具体就不过多介绍，大家可以看一下官方文档，也没必要细究，用到哪个知道怎么查就行了。

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

Python脚本

造轮子不如站在巨人的肩膀上，发现已经有博主写过了，不过是用Python2编写的，print的地方改一下就行了。

使用Python怎么编写一个自动清理系统垃圾功能

import os
del_extension = {
  '.tmp': '临时文件',
  '._mp': '临时文件_mp',
  '.log': '日志文件',
  '.gid': '临时帮助文件',
  '.chk': '磁盘检查文件',
  '.old': '临时备份文件',
  '.xlk': 'Excel备份文件',
  '.bak': '临时备份文件bak'
}
del_userprofile = ['cookies', 'recent', 'Temporary Internet Files', 'Temp']
del_windir = ['prefetch', 'temp']
SYS_DRIVE = os.environ['systemdrive'] + '\\'
USER_PROFILE = os.environ['userprofile']
WIN_DIR = os.environ['windir']

def del_dir_or_file(root):
  try:
    if os.path.isfile(root):
      os.remove(root)
      print ("file",root,"removed")
    elif os.path.isdir(root):
      os.rmdir(root)
      print("dir",root,"removed")

  except WindowsError:
    print("failure",root,"can't remove")

def formatSize(b):
  try:
    kb = b // 1024
  except:
    print("传入字节格式不对")
    return "Error"
  if kb > 1024:
    M = kb // 1024
    if M > 1024:
      G = M // 1024
      return "%dG" % G
    else:
      return "%dM" % M
  else:
    return "%dkb" % kb

class DiskClean(object):
  def __init__(self):
    self.del_info = {}
    self.del_file_paths = []
    self.total_size = 0
    for i,j in del_extension.items():
      self.del_info[i] = dict(name = j,count = 0 )

  def scanf(self):
    for roots,dirs,files in os.walk(USER_PROFILE):
      for files_item in files:
        file_extension = os.path.splitext(files_item)[1]
        if file_extension in self.del_info:
          file_full_path = os.path.join(roots,files_item)
          self.del_file_paths.append(file_full_path)
          self.del_info[file_extension]['count'] += 1
          self.total_size += os.path.getsize(file_full_path)

  def show(self):
    re = formatSize(self.total_size)
    for i in self.del_info:
      print(self.del_info[i]["name"],"共计",self.del_info[i]["count"],"个")
    return re

  def delete_files(self):
    for i in self.del_file_paths:
      print(i)
      del_dir_or_file(i)
if __name__ == "__main__":
  print("初始化清理垃圾程序")
  cleaner = DiskClean()
  print("开始扫描垃圾文件请耐心等待\n")
  cleaner.scanf()
  print("扫描成功，结果如下")
  re = cleaner.show()
  cleaner.delete_files()