"""Data extraction functions."""

from sqlalchemy import text

from availability_etl import constants
from availability_etl.connectors import microservice_db


def extracted():
    """Iterator of extracted data rows.

    Yields:
        dict: A dictionary with extracted data row.

    Raises:
        StopIteration: There are no more items to yield.
    """
    with microservice_db.get_microservice_connection() as conn:
        result = conn.execute(
            text(constants.SQL_MS_SELECT_DATA))
        for row in result:
            yield dict(row)