Python实现Linux环境下的ls命令

最新推荐文章于 2025-11-05 19:16:48 发布

原创最新推荐文章于 2025-11-05 19:16:48 发布 · 5.9k 阅读

1 ·

CC 4.0 BY-SA版权

本文介绍了一个Python脚本，该脚本能在Windows环境下利用正则表达式搜索指定目录下的文件，并显示匹配文件的数量及大小。此外，脚本还提供了删除匹配文件的功能。

部署运行你感兴趣的模型镜像

在Linux下使用ls命令结合正则表达式，能够高效地进行文件搜索，并通过参数操作文件，于是就想用Python实现这个功能以便在Windows上使用

import os
import re
import sys
path = os.getcwd()
substr = raw_input('The sub-string of the file (Support for Regular Expression): ')
reg = r'' + substr
count = 0
totalSize = 0
fileList = os.listdir(path)

def getFileSize(filePath):
	return os.path.getsize(filePath)

def humanReadFileSize(fileByte):
	# fileByte = str(getFileSize(filePath))
	if fileByte == 0:
		return ''
	elif fileByte < 1024:
		# B
		fileSize = str(fileByte) + ' B'
	elif fileByte < 1024*1024:
		# KB
		fileSize = str(fileByte/1024.0) + ' KB'
	elif fileByte < 1024*1024*1024:
		# MB
		fileSize = str(fileByte/1024.0/1024.0) + ' MB'
	else:
		# GB
		fileSize = str(fileByte/1024.0/1024.0/1024.0) + ' GB'
	return fileSize

for fileName in fileList:
	if re.match(reg, fileName):
		count = count+1
		filePath = os.path.join(path, fileName)
		fileSize = getFileSize(filePath)

		totalSize = totalSize+fileSize
		print '\t', count, ' - ', humanReadFileSize(fileSize).ljust(30, ' '), fileName
		print '-'*160
		if len(sys.argv)!=1:
			if sys.argv[1] == 'remove':
				os.remove(filePath)

print '\n\t\t%d file(s) in total, total size: %s' % (count, humanReadFileSize(totalSize))

您可能感兴趣的与本文相关的镜像