blob: f4e147df0ef188a0741db3468d15d9341e792167 (
plain)
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
|
import hashlib
import sys
from string import Template
args = sys.argv
version = args[1]
template_file_path = args[2]
generated_file_path = args[3]
# SHA512, SHA256, or SHA1
hash_type = args[4]
# Deployment files
deployment_file_path_1 = args[5]
deployment_file_path_2 = args[6] if len(args) > 6 else None
deployment_file_path_3 = args[7] if len(args) > 7 else None
print("Generating package for file: %s" % deployment_file_path_1)
if deployment_file_path_2 is not None:
print("and for file: %s" % deployment_file_path_2)
if deployment_file_path_3 is not None:
print("and for file: %s" % deployment_file_path_3)
print(" VERSION: %s" % version)
print(" TEMPLATE PATH: %s" % template_file_path)
print(" SAVING AT: %s" % generated_file_path)
print(" USING HASH TYPE: %s" % hash_type)
def get_hash(deployment_file):
if str.lower(hash_type) == "sha512":
deployment_hash = hashlib.sha512(deployment_file.read()).hexdigest()
elif str.lower(hash_type) == "sha256":
deployment_hash = hashlib.sha256(deployment_file.read()).hexdigest()
elif str.lower(hash_type) == "sha1":
deployment_hash = hashlib.sha1(deployment_file.read()).hexdigest()
else:
print(
'Unsupported hash format "%s". Please use SHA512, SHA256, or SHA1.',
hash_type,
)
exit(1)
print("Generated hash: %s" % str(deployment_hash))
return deployment_hash
with open(deployment_file_path_1, "rb") as deployment_file_1:
deployment_hash_1 = get_hash(deployment_file_1)
deployment_hash_2 = None
if deployment_file_path_2 is not None:
with open(deployment_file_path_2, "rb") as deployment_file_2:
deployment_hash_2 = get_hash(deployment_file_2)
deployment_hash_3 = None
if deployment_file_path_3 is not None:
with open(deployment_file_path_3, "rb") as deployment_file_3:
deployment_hash_3 = get_hash(deployment_file_3)
with open(template_file_path, "r") as template_file:
template = Template(template_file.read())
substitutes = dict()
substitutes["version"] = version
substitutes["hash1"] = deployment_hash_1
if deployment_hash_2 is not None:
substitutes["hash2"] = deployment_hash_2
if deployment_hash_3 is not None:
substitutes["hash3"] = deployment_hash_3
substitute = template.safe_substitute(substitutes)
print("\n================== Generated package file ==================\n")
print(substitute)
print("\n============================================================\n")
with open(generated_file_path, "w") as generated_file:
generated_file.write(substitute)
|
