def mutate(root, info, question_data):
errors = {}
current_user = get_jwt_identity()
user = UserModel.find_by_id(current_user["id"])
if not user:
errors["user"] = "******"
category = CategoryModel.find_by_id(question_data["category_id"])
if not category:
errors["category"] = "not found"
location = CityModel.find_by_id(question_data["location_id"])
if not location:
errors["location"] = "not found"
if errors:
raise GraphQLError(json.dumps(errors))
del question_data["category_id"]
del question_data["location_id"]
question = QuestionModel(created_by=user,
category=category,
location=location,
**question_data)
question.save()
return CreateQuestion(question=question, ok=True)
def resolve_category(root, info, _id):
category = CategoryModel.find_by_id(_id)
return category
