Add script for giving a user CCPO permissions from the command line

This commit is contained in:
leigh-mil 2019-11-20 16:27:13 -05:00
parent 3f146c7da8
commit 778cda0924

35
script/add_ccpo_user.py Normal file
View File

@ -0,0 +1,35 @@
import os
import sys
parent_dir = os.path.abspath(os.path.join(os.path.dirname(__file__), ".."))
sys.path.append(parent_dir)
from atst.app import make_config, make_app
from atst.domain.exceptions import NotFoundError
from atst.domain.users import Users
def grant_ccpo_perms(dod_id):
try:
user = Users.get_by_dod_id(dod_id)
if user.permission_sets:
print("%s (DoD ID: %s) already CCPO user." % (user.full_name, user.dod_id))
else:
Users.give_ccpo_perms(user)
print(
"CCPO permissions successfully granted to %s (DoD ID: %s)."
% (user.full_name, user.dod_id)
)
except NotFoundError:
print("User not found.")
if __name__ == "__main__":
config = make_config({"DISABLE_CRL_CHECK": True, "DEBUG": False})
app = make_app(config)
with app.app_context():
dod_id = sys.argv[1]
grant_ccpo_perms(dod_id)