summaryrefslogtreecommitdiff
path: root/integrity.py
diff options
context:
space:
mode:
Diffstat (limited to 'integrity.py')
-rw-r--r--integrity.py34
1 files changed, 0 insertions, 34 deletions
diff --git a/integrity.py b/integrity.py
deleted file mode 100644
index 14bfe66..0000000
--- a/integrity.py
+++ /dev/null
@@ -1,34 +0,0 @@
-import sys
-import hashlib
-import base64
-from pathlib import Path
-import requests
-
-def file_hash(file_path: Path) -> str:
- sha256_hash = hashlib.sha256()
- with open(file_path, "rb") as file:
- while True:
- data = file.read(65536)
- if not data:
- break
- sha256_hash.update(data)
- return "sha256-" + base64.b64encode(sha256_hash.digest()).decode()
-
-def url_hash(url: str) -> str:
- sha256_hash = hashlib.sha256()
- with requests.get(url, stream=True) as resp:
- resp.raise_for_status()
- while True:
- data = resp.raw.read(65536)
- if not data:
- break
- sha256_hash.update(data)
- return "sha256-" + base64.b64encode(sha256_hash.digest()).decode()
-
-if __name__ == "__main__" and len(sys.argv) > 1:
- path = Path(sys.argv[1])
- if path.exists():
- print(file_hash(path))
- else:
- print(url_hash(sys.argv[1]))
-