Skip to content
This repository has been archived by the owner on Apr 26, 2024. It is now read-only.

Commit

Permalink
Add a flag to the synapse_review_recent_signups script to ignore an…
Browse files Browse the repository at this point in the history
…d filter appservice users. (#11675)
  • Loading branch information
lukasdenk committed Jan 17, 2022
1 parent cefd4b8 commit d8be992
Show file tree
Hide file tree
Showing 3 changed files with 23 additions and 8 deletions.
1 change: 1 addition & 0 deletions changelog.d/11675.feature
Original file line number Diff line number Diff line change
@@ -0,0 +1 @@
Add a flag to the `synapse_review_recent_signups` script to ignore and filter appservice users.
18 changes: 15 additions & 3 deletions synapse/_scripts/review_recent_signups.py
Original file line number Diff line number Diff line change
Expand Up @@ -46,7 +46,9 @@ class UserInfo:
ips: List[str] = attr.Factory(list)


def get_recent_users(txn: LoggingTransaction, since_ms: int) -> List[UserInfo]:
def get_recent_users(
txn: LoggingTransaction, since_ms: int, exclude_app_service: bool
) -> List[UserInfo]:
"""Fetches recently registered users and some info on them."""

sql = """
Expand All @@ -56,6 +58,9 @@ def get_recent_users(txn: LoggingTransaction, since_ms: int) -> List[UserInfo]:
AND deactivated = 0
"""

if exclude_app_service:
sql += " AND appservice_id IS NULL"

txn.execute(sql, (since_ms / 1000,))

user_infos = [UserInfo(user_id, creation_ts) for user_id, creation_ts in txn]
Expand Down Expand Up @@ -113,14 +118,20 @@ def main() -> None:
"-e",
"--exclude-emails",
action="store_true",
help="Exclude users that have validated email addresses",
help="Exclude users that have validated email addresses.",
)
parser.add_argument(
"-u",
"--only-users",
action="store_true",
help="Only print user IDs that match.",
)
parser.add_argument(
"-a",
"--exclude-app-service",
help="Exclude appservice users.",
action="store_true",
)

config = ReviewConfig()

Expand All @@ -133,6 +144,7 @@ def main() -> None:

since_ms = time.time() * 1000 - Config.parse_duration(config_args.since)
exclude_users_with_email = config_args.exclude_emails
exclude_users_with_appservice = config_args.exclude_app_service
include_context = not config_args.only_users

for database_config in config.database.databases:
Expand All @@ -143,7 +155,7 @@ def main() -> None:

with make_conn(database_config, engine, "review_recent_signups") as db_conn:
# This generates a type of Cursor, not LoggingTransaction.
user_infos = get_recent_users(db_conn.cursor(), since_ms) # type: ignore[arg-type]
user_infos = get_recent_users(db_conn.cursor(), since_ms, exclude_users_with_appservice) # type: ignore[arg-type]

for user_info in user_infos:
if exclude_users_with_email and user_info.emails:
Expand Down
12 changes: 7 additions & 5 deletions synapse/handlers/register.py
Original file line number Diff line number Diff line change
Expand Up @@ -979,16 +979,18 @@ async def _register_email_threepid(
if (
self.hs.config.email.email_enable_notifs
and self.hs.config.email.email_notif_for_new_users
and token
):
# Pull the ID of the access token back out of the db
# It would really make more sense for this to be passed
# up when the access token is saved, but that's quite an
# invasive change I'd rather do separately.
user_tuple = await self.store.get_user_by_access_token(token)
# The token better still exist.
assert user_tuple
token_id = user_tuple.token_id
if token:
user_tuple = await self.store.get_user_by_access_token(token)
# The token better still exist.
assert user_tuple
token_id = user_tuple.token_id
else:
token_id = None

await self.pusher_pool.add_pusher(
user_id=user_id,
Expand Down

0 comments on commit d8be992

Please sign in to comment.