Move environment role method into domain

This commit is contained in:
Montana
2018-09-21 11:14:30 -04:00
parent 6d50c228af
commit e12c1f42db
4 changed files with 21 additions and 4 deletions

View File

@@ -0,0 +1,16 @@
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