add a validate method to HashedStream

1 files changed, 1 insertions, 4 deletions

diff --git a/importpkg.py b/importpkg.py
index 2cdefc4..f72cf03 100755
--- a/importpkg.py
+++ b/importpkg.py
@@ -107,10 +107,7 @@ def main():
         stdin = HashedStream(stdin, hashlib.sha256())
     process_package(stdin, hash_functions, dumper.represent)
     if options.hash:
-        while stdin.read(4096):
-            pass
-        if stdin.hexdigest() != options.hash:
-            raise ValueError("hash sum mismatch")
+        stdin.validate(options.hash)
     dumper.represent("commit")
     dumper.close()