[Patch #476612] Add attributes from PEP247 to the md5 and sha modules

akuchling · akuchling · commit 75fec2c8edd4 · 2001-11-02T21:41:00.000Z
diff --git a/Modules/md5module.c b/Modules/md5module.c
@@ -161,6 +161,10 @@ static PyMethodDef md5_methods[] = {
 static PyObject *
 md5_getattr(md5object *self, char *name)
 {
+        if (strcmp(name, "digest_size") == 0) {
+    		return PyInt_FromLong(16);
+        }
+
 	return Py_FindMethod(md5_methods, (PyObject *)self, name);
 }
 
@@ -264,11 +268,13 @@ static PyMethodDef md5_functions[] = {
 DL_EXPORT(void)
 initmd5(void)
 {
-	PyObject *m, *d;
+	PyObject *m, *d, *i;
 
         MD5type.ob_type = &PyType_Type;
 	m = Py_InitModule3("md5", md5_functions, module_doc);
 	d = PyModule_GetDict(m);
 	PyDict_SetItemString(d, "MD5Type", (PyObject *)&MD5type);
+        if ( (i = PyInt_FromLong(16)) != NULL) 
+        	PyDict_SetItemString(d, "digest_size", i);
 	/* No need to check the error here, the caller will do that */
 }
diff --git a/Modules/shamodule.c b/Modules/shamodule.c
@@ -5,7 +5,7 @@
 /* See below for information about the original code this module was
    based upon. Additional work performed by:
 
-   Andrew Kuchling (amk1@bigfoot.com)
+   Andrew Kuchling (akuchlin@mems-exchange.org)
    Greg Stein (gstein@lyra.org)
 */
 
@@ -458,8 +458,8 @@ SHA_getattr(PyObject *self, char *name)
 {
     if (strcmp(name, "blocksize")==0)
         return PyInt_FromLong(1);
-    if (strcmp(name, "digestsize")==0)
-        return PyInt_FromLong(20);
+    if (strcmp(name, "digest_size")==0 || strcmp(name, "digestsize")==0)
+    	return PyInt_FromLong(20);
 
     return Py_FindMethod(SHA_methods, self, name);
 }
@@ -542,4 +542,5 @@ initsha(void)
                                 functions require an integral number of
                                 blocks */ 
     insint("digestsize", 20);
+    insint("digest_size", 20);
 }

Original file line number	Diff line number	Diff line change
`@@ -161,6 +161,10 @@ static PyMethodDef md5_methods[] = {`
`161`	`161`	`static PyObject *`
`162`	`162`	`md5_getattr(md5object self, char name)`
`163`	`163`	`{`
	`164`	`+ if (strcmp(name, "digest_size") == 0) {`
	`165`	`+ return PyInt_FromLong(16);`
	`166`	`+ }`
	`167`	`+`
`164`	`168`	`return Py_FindMethod(md5_methods, (PyObject *)self, name);`
`165`	`169`	`}`
`166`	`170`
`@@ -264,11 +268,13 @@ static PyMethodDef md5_functions[] = {`
`264`	`268`	`DL_EXPORT(void)`
`265`	`269`	`initmd5(void)`
`266`	`270`	`{`
`267`		`- PyObject m, d;`
	`271`	`+ PyObject m, d, *i;`
`268`	`272`
`269`	`273`	`MD5type.ob_type = &PyType_Type;`
`270`	`274`	`m = Py_InitModule3("md5", md5_functions, module_doc);`
`271`	`275`	`d = PyModule_GetDict(m);`
`272`	`276`	`PyDict_SetItemString(d, "MD5Type", (PyObject *)&MD5type);`
	`277`	`+ if ( (i = PyInt_FromLong(16)) != NULL)`
	`278`	`+ PyDict_SetItemString(d, "digest_size", i);`
`273`	`279`	`/* No need to check the error here, the caller will do that */`
`274`	`280`	`}`