def mutate(self, info, referee_data=None, sport_data=None):
"""Create Referee Graphql"""
ref_data = input_to_dictionary(referee_data)
referee = RefereeModel(**ref_data)
referee_db = DB.session.query(RefereeModel).filter_by(
email=ref_data['email']).first()
if referee_db:
print('need to update')
referee_db = referee
else:
DB.session.add(referee)
data = input_to_dictionary(sport_data)
# Description isn't part of RefereeSport Model, so pull out the field
description = data.pop('description')
# Level returns as string, need to get Level Object
level = DB.session.query(Level).filter_by(description=data['level']).first()
if level:
data['level'] = level
else:
print('need no level logic')
sport = RefereeSport(**data)
sport_db = DB.session.query(Sport, RefereeSport).join(Sport).filter(
Sport.description == description).first()
if sport_db:
print('need to update')
else:
DB.session.add(sport)
DB.session.commit()
return CreateReferee(referee=referee)
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
state_data = {'name': input['state_province']}
state_province = check_state_province(state_data)
data['state_province'] = state_province
return CreateLocation(Location=check_location(data))
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
province = db.session.query(StateProvinceModel).filter_by(id=data['id'])
province.delete()
db.session.commit()
return DeleteStateProvince(ok=True)
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
location = db.session.query(LocationModel).filter_by(id=data['id'])
location.delete()
db.session.commit()
return DeleteLocation(ok=True)
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
country = db.session.query(CountryModel).filter_by(id=data['id'])
country.delete()
db.session.commit()
return DeleteCountry(ok=True)
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
country = CountryModel(**data)
db.session.add(country)
db.session.commit()
return CreateCountry(country=country)
def mutate(self, info, mutate_input):
""" Mutate for delete sport """
data = input_to_dictionary(mutate_input)
sport = DB.session.query(SportModel).filter_by(id=data['id'])
sport.delete(data['id'])
DB.session.commit()
return DeleteSport(ok=True)
def mutate(self, info, referee_data):
"""Update Referee Graphql"""
data = input_to_dictionary(referee_data)
referee = DB.session.query(RefereeModel).filter_by(id=data['id']).first()
referee.update(data)
DB.session.commit()
referee = DB.session.query(RefereeModel).filter_by(id=data['id']).first()
return UpdateReferee(referee=referee)
def mutate(self, info, sport_data):
"""Sport Graphql Update mutation"""
data = input_to_dictionary(sport_data)
sport = DB.session.query(SportModel).filter_by(id=data['id'])
sport.update(data)
DB.session.commit()
sport = DB.session.query(SportModel).filter_by(id=data['id']).first()
return UpdateSport(sport=sport)
def mutate(self, info, coach_data):
"""Update Coach Graphql"""
data = input_to_dictionary(coach_data)
coach = DB.session.query(CoachModel).filter_by(id=data['id']).first()
coach.update(data)
DB.session.commit()
coach = DB.session.query(CoachModel).filter_by(id=data['id']).first()
return UpdateCoach(coach=coach)
def mutate(self, info, team_data):
"""Mutation method for Update Team"""
data = input_to_dictionary(team_data)
team = DB.session.query(TeamModel).filter_by(id=data['id'])
team.update(data)
DB.session.commit()
team = DB.session.query(TeamModel).filter_by(id=data['id']).first()
return UpdateTeam(team=team)
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
country = db.session.query(CountryModel).filter_by(id=data['id'])
country.update(data)
db.session.commit()
country = db.session.query(CountryModel).filter_by(
id=data['id']).first()
return UpdateCountry(country=country)
def mutate(self, info, player_data):
"""Update Player Graphql"""
data = input_to_dictionary(player_data)
player = DB.session.query(PlayerModel).filter_by(id=data['id']).first()
player.update(data)
DB.session.commit()
player = DB.session.query(PlayerModel).filter_by(id=data['id']).first()
return UpdatePlayer(player=player)
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
location = db.session.query(LocationModel).\
filter_by(id=data['id'])
location.update(data)
db.session.commit()
location = db.session.query(LocationModel).\
filter_by(id=data['id']).first()
return UpdateLocation(Location=location)
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
state_province = db.session.query(StateProvinceModel).\
filter_by(id=data['id'])
state_province.update(data)
db.session.commit()
state_province = db.session.query(StateProvinceModel).\
filter_by(id=data['id']).first()
return UpdateStateProvince(StateProvince=state_province)
def mutate(self, info, player_data=None):
"""Create Player Graphql"""
data = input_to_dictionary(player_data)
player = PlayerModel(**data)
player_db = DB.session.query(PlayerModel).filter_by(
description=data['description']).first()
if player_db:
print('need to update')
player_db = player
else:
DB.session.add(player)
DB.session.commit()
return CreatePlayer(player=player)
def mutate(self, info, team_data=None):
"""Mutation method for Create Team"""
data = input_to_dictionary(team_data)
team = TeamModel(**data)
team_db = DB.session.query(TeamModel).filter_by(
description=data['description']).first()
if team_db:
print('need to update')
team_db = team
else:
DB.session.add(team)
DB.session.commit()
return CreateTeam(team=team)
def mutate(self, info, sport_data=None):
"""Sport Graphql mutation"""
data = input_to_dictionary(sport_data)
sport = SportModel(**data)
sport_db = DB.session.query(SportModel).filter_by(
description=data['description']).first()
if sport_db:
print('need to update')
sport_db = sport
else:
DB.session.add(sport)
DB.session.commit()
return CreateSport(sport=sport)
def mutate(self, info, coach_data=None):
"""Create Coach Graphql"""
data = input_to_dictionary(coach_data)
coach = CoachModel(**data)
coach_db = DB.session.query(CoachModel).filter_by(
description=data['description']).first()
if coach_db:
print('need to update')
coach_db = coach
else:
DB.session.add(coach)
DB.session.commit()
return CreateCoach(coach=coach)
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
user = UserModel()
user.email = data['email']
user.password = data['password']
return CheckAuth(Auth=Auth)
def mutate(self, info, input_value):
data = utils.input_to_dictionary(input_value)
return CreateStateProvince(StateProvince=check_state_province(data))
