52 lines
1.6 KiB
Python
52 lines
1.6 KiB
Python
|
import datetime
|
||
|
|
from .._compat import PY2, basestring, to_bytes
|
||
|
|
from ..adapters.mongo import Mongo, MongoBlob
|
||
|
|
from ..helpers.classes import Reference
|
||
|
|
from ..objects import Row
|
||
|
|
from .base import NoSQLRepresenter
|
||
|
|
from . import representers, for_type
|
||
|
|
|
||
|
|
|
||
|
|
@representers.register_for(Mongo)
|
||
|
|
class MongoRepresenter(NoSQLRepresenter):
|
||
|
|
@for_type('id')
|
||
|
|
def _id(self, value):
|
||
|
|
return self.adapter.object_id(value)
|
||
|
|
|
||
|
|
@for_type('reference')
|
||
|
|
def _reference(self, value):
|
||
|
|
if isinstance(value, (Row, Reference)):
|
||
|
|
value = value['id']
|
||
|
|
return self.adapter.object_id(value)
|
||
|
|
|
||
|
|
@for_type('date')
|
||
|
|
def _date(self, value):
|
||
|
|
# this piece of data can be stripped off based on the fieldtype
|
||
|
|
t = datetime.time(0, 0, 0)
|
||
|
|
# mongodb doesn't have a date object and so it must datetime,
|
||
|
|
# string or integer
|
||
|
|
return datetime.datetime.combine(value, t)
|
||
|
|
|
||
|
|
@for_type('time')
|
||
|
|
def _time(self, value):
|
||
|
|
# this piece of data can be stripped off based on the fieldtype
|
||
|
|
d = datetime.date(2000, 1, 1)
|
||
|
|
# mongodb doesn't have a time object and so it must datetime,
|
||
|
|
# string or integer
|
||
|
|
return datetime.datetime.combine(d, value)
|
||
|
|
|
||
|
|
@for_type('datetime')
|
||
|
|
def _datetime(self, value):
|
||
|
|
return value
|
||
|
|
|
||
|
|
@for_type('blob')
|
||
|
|
def _blob(self, value):
|
||
|
|
if isinstance(value, basestring) and value == '':
|
||
|
|
value = None
|
||
|
|
return MongoBlob(value) if PY2 else to_bytes(value)
|
||
|
|
|
||
|
|
@for_type('list:reference')
|
||
|
|
def _list_reference(self, value):
|
||
|
|
values = self._represent_list(value)
|
||
|
|
return list(map(self.adapter.object_id, values))
|