X-Git-Url: https://git.jsancho.org/?p=mojodb.git;a=blobdiff_plain;f=cursor.py;fp=cursor.py;h=ca9e2a4cc0f902dd8a3261b854ba35d64207b97e;hp=202c6c78aabb6429a8b8cd012b616e9f921a6169;hb=4a34db8a057d135225e70ede89b767e89f827c8f;hpb=2758cc7948327b7a0256a7ba3291024afea02552 diff --git a/cursor.py b/cursor.py index 202c6c7..ca9e2a4 100644 --- a/cursor.py +++ b/cursor.py @@ -89,7 +89,9 @@ class Cursor(object): query['where'] = [] for k, v in self.spec.iteritems(): table_f = '%s$%s' % (self.collection.table_name, k) - if type(v) in (int, float): + if k == u'_id': + query['where'].append(((table_id, 'id'), '=', cPickle.dumps(v))) + elif type(v) in (int, float): field_q = self._get_cursor_field(table_id, table_f, field_name='number') query['where'].append((field_q, '=', v)) else: @@ -115,9 +117,9 @@ class Cursor(object): raise StopIteration else: document = {} - if '_id' in self.fields: - document['_id'] = res[0] - fields_without_id = filter(lambda x: x != '_id', self.fields) + if u'_id' in self.fields: + document[u'_id'] = cPickle.loads(res[0]) + fields_without_id = filter(lambda x: x != u'_id', self.fields) for i in xrange(len(fields_without_id)): if not res[i + 1] is None: document[fields_without_id[i]] = cPickle.loads(res[i + 1])