async def select_pool(ctx):
click.clear()
click.echo("==================")
click.echo(" Download Pools ")
click.echo("==================")
click.echo("")
click.echo("Please give me the URLs and/or pool ID.")
click.echo("When you're done, you can give me an empty line.")
pools = []
while True:
# fmt: off
response = click.prompt("",
prompt_suffix="> ",
default="",
show_default=False)
# fmt: on
if not response or not response.strip():
if not pools:
continue
break
pool_id = get_pool_id(response)
if not pool_id:
click.secho("Please send valid URL or pool ID!")
continue
obj = await get_pool(ctx, pool_id)
if obj:
print_pool(obj)
pools.append(obj) # noqa
else:
click.secho(f'Pool "{pool_id}" was not found.') # noqa
output = click.prompt("Where will the images be saved? ",
type=click.Path(),
default=".")
jobs = click.prompt(
"How many concurrent jobs will be done? "
"(If you don't know what that means, just leave it as is.)",
type=int,
default=4,
)
type_ = click.prompt(
"Which quality do you want to download? ",
type=click.Choice(["sample", "file", "preview"]),
default="file",
)
await ctx.invoke(pool,
pool_id=-1,
output=output,
jobs=jobs,
type=type_,
pools=pools)
async def interactive(ctx):
while True:
click.clear()
click.echo("==================")
click.echo(" Interactive Mode ")
click.echo("==================")
click.echo("")
click.echo("What do you want to do?")
click.echo("1. Download post(s).")
click.echo("2. Download pool(s).")
click.echo("Select: [1/2] ", nl=False)
c = click.getchar()
await func_table.get(c, invalid_input)(ctx)
click.confirm("Do you want to do anything else?", abort=True)
def clear():
"""Clears the entire screen."""
click.clear()