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

Use if/else to avoid multiple queries

Browse Source
This commit is contained in:
leigh-mil 2019-03-26 10:36:23 -04:00
parent b5571000fe
commit 0d30b81ec5
1 changed files with 7 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
atst/domain/authz/decorator.py
View File

@ -13,23 +13,23 @@ from atst.domain.exceptions import UnauthorizedError
def check_access(permission, message, exception, *args, **kwargs): def check_access(permission, message, exception, *args, **kwargs):
access_args = {"message": message} access_args = {"message": message}
if "portfolio_id" in kwargs:
access_args["portfolio"] = Portfolios.get(
g.current_user, kwargs["portfolio_id"]
)
if "application_id" in kwargs: if "application_id" in kwargs:
application = Applications.get(kwargs["application_id"]) application = Applications.get(kwargs["application_id"])
access_args["portfolio"] = application.portfolio access_args["portfolio"] = application.portfolio
if "task_order_id" in kwargs: elif "task_order_id" in kwargs:
task_order = TaskOrders.get(kwargs["task_order_id"]) task_order = TaskOrders.get(kwargs["task_order_id"])
access_args["portfolio"] = task_order.portfolio access_args["portfolio"] = task_order.portfolio
if "token" in kwargs: elif "token" in kwargs:
invite = Invitations._get(kwargs["token"]) invite = Invitations._get(kwargs["token"])
access_args["portfolio"] = invite.portfolio_role.portfolio access_args["portfolio"] = invite.portfolio_role.portfolio
elif "portfolio_id" in kwargs:
access_args["portfolio"] = Portfolios.get(
g.current_user, kwargs["portfolio_id"]
)
if exception is not None and exception(g.current_user, **access_args, **kwargs): if exception is not None and exception(g.current_user, **access_args, **kwargs):
return True return True