def __init__(self, args): self.name = args['name'] self.capacity = args['capacity'] self.number_players = args['number_players'] self.pitch_postcode = args['pitch_postcode'] self.time = args['time'] self.created_by = current_user().id self.validate_capacity_greater_than_players() self._update_enrollments(args)
def post(self): if not current_user(): return 'You need to be logged in', 403 args = self.reqparse.parse_args() try: team = Team(args) db.session.add(team) db.session.commit() except Exception as e: return str(e), 400 return 'Team created successfully', 201
def post(self, team_id): if not current_user(): return 'You need to be logged in', 403 args = self.reqparse.parse_args() args['team_id'] = team_id try: enrollment = Enrollment(args) db.session.add(enrollment) db.session.commit() except Exception as e: return str(e), 400 new_enrollment_email(team_id) return 'Enrolled successfully', 201
def _append_to_user(self): user = User.query.get(current_user().id) user.teams.append(self)
def get(self): if 'user_id' in session: return {'user': marshal(current_user(), user_fields)} return 'No session set', 200
def test_current_user(self): session.clear() user = current_user() self.assertEqual(user, None)