diff --git a/orm_mongodb.py b/orm_mongodb.py index b20bdc5..714228c 100644 --- a/orm_mongodb.py +++ b/orm_mongodb.py @@ -28,6 +28,7 @@ import gridfs from bson.objectid import ObjectId from datetime import datetime +from numbers import Number #mongodb stuff try: @@ -232,7 +233,7 @@ def read(self, cr, user, ids, fields=None, context=None, if not fields: fields = self._columns.keys() select = ids - if isinstance(ids, (int, long)): + if isinstance(ids, Number): select = [ids] result = self._read_flat(cr, user, select, fields, context, load) @@ -251,7 +252,7 @@ def read(self, cr, user, ids, fields=None, context=None, else: continue - if isinstance(ids, (int, long)): + if isinstance(ids, Number): return result and result[0] or False return result