def upsert(name, triangle):
""" Handles upsert funtionality of mutations
Parameters
----------
name :
The name of the Triangle
triangle:
The triangle instance
"""
bytes_container = pickle.dumps(triangle)
try:
# If table exists
exists = db_session.query(ModelTriangle).filter_by(name=name)
if exists.first():
exists.update({'edited': datetime.utcnow(),
'data': bytes_container})
else:
data = {'name': name,
'data': bytes_container,
'created': datetime.utcnow(),
'edited': datetime.utcnow()}
db_session.add(ModelTriangle(**data))
except:
data = {'name': name,
'data': bytes_container,
'created': datetime.utcnow(),
'edited': datetime.utcnow()}
db_session.add(ModelTriangle(**data))
db_session.commit()
return db_session.query(ModelTriangle).filter_by(name=name).first()
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
#data['password'] = hash(data['password']) this is where I will return a JWT
service = ModelService(**data)
db_session.add(service)
db_session.commit()
return CreateService(service=service)
def mutate(self, info, input):
"""hjk"""
data = utils.input_to_dictionary(input)
user = UserModel(**data)
db_session.add(user)
db_session.commit()
return CreateUser(user=user)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
task = ModelTask(**data)
db_session.add(task)
db_session.commit()
return CreateTask(task=task)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data['match_date'] = datetime.strptime(data['match_date'], '%Y-%m-%d')
match = ModelMatches(**data)
db_session.add(match)
db_session.commit()
return CreateMatch(match=match)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data['created'] = datetime.utcnow()
card = ModelCard(**data)
db_session.add(card)
db_session.commit()
return CreateCard(card=card)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data["created"] = datetime.utcnow()
data["edited"] = datetime.utcnow()
customer = ModelCustomer(**data)
db_session.add(customer)
db_session.commit()
return CreateCustomer(customer=customer)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data['created'] = datetime.utcnow()
data['edited'] = datetime.utcnow()
planet = ModelPlanet(**data)
db_session.add(planet)
db_session.commit()
return CreatePlanet(planet=planet)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data['created'] = datetime.utcnow()
data['edited'] = datetime.utcnow()
person = ModelPeople(**data)
db_session.add(person)
db_session.commit()
return CreatePerson(person=person)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data["created"] = datetime.utcnow()
data["edited"] = datetime.utcnow()
agent = ModelAgent(**data)
db_session.add(agent)
db_session.commit()
return CreateAgent(agent=agent)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data['created'] = datetime.utcnow()
data['edited'] = datetime.utcnow()
person = ModelPeople(**data)
db_session.add(person)
db_session.commit()
return CreatePerson(person=person)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data["time_posted"] = datetime.utcnow()
data["queue_pos"] = len(
db_session.query(QuestionModel).filter(
QuestionModel.course_id == data["course_id"]).filter(
QuestionModel.queue_pos > 0).all()) + 1
question = QuestionModel(**data)
db_session.add(question)
db_session.commit()
return CreateQuestion(question=question)
def save_user_data(kwargs):
email = kwargs['data']['email']
if db_session.query(kwargs['model']).filter_by(email=email).first(
): #checks to see if user already made an account with the given email
return "Email already in use"
kwargs['data']['passphrase'] = hash_passphrase(
kwargs['data']['passphrase'])
kwargs['data']['id'] = create_uid(kwargs['model'])
new_model = kwargs['model'](**kwargs['data'])
db_session.add(new_model)
db_session.commit()
return "Account Created"
def save_services_data(kwargs):
query_result = db_session.query(
kwargs['model']).filter_by(user_id=kwargs['user_id'])
if not query_result.first(): #.first() will return None if not found
kwargs['data']['id'] = create_uid(kwargs['model'])
kwargs['data']['user_id'] = kwargs['user_id']
new_model = kwargs['model'](**kwargs['data'])
db_session.add(new_model)
else:
query_result.update(
kwargs["data"]
) #if we use .first() here then the object will not have the update attribute
db_session.commit()
return "Data Added"
