pk/atst
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Catch error if demo user does not exist

Browse Source
This commit is contained in:
Patrick Smith 2018-10-12 14:35:39 -04:00
parent f65474a129
commit 9d03c3e856
1 changed files with 7 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
script/remove_sample_data.py
View File

@ -19,6 +19,7 @@ from atst.models.workspace import Workspace
from atst.models.mixins import AuditableMixin
from atst.domain.environments import Environments
from atst.domain.exceptions import NotFoundError
from atst.domain.reports import MONTHLY_SPEND_AARDVARK, MONTHLY_SPEND_BELUGA
from atst.domain.requests import Requests
from atst.domain.users import Users
@ -41,8 +42,12 @@ dod_ids = [
def create_demo_workspace(name, data):
workspace_owner = Users.get_by_dod_id("6786786786") # Other
try:
workspace_owner = Users.get_by_dod_id("678678678") # Other
auditor = Users.get_by_dod_id("3453453453") # Sally
except NotFoundError:
print("Could not find demo users; will not create demo workspace {}".format(name))
return
request = RequestFactory.build(creator=workspace_owner)
request.task_order = TaskOrderFactory.build()