tabcmd/tabcmd/commands/site/list_command.py at b102fde13f06d02e86b94de31d785f1c6019d8ac · tableau/tabcmd · GitHub

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
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
from tabcmd.commands.auth.session import Session
from tabcmd.commands.constants import Errors
from tabcmd.commands.server import Server
from tabcmd.execution.localize import _
from tabcmd.execution.logger_config import log


class ListCommand(Server):
    """
    Command to return a list of content the user can access
    """

    name: str = "list"
    description: str = _("tabcmd.listing.short_description")

    @staticmethod
    def define_args(list_parser):
        args_group = list_parser.add_argument_group(title=ListCommand.name)
        args_group.add_argument(
            "content", choices=["projects", "workbooks", "datasources", "flows"], help=_("tabcmd.options.select_type")
        )

        format_group = list_parser.add_mutually_exclusive_group()
        # TODO: should this be saved directly to csv?
        format_group.add_argument("--machine", action="store_true", help=_("tabcmd.listing.help.machine"))

        data_group = list_parser.add_argument_group(title=_("tabcmd.listing.group.attributes"))
        # data_group.add_argument("-i", "--id", action="store_true", help="Show item id") # default true
        data_group.add_argument("-n", "--name", action="store_true", help=_("tabcmd.listing.help.name"))  # default true
        data_group.add_argument("-o", "--owner", action="store_true", help=_("tabcmd.listing.help.owner"))
        data_group.add_argument("-d", "--details", action="store_true", help=_("tabcmd.listing.help.details"))
        data_group.add_argument("-a", "--address", action="store_true", help=_("tabcmd.listing.help.address"))

    @classmethod
    def run_command(cls, args):
        logger = log(__name__, args.logging_level)
        logger.debug(_("tabcmd.launching"))
        session = Session()
        server = session.create_session(args, logger)
        content_type = args.content

        try:
            logger.info(_("tabcmd.listing.header").format(content_type, session.username))

            if content_type == "projects":
                items = server.projects.all()
            elif content_type == "workbooks":
                items = server.workbooks.all()
            elif content_type == "datasources":
                items = server.datasources.all()
            elif content_type == "flows":
                items = server.flows.all()

            if not items or len(items) == 0:
                logger.info(_("tabcmd.listing.none"))
                exit(0)

            logger.info(ListCommand.show_header(args, content_type))
            for item in items:
                if args.machine:
                    id = item.id
                    name = ", " + item.name if args.name else ""
                    owner = ", " + item.owner_id if args.owner else ""
                    url = ""
                    if args.address and content_type in ["workbooks", "datasources"]:
                        url = ", " + item.content_url
                    children = (
                        ", " + ListCommand.format_children_listing(args, server, content_type, item)
                        if args.details
                        else ""
                    )

                else:
                    id = _("tabcmd.listing.label.id").format(item.id)
                    name = ", " + _("tabcmd.listing.label.name").format(item.name) if args.name else ""
                    owner = ", " + _("tabcmd.listing.label.owner").format(item.owner_id) if args.owner else ""

                    url = ""
                    if args.address and content_type in ["workbooks", "datasources"]:
                        url = ", " + item.content_url
                    children = (
                        ListCommand.format_children_listing(args, server, content_type, item) if args.details else ""
                    )

                logger.info("{0}{1}{2}{3}{4}".format(id, name, owner, url, children))

            # TODO: do we want this line if it is csv output?
            logger.info(_("tabcmd.listing.summary").format(len(items), content_type))
        except Exception as e:
            Errors.exit_with_error(logger, exception=e)

    @staticmethod
    def format_children_listing(args, server, content_type, item):
        if args.details:
            if content_type == "workbooks":
                server.workbooks.populate_views(item)
                child_items = item.views[:10]
                children = ", " + _("tabcmd.listing.label.views") + ", ".join(map(lambda x: x.name, child_items))
                return children
        return ""

    @staticmethod
    def show_header(args, content_type):
        id = _("tabcmd.listing.header.id")
        name = ", " + _("tabcmd.listing.header.name") if args.name else ""
        owner = ", " + _("tabcmd.listing.header.owner") if args.owner else ""
        url = (
            ", " + _("tabcmd.listing.header.url")
            if args.address and content_type in ["workbooks", "datasources"]
            else ""
        )
        children = ", " + _("tabcmd.listing.header.children") if args.details and content_type == "workbooks" else ""
        return "{0}{1}{2}{3}{4}".format(id, name, owner, url, children)