def remove_team_member_context(env):
team_id = int(flask.request.form.get('team_id'))
user_id = int(flask.request.form.get('user_id'))
if team_id is None or user_id is None:
flask.abort(401)
team = get_or_404(env.api.teams.get_single, team_id=team_id)
membership = first_or_none(get_or_500(env.api.team_memberships.get, team_id=team.id, user_id=user_id))
return dict(team=team, membership=membership)
def only_team_owner(env, kwargs):
logged_user = flask_login.current_user
if 'team' in kwargs:
user_id = kwargs['team'].founder.id
elif 'team_id' in flask.request.view_args:
team = get_or_404(env.api.teams.get_single, team_id=flask.request.view_args['team_id'])
user_id = team.founder.id
else:
raise EnvironmentError('No team or team_id in view params')
return logged_user.is_authenticated() and logged_user.id == user_id
def create_edit_context(env, game_id):
user = flask_login.current_user
game = get_or_404(env.api.games.get_single, game_id=game_id)
game_ownership = None
form = None
if user.is_authenticated():
game_ownerships_by_game_id = hash_by(lambda go: go.game.id, user.game_ownerships)
game_ownership = game_ownerships_by_game_id.get(game_id)
if game_ownership:
form = GameUpdateForm(env.api, user.token, game_ownership.id)
else:
form = GameJoinForm(env.api, user.token, game_id, user.id)
return dict(game_ownership=game_ownership, form=form, game=game)
def user_edit_friendship_context(env, user_id):
me = flask_login.current_user
return dict(
user=get_or_404(env.api.users.get_single, user_id=user_id, token=me.token),
friendship_invite=first_or_none(get_or_500(env.api.friendship_invites.get, token=me.token, from_user_id=me.id, to_user_id=user_id)),
friendship=first_or_none(get_or_500(env.api.friendships.get, token=me.token, user_id=me.id, to_user_id=user_id)))
def create_edit_context(env, team_id):
user = flask_login.current_user
team = get_or_404(env.api.teams.get_single, team_id=team_id)
change_logo_form = ChangeTeamLogoForm(team.id, user.token, env.api)
change_basic_form = EditTeamInfoForm(user, team, env.api)
return dict(team=team, logo_form=change_logo_form, basic_form=change_basic_form)
