Use set for uniqueness

This commit is contained in:
Jason Robinson
2026-06-25 17:39:42 +03:00
parent 3e4301ad01
commit 5f761ab5b4
3 changed files with 10 additions and 10 deletions
+7 -7
View File
@@ -103,14 +103,14 @@ class ProfileHandler:
self._worker_locks = hs.get_worker_locks_handler()
async def _record_profile_updates(
self, user_id: UserID, updated_fields: list[str]
self, user_id: UserID, updated_fields: set[str]
) -> None:
"""
Record user profile updates to our stream updates table.
Args:
user_id: The user whose profile has had updates.
updated_fields: A list of the names of the fields that were updated.
updated_fields: A set of the names of the fields that were updated.
Returns:
None
@@ -300,7 +300,7 @@ class ProfileHandler:
await self.store.set_profile_displayname(target_user, displayname_to_set)
await self._record_profile_updates(
target_user,
[ProfileFields.DISPLAYNAME],
{ProfileFields.DISPLAYNAME},
)
profile = await self.store.get_profileinfo(target_user)
@@ -413,7 +413,7 @@ class ProfileHandler:
await self.store.set_profile_avatar_url(target_user, avatar_url_to_set)
await self._record_profile_updates(
target_user,
[ProfileFields.AVATAR_URL],
{ProfileFields.AVATAR_URL},
)
profile = await self.store.get_profileinfo(target_user)
@@ -545,7 +545,7 @@ class ProfileHandler:
await self.store.delete_profile(target_user)
await self._record_profile_updates(
target_user, [field_name for field_name, _value in profile_updates]
target_user, {field_name for field_name, _value in profile_updates}
)
await self._third_party_rules.on_profile_update(
@@ -746,7 +746,7 @@ class ProfileHandler:
raise AuthError(403, "Cannot set another user's profile")
await self.store.set_profile_field(target_user, field_name, new_value)
await self._record_profile_updates(target_user, [field_name])
await self._record_profile_updates(target_user, {field_name})
# Custom fields do not propagate into the user directory *or* rooms.
profile = await self.store.get_profileinfo(target_user)
@@ -782,7 +782,7 @@ class ProfileHandler:
raise AuthError(400, "Cannot set another user's profile")
await self.store.delete_profile_field(target_user, field_name)
await self._record_profile_updates(target_user, [field_name])
await self._record_profile_updates(target_user, {field_name})
# Custom fields do not propagate into the user directory *or* rooms.
profile = await self.store.get_profileinfo(target_user)
+1 -1
View File
@@ -592,7 +592,7 @@ class ProfileWorkerStore(SQLBaseStore):
self,
user_id: UserID,
action: str,
updated_fields: list[str] | None,
updated_fields: set[str] | None,
) -> int:
"""Persist profile update markers and return the last stream ID."""
assert self._can_write_to_profile_updates
+2 -2
View File
@@ -174,7 +174,7 @@ class ProfileStoreTestCase(unittest.HomeserverTestCase):
stream_id = self.get_success(
self.store.add_profile_updates(
user_id=UserID.from_string("@user:test"),
updated_fields=[next(field_name_gen)],
updated_fields={next(field_name_gen)},
action=ProfileUpdateAction.UPDATE.value,
)
)
@@ -202,7 +202,7 @@ class ProfileStoreTestCase(unittest.HomeserverTestCase):
stream_id = self.get_success(
self.store.add_profile_updates(
user_id=UserID.from_string("@user:test"),
updated_fields=[next(field_name_gen)],
updated_fields={next(field_name_gen)},
action=ProfileUpdateAction.UPDATE.value,
)
)