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

use first_or_none utility for finding portfolio role

Browse Source
This commit is contained in:
dandds 2019-03-14 14:24:41 -04:00
parent ad56ddc60e
commit 79cb7ac2eb
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
atst/domain/authz.py
View File

@ -1,3 +1,4 @@
from atst.utils import first_or_none
from atst.models.permissions import Permissions
from atst.domain.exceptions import UnauthorizedError
@ -5,8 +6,8 @@ from atst.domain.exceptions import UnauthorizedError
class Authorization(object):
@classmethod
def has_portfolio_permission(cls, user, portfolio, permission):
port_role = next(
(pr for pr in user.portfolio_roles if pr.portfolio == portfolio), None
port_role = first_or_none(
lambda pr: pr.portfolio == portfolio, user.portfolio_roles
)
if port_role:
return permission in port_role.permissions