Как добавить фильтрацию по идентификатору в распознаватель пользователей?
У меня есть schema.py, которая выглядит так:
import graphene
from app import db
from graphene import relay
from app.models import User as UserModel, Event as EventModel
from graphene_sqlalchemy import SQLAlchemyConnectionField, SQLAlchemyObjectType
class User(SQLAlchemyObjectType):
class Meta:
model = UserModel
interfaces = (relay.Node, )
class UserConnection(relay.Connection):
class Meta:
node = User
class Event(SQLAlchemyObjectType):
class Meta:
model = EventModel
interfaces = (relay.Node, )
class EventConnections(relay.Connection):
class Meta:
node = Event
class CreateUser(graphene.Mutation):
class Arguments:
username = graphene.String(required=True)
fname = graphene.String(required=True)
surname = graphene.String(required=True)
email = graphene.String(required=True)
password = graphene.String(required=True)
user = graphene.Field(lambda: User)
def mutate(self, info, username, fname, surname, email, password):
user = UserModel.query.filter_by(username=username).first()
if user is None:
user = UserModel(username=username, fname=fname, surname=surname, email=email, password=password)
else:
return None
db.session.add(user)
db.session.commit()
return CreateUser(user=user)
class Query(graphene.ObjectType):
node = relay.Node.Field()
# queries that return individual models
user = graphene.Field(lambda: User, username=graphene.String())
event = graphene.Field(lambda: Event, title=graphene.String())
# queries that return all models of given type
all_users = SQLAlchemyConnectionField(UserConnection)
all_events = SQLAlchemyConnectionField(EventConnections)
# resolvers
def resolve_user(self, info, **kwargs):
query = User.get_query(info)
# id = kwargs.get("id")
username = kwargs.get('username')
return query.filter(UserModel.username == username).first()
def resolve_event(self, info, **kwargs):
query = Event.get_query(info)
# id = kwargs.get("id")
title = kwargs.get('title')
return query.filter(EventModel.title == title).first()
class Mutation(graphene.ObjectType):
create_user = CreateUser.Field()
schema = graphene.Schema(query=Query, mutation=Mutation, types=[User, Event])
и мои модели выглядят так:
from app import db
from datetime import datetime
class User(db.Model):
id = db.Column(db.Integer, primary_key=True)
username = db.Column(db.String(30), unique=True, nullable=False)
fname = db.Column(db.String(20), nullable=False)
surname = db.Column(db.String(35), nullable=False)
is_verified = db.Column(db.Boolean(), default=False)
profile_pic = db.Column(db.Text(), default="default.jpg")
email = db.Column(db.String(120), nullable=False)
password = db.Column(db.String(50), nullable=False)
events = db.relationship("Event", backref="organizer", lazy=True)
def __repr__(self):
print(f"User({self.id}, {self.name}, {self.surname}, {self.email}")
class Event(db.Model):
id = db.Column(db.Integer, primary_key=True)
title = db.Column(db.String(120), nullable=False)
date_posted = db.Column(db.String(50), nullable=False, default=datetime.now().isoformat())
description = db.Column(db.Text)
user_id = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
Итак, если я хочу получить всех пользователей, я набираю это:
query{
allUsers{
edges{
node{
id
username
email
}
}
}
}
Когда я получаю идентификатор, я получаю что-то вроде "id": "VXNlcjoy"
но в моих моделях id - это int. Если я изменю свой преобразователь на фильтрацию по ранее упомянутому идентификатору, sqlalchemy выдаст мне ошибку, потому что идентификатор в базе данных имеет целое число. Затем я изменяю свой преобразователь, чтобы принимать идентификаторы как целые числа и фильтровать по истинным идентификаторам. После этого изменения фильтрация с помощью резольвера работает, но происходит следующее:
{
user(id: 1){
id
}
}
Отклик:
{
"data": {
"user": {
"id": "VXNlcjoy"
}
}
}
Я немного подумал и думаю, что есть два способа об этом:
- заставить graphql вернуть мне тот целочисленный идентификатор, который находится в базе данных
- каким-то образом заставить преобразователь запрашивать пользователей и события по строковому идентификатору, который дает graphql ("VXNlcjoy")
Как я могу этого добиться?
1 ответ
Я нашел это, я переопределил зарезервированную переменную graphql с именем id
который используется для кеширования и разбивки на страницы. В models.py я переименовал столбецid
к uuid
и в таблице Event
user_id = db.Column(db.Integer, db.ForeignKey("user.id"), nullable=False)
к
user_uuid = db.Column(db.Integer, db.ForeignKey("user.uuid"), nullable=False)
и получил это:
mutation{
createUser(email:"saki1@saki.com", fname:"saki",surname:"sakal",username:"saki709", password:"sakisaki"){
user{
uuid
id
}
}
}
Отклик:
{
"data": {
"createUser": {
"user": {
"uuid": "1",
"id": "VXNlcjox"
}
}
}
}
TL;DR - id переименован в uuid, потому что у graphql есть собственный идентификатор переменной