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

reformat

Browse Source
This commit is contained in:
Kenneth Kehl
2023-08-29 14:54:30 -07:00
parent 19dcd7a48b
commit 1ecb747c6d
588 changed files with 34100 additions and 23589 deletions
Download Patch File Download Diff File Expand all files Collapse all files

485
app/commands.py
View File

@@ -66,7 +66,7 @@ from app.models import (
from app.utils import get_midnight_in_utc
@click.group(name='command', help='Additional commands')
@click.group(name="command", help="Additional commands")
def command_group():
pass
@@ -83,7 +83,7 @@ class notify_command:
# in the test environment the app context is already provided and having
# another will lead to the test db connection being closed prematurely
if getenv('NOTIFY_ENVIRONMENT', '') != 'test':
if getenv("NOTIFY_ENVIRONMENT", "") != "test":
# with_appcontext ensures the config is loaded, db connected, etc.
decorators.insert(0, flask.cli.with_appcontext)
@@ -99,27 +99,38 @@ class notify_command:
@notify_command()
@click.option('-u', '--user_email_prefix', required=True, help="""
@click.option(
"-u",
"--user_email_prefix",
required=True,
help="""
Functional test user email prefix. eg "notify-test-preview"
""") # noqa
""",
) # noqa
def purge_functional_test_data(user_email_prefix):
"""
Remove non-seeded functional test data
users, services, etc. Give an email prefix. Probably "notify-tests-preview".
"""
if getenv('NOTIFY_ENVIRONMENT', '') not in ['development', 'test']:
current_app.logger.error('Can only be run in development')
if 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:
# Make sure the full email includes a uuid in it
# Just in case someone decides to use a similar email address.
try:
uuid.UUID(usr.email_address.split("@")[0].split('+')[1])
uuid.UUID(usr.email_address.split("@")[0].split("+")[1])
except ValueError:
print("Skipping {} as the user email doesn't contain a UUID.".format(usr.email_address))
print(
"Skipping {} as the user email doesn't contain a UUID.".format(
usr.email_address
)
)
else:
services = dao_fetch_all_services_by_user(usr.id)
if services:
@@ -127,7 +138,9 @@ def purge_functional_test_data(user_email_prefix):
for service in services:
delete_service_and_all_associated_db_objects(service)
else:
services_created_by_this_user = dao_fetch_all_services_created_by_user(usr.id)
services_created_by_this_user = dao_fetch_all_services_created_by_user(
usr.id
)
if services_created_by_this_user:
# user is not part of any services but may still have been the one to create the service
# sometimes things get in this state if the tests fail half way through
@@ -141,9 +154,13 @@ def purge_functional_test_data(user_email_prefix):
delete_model_user(usr)
@notify_command(name='insert-inbound-numbers')
@click.option('-f', '--file_name', required=True,
help="""Full path of the file to upload, file is a contains inbound numbers, one number per line.""")
@notify_command(name="insert-inbound-numbers")
@click.option(
"-f",
"--file_name",
required=True,
help="""Full path of the file to upload, file is a contains inbound numbers, one number per line.""",
)
def insert_inbound_numbers_from_file(file_name):
# TODO maintainability what is the purpose of this command? Who would use it and why?
@@ -163,34 +180,41 @@ def setup_commands(application):
application.cli.add_command(command_group)
@notify_command(name='rebuild-ft-billing-for-day')
@click.option('-s', '--service_id', required=False, type=click.UUID)
@click.option('-d', '--day', help="The date to recalculate, as YYYY-MM-DD", required=True,
type=click_dt(format='%Y-%m-%d'))
@notify_command(name="rebuild-ft-billing-for-day")
@click.option("-s", "--service_id", required=False, type=click.UUID)
@click.option(
"-d",
"--day",
help="The date to recalculate, as YYYY-MM-DD",
required=True,
type=click_dt(format="%Y-%m-%d"),
)
def rebuild_ft_billing_for_day(service_id, day):
# TODO maintainability what is the purpose of this command? Who would use it and why?
"""
Rebuild the data in ft_billing for the given service_id and date
"""
def rebuild_ft_data(process_day, service):
deleted_rows = delete_billing_data_for_service_for_day(process_day, service)
current_app.logger.info('deleted {} existing billing rows for {} on {}'.format(
deleted_rows,
service,
process_day
))
transit_data = fetch_billing_data_for_day(process_day=process_day, service_id=service)
current_app.logger.info(
"deleted {} existing billing rows for {} on {}".format(
deleted_rows, service, process_day
)
)
transit_data = fetch_billing_data_for_day(
process_day=process_day, service_id=service
)
# transit_data = every row that should exist
for data in transit_data:
# upsert existing rows
update_fact_billing(data, process_day)
current_app.logger.info('added/updated {} billing rows for {} on {}'.format(
len(transit_data),
service,
process_day
))
current_app.logger.info(
"added/updated {} billing rows for {} on {}".format(
len(transit_data), service, process_day
)
)
if service_id:
# confirm the service exists
@@ -198,21 +222,37 @@ def rebuild_ft_billing_for_day(service_id, day):
rebuild_ft_data(day, service_id)
else:
services = get_service_ids_that_need_billing_populated(
get_midnight_in_utc(day),
get_midnight_in_utc(day + timedelta(days=1))
get_midnight_in_utc(day), get_midnight_in_utc(day + timedelta(days=1))
)
for row in services:
rebuild_ft_data(day, row.service_id)
@notify_command(name='bulk-invite-user-to-service')
@click.option('-f', '--file_name', required=True,
help="Full path of the file containing a list of email address for people to invite to a service")
@click.option('-s', '--service_id', required=True, help='The id of the service that the invite is for')
@click.option('-u', '--user_id', required=True, help='The id of the user that the invite is from')
@click.option('-a', '--auth_type', required=False,
help='The authentication type for the user, sms_auth or email_auth. Defaults to sms_auth if not provided')
@click.option('-p', '--permissions', required=True, help='Comma separated list of permissions.')
@notify_command(name="bulk-invite-user-to-service")
@click.option(
"-f",
"--file_name",
required=True,
help="Full path of the file containing a list of email address for people to invite to a service",
)
@click.option(
"-s",
"--service_id",
required=True,
help="The id of the service that the invite is for",
)
@click.option(
"-u", "--user_id", required=True, help="The id of the user that the invite is from"
)
@click.option(
"-a",
"--auth_type",
required=False,
help="The authentication type for the user, sms_auth or email_auth. Defaults to sms_auth if not provided",
)
@click.option(
"-p", "--permissions", required=True, help="Comma separated list of permissions."
)
def bulk_invite_user_to_service(file_name, service_id, user_id, auth_type, permissions):
# permissions
# manage_users | manage_templates | manage_settings
@@ -222,39 +262,61 @@ def bulk_invite_user_to_service(file_name, service_id, user_id, auth_type, permi
# view_activity
# "send_texts,send_emails,view_activity"
from app.service_invite.rest import create_invited_user
file = open(file_name)
for email_address in file:
data = {
'service': service_id,
'email_address': email_address.strip(),
'from_user': user_id,
'permissions': permissions,
'auth_type': auth_type,
'invite_link_host': current_app.config['ADMIN_BASE_URL']
"service": service_id,
"email_address": email_address.strip(),
"from_user": user_id,
"permissions": permissions,
"auth_type": auth_type,
"invite_link_host": current_app.config["ADMIN_BASE_URL"],
}
with current_app.test_request_context(
path='/service/{}/invite/'.format(service_id),
method='POST',
path="/service/{}/invite/".format(service_id),
method="POST",
data=json.dumps(data),
headers={"Content-Type": "application/json"}
headers={"Content-Type": "application/json"},
):
try:
response = create_invited_user(service_id)
if response[1] != 201:
print("*** ERROR occurred for email address: {}".format(email_address.strip()))
print(
"*** ERROR occurred for email address: {}".format(
email_address.strip()
)
)
print(response[0].get_data(as_text=True))
except Exception as e:
print("*** ERROR occurred for email address: {}. \n{}".format(email_address.strip(), e))
print(
"*** ERROR occurred for email address: {}. \n{}".format(
email_address.strip(), e
)
)
file.close()
@notify_command(name='archive-jobs-created-between-dates')
@click.option('-s', '--start_date', required=True, help="start date inclusive", type=click_dt(format='%Y-%m-%d'))
@click.option('-e', '--end_date', required=True, help="end date inclusive", type=click_dt(format='%Y-%m-%d'))
@notify_command(name="archive-jobs-created-between-dates")
@click.option(
"-s",
"--start_date",
required=True,
help="start date inclusive",
type=click_dt(format="%Y-%m-%d"),
)
@click.option(
"-e",
"--end_date",
required=True,
help="end date inclusive",
type=click_dt(format="%Y-%m-%d"),
)
def update_jobs_archived_flag(start_date, end_date):
current_app.logger.info('Archiving jobs created between {} to {}'.format(start_date, end_date))
current_app.logger.info(
"Archiving jobs created between {} to {}".format(start_date, end_date)
)
process_date = start_date
total_updated = 0
@@ -267,20 +329,29 @@ def update_jobs_archived_flag(start_date, end_date):
created_at >= (date :start + time '00:00:00')
and created_at < (date :end + time '00:00:00')
"""
result = db.session.execute(sql, {"start": process_date, "end": process_date + timedelta(days=1)})
result = db.session.execute(
sql, {"start": process_date, "end": process_date + timedelta(days=1)}
)
db.session.commit()
current_app.logger.info('jobs: --- Completed took {}ms. Archived {} jobs for {}'.format(
datetime.now() - start_time, result.rowcount, process_date))
current_app.logger.info(
"jobs: --- Completed took {}ms. Archived {} jobs for {}".format(
datetime.now() - start_time, result.rowcount, process_date
)
)
process_date += timedelta(days=1)
total_updated += result.rowcount
current_app.logger.info('Total archived jobs = {}'.format(total_updated))
current_app.logger.info("Total archived jobs = {}".format(total_updated))
@notify_command(name='populate-organizations-from-file')
@click.option('-f', '--file_name', required=True,
help="Pipe delimited file containing organization name, sector, agreement_signed, domains")
@notify_command(name="populate-organizations-from-file")
@click.option(
"-f",
"--file_name",
required=True,
help="Pipe delimited file containing organization name, sector, agreement_signed, domains",
)
def populate_organizations_from_file(file_name):
# [0] organization name:: name of the organization insert if organization is missing.
# [1] sector:: Federal | State only
@@ -291,29 +362,31 @@ def populate_organizations_from_file(file_name):
# The expectation is that the organization, organization_to_service
# and user_to_organization will be cleared before running this command.
# Ignoring duplicates allows us to run the command again with the same file or same file with new rows.
with open(file_name, 'r') as f:
with open(file_name, "r") as f:
def boolean_or_none(field):
if field == '1':
if field == "1":
return True
elif field == '0':
elif field == "0":
return False
elif field == '':
elif field == "":
return None
for line in itertools.islice(f, 1, None):
columns = line.split('|')
columns = line.split("|")
print(columns)
email_branding = None
email_branding_column = columns[5].strip()
if len(email_branding_column) > 0:
email_branding = EmailBranding.query.filter(EmailBranding.name == email_branding_column).one()
email_branding = EmailBranding.query.filter(
EmailBranding.name == email_branding_column
).one()
data = {
'name': columns[0],
'active': True,
'agreement_signed': boolean_or_none(columns[3]),
'organization_type': columns[1].lower(),
'email_branding_id': email_branding.id if email_branding else None
"name": columns[0],
"active": True,
"agreement_signed": boolean_or_none(columns[3]),
"organization_type": columns[1].lower(),
"email_branding_id": email_branding.id if email_branding else None,
}
org = Organization(**data)
try:
@@ -322,7 +395,7 @@ def populate_organizations_from_file(file_name):
except IntegrityError:
print("duplicate org", org.name)
db.session.rollback()
domains = columns[4].split(',')
domains = columns[4].split(",")
for d in domains:
if len(d.strip()) > 0:
domain = Domain(domain=d.strip(), organization_id=org.id)
@@ -334,10 +407,14 @@ def populate_organizations_from_file(file_name):
db.session.rollback()
@notify_command(name='populate-organization-agreement-details-from-file')
@click.option('-f', '--file_name', required=True,
help="CSV file containing id, agreement_signed_version, "
"agreement_signed_on_behalf_of_name, agreement_signed_at")
@notify_command(name="populate-organization-agreement-details-from-file")
@click.option(
"-f",
"--file_name",
required=True,
help="CSV file containing id, agreement_signed_version, "
"agreement_signed_on_behalf_of_name, agreement_signed_at",
)
def populate_organization_agreement_details_from_file(file_name):
"""
The input file should be a comma separated CSV file with a header row and 4 columns
@@ -355,7 +432,7 @@ def populate_organization_agreement_details_from_file(file_name):
org = dao_get_organization_by_id(row[0])
current_app.logger.info(f"Updating {org.name}")
if not org.agreement_signed:
raise RuntimeError('Agreement was not signed')
raise RuntimeError("Agreement was not signed")
org.agreement_signed_version = float(row[1])
org.agreement_signed_on_behalf_of_name = row[2].strip()
@@ -365,25 +442,31 @@ def populate_organization_agreement_details_from_file(file_name):
db.session.commit()
@notify_command(name='associate-services-to-organizations')
@notify_command(name="associate-services-to-organizations")
def associate_services_to_organizations():
services = Service.get_history_model().query.filter_by(
version=1
).all()
services = Service.get_history_model().query.filter_by(version=1).all()
for s in services:
created_by_user = User.query.filter_by(id=s.created_by_id).first()
organization = dao_get_organization_by_email_address(created_by_user.email_address)
organization = dao_get_organization_by_email_address(
created_by_user.email_address
)
service = dao_fetch_service_by_id(service_id=s.id)
if organization:
dao_add_service_to_organization(service=service, organization_id=organization.id)
dao_add_service_to_organization(
service=service, organization_id=organization.id
)
print("finished associating services to organizations")
@notify_command(name='populate-service-volume-intentions')
@click.option('-f', '--file_name', required=True,
help="Pipe delimited file containing service_id, SMS, email")
@notify_command(name="populate-service-volume-intentions")
@click.option(
"-f",
"--file_name",
required=True,
help="Pipe delimited file containing service_id, SMS, email",
)
def populate_service_volume_intentions(file_name):
# [0] service_id
# [1] SMS:: volume intentions for service
@@ -391,9 +474,9 @@ def populate_service_volume_intentions(file_name):
# TODO maintainability what is the purpose of this command? Who would use it and why?
with open(file_name, 'r') as f:
with open(file_name, "r") as f:
for line in itertools.islice(f, 1, None):
columns = line.split(',')
columns = line.split(",")
print(columns)
service = dao_fetch_service_by_id(columns[0])
service.volume_sms = columns[1]
@@ -402,14 +485,17 @@ def populate_service_volume_intentions(file_name):
print("populate-service-volume-intentions complete")
@notify_command(name='populate-go-live')
@click.option('-f', '--file_name', required=True, help='CSV file containing live service data')
@notify_command(name="populate-go-live")
@click.option(
"-f", "--file_name", required=True, help="CSV file containing live service data"
)
def populate_go_live(file_name):
# 0 - count, 1- Link, 2- Service ID, 3- DEPT, 4- Service Name, 5- Main contact,
# 6- Contact detail, 7-MOU, 8- LIVE date, 9- SMS, 10 - Email, 11 - Letters, 12 -CRM, 13 - Blue badge
import csv
print("Populate go live user and date")
with open(file_name, 'r') as f:
with open(file_name, "r") as f:
rows = csv.reader(
f,
quoting=csv.QUOTE_MINIMAL,
@@ -420,7 +506,7 @@ def populate_go_live(file_name):
print(index, row)
service_id = row[2]
go_live_email = row[6]
go_live_date = datetime.strptime(row[8], '%d/%m/%Y') + timedelta(hours=12)
go_live_date = datetime.strptime(row[8], "%d/%m/%Y") + timedelta(hours=12)
print(service_id, go_live_email, go_live_date)
try:
if go_live_email:
@@ -440,7 +526,7 @@ def populate_go_live(file_name):
dao_update_service(service)
@notify_command(name='fix-billable-units')
@notify_command(name="fix-billable-units")
def fix_billable_units():
query = Notification.query.filter(
Notification.notification_type == SMS_TYPE,
@@ -451,7 +537,9 @@ def fix_billable_units():
)
for notification in query.all():
template_model = dao_get_template_by_id(notification.template_id, notification.template_version)
template_model = dao_get_template_by_id(
notification.template_id, notification.template_version
)
template = SMSMessageTemplate(
template_model.__dict__,
@@ -459,26 +547,28 @@ def fix_billable_units():
prefix=notification.service.name,
show_prefix=notification.service.prefix_sms,
)
print("Updating notification: {} with {} billable_units".format(notification.id, template.fragment_count))
print(
"Updating notification: {} with {} billable_units".format(
notification.id, template.fragment_count
)
)
Notification.query.filter(
Notification.id == notification.id
).update(
Notification.query.filter(Notification.id == notification.id).update(
{"billable_units": template.fragment_count}
)
db.session.commit()
print("End fix_billable_units")
@notify_command(name='delete-unfinished-jobs')
@notify_command(name="delete-unfinished-jobs")
def delete_unfinished_jobs():
cleanup_unfinished_jobs()
print("End cleanup_unfinished_jobs")
@notify_command(name='process-row-from-job')
@click.option('-j', '--job_id', required=True, help='Job id')
@click.option('-n', '--job_row_number', type=int, required=True, help='Job id')
@notify_command(name="process-row-from-job")
@click.option("-j", "--job_id", required=True, help="Job id")
@click.option("-n", "--job_row_number", type=int, required=True, help="Job id")
def process_row_from_job(job_id, job_row_number):
job = dao_get_job_by_id(job_id)
db_template = dao_get_template_by_id(job.template_id, job.template_version)
@@ -486,19 +576,27 @@ def process_row_from_job(job_id, job_row_number):
template = db_template._as_utils_template()
for row in RecipientCSV(
s3.get_job_from_s3(str(job.service_id), str(job.id)),
template_type=template.template_type,
placeholders=template.placeholders
s3.get_job_from_s3(str(job.service_id), str(job.id)),
template_type=template.template_type,
placeholders=template.placeholders,
).get_rows():
if row.index == job_row_number:
notification_id = process_row(row, template, job, job.service)
current_app.logger.info("Process row {} for job {} created notification_id: {}".format(
job_row_number, job_id, notification_id))
current_app.logger.info(
"Process row {} for job {} created notification_id: {}".format(
job_row_number, job_id, notification_id
)
)
@notify_command(name='populate-annual-billing-with-the-previous-years-allowance')
@click.option('-y', '--year', required=True, type=int,
help="""The year to populate the annual billing data for, i.e. 2019""")
@notify_command(name="populate-annual-billing-with-the-previous-years-allowance")
@click.option(
"-y",
"--year",
required=True,
type=int,
help="""The year to populate the annual billing data for, i.e. 2019""",
)
def populate_annual_billing_with_the_previous_years_allowance(year):
"""
add annual_billing for given year.
@@ -518,20 +616,36 @@ def populate_annual_billing_with_the_previous_years_allowance(year):
where service_id = :service_id
order by financial_year_start desc limit 1
"""
free_allowance_rows = db.session.execute(latest_annual_billing, {"service_id": row.id})
free_allowance = [x[0]for x in free_allowance_rows]
print("create free limit of {} for service: {}".format(free_allowance[0], row.id))
dao_create_or_update_annual_billing_for_year(service_id=row.id,
free_sms_fragment_limit=free_allowance[0],
financial_year_start=int(year))
free_allowance_rows = db.session.execute(
latest_annual_billing, {"service_id": row.id}
)
free_allowance = [x[0] for x in free_allowance_rows]
print(
"create free limit of {} for service: {}".format(free_allowance[0], row.id)
)
dao_create_or_update_annual_billing_for_year(
service_id=row.id,
free_sms_fragment_limit=free_allowance[0],
financial_year_start=int(year),
)
@notify_command(name='populate-annual-billing-with-defaults')
@click.option('-y', '--year', required=True, type=int,
help="""The year to populate the annual billing data for, i.e. 2021""")
@click.option('-m', '--missing-services-only', default=True, type=bool,
help="""If true then only populate services missing from annual billing for the year.
If false populate the default values for all active services.""")
@notify_command(name="populate-annual-billing-with-defaults")
@click.option(
"-y",
"--year",
required=True,
type=int,
help="""The year to populate the annual billing data for, i.e. 2021""",
)
@click.option(
"-m",
"--missing-services-only",
default=True,
type=bool,
help="""If true then only populate services missing from annual billing for the year.
If false populate the default values for all active services.""",
)
def populate_annual_billing_with_defaults(year, missing_services_only):
"""
Add or update annual billing with free allowance defaults for all active services.
@@ -544,71 +658,85 @@ def populate_annual_billing_with_defaults(year, missing_services_only):
This is useful to ensure all services start the new year with the correct annual billing.
"""
if missing_services_only:
active_services = Service.query.filter(
Service.active
).outerjoin(
AnnualBilling, and_(Service.id == AnnualBilling.service_id, AnnualBilling.financial_year_start == year)
).filter(
AnnualBilling.id == None # noqa
).all()
active_services = (
Service.query.filter(Service.active)
.outerjoin(
AnnualBilling,
and_(
Service.id == AnnualBilling.service_id,
AnnualBilling.financial_year_start == year,
),
)
.filter(AnnualBilling.id == None) # noqa
.all()
)
else:
active_services = Service.query.filter(
Service.active
).all()
active_services = Service.query.filter(Service.active).all()
previous_year = year - 1
services_with_zero_free_allowance = db.session.query(AnnualBilling.service_id).filter(
AnnualBilling.financial_year_start == previous_year,
AnnualBilling.free_sms_fragment_limit == 0
).all()
services_with_zero_free_allowance = (
db.session.query(AnnualBilling.service_id)
.filter(
AnnualBilling.financial_year_start == previous_year,
AnnualBilling.free_sms_fragment_limit == 0,
)
.all()
)
for service in active_services:
# If a service has free_sms_fragment_limit for the previous year
# set the free allowance for this year to 0 as well.
# Else use the default free allowance for the service.
if service.id in [x.service_id for x in services_with_zero_free_allowance]:
print(f'update service {service.id} to 0')
print(f"update service {service.id} to 0")
dao_create_or_update_annual_billing_for_year(
service_id=service.id,
free_sms_fragment_limit=0,
financial_year_start=year
financial_year_start=year,
)
else:
print(f'update service {service.id} with default')
print(f"update service {service.id} with default")
set_default_free_allowance_for_service(service, year)
# We use noqa to protect this method from the vulture dead code check. Otherwise, the params ctx and param
# will trigger vulture and cause a build failure.
def validate_mobile(ctx, param, value): # noqa
if (len(''.join(i for i in value if i.isdigit())) != 10):
def validate_mobile(ctx, param, value): # noqa
if len("".join(i for i in value if i.isdigit())) != 10:
raise click.BadParameter("mobile number must have 10 digits")
else:
return value
@notify_command(name='create-test-user')
@click.option('-n', '--name', required=True, prompt=True)
@click.option('-e', '--email', required=True, prompt=True) # TODO: require valid email
@click.option('-m', '--mobile_number',
required=True, prompt=True, callback=validate_mobile)
@click.option('-p', '--password',
required=True, prompt=True, hide_input=True, confirmation_prompt=True)
@click.option('-a', '--auth_type', default="sms_auth")
@click.option('-s', '--state', default="active")
@click.option('-d', '--admin', default=False, type=bool)
@notify_command(name="create-test-user")
@click.option("-n", "--name", required=True, prompt=True)
@click.option("-e", "--email", required=True, prompt=True) # TODO: require valid email
@click.option(
"-m", "--mobile_number", required=True, prompt=True, callback=validate_mobile
)
@click.option(
"-p",
"--password",
required=True,
prompt=True,
hide_input=True,
confirmation_prompt=True,
)
@click.option("-a", "--auth_type", default="sms_auth")
@click.option("-s", "--state", default="active")
@click.option("-d", "--admin", default=False, type=bool)
def create_test_user(name, email, mobile_number, password, auth_type, state, admin):
if getenv('NOTIFY_ENVIRONMENT', '') not in ['development', 'test']:
current_app.logger.error('Can only be run in development')
if getenv("NOTIFY_ENVIRONMENT", "") not in ["development", "test"]:
current_app.logger.error("Can only be run in development")
return
data = {
'name': name,
'email_address': email,
'mobile_number': mobile_number,
'password': password,
'auth_type': auth_type,
'state': state, # skip the email verification for our test user
'platform_admin': admin,
"name": name,
"email_address": email,
"mobile_number": mobile_number,
"password": password,
"auth_type": auth_type,
"state": state, # skip the email verification for our test user
"platform_admin": admin,
}
user = User(**data)
try:
@@ -619,19 +747,23 @@ def create_test_user(name, email, mobile_number, password, auth_type, state, adm
db.session.rollback()
@notify_command(name='create-admin-jwt')
@notify_command(name="create-admin-jwt")
def create_admin_jwt():
if getenv('NOTIFY_ENVIRONMENT', '') != 'development':
current_app.logger.error('Can only be run in development')
if getenv("NOTIFY_ENVIRONMENT", "") != "development":
current_app.logger.error("Can only be run in development")
return
print(create_jwt_token(current_app.config['SECRET_KEY'], current_app.config['ADMIN_CLIENT_ID']))
print(
create_jwt_token(
current_app.config["SECRET_KEY"], current_app.config["ADMIN_CLIENT_ID"]
)
)
@notify_command(name='create-user-jwt')
@click.option('-t', '--token', required=True, prompt=False)
@notify_command(name="create-user-jwt")
@click.option("-t", "--token", required=True, prompt=False)
def create_user_jwt(token):
if getenv('NOTIFY_ENVIRONMENT', '') != 'development':
current_app.logger.error('Can only be run in development')
if getenv("NOTIFY_ENVIRONMENT", "") != "development":
current_app.logger.error("Can only be run in development")
return
service_id = token[-73:-37]
api_key = token[-36:]
@@ -639,25 +771,24 @@ def create_user_jwt(token):
def _update_template(id, name, template_type, content, subject):
template = Template.query.filter_by(id=id).first()
template.name = name
template.template_type = template_type
template.content = '\n'.join(content)
template.content = "\n".join(content)
template.subject = subject
history = TemplateHistory.query.filter_by(id=id).first()
history.name = name
history.template_type = template_type
history.content = '\n'.join(content)
history.content = "\n".join(content)
history.subject = subject
db.session.commit()
@notify_command(name='update-templates')
@notify_command(name="update-templates")
def update_templates():
with open(current_app.config['CONFIG_FILES'] + '/templates.json') as f:
with open(current_app.config["CONFIG_FILES"] + "/templates.json") as f:
data = json.load(f)
for d in data:
_update_template(d['id'], d['name'], d['type'], d['content'], d['subject'])
_update_template(d["id"], d["name"], d["type"], d["content"], d["subject"])