pygraphy test code

test.py

import uvicorn
import pygraphy

from fastapi import FastAPI
from typing import Optional, List

app = FastAPI()


class Foo(pygraphy.Object):
    a: str


class Bar(pygraphy.Object):
    b: int


class FooBar(pygraphy.Union):
    members = (Foo, Bar)


class GeoInput(pygraphy.Input):
    lat: float
    lng: float
    string: str  # new

    @property
    def latlng(self):
        return "({},{})".format(self.lat, self.lng)


class Address(pygraphy.Object):
    latlng: str

    @pygraphy.field
    def foobar(self) -> List[FooBar]:
        return [Foo(a="test") for _ in range(5)]


class Query(pygraphy.Query):
    @pygraphy.field
    def address(self, geo: GeoInput) -> Address:
        print(geo.string)
        return Address(latlng=geo.latlng)


class Mutation(pygraphy.Object):
    @pygraphy.field
    def create_address(self, geo: GeoInput) -> Address:
        return Address(latlng=geo.latlng)


@app.route("/")
class Schema(pygraphy.Schema):
    query: Optional[Query]
    mutation: Optional[Mutation]


if __name__ == "__main__":
    uvicorn.run(app, host="0.0.0.0", port=8000)