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):
        if not isinstance(value, list):
            return self.adapter.object_id(value)
        values = self._represent_list(value)
        return list(map(self.adapter.object_id, values))