Small tweaks (#17168)

* Clean up repeated code in auth

* Ensure review status is migrated for anonymous users
This commit is contained in:
Josh Hawkins
2025-03-15 08:11:45 -05:00
committed by GitHub
parent 19342c8768
commit d87268acfe
2 changed files with 5 additions and 3 deletions

View File

@@ -253,12 +253,12 @@ def auth(request: Request):
# pass the user header value from the upstream proxy if a mapping is specified
# or use anonymous if none are specified
user_header = proxy_config.header_map.user
role_header = proxy_config.header_map.role
success_response.headers["remote-user"] = (
request.headers.get(user_header, default="anonymous")
if user_header
else "anonymous"
)
role_header = proxy_config.header_map.role
role = (
request.headers.get(role_header, default="viewer")

View File

@@ -62,11 +62,13 @@ def migrate(migrator, database, fake=False, **kwargs):
'SELECT "id" FROM "reviewsegment" WHERE "has_been_reviewed" = 1'
)
reviewed_segment_ids = [row[0] for row in cursor.fetchall()]
# also migrate for anonymous (unauthenticated users)
usernames = [user.username for user in all_users] + ["anonymous"]
for segment_id in reviewed_segment_ids:
for user in all_users:
for username in usernames:
UserReviewStatus.create(
user_id=user.username,
user_id=username,
review_segment=segment_id,
has_been_reviewed=True,
)