Skip to content

Commit

Permalink
fix: Impersonation suggestion (#27018)
Browse files Browse the repository at this point in the history
  • Loading branch information
benjackwhite authored Dec 19, 2024
1 parent 1b9b3e0 commit 877202d
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 9 deletions.
11 changes: 4 additions & 7 deletions posthog/middleware.py
Original file line number Diff line number Diff line change
Expand Up @@ -278,13 +278,10 @@ def can_switch_to_team(self, new_team: Team, request: HttpRequest):

# :KLUDGE: This is more inefficient than needed, doing several expensive lookups
# However this should be a rare operation!
if not user_access_control.check_access_level_for_object(new_team, "member"):
# Do something to indicate that they don't have access to the team...
return False

# :KLUDGE: This is more inefficient than needed, doing several expensive lookups
# However this should be a rare operation!
if user_permissions.team(new_team).effective_membership_level is None:
if (
not user_access_control.check_access_level_for_object(new_team, "member")
and user_permissions.team(new_team).effective_membership_level is None
):
if user.is_staff:
# Staff users get a popup with suggested users to log in as, facilating support
request.suggested_users_with_access = UserBasicSerializer( # type: ignore
Expand Down
4 changes: 2 additions & 2 deletions posthog/test/test_middleware.py
Original file line number Diff line number Diff line change
Expand Up @@ -164,7 +164,7 @@ def setUp(self):
def test_project_switched_when_accessing_dashboard_of_another_accessible_team(self):
dashboard = Dashboard.objects.create(team=self.second_team)

with self.assertNumQueries(self.base_app_num_queries + 7): # AutoProjectMiddleware adds 4 queries
with self.assertNumQueries(self.base_app_num_queries + 6): # AutoProjectMiddleware adds 4 queries
response_app = self.client.get(f"/dashboard/{dashboard.id}")
response_users_api = self.client.get(f"/api/users/@me/")
response_users_api_data = response_users_api.json()
Expand Down Expand Up @@ -282,7 +282,7 @@ def test_project_switched_when_accessing_cohort_of_another_accessible_team(self)
def test_project_switched_when_accessing_feature_flag_of_another_accessible_team(self):
feature_flag = FeatureFlag.objects.create(team=self.second_team, created_by=self.user)

with self.assertNumQueries(self.base_app_num_queries + 7):
with self.assertNumQueries(self.base_app_num_queries + 6):
response_app = self.client.get(f"/feature_flags/{feature_flag.id}")
response_users_api = self.client.get(f"/api/users/@me/")
response_users_api_data = response_users_api.json()
Expand Down

0 comments on commit 877202d

Please sign in to comment.