Fixing a deprecation warning

[doneit] / src / rtm_api.py

diff --git a/src/rtm_api.py b/src/rtm_api.py
index d513fc8..b617632 100644 (file)
--- a/src/rtm_api.py
+++ b/src/rtm_api.py
@@ -3,12 +3,15 @@
 Python library for Remember The Milk API
 
 @note For help, see http://www.rememberthemilk.com/services/api/methods/
+
+@bug Under random situations, the response comes back incomplete.  Maybe a race condition, but how?
 """
 
 import weakref
 import warnings
 import urllib
-from md5 import md5
+import urllib2
+import hashlib
 
 _use_simplejson = False
 try:
@@ -32,6 +35,10 @@ class RTMAPIError(RTMError):
        pass
 
 
+class RTMParseError(RTMError):
+       pass
+
+
 class AuthStateMachine(object):
        """If the state is in those setup for the machine, then return
        the datum sent.  Along the way, it is an automatic call if the
@@ -76,14 +83,14 @@ class RTMapi(object):
        def _sign(self, params):
                "Sign the parameters with MD5 hash"
                pairs = ''.join(['%s%s' % (k, v) for (k, v) in sortedItems(params)])
-               return md5(self._secret+pairs).hexdigest()
+               return hashlib.md5(self._secret+pairs).hexdigest()
 
        @staticmethod
        def open_url(url, queryArgs=None):
                if queryArgs:
                        url += '?' + urllib.urlencode(queryArgs)
                warnings.warn("Performing download of %s" % url, stacklevel=5)
-               return urllib.urlopen(url)
+               return urllib2.urlopen(url)
 
        def get(self, **params):
                "Get the XML response for the passed `params`."
@@ -91,12 +98,9 @@ class RTMapi(object):
                params['format'] = 'json'
                params['api_sig'] = self._sign(params)
 
-               json = self.open_url(SERVICE_URL, params).read()
-
-               if _use_simplejson:
-                       data = DottedDict('ROOT', simplejson.loads(json))
-               else:
-                       data = DottedDict('ROOT', safer_eval(json))
+               connection = self.open_url(SERVICE_URL, params)
+               json = connection.read()
+               data = DottedDict('ROOT', parse_json(json))
                rsp = data.rsp
 
                if rsp.stat == 'fail':
@@ -207,7 +211,21 @@ class DottedDict(object):
 
 
 def safer_eval(string):
-       return eval(string, {}, {})
+       try:
+               return eval(string, {}, {})
+       except SyntaxError, e:
+               print "="*60
+               print string
+               print "="*60
+               newE = RTMParseError("Error parseing json")
+               newE.error = e
+               raise newE
+
+
+if _use_simplejson:
+       parse_json = simplejson.loads
+else:
+       parse_json = safer_eval
 
 
 API = {