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