您的位置：首页 > 脚本大全 > > 正文

python怎么判断文件大小（python3实现指定目录下文件sha256及文件大小统计）

更多时间：2022-01-25 00:11:28 类别：脚本大全浏览量：1958

python怎么判断文件大小

python3实现指定目录下文件sha256及文件大小统计

有时会统计某个目录下有哪些文件，每个文件的sha256及文件大小等相关信息，这里用python3写了个脚本用来实现此功能，此脚本可跨平台，同时支持windows和linux，脚本(get_dir_file_info.py)内容如下：

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

20

21

22

23

24

25

26

27

28

29

30

31

32

33

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

58

59

60

61

62

63

64

65

66

67

68

69

70

71

72

73

74

75

76

77

78

79

80

81

82

83

84

85

86

87

88

89 import os

import sys

import hashlib

def usage():

''' usage description '''

num = len(sys.argv)

if num != 3:

print("error: please input two parameters")

print("for example: {} path_name save_file_name".format(sys.argv[0]))

sys.exit(1)

def getfileslist():

''' get file list '''

input_path_name = sys.argv[1]

result = list()

for dirpath, dirnames, filenames in os.walk(input_path_name, followlinks=true):

for filename in filenames:

file_path = os.path.join(dirpath, filename)

result.append(file_path)

result.sort()

return result

def calcfilesha256(filname):

''' calculate file sha256 '''

with open(filname, "rb") as f:

sha256obj = hashlib.sha256()

sha256obj.update(f.read())

hash_value = sha256obj.hexdigest()

return hash_value

def calcfilesize(filename):

''' calculate file size '''

return os.stat(filename).st_size

def getfilecontent():

''' get file contnet '''

files_list = getfileslist()

result = list()

for f in files_list:

hash = calcfilesha256(f)

size = calcfilesize(f)

file_name = os.path.basename(os.path.realpath(f))

path_name = os.path.dirname(os.path.realpath(f))

dictionary = {"path": path_name, "filename": file_name, "sha256": hash, "size": size}

#print("result: {}".format(dictionary))

result.append(dictionary)

return result

def writetofile(contents):

''' write content to the specified file '''

fp = open(sys.argv[2], "w")

for content in contents:

#print("content:", content)

str0 = str(content)

str1 = str0.replace("\\\\", "/")

fp.write(str1)

fp.write("\n")

fp.close()

def replacestr(src_str, new_str):

''' replace source string with new string '''

contents = list()

fp = open(sys.argv[2], "r")

line = fp.readline()

while line:

contents.append(line)

line = fp.readline()

fp.close()

fp = open(sys.argv[2], "w")

for content in contents:

str0 = content.replace(src_str, new_str)

fp.write(str0)

fp.close()

def main():

usage()

writetofile(getfilecontent())

replacestr(" ", "")

replacestr("'", "\"")

if __name__ == "__main__":

main()