Как добавить фильтрацию по идентификатору в распознаватель пользователей?

У меня есть 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 есть собственный идентификатор переменной

Другие вопросы по тегам