mirror of
https://github.com/GSA/notifications-api.git
synced 2026-02-01 07:35:34 -05:00
Adding new data models for letter branding.
This commit is contained in:
@@ -50,6 +50,7 @@ from app.models import (
|
||||
Complaint,
|
||||
InvitedUser,
|
||||
TemplateFolder,
|
||||
LetterBranding
|
||||
)
|
||||
|
||||
|
||||
@@ -724,3 +725,14 @@ def create_template_folder(service, name='foo', parent=None):
|
||||
db.session.add(tf)
|
||||
db.session.commit()
|
||||
return tf
|
||||
|
||||
|
||||
def create_letter_branding(name='HM Government', filename='hm-government', domain=None, platform_default=True):
|
||||
test_domain_branding = LetterBranding(name=name,
|
||||
filename=filename,
|
||||
domain=domain,
|
||||
platform_default=platform_default
|
||||
)
|
||||
db.session.add(test_domain_branding)
|
||||
db.session.commit()
|
||||
return test_domain_branding
|
||||
|
||||
Reference in New Issue
Block a user