darkhelm/notifications-api
1
0
Fork 0
mirror of https://github.com/GSA/notifications-api.git synced 2026-02-02 17:31:14 -05:00
Code Issues Packages Projects Releases Wiki Activity

add env check before purging data in command

Browse Source
This commit is contained in:
stvnrlly
2022-10-20 14:04:38 -04:00
parent cb9e098c65
commit 2ce19fd502
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
app/commands.py
View File

@@ -113,6 +113,10 @@ def purge_functional_test_data(user_email_prefix):
users, services, etc. Give an email prefix. Probably "notify-tests-preview". users, services, etc. Give an email prefix. Probably "notify-tests-preview".
""" """
if os.getenv('NOTIFY_ENVIRONMENT', '') not in ['development', 'test']:
current_app.logger.error('Can only be run in development')
return
users = User.query.filter(User.email_address.like("{}%".format(user_email_prefix))).all() users = User.query.filter(User.email_address.like("{}%".format(user_email_prefix))).all()
for usr in users: for usr in users:
# Make sure the full email includes a uuid in it # Make sure the full email includes a uuid in it