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)
person = db_session.query(ModelPeople).filter_by(id=data['id']).first()
print("mutation Delete")
db_session.delete(person)
db_session.commit()
person = db_session.query(ModelPeople).filter_by(id=data['id']).first()
return DeletePerson(person=person)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
match = ModelMatches(**data)
db_session.query(ModelMatches).filter_by(
match_id=data['match_id']).delete()
db_session.commit()
return DeleteMatch(match=match)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
user = db_session.query(UserModel).filter_by(netid=data["netid"])
user.update(data)
db_session.commit()
user = db_session.query(UserModel).filter_by(
netid=data["netid"]).first()
return UpdateUser(user=user)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
question = db_session.query(QuestionModel).filter_by(id=data["id"])
question.update(data)
db_session.commit()
question = db_session.query(QuestionModel).filter_by(
id=data["id"]).first()
return UpdateQuestion(question=question)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
task = db_session.query(ModelTask).filter_by(id=data['id'])
task.update(data)
db_session.commit()
task = db_session.query(ModelTask).filter_by(id=data['id']).first()
return UpdateTask(task=task)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
card = db_session.query(ModelCard).filter_by(id=data['id'])
card.update(data)
db_session.commit()
card = db_session.query(ModelCard).filter_by(id=data['id']).first()
return UpdateCard(card=card)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data["edited"] = datetime.utcnow()
agent = db_session.query(ModelAgent).filter_by(id=data["id"])
agent.update(data)
db_session.commit()
agent = db_session.query(ModelAgent).filter_by(id=data["id"]).first()
return UpdateAgent(agent=agent)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data['edited'] = datetime.utcnow()
person = db_session.query(ModelPeople).filter_by(id=data['id'])
people.update(data)
db_session.commit()
person = db_session.query(ModelPeople).filter_by(id=data['id']).first()
return UpdatePerson(person=person)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data['edited'] = datetime.utcnow()
planet = db_session.query(ModelPlanet).filter_by(id=data['id'])
planet.update(data)
db_session.commit()
planet = db_session.query(ModelPlanet).filter_by(id=data['id']).first()
return UpdatePlanet(planet=planet)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
data["edited"] = datetime.utcnow()
customer = db_session.query(ModelCustomer).filter_by(id=data["id"])
customer.update(data)
db_session.commit()
customer = db_session.query(ModelCustomer).filter_by(
id=data["id"]).first()
return UpdaterCustomer(customer=customer)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
#filter_by refers to the columns or rows in the attributes
service = db_session.query(ModelService).filter_by(
service=data['service'])
service.update(data)
db_session.commit()
service = db_session.query(ModelService).filter_by(
service=data['service']).first()
return UpdateService(service=service)
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
if "match_date" in data.keys():
data['match_date'] = datetime.strptime(data['match_date'],
'%Y-%m-%d')
match = db_session.query(ModelMatches).filter_by(
match_id=data['match_id'])
match.update(data)
db_session.commit()
match = db_session.query(ModelMatches).filter_by(
match_id=data['match_id']).first()
return UpdateMatch(match=match)
def check_passphrase(email, passphrase):
user = db_session.query(ModelUser).filter_by(email=email).scalar()
if bcrypt.checkpw(passphrase.encode(), user.passphrase):
return user.id
else:
return None
def mutate(self, info, **kwargs):
card = db_session.query(ModelCard).filter_by(
id=kwargs.get('id')).first()
db_session.delete(card)
db_session.commit()
return DeleteCard(status="OK")
def resolve_findMatch(args: Dict,
info: graphql.execution.base.ResolveInfo,
date: Union[str, None] = None):
date = datetime.strptime(date, '%Y-%m-%d')
match = db_session.query(ModelMatches).filter_by(match_date=date).all()
return match
def create_uid(model): min_ = 10000 max_ = 10000000 uid = randint(min_, max_) while db_session.query(model).filter_by(id=uid).first() is not None: uid = randint(min_, max_) return uid
def mutate(self, info, input):
data = utils.input_to_dictionary(input)
service = db_session.query(ModelService).filter_by(
service=data['service'])
service.delete()
db_session.commit()
return RemoveService(
service="Removed {} service".format(data['service']))
#return RemoveService(service=service)
def mutate(self, info, name, **kwargs):
triangle = db_session.query(ModelTriangle).filter_by(name=name)
updated_tri = AbstractTriangleMutation.transform(
pickle.loads(triangle.first().data), **kwargs)
assign_to = kwargs.get('assign_to', None)
if assign_to:
return AbstractTriangleMutation(triangle=upsert(assign_to, updated_tri))
else:
bytes_container = pickle.dumps(updated_tri)
triangle.update({'data': bytes_container})
return AbstractTriangleMutation(triangle=triangle.first())
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 resolve_hub_keywords_list(self, info, **args):
keywords = []
q = db_session.query(hub_db.HubProjects).filter(
getattr(hub_db.HubProjects, "show").is_(True)
)
c = Counter([key for data in q for key in data.keywords])
for key, count in c.items():
keywords.append(HubKeywords(key, count))
return keywords
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"
def resolve_nace_industry(self, info, **args):
return (db_session.query(albania_db.AlbaniaNACEIndustry).filter(
getattr(albania_db.AlbaniaNACEIndustry, "nace_id") ==
args["nace_id"]).one())
def resolve_nace_industry_list(self, info, **args):
return db_session.query(albania_db.AlbaniaNACEIndustry)
def resolve_find_service(self, info, service): service_name = db_session.query(ModelService).filter_by(service=service) return service_name
def resolve_script(self, info, **args):
return db_session.query(albania_db.AlbaniaScript)
def resolve_find_user(self, info, email): service_name = db_session.query(ModelUser).filter_by(email=email) return service_name
def mutate(self, info, name):
db_session.query(ModelTriangle).filter_by(name=name).delete()
db_session.commit()
return DeleteTriangle(ok='ok')
def resolve_jordan_industry(self, info, **args):
return (db_session.query(jordan_db.JordanIndustry).filter(
getattr(jordan_db.JordanIndustry, "industry_code") ==
args["industry_code"]).one())
