aljaz/CTFd
1
0
Fork 0
mirror of https://github.com/aljazceru/CTFd.git synced 2025-12-17 14:04:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

Rough implementation of user registration limit

Browse Source
This commit is contained in:
Kevin Chung
2023-06-13 17:02:30 -04:00
parent dd20786b84
commit 5e8ff5d892
3 changed files with 29 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
CTFd/auth.py
View File

@@ -211,6 +211,14 @@ def register():
valid_email = validators.validate_email(email_address) valid_email = validators.validate_email(email_address)
team_name_email_check = validators.validate_email(name) team_name_email_check = validators.validate_email(name)
num_users_limit = int(get_config("num_users", default=0))
num_users = Users.query.filter_by(banned=False, hidden=False).count()
if num_users_limit and num_users >= num_users_limit:
abort(
403,
description=f"Reached the maximum number of users ({num_users_limit}).",
)
if get_config("registration_code"): if get_config("registration_code"):
if ( if (
registration_code.lower() registration_code.lower()
@@ -490,6 +498,15 @@ def oauth_redirect():
user = Users.query.filter_by(email=user_email).first() user = Users.query.filter_by(email=user_email).first()
if user is None: if user is None:
# Respect the user count limit
num_users_limit = int(get_config("num_users", default=0))
num_users = Users.query.filter_by(banned=False, hidden=False).count()
if num_users_limit and num_users >= num_users_limit:
abort(
403,
description=f"Reached the maximum number of users ({num_users_limit}).",
)
# Check if we are allowing registration before creating users # Check if we are allowing registration before creating users
if registration_visible() or mlc_registration(): if registration_visible() or mlc_registration():
user = Users( user = Users(

4
CTFd/forms/config.py
View File

@@ -48,6 +48,10 @@ class AccountSettingsForm(BaseForm):
widget=NumberInput(min=0), widget=NumberInput(min=0),
description="Max number of teams (Teams mode only)", description="Max number of teams (Teams mode only)",
) )
num_users = IntegerField(
widget=NumberInput(min=0),
description="Max number of users",
)
verify_emails = SelectField( verify_emails = SelectField(
"Verify Emails", "Verify Emails",
description="Control whether users must confirm their email addresses before playing", description="Control whether users must confirm their email addresses before playing",

8
CTFd/themes/admin/templates/configs/accounts.html
View File

@@ -46,6 +46,14 @@
</small> </small>
</div> </div>
<div class="form-group">
{{ form.num_users.label }}
{{ form.num_users(class="form-control", value=num_users) }}
<small class="form-text text-muted">
{{ form.num_users.description }}
</small>
</div>
<div class="form-group"> <div class="form-group">
{{ form.team_disbanding.label }} {{ form.team_disbanding.label }}
{{ form.team_disbanding(class="form-control", value=team_disbanding) }} {{ form.team_disbanding(class="form-control", value=team_disbanding) }}