Move some methods from the API client to the model

They make more sense being on the model, and it doesn’t make any sense
to duplicate them.
This commit is contained in:
Chris Hill-Scott
2019-06-07 09:14:14 +01:00
parent b233e3ca24
commit 88e36d6841
4 changed files with 8 additions and 15 deletions

View File

@@ -195,18 +195,5 @@ class UserApiClient(NotifyAdminAPIClient):
), [])
return sorted(all_services, key=lambda service: service['name'])
def user_has_live_services(self, user):
return any(
not service['restricted'] for service in self.get_services_for_user(user)
)
def get_service_ids_for_user(self, user):
return {
service['id'] for service in self.get_services_for_user(user)
}
def user_belongs_to_service(self, user, service_id):
return str(service_id) in self.get_service_ids_for_user(user)
user_api_client = UserApiClient()