trying_gql.py

from dataclasses import dataclass
import sys
import time
from typing import Any, ClassVar, Dict, List
import asyncio
from gql import gql, Client, AIOHTTPTransport
from graphql.language.ast import DocumentNode


@dataclass
class Artist:
    '''Basic Artist representation.'''

    name: str
    slug: str
    qwant_url: str


class QwantMusicClient:
    TRANSPORT: ClassVar[AIOHTTPTransport] = AIOHTTPTransport(
        url='https://qwant-music.herokuapp.com/graphql/'
    )
    QUERY: ClassVar[DocumentNode] = gql(
        '''
        query($name: String!) {
            artist(name: $name) {
                name
                slug
                qwantUrl
            }
        }
        '''
    )

    @classmethod
    async def fetch_artists(cls, artists: List[str]) -> List[Artist]:
        """Fetch Artists from the API asynchronously and print the lasting
        time.

        Params:
            artists: The list of the artist name to fetch

        Return:
            The list of instanciated Artists
        """
        result: List[Artist] = []
        async with Client(
            transport=cls.TRANSPORT, fetch_schema_from_transport=True
        ) as session:

            async def search(artist_name: str) -> None:
                start: float = time.time()
                res: Dict[str, Any] = await session.execute(
                    cls.QUERY, variable_values={'name': artist_name}
                )
                res['artist']['qwant_url'] = res['artist'].pop('qwantUrl')
                artist: Artist = Artist(**res['artist'])
                end: float = time.time()
                print(f'Lasted {end - start:.2f} second(s) for {artist}')
                result.append(artist)

            coros = [search(artist) for artist in artists]
            await asyncio.gather(*coros)
            return result


async def main() -> None:
    '''Main function'''
    artists: List[str]
    if not (artists := sys.argv[1:]):
        artists = ['Flight Facilities', 'Roosevelt', 'Satin Jackets', 'Bonobo']
    await QwantMusicClient.fetch_artists(artists)


if __name__ == '__main__':
    asyncio.run(main())

> python .\trying_gql.py
Lasted 0.55 second(s) for Artist(name='Roosevelt', slug='roosevelt', qwant_url='https://www.qwant.com/music/artist/roosevelt')
Lasted 0.59 second(s) for Artist(name='Flight Facilities', slug='flight-facilities', qwant_url='https://www.qwant.com/music/artist/flight-facilities')
Lasted 0.63 second(s) for Artist(name='Satin Jackets', slug='satin-jackets', qwant_url='https://www.qwant.com/music/artist/satin-jackets')
Lasted 0.64 second(s) for Artist(name='Bonobo', slug='bonobo', qwant_url='https://www.qwant.com/music/artist/bonobo')
> python .\trying_gql.py
Lasted 0.58 second(s) for Artist(name='Flight Facilities', slug='flight-facilities', qwant_url='https://www.qwant.com/music/artist/flight-facilities')
Lasted 0.63 second(s) for Artist(name='Roosevelt', slug='roosevelt', qwant_url='https://www.qwant.com/music/artist/roosevelt')
Lasted 0.67 second(s) for Artist(name='Bonobo', slug='bonobo', qwant_url='https://www.qwant.com/music/artist/bonobo')
Lasted 0.72 second(s) for Artist(name='Satin Jackets', slug='satin-jackets', qwant_url='https://www.qwant.com/music/artist/satin-jackets')
> python .\trying_gql.py
Lasted 0.57 second(s) for Artist(name='Flight Facilities', slug='flight-facilities', qwant_url='https://www.qwant.com/music/artist/flight-facilities')
Lasted 0.57 second(s) for Artist(name='Bonobo', slug='bonobo', qwant_url='https://www.qwant.com/music/artist/bonobo')
Lasted 0.64 second(s) for Artist(name='Satin Jackets', slug='satin-jackets', qwant_url='https://www.qwant.com/music/artist/satin-jackets')
Lasted 0.68 second(s) for Artist(name='Roosevelt', slug='roosevelt', qwant_url='https://www.qwant.com/music/artist/roosevelt')
>