폴더 크기 구하기 예제

folderSize.py

import operator
import sys
import os
from PyQt5.QtWidgets import *

import pandas as pd

gExcelFile = 'foldersize.xlsx'


class CFolderSize:
    def getFolderSize(self, path, caller, parentPath):
        if path not in caller.dicFolder:
            caller.dicFolder[path] = {'size': 0, 'folders': 0, 'files': 0, 'error': False}
        if (parentPath != ''):
            if parentPath not in caller.dicFolder:
                caller.dicFolder[parentPath] = {'size': 0, 'folders': 0, 'files': 0, 'error': False}

        try:
            for fname in os.listdir(path):
                fullname = os.path.join(path, fname)

                if os.path.isdir(fullname):
                    caller.dicFolder[path]['folders'] += 1
                    if False == self.getFolderSize(fullname, caller, path):
                        # if parentPath != '@root@':
                        #     caller.dicFolder[parentPath]['error'] = True
                        continue

                if os.path.isfile(fullname):
                    caller.dicFolder[path]['files'] += 1
                    size = os.path.getsize(fullname)
                    caller.dicFolder[path]['size'] += size
                    caller.dicFile[fullname] = size
        except:
            print('exception 발생', path, parentPath)
            caller.dicFolder[path]['error'] = True
            return False

        if (parentPath != '@root@'):
            caller.dicFolder[parentPath]['size'] += caller.dicFolder[path]['size']
        print(path, 'folder size=', caller.dicFolder[path]['size'],
              'folders#= ', caller.dicFolder[path]['folders'],
              'files#= ', caller.dicFolder[path]['files'])
        return True

    def convertSizeToString(self, size):
        if size == 0:
            return ''
        gb, rest = divmod(size, 1000000000)
        if (gb > 0):
            return '%s GB (%s)' % (format(gb, ','), format(size, ','))
        mb, rest = divmod(rest, 1000000)
        if (mb > 0):
            return '%s MB (%s)' % (format(mb, ','), format(size, ','))

        kb, rest = divmod(rest, 1000000)
        if (kb > 0):
            return '%s KB (%s)' % (format(kb, ','), format(size, ','))

        return '%s' % (format(size, ','))


class MyWindow(QMainWindow):
    def __init__(self):
        super().__init__()

        self.dicFolder = {}
        self.dicFile = {}

        self.dicSortedFolder = []
        self.dicSortedFile = []

        # 윈도우 버튼 배치
        self.setWindowTitle("Folder Size")
        nH = 20

        self.pathEdit = QLineEdit("", self)
        self.pathEdit.move(20, nH)
        self.pathEdit.resize(300, 30)
        #        self.pathEdit.textChanged.connect(self.pathEditChanged)
        self.pathEdit.setText('00660')
        self.label = QLabel('폴더', self)
        self.label.move(340, nH)
        nH += 50

        btnRun = QPushButton("Run", self)
        btnRun.move(20, nH)
        btnRun.clicked.connect(self.btnRun_clicked)
        nH += 50

        self.setGeometry(300, 300, 400, nH)
        self.pathEdit.setText('c:\\')

    def btnRun_clicked(self):
        findPath = self.pathEdit.text()
        if findPath[-1] != "\\":
            findPath += "\\"

        if not os.path.isdir(findPath):
            print('존재하지 않는 폴더입니다.', findPath)
            return

        objFolder = CFolderSize()
        self.dicFolder = {}
        self.dicFile = {}
        self.dicSortedFolder = []
        self.dicSortedFile = []

        objFolder.getFolderSize(findPath, self, '@root@')

        # print('-----------------------------------------')
        # max = 20
        # if max > len(self.dicFolder) :
        #     max = len(self.dicFolder)
        # print('TOP 20 BIG SIZE FOLDERS')
        self.dicSortedFolder = sorted(self.dicFolder.items(), key=lambda x: x[1]['size'], reverse=True)
        # for i in range(0, max) :
        #     print(self.dicSortedFolder[i][0], objFolder.convertSizeToString(self.dicSortedFolder[i][1]['size']))

        # print('-----------------------------------------')
        # print('TOP 20 BIG SIZE FILES')
        # max = 20
        # print(self.dicFile)
        # if max > len(self.dicFile) :
        #     max = len(self.dicFile)
        self.dicSortedFile = sorted(self.dicFile.items(), key=operator.itemgetter(1), reverse=True)
        # for i in range(0, max) :
        #     print(self.dicSortedFile[i][0], objFolder.convertSizeToString(self.dicSortedFile[i][1]))

        print('엑셀 내보내기 시작')
        self.exportExcel(self.dicSortedFolder, self.dicSortedFile)

    def exportExcel(self, folderlist, filelist):
        i = 0
        writer = pd.ExcelWriter(gExcelFile, engine='xlsxwriter')
        df = pd.DataFrame(columns=['path', 'size', '폴더수', '파일수'])
        for folder in folderlist:
            i += 1
            if (i > 10000) :
                print('폴더리스트 저장은 10000개만 저장, 전체 폴더 개수 %d' %(len(folderlist)))
                break
            item = {}
            item['path'] = folder[0]
            item['size'] = folder[1]['size']
            item['폴더수'] = folder[1]['folders']
            item['파일수'] = folder[1]['files']
            df.loc[len(df)] = item

        # Convert the dataframe to an XlsxWriter Excel object.
        df.to_excel(writer, sheet_name='folders')

        i = 0
        df2 = pd.DataFrame(columns=['filename', 'size', 'fullpath'])
        for file in filelist:
            i += 1
            if (i > 10000) :
                print('파일리스트 저장은 10000 개만 저장, 전체 파일 수 %d' %(len(filelist)))
                break

            item = {}
            path = file[0]
            item['filename'] = os.path.basename(path)
            item['size'] = file[1]
            item['fullpath'] = path
            df2.loc[len(df2)] = item
        df2.to_excel(writer, sheet_name='files')

        # Close the Pandas Excel writer and output the Excel file.
        writer.save()
        os.startfile(gExcelFile)
        return


if __name__ == "__main__":
    app = QApplication(sys.argv)
    myWindow = MyWindow()
    myWindow.show()
    app.exec_()