how to personalize resolve DjangoObjectType

[yoelvysmartinez]

class Country(models.Model):
    oid = models.AutoField(primary_key=True)
    code = models.CharField(max_length=5, null=True)
    name = models.CharField(max_length=100, null=True)

    class Meta:
        db_table = ('countries')

    def __str__(self):
        return self.name


class City(models.Model):
    oid = models.AutoField(primary_key=True)
    code = models.CharField(max_length=5, null=True)
    name = models.CharField(max_length=100, null=True)
    country = models.ForeignKey(
        Country, db_column='countries_oid', blank=False, null=False, on_delete=models.CASCADE, related_name='cities')

    class Meta:
        db_table = ('cities')

    def __str__(self):
        return self.name

My schema

class CountryNode(DjangoObjectType):
    class Meta:
        model = Country
        filter_fields = {
            'code': ['icontains'],
            'name': ['iexact'],
        }
        interfaces = (relay.Node, )

    def resolve_cities(self, info):
        return None


class CityNode(DjangoObjectType):
    class Meta:
        model = City
        filter_fields = {
            'code': ['icontains'],
            'name': ['icontains'],
        }
        interfaces = (relay.Node, )

My question is how to personalize resolve_cities in CountryNode class.

Something like this

class CountryNode(DjangoObjectType):
    class Meta:
        model = Country
        filter_fields = {
            'code': ['icontains'],
            'name': ['iexact'],
        }
        interfaces = (relay.Node, )

    def resolve_cities(self, info):
        if not info.context.user.is_authenticated():
            return City.objects.none()
        else:
            return City.objects.all()

[yoelvysmartinez]

I've done it with Object Type but I do not want to lose the relay pagination

[louisdvs]

You shouldn't be trying to add a resolver for Cities on the Country Node. The only reason you would add a field and resolver directly on the Type definition is if it is not defined in the django model.

You should be able to add the custom resolver in your query, like:

class Query(object):
    city = relay.Node.Field(CityNode)
    all_cities = DjangoFilterConnectionField(CityNode)

    def resolve_all_cities(self, info):
        if not info.context.user.is_authenticated():
            return City.objects.none()
        else:
            return City.objects.all()`

Hope this helps. Cheers

[phalt]

Hey @yoelvys - it looks like @louisdvs is providing the correct solution. Closing as resolved.