p0003 os文件操作

import os


class :
    def __init__(self):
        self.path = None
        self.fileList = []
        self.fileAllSize = 0
        self.unitList = ["B", "kB", "MB", "GB", "TB"]

    def getFileNumber(self, path):
        if os.path.exists(path):
            for name in os.listdir(path):
                name = os.path.join(path, name)
                if os.path.isdir(name):
                    self.getFileNumber(name)
                else:
                    self.fileList.append(name)
                
        else:
            raise TypeError("路径输入错误!")

    def getFileSize(self, path):
        if os.path.exists(path):
            for name in os.listdir(path):
                name = os.path.join(path, name)
                if os.path.isdir(name):
                    self.getFileSize(name)
                else:
                    self.fileAllSize += os.path.getsize(name)
        else:
            raise TypeError("路径输入错误!")

    def uintChange(self):
        index = len(self.unitList)
        while True:
            if self.fileAllSize < 1024:
                print("文件大小为%dbytes" % self.fileAllSize)
                break
            else:
                if self.fileAllSize > 1024 ** index:
                    print("文件总大小为:%.2f%s" % (self.fileAllSize /
                                             1024 ** index, self.unitList[index]))
                    break
                else:
                    index -= 1


fileP = fileProcess()
# C:UsersAdministratorDesktopwinhex
# a.getFileNumber(input("请输入要获取的文件路径:"))
fileP.path = input("请输入要获取的文件路径:")
fileP.getFileSize(fileP.path)
fileP.getFileNumber(fileP.path)
print("文件数量:%d" % len(fileP.fileList))
fileP.uintChange()