24 lines
728 B
Python
24 lines
728 B
Python
from atst.models.environment_role import EnvironmentRole
|
|
from atst.database import db
|
|
|
|
|
|
class EnvironmentRoles(object):
|
|
@classmethod
|
|
def get(cls, user_id, environment_id):
|
|
existing_env_role = (
|
|
db.session.query(EnvironmentRole)
|
|
.filter(
|
|
EnvironmentRole.user_id == user_id,
|
|
EnvironmentRole.environment_id == environment_id,
|
|
)
|
|
.one_or_none()
|
|
)
|
|
return existing_env_role
|
|
|
|
@classmethod
|
|
def delete(cls, user_id, environment_id):
|
|
existing_env_role = EnvironmentRoles.get(user_id, environment_id)
|
|
if existing_env_role:
|
|
db.session.delete(existing_env_role)
|
|
db.session.commit()
|