def add_user(self, user, identity):
r = TeamUserRelationship.get(team_id=self.id,
user_id=user.name)
if not r:
TeamUserRelationship.create(team_id=self.id,
user_id=user.name,
identity=identity)
return True
def remove_member(self, username):
r = TeamUserRelationship.get(team_id=self.id,
user_id=username)
if r:
if r.identity != TEAM_MEMBER:
return False
r.delete()
return True
def test_add_and_delete_team_user_relationship(self):
team_id = 1111
user_id = "chengeng"
identity = 2
rl = TeamUserRelationship.create(team_id=team_id,
user_id=user_id,
identity=identity)
relationship = TeamUserRelationship.get(team_id=team_id,
user_id=user_id)
ok_(rl.id == relationship.id)
rl.delete()
def test_add_and_delete_team_user_relationship(self):
team_id = 1111
user_id = "chengeng"
identity = 2
rl = TeamUserRelationship.create(team_id=team_id,
user_id=user_id,
identity=identity)
relationship = TeamUserRelationship.get(team_id=team_id,
user_id=user_id)
ok_(rl.id == relationship.id)
rl.delete()
def add_member(self, username):
r = TeamUserRelationship.get(team_id=self.id,
user_id=username)
if not r:
r = TeamUserRelationship.create(team_id=self.id,
user_id=username,
identity=TEAM_MEMBER)
return True
if r.identity != TEAM_MEMBER:
r.identity = TEAM_MEMBER
r.save()
return True
def remove_user(self, user):
r = TeamUserRelationship.get(team_id=self.id,
user_id=user.name)
if r:
r.delete()
return True
def had_joined(self, user_id):
r = TeamUserRelationship.get(team_id=self.id, user_id=user_id)
return r or False
def is_member(self, user_id):
r = TeamUserRelationship.get(team_id=self.id, user_id=user_id)
return True if r else False
def is_owner(self, user_id):
r = TeamUserRelationship.get(team_id=self.id, user_id=user_id)
return r and r.is_owner or False
