add as_of_time to mqlread*

git-svn-id: http://freebase-python.googlecode.com/svn/trunk@81 5914aa95-5b3a-0410-a3b5-7b719e7fe9b2

1 files changed, 12 insertions, 3 deletions

diff --git a/freebase-api/freebase/api/session.py b/freebase-api/freebase/api/session.py
index 511547c..e6b9c00 100644
--- a/freebase-api/freebase/api/session.py
+++ b/freebase-api/freebase/api/session.py
@@ -396,13 +396,16 @@ class HTTPMetawebSession(MetawebSession):
         self.log.debug('LOGIN COOKIES: %s', self.cookiejar)
 
 
-    def mqlreaditer(self, sq):
+    def mqlreaditer(self, sq, asof=None):
         """read a structure query"""
 
         cursor = True
 
         while 1:
             subq = dict(query=[sq], cursor=cursor, escape=False)
+            if asof:
+                subq['as_of_time'] = asof
+
             qstr = simplejson.dumps(subq)
 
             service = '/api/service/mqlread'
@@ -418,9 +421,12 @@ class HTTPMetawebSession(MetawebSession):
             else:
                 return
 
-    def mqlread(self, sq):
+    def mqlread(self, sq, asof=None):
         """read a structure query"""
         subq = dict(query=sq, escape=False)
+        if asof:
+            subq['as_of_time'] = asof
+
         if isinstance(sq, list):
             subq['cursor'] = True
 
@@ -435,12 +441,15 @@ class HTTPMetawebSession(MetawebSession):
 
         return self._mqlresult(r)
 
-    def mqlreadmulti(self, queries):
+    def mqlreadmulti(self, queries, asof=None):
         """read a structure query"""
         keys = [('q%d' % i) for i,v in enumerate(queries)];
         envelope = {}
         for i,sq in enumerate(queries):
             subq = dict(query=sq, escape=False)
+            if asof:
+                subq['as_of_time'] = asof
+
             # XXX put this back once mqlreadmulti is working in general
             #if isinstance(sq, list):
             #    subq['cursor'] = True