Remove unused PortfolioRoles methods
This commit is contained in:
@@ -1,22 +1,13 @@
|
|||||||
from sqlalchemy.orm.exc import NoResultFound
|
from sqlalchemy.orm.exc import NoResultFound
|
||||||
|
|
||||||
from atst.database import db
|
from atst.database import db
|
||||||
from atst.models.portfolio_role import (
|
from atst.models.portfolio_role import PortfolioRole, Status as PortfolioRoleStatus
|
||||||
PortfolioRole,
|
|
||||||
Status as PortfolioRoleStatus,
|
|
||||||
MEMBER_STATUSES,
|
|
||||||
)
|
|
||||||
from atst.models.user import User
|
from atst.models.user import User
|
||||||
|
|
||||||
from .permission_sets import PermissionSets
|
from .permission_sets import PermissionSets
|
||||||
from .exceptions import NotFoundError
|
from .exceptions import NotFoundError
|
||||||
|
|
||||||
|
|
||||||
MEMBER_STATUS_CHOICES = [
|
|
||||||
dict(name=key, display_name=value) for key, value in MEMBER_STATUSES.items()
|
|
||||||
]
|
|
||||||
|
|
||||||
|
|
||||||
class PortfolioRoles(object):
|
class PortfolioRoles(object):
|
||||||
@classmethod
|
@classmethod
|
||||||
def get(cls, portfolio_id, user_id):
|
def get(cls, portfolio_id, user_id):
|
||||||
@@ -39,34 +30,6 @@ class PortfolioRoles(object):
|
|||||||
except NoResultFound:
|
except NoResultFound:
|
||||||
raise NotFoundError("portfolio_role")
|
raise NotFoundError("portfolio_role")
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def _get_active_portfolio_role(cls, portfolio_id, user_id):
|
|
||||||
try:
|
|
||||||
return (
|
|
||||||
db.session.query(PortfolioRole)
|
|
||||||
.join(User)
|
|
||||||
.filter(User.id == user_id, PortfolioRole.portfolio_id == portfolio_id)
|
|
||||||
.filter(PortfolioRole.status == PortfolioRoleStatus.ACTIVE)
|
|
||||||
.one()
|
|
||||||
)
|
|
||||||
except NoResultFound:
|
|
||||||
return None
|
|
||||||
|
|
||||||
@classmethod
|
|
||||||
def _get_portfolio_role(cls, user, portfolio_id):
|
|
||||||
try:
|
|
||||||
existing_portfolio_role = (
|
|
||||||
db.session.query(PortfolioRole)
|
|
||||||
.filter(
|
|
||||||
PortfolioRole.user == user,
|
|
||||||
PortfolioRole.portfolio_id == portfolio_id,
|
|
||||||
)
|
|
||||||
.one()
|
|
||||||
)
|
|
||||||
return existing_portfolio_role
|
|
||||||
except NoResultFound:
|
|
||||||
raise NotFoundError("portfolio role")
|
|
||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def add(cls, user, portfolio_id, permission_sets=None):
|
def add(cls, user, portfolio_id, permission_sets=None):
|
||||||
new_portfolio_role = None
|
new_portfolio_role = None
|
||||||
|
Reference in New Issue
Block a user