"""Common generators for SoS flows."""

from analytics_aggregation import sos_config


def granular_copy_sos_data_params(context):
    """Generator for granular copy params."""
    skip_placements = context.get('skip_placements', False)
    if skip_placements:
        copy_params = sos_config.GRANULAR_COPY_TOTALS_PARAMS
    else:
        copy_params = sos_config.GRANULAR_COPY_PARAMS
    for granular_copy_data_params in copy_params:
        yield granular_copy_data_params