import asyncio

import logging

from gql import Client, gql

from gql.transport.websockets import WebsocketsTransport

logging.basicConfig(level=logging.INFO)

async def main():

transport = WebsocketsTransport(url="wss://countries.trevorblades.com/graphql")

# Using `async with` on the client will start a connection on the transport

# and provide a `session` variable to execute queries on this connection

async with Client(

transport=transport,

fetch_schema_from_transport=True,

) as session:

# Execute single query

query = gql("""

query getContinents {

continents {

code

name

}

}

""")

result = await session.execute(query)

print(result)

# Request subscription

subscription = gql("""

subscription {

somethingChanged {

id

}

}

""")

async for result in session.subscribe(subscription):

print(result)

asyncio.run(main())