Refactor the hash helper script to use pathlib and CLI args

This commit is contained in:
Sviatoslav Sydorenko 2022-01-09 00:24:29 +01:00 committed by GitHub
parent 8682135dac
commit 0575dc8eab
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 8 additions and 7 deletions

View file

@ -1,20 +1,21 @@
import os
import hashlib import hashlib
import pathlib
import sys
sha256 = hashlib.sha256() sha256 = hashlib.sha256()
md5 = hashlib.md5() md5 = hashlib.md5()
blake2_256 = hashlib.blake2b(digest_size=256 // 8) blake2_256 = hashlib.blake2b(digest_size=256 // 8)
file_list = os.listdir(os.path.abspath(os.getenv("INPUT_PACKAGES_DIR"))) packages_dir = pathlib.Path(sys.argv[1]).resolve().absolute()
file_iterable = packages_dir.iterdir()
print("Showing hash values of files to be uploaded:") print("Showing hash values of files to be uploaded:")
for file in file_list: for file_object in file_iterable:
print(file) print(file_object)
print("") print("")
with open(os.path.abspath(os.path.join(os.getenv("INPUT_PACKAGES_DIR"), file)), "rb") as f: content = file_object.read_bytes()
content = f.read()
sha256.update(content) sha256.update(content)
md5.update(content) md5.update(content)

View file

@ -45,7 +45,7 @@ if [[ ${INPUT_VERBOSE,,} != "false" ]] ; then
fi fi
if [[ ${INPUT_PRINT_HASH,,} != "false" || ${INPUT_VERBOSE,,} != "false" ]] ; then if [[ ${INPUT_PRINT_HASH,,} != "false" || ${INPUT_VERBOSE,,} != "false" ]] ; then
python /app/print-hash.py python /app/print-hash.py "${INPUT_PACKAGES_DIR%%/}"
fi fi
TWINE_USERNAME="$INPUT_USER" \ TWINE_USERNAME="$INPUT_USER" \