import click

from atlas_um.dna_accounts.services import ImportDNAAccountFromAuth0Service


@click.argument("product")
@click.option("--user_id", "-u", multiple=True)
@click.option("--import_all", "-a", is_flag=True)
@click.option("--batch_from", "-f", default=None, type=click.INT)
@click.option("--batch_to", "-t", default=None, type=click.INT)
def import_users_by_product(
    product, user_id, import_all, batch_from, batch_to
):
    if not user_id and not import_all:
        click.echo("--user_id or --import_all flag is required")
        return
    ImportDNAAccountFromAuth0Service.import_all_by_product(
        product, user_id, batch_from, batch_to
    )