X-Git-Url: https://osm.etsi.org/gitweb/?a=blobdiff_plain;ds=sidebyside;f=authosm%2Fmodels.py;h=74c1f1eb396be5da893f6bbf240784da31816b95;hb=refs%2Ftags%2Fv6.0.4;hp=d03e6d25b15f314be7a1ff8f26b971fe9ca77fca;hpb=8efc923308ffeb9800e75c7b74db0faabd82e20e;p=osm%2FLW-UI.git diff --git a/authosm/models.py b/authosm/models.py index d03e6d2..74c1f1e 100644 --- a/authosm/models.py +++ b/authosm/models.py @@ -69,6 +69,7 @@ class AbstractOsmUser(AbstractBaseUser, PermissionsMixin): psw = models.CharField(_('psw'), max_length=36) token = models.CharField(_('token'), max_length=255) project_id = models.CharField(_('project_id'), max_length=36) + project_name = models.CharField(_('project_name'), max_length=36, default='') token_expires = models.FloatField(_('token_expires'), max_length=36) objects = OsmUserManager() @@ -91,11 +92,11 @@ class AbstractOsmUser(AbstractBaseUser, PermissionsMixin): def get_projects(self): client = Client() - result = client.get_user_info(self.get_token(), self.username) - if 'error' in result and result['error'] is True: + user_info = client.get_user_info(self.get_token(), self.username) + if 'error' in user_info and user_info['error'] is True: return [] else: - return result['data']['projects'] + return user_info['data']['project_role_mappings'] def switch_project(self, project_id): client = Client() @@ -105,6 +106,7 @@ class AbstractOsmUser(AbstractBaseUser, PermissionsMixin): else: self.token = result['data']['id'] self.project_id = result['data']['project_id'] + self.project_name = result['data']['project_name'] self.token_expires = result['data']['expires'] self.save() return True