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
|
import os
import subprocess
import filecmp
import hashlib
import argparse
from os import listdir
from os.path import isfile, join
qtver='5.13'
usehashforfiles = ['YUView.exe']
filter = ['versioninfo.txt']
def get_git_revision_hash():
return subprocess.check_output(['git', 'rev-parse', 'HEAD'])
def get_git_revision_short_hash():
return subprocess.check_output(['git', 'rev-parse', '--short', 'HEAD'])
def get_git_revision():
return subprocess.check_output(['git', 'describe', '--tags'])
def md5(fname):
hash_md5 = hashlib.md5()
with open(fname, "rb") as f:
for chunk in iter(lambda: f.read(4096), b""):
hash_md5.update(chunk)
return hash_md5.hexdigest()
def getAllSubFiles(rootDir):
fileSet = set()
for dir_, _, files in os.walk(rootDir):
for fileName in files:
relDir = os.path.relpath(dir_, rootDir)
relFile = os.path.join(relDir, fileName)
fileSet.add(relFile)
return fileSet
def md5andversionallfiles(mypath,outfile):
versionfile = open(outfile,'w')
versionfile.write('Last Commit: ' + str(get_git_revision().decode('utf-8')))
versionfile.write('Last Commit Hash: ' + str(get_git_revision_hash().decode('utf-8')))
versionfile.write('Last Commit Hash Short: ' + str(get_git_revision_short_hash().decode('utf-8')))
onlyfiles = getAllSubFiles(mypath)
#onlyfiles = [f for f in listdir(mypath) if isfile(join(mypath, f))]
for f in onlyfiles:
fileclean = f.lstrip(".\\")
if fileclean not in filter:
md5hash = md5(join(mypath,f))
filesize = os.path.getsize(join(mypath,f))
if fileclean in usehashforfiles:
version = str(get_git_revision_hash().decode('utf-8'))
else:
version = qtver
out = fileclean + ", " + version.rstrip() + ", " + md5hash + ", " + str(filesize) + "\n"
versionfile.write(out)
versionfile.close()
def main():
parser = argparse.ArgumentParser(description="Computes md5 sums for all files and also adds some git information.")
parser.add_argument('-d', '--directory', help='Specifies the input directory.', type=str, required=True)
parser.add_argument('-o', '--output', help='Specified the output file.', type=str, required=True)
args = parser.parse_args()
if args.directory is not None and args.output is not None:
md5andversionallfiles(args.directory,args.output)
else:
print('An error occurred. Did you specify the path and the output file?')
if __name__ == "__main__":
main()
|
