快捷導(dǎo)航

Python如何批量獲取文件夾的大小并保存

更新時間：2020年03月31日 11:57:53 作者：Johnthegreat

這篇文章主要介紹了Python如何批量獲取文件夾的大小并保存,文中通過示例代碼介紹的非常詳細，對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值,需要的朋友可以參考下

很多時候，查看一個文件夾下的每個文件大小可以輕易的做到，因為文件后面就是文件尺寸，但是如果需要查看一個文件夾下面所有的文件夾對應(yīng)的尺寸，就發(fā)現(xiàn)需要把鼠標放到對應(yīng)的文件夾上，稍等片刻才會出結(jié)果。

有時候，我們需要查看幾十個甚至于上百個文件夾，找出包含文件最多，空間占用最大的那個，就比較麻煩了。這段代碼是我以前的代碼，可以按大小排序輸出文件夾大小到txt文件，供使用的方便。

格式化當(dāng)時花了很長時間，最后發(fā)現(xiàn)使用‘YaHei.Consolas'字體可以解決，對齊后輸出結(jié)果看起來還算舒服。

上代碼：

import os
import datetime
 
 
def get_folder_size(path):
  folder_size = 0
 
  if not os.path.exists(path):
    return folder_size
 
  if os.path.isfile(path):
    folder_size = os.path.getsize(path)
    return folder_size
  try:
    if os.path.isdir(path):
      with os.scandir(path) as directory_lists:
        for directory_list in directory_lists:
          if directory_list.is_dir():
            sub_folder_size = get_folder_size(directory_list.path) # 遞歸獲取大小
            folder_size += sub_folder_size
          elif directory_list.is_file():
            file_size = os.path.getsize(directory_list.path)
            folder_size += file_size
 
        return folder_size
  except:
    pass
 
 
# 以下主要是為了格式化輸出
def get_file_length(file_name):
  characters = list(file_name)
  ascii_length = 0
  utf8_length = 0
 
  for character in characters:
    if ord(character) < 128:
      ascii_length += 1
    else:
      utf8_length += 2
 
  return ascii_length + utf8_length
 
 
def main(basedir):
  with os.scandir(basedir) as dirs:
    directory_size = []
    for dir in dirs:
      try:
        if not dir.is_file():
          dirsize = round(get_folder_size(dir.path) / 1000000) # return the file size in Mb
          resformat = [dir.name, dirsize]
          directory_size.append(resformat)
      except:
        pass
    results = sorted(directory_size, key=lambda x: x[1], reverse=True) # return a list ordered by size
    results = [[i[0], '文件夾大小：' + str(i[1]) + ' Mb'] for i in results]
 
    with open(basedir + os.sep + datetime.date.today().isoformat() + '.txt', 'a+') as f:
      for result in results:
        # 按照50的寬度格式化輸出結(jié)果
        len1 = 50 - get_file_length(result[0]) + len(result[0])
        len2 = 25 - get_file_length(result[1]) + len(result[1])
        f.writelines('{:<{len1}s} {:>{len2}s}\n'.format(result[0], result[1], len1=len1, len2=len2))
      print('The result was successfully saved in the directory with date as file name.')
 
 
if __name__ == "__main__":
  basedir = input("Please input the directory you would like to know the sizes: ")
  main(basedir)

如果輸入相應(yīng)的文件夾路徑，輸出結(jié)果如下：