def post(self, team_id):
if TeamModel.find_by_id(team_id):
return {"msg": "Team already exist"}, 400
data = self.parser.parse_args()
new_team = TeamModel(team_id, **data)
new_team.save_to_db()
return new_team.json(), 201
def put(self, team_id):
data = self.parser.parse_args()
team = TeamModel.find_by_id(team_id)
if team is None:
team = TeamModel(team_id, **data)
else:
team.team_name = data["team_name"]
team.save_to_db()
return team.json()
def __init__(self, name, back_number, team_id):
self.name = name
self.back_number = back_number
self.team_id = team_id
self.division = TeamModel.find_by_id(team_id).division
def delete(self, team_id):
team = TeamModel.find_by_id(team_id)
if team:
team.delete_from_db()
return {"msg": "Team Deleted"}, 200
return {"msg": "Team Not Found"}, 404
def get(self, team_id):
team = TeamModel.find_by_id(team_id)
if team:
return team.json(), 200
return {"msg": "Team Not Found"}, 404
