[gentoo-commits] proj/R_overlay:master commit in: roverlay/

[André Erdmann]

commit:     057e5f51ae4cd402a9a80105aa1dcd8993055a77
Author:     André Erdmann <dywi <AT> mailerd <DOT> de>
AuthorDate: Wed Aug  1 07:25:51 2012 +0000
Commit:     André Erdmann <dywi <AT> mailerd <DOT> de>
CommitDate: Wed Aug  1 07:25:51 2012 +0000
URL:        http://git.overlays.gentoo.org/gitweb/?p=proj/R_overlay.git;a=commit;h=057e5f51

digest.py

---
 roverlay/digest.py |   35 +++++++++++++++++++++++++++++++++++
 1 files changed, 35 insertions(+), 0 deletions(-)

diff --git a/roverlay/digest.py b/roverlay/digest.py
new file mode 100644
index 0000000..4ccbca2
--- /dev/null
+++ b/roverlay/digest.py
@@ -0,0 +1,35 @@
+import hashlib
+
+def md5sum_file ( fh, binary_digest=False ):
+	"""Returns the md5 sum for an already opened file."""
+	md5 = hashlib.md5()
+	blocksize = 16384
+
+	block = fh.read ( blocksize )
+	while block:
+		md5.update ( block )
+		block = fh.read ( blocksize )
+
+	return md5.digest() if binary_digest else md5.hexdigest()
+# --- end of md5sum_file (...) ---
+
+
+_DIGEST_MAP = dict (
+	md5 = md5sum_file,
+)
+
+def digest_supported ( digest_type ):
+	"""Returns True if the given digest type is supported, else False."""
+	return digest_type in _DIGEST_MAP
+# --- digest_supported (...) ---
+
+def dodigest_file ( _file, digest_type, binary_digest=False ):
+	ret = None
+	with open ( _file, mode='rb' ) as fh:
+		ret = _DIGEST_MAP [digest_type] ( fh, binary_digest=binary_digest )
+	return ret
+# --- end of dodigest_file (...) ---
+
+def digest_compare ( _file, digest, digest_type, binary_digest=False ):
+	return digest == dodigest_file ( _file, digest_type, binary_digest )
+# --- end of digest_compare (...) ---