"""Module for work with art_relations database."""

from typing import Iterable

import config
from lambdacommon.util import mysql_connection


def execute_query(query: str, args: dict) -> Iterable[dict]:
    """Execute sql query against art_relations database.

    Args:
        query: sql query to execute
        args: query arguments

    Returns:
        Iterable of query results
    """
    with mysql_connection(**config.connection_info) as connection:
        with connection.cursor() as cursor:
            cursor.execute(query, args=args)
            for record in cursor.fetchall():
                yield record