Allow passing the module object itself to json.use().

cmlenz · cmlenz · commit ef2b55fdbe9c · 2009-07-01T14:51:13.000Z
--HG--
extra : convert_revision : svn%3A7a298fb0-333a-0410-83e7-658617cd9cf3/trunk%40172
diff --git a/couchdb/json.py b/couchdb/json.py
@@ -79,15 +79,18 @@ def use(module=None, decode=None, encode=None):
     corresponding JSON string. Exceptions raised by decoding and encoding
     should be propagated up unaltered.
     
-    :param module: the name of the JSON library module to use.
-    :type module: str
+    :param module: the name of the JSON library module to use, or the module
+                   object itself
+    :type module: str or module
     :param decode: a function for decoding JSON strings
     :type decode: callable
     :param encode: a function for encoding objects as JSON strings
     :type encode: callable
     """
     global _decode, _encode, _initialized, _using
     if module is not None:
+        if not isinstance(module, basestring):
+            module = module.__name__
         if module not in ('cjson', 'json', 'simplejson'):
             raise ValueError('Unsupported JSON module %s' % module)
         _using = module
