写一个命令行工具计算文件大小（使用argparse模块实现命令行工具）

最新推荐文章于 2022-11-06 20:19:58 发布

QY&ZY

最新推荐文章于 2022-11-06 20:19:58 发布

阅读量153

点赞数

分类专栏：博文笔记文章标签： python

本文链接：https://blog.csdn.net/weixin_45883153/article/details/117912584

版权

博文同时被 2 个专栏收录

6 篇文章 0 订阅

订阅专栏

笔记

6 篇文章 0 订阅

订阅专栏

这是一个Python脚本，实现了一个命令行工具，能够根据用户输入的参数计算文件或目录的大小。通过`-d`选项计算目录大小，遍历目录下所有文件；使用`-f`选项计算单个文件的大小。程序使用了`os`和`argparse`模块，确保了路径的有效性并提供了友好的输出。

摘要由CSDN通过智能技术生成

题目：

写一个命令行工具可以计算文件和目录大小，输入文件路径中带-d是计算目录大小，带-f 是计算文件大小

导入模块

import os
import argparse

获取目录大小

def get_dir_size(path):
foldSize = 0   
if os.path.isdir(path):
        for curDir, subDir, files in os.walk(path)://用walk方法比较快
           for file in files:
               file_path = os.path.join(curDir, file)
                foldSize += os.path.getsize(file_path)//递归
        print("Folder size: %.1f" % foldSize)
 else:
      print("Invalid directory")

获取文件大小

def get_file_size(path):
   if os.path.isfile(path):
       fileSzie = os.path.getsize(path)
       print("File Size: %.1f" % fileSzie)
   else:    
       print("Invalid file path")

主函数

if __name__ == '__main__':
   parser = argparse.ArgumentParser()
   parser.add_argument('-d', '--directory', default=,type=str,help='Directory need to be calculate')
   parser.add_argument('-f', '--file', help='File need to be calculate')
   args = parser.parse_args()
  if args.directory:
      get_dir_size(args.directory)
  if args.file:
    get_file_size(args.file)

新人小白，个人学习笔记而已。大神勿喷