Update Cryptsy.py

ScriptProdigy · ScriptProdigy · commit a74cdde4e5c3 · 2013-06-28T17:54:15.000-04:00
Implemented post processing of returns. If a datetime exists but not a timestamp then one will be generated based on the datetime.
diff --git a/Cryptsy.py b/Cryptsy.py
@@ -4,11 +4,26 @@
 import time
 import hmac,hashlib
 
+def createTimeStamp(datestr, format="%Y-%m-%d %H:%M:%S"):
+    return time.mktime(time.strptime(datestr, format))
+
 class Cryptsy:
     def __init__(self, APIKey, Secret):
         self.APIKey = APIKey
         self.Secret = Secret
 
+    def post_process(self, before):
+        after = before
+
+        # Add timestamps if there isnt one but is a datetime
+        if(isinstance(after['return'], list)):
+            for x in xrange(0, len(after['return'])):
+                if(isinstance(after['return'][x], dict)):
+                    if('datetime' in after['return'][x] and 'timestamp' not in after['return'][x]):
+                        after['return'][x]['timestamp'] = float(createTimeStamp(after['return'][x]['datetime']))
+                        
+        return after
+
     def api_query(self, method, req={}):
         if(method=="marketdata" or method=="orderdata"):
             ret = urllib2.urlopen(urllib2.Request('https://www.cryptsy.com/api.php?method=' + method))
@@ -25,7 +40,8 @@ def api_query(self, method, req={}):
             }
 
             ret = urllib2.urlopen(urllib2.Request('https://www.cryptsy.com/api', post_data, headers))
-            return json.loads(ret.read())
+            jsonRet = json.loads(ret.read())
+            return self.post_process(jsonRet)
 
     def getMarketData(self):
         return self.api_query("marketdata")
