larxius commited on
Commit
6ed7332
·
verified ·
1 Parent(s): 3f34eb4

Update backend_structured/routes.py

Browse files
Files changed (1) hide show
  1. backend_structured/routes.py +3 -1
backend_structured/routes.py CHANGED
@@ -1010,6 +1010,7 @@ def impersonate_org(current_user, org_id):
1010
  db.session.commit()
1011
 
1012
  tokens = _generate_tokens(target_user.id)
 
1013
  return jsonify({
1014
  'message': f'Impersonating {target_user.email}',
1015
  'access_token': tokens['access_token'],
@@ -1017,7 +1018,8 @@ def impersonate_org(current_user, org_id):
1017
  'id': target_user.id,
1018
  'email': target_user.email,
1019
  'role': target_user.role,
1020
- 'org_id': target_user.org_id
 
1021
  }
1022
  }), 200
1023
 
 
1010
  db.session.commit()
1011
 
1012
  tokens = _generate_tokens(target_user.id)
1013
+ target_org = db.session.get(Organization, target_user.org_id) if target_user.org_id else None
1014
  return jsonify({
1015
  'message': f'Impersonating {target_user.email}',
1016
  'access_token': tokens['access_token'],
 
1018
  'id': target_user.id,
1019
  'email': target_user.email,
1020
  'role': target_user.role,
1021
+ 'org_id': target_user.org_id,
1022
+ 'org_name': target_org.name if target_org else 'Organization'
1023
  }
1024
  }), 200
1025