summaryrefslogtreecommitdiffstats
path: root/eos_downloader/cli/cli.py
blob: ddd0deae4f2939043c15903c5dd7ffa2a30b198e (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
#!/usr/bin/env python
# coding: utf-8 -*-
# pylint: disable=no-value-for-parameter
# pylint: disable=cyclic-import
# pylint: disable=too-many-arguments
# pylint: disable=unused-argument


"""
ARDL CLI Baseline.
"""

import click
from rich.console import Console
import eos_downloader
from eos_downloader.cli.get import commands as get_commands
from eos_downloader.cli.debug import commands as debug_commands
from eos_downloader.cli.info import commands as info_commands


@click.group()
@click.pass_context
@click.option('--token', show_envvar=True, default=None, help='Arista Token from your customer account')
def ardl(ctx: click.Context, token: str) -> None:
    """Arista Network Download CLI"""
    ctx.ensure_object(dict)
    ctx.obj['token'] = token


@click.command()
def version() -> None:
    """Display version of ardl"""
    console = Console()
    console.print(f'ardl is running version {eos_downloader.__version__}')


@ardl.group(no_args_is_help=True)
@click.pass_context
def get(ctx: click.Context) -> None:
    # pylint: disable=redefined-builtin
    """Download Arista from Arista website"""


@ardl.group(no_args_is_help=True)
@click.pass_context
def info(ctx: click.Context) -> None:
    # pylint: disable=redefined-builtin
    """List information from Arista website"""


@ardl.group(no_args_is_help=True)
@click.pass_context
def debug(ctx: click.Context) -> None:
    # pylint: disable=redefined-builtin
    """Debug commands to work with ardl"""

# ANTA CLI Execution


def cli() -> None:
    """Load ANTA CLI"""
    # Load group commands
    get.add_command(get_commands.eos)
    get.add_command(get_commands.cvp)
    info.add_command(info_commands.eos_versions)
    debug.add_command(debug_commands.xml)
    ardl.add_command(version)
    # Load CLI
    ardl(
        obj={},
        auto_envvar_prefix='arista'
    )


if __name__ == '__main__':
    cli()