Python如何读取文件夹所在磁盘的大小
在Python中，要读取文件夹所在磁盘的大小，我们可以使用os模块来实现。os模块提供了许多操作系统相关的功能，包括文件和目录操作。
第一步：获取文件夹路径
首先，我们需要获取文件夹的路径。可以通过用户输入、配置文件等方式来获取。
import os
folder_path = input("请输入文件夹路径：")
第二步：遍历文件夹
接下来，我们需要遍历文件夹中的所有文件和子文件夹。我们可以使用os.walk()函数来实现递归遍历。
total_size = 0
for root, dirs, files in os.walk(folder_path):
for file in files:
file_path = os.path.join(root, file)
file_size = os.path.getsize(file_path)
total_size += file_size
在上述代码中，os.walk()函数返回一个生成器，每次迭代返回一个三元组(root, dirs, files)，分别表示当前目录的路径、子目录列表和文件列表。我们遍历文件列表，并使用os.path.join()函数来获取每个文件的完整路径。
然后，我们使用os.path.getsize()函数来获取每个文件的大小，并将其累加到total_size变量中。
第三步：格式化文件夹大小
最后，我们需要将文件夹大小格式化为易读的形式，例如以字节、千字节、兆字节或千兆字节表示。
def format_size(size):
# 定义单位
units = ['Bytes', 'KB', 'MB', 'GB', 'TB']
# 迭代计算文件夹大小
for unit in units:
if size < 1024:
return f"{size:.2f} {unit}"
size /= 1024
return f"{size:.2f} PB"
formatted_size = format_size(total_size)
在上述代码中，我们定义了一个format_size()函数，用于将文件夹大小格式化为易读的形式。我们使用一个循环来判断文件夹大小是否超过了当前单位（1024），如果超过了则将其转换到下一个单位。最后，我们使用f-string来将文件夹大小和单位拼接起来。
完整代码
下面是完整的代码示例：
import os
folder_path = input("请输入文件夹路径：")
total_size = 0
for root, dirs, files in os.walk(folder_path):
for file in files:
file_path = os.path.join(root, file)
file_size = os.path.getsize(file_path)
total_size += file_size
def format_size(size):
# 定义单位
units = ['Bytes', 'KB', 'MB', 'GB', 'TB']
# 迭代计算文件夹大小
for unit in units:
if size < 1024:
return f"{size:.2f} {unit}"
size /= 1024
return f"{size:.2f} PB"
formatted_size = format_size(total_size)
print(f"文件夹大小为：{formatted_size}")
序列图
下面是使用Mermaid语法绘制的序列图，展示了代码的执行流程：
sequenceDiagram
participant User
participant Python
participant OS
User->>Python: 输入文件夹路径
Python->>OS: 调用os.walk()
loop 遍历文件列表
OS-->>Python: 返回文件路径
Python->>OS: 获取文件大小
Python->>Python: 累加文件大小
end
Python-->>User: 输出文件夹大小
在上述序列图中，用户输入文件夹路径后，Python调用os.walk()来遍历文件夹中的文件列表。然后，Python获取每个文件的大小并累加起来。最后，Python将格式化后的文件夹大小输出给用户。
状态图
下面是使用Mermaid语法绘制的状态图，展示了format_size()函数的逻辑：
stateDiagram
[*] --> CheckSize
CheckSize --> FormatBytes: 文件夹大小 < 1024
CheckSize --> FormatKB: 文件夹大小 >= 1024
CheckSize --> FormatMB: 文件夹大小 >= 1024^2
CheckSize --> Format
