@classmethod
def get_collection(cls):
    """Return a handle to the database collection for the class"""
    if cls._context_manager.is_draft:
        return getattr(cls.get_db(), '{collection}__draft__'.format(
            collection=cls._collection))
    return getattr(cls.get_db(), cls._collection)
