from gql import gql, Client
from gql.transport.aiohttp import AIOHTTPTransport

transport = AIOHTTPTransport(url="https://gql.ivazh.ru/graphql/")


def get_classifier():
    client = Client(transport=transport, fetch_schema_from_transport=True)
    query = gql(
        """
            query getClassifier {
              getClassifier(name: "ood")
            }
        """
    )
    result = client.execute(query)
    return result['getClassifier']


def get_catalog():
    client = Client(transport=transport, fetch_schema_from_transport=True)
    query = gql(
        """
            query getCatalog {
              getCatalog(name: "ood")
            }
        """
    )
    result = client.execute(query)
    return result['getCatalog']


def get_object(oid: str):
    client = Client(transport=transport, fetch_schema_from_transport=True)
    query = gql(
        """
            query getObjects($oid: String!) {
              getObject(name: "ood", oid: $oid)
            }
        """
    )
    params = {'oid': oid}
    result = client.execute(query, variable_values=params)
    return result['getObject']