class UserFacade(BaseFacade):
def __init__(self, context):
self._repository = UserRepository()
super(UserFacade, self).__init__(context)
def get(self, entity_id):
entity = self._repository.get(entity_id)
if entity is None:
return None
return obj_map(entity, UserViewModel, rules={
'photos':lambda e: map(lambda f:obj_map(f, UploadFileViewModel), e.uploadFiles)
})
@staticmethod
def get_from_auth_id(auth_id):
entity = UserRepository().get_from_auth_id(auth_id)
if entity is None:
return None
return obj_map(entity, UserViewModel, rules={
'photos':lambda e: map(lambda f:obj_map(f, UploadFileViewModel), e.uploadFiles)
})
def save(self, viewModel):
entity = obj_map(viewModel, User)
self._repository.save(entity)
def add_file(self, file_id, file_type):
entity = self.get(self._context.user_id)
if entity is None:
return
return self._repository.add_file(self._context.user_id, file_id, file_type)
def delete_all_files(self):
entity = self.get(self._context.user_id)
if entity is None:
return
return self._repository.delete_all_files(self._context.user_id)
def __init__(self, context):
self._repository = UserRepository()
super(UserFacade, self).__init__(context)
