mreg-cli 1.0.0__py3-none-any.whl

mreg_cli/commands/label.py

@@ -0,0 +1,146 @@
+"""Label-related commands for mreg_cli."""
+
+from __future__ import annotations
+
+import argparse
+from typing import Any
+
+from mreg_cli.api.models import Label
+from mreg_cli.commands.base import BaseCommand
+from mreg_cli.commands.registry import CommandRegistry
+from mreg_cli.exceptions import EntityNotFound, InputFailure
+from mreg_cli.outputmanager import OutputManager
+from mreg_cli.types import Flag
+
+command_registry = CommandRegistry()
+
+
+class LabelCommands(BaseCommand):
+    """Label commands for the CLI."""
+
+    def __init__(self, cli: Any) -> None:
+        """Initialize the label commands."""
+        super().__init__(cli, command_registry, "label", "Manage host labels.", "Manage labels")
+
+
+@command_registry.register_command(
+    prog="add",
+    description="Add a label",
+    short_desc="Add a label",
+    flags=[
+        Flag("name", short_desc="Label name", description="The name of the new label"),
+        Flag("description", description="The purpose of the label"),
+    ],
+)
+def label_add(args: argparse.Namespace) -> None:
+    """Add a label.
+
+    :param args: argparse.Namespace (name, description)
+    """
+    if " " in args.name:
+        raise InputFailure("The label name can't contain spaces.")
+
+    # We can't do a fetch_after_create here because the API is... broken.
+    # https://github.com/unioslo/mreg/blob/eed5c154bcc47b1dea474feabad46125ebde0aec/mreg/api/v1/views_labels.py#L30
+    # https://github.com/unioslo/mreg/blob/eed5c154bcc47b1dea474feabad46125ebde0aec/mreg/api/v1/views.py#L187
+    Label.create({"name": args.name, "description": args.description}, fetch_after_create=False)
+    OutputManager().add_ok(f'Added label "{args.name}"')
+
+
+@command_registry.register_command(
+    prog="list", description="List labels", short_desc="List labels", flags=[]
+)
+def label_list(_: argparse.Namespace) -> None:
+    """List labels."""
+    labels = Label.get_all()
+    if not labels:
+        OutputManager().add_line("No labels")
+        return
+    OutputManager().add_formatted_table(("Name", "Description"), ("name", "description"), labels)
+
+
+@command_registry.register_command(
+    prog="remove",
+    description="Remove a label",
+    short_desc="Remove a label",
+    flags=[
+        Flag(
+            "name",
+            short_desc="Label name",
+            description="The name of the label to remove",
+        )
+    ],
+)
+def label_delete(args: argparse.Namespace) -> None:
+    """Delete a label.
+
+    :param args: argparse.Namespace (name)
+    """
+    label = Label.get_by_name_or_raise(args.name)
+    if not label:
+        raise EntityNotFound(f'Label "{args.name}" does not exist.')
+
+    label.delete()
+    OutputManager().add_ok(f'Removed label "{args.name}"')
+
+
+@command_registry.register_command(
+    prog="info",
+    description="Show details about a label",
+    short_desc="Label details",
+    flags=[Flag("name", short_desc="Label name", description="The name of the label")],
+)
+def label_info(args: argparse.Namespace) -> None:
+    """Show details about a label.
+
+    :param args: argparse.Namespace (name)
+    """
+    Label.get_by_name_or_raise(args.name).output()
+
+
+@command_registry.register_command(
+    prog="rename",
+    description="Rename a label and/or change the description",
+    short_desc="Rename a label",
+    flags=[
+        Flag(
+            "oldname",
+            short_desc="Old name",
+            description="The old (current) name of the label",
+        ),
+        Flag("newname", short_desc="New name", description="The new name of the label"),
+    ],
+)
+def label_rename(args: argparse.Namespace) -> None:
+    """Rename a label.
+
+    :param args: argparse.Namespace (oldname, newname)
+    """
+    Label.get_by_name_or_raise(args.oldname).rename(args.newname)
+    OutputManager().add_ok(f'Renamed label "{args.oldname}" to "{args.newname}"')
+
+
+@command_registry.register_command(
+    prog="set_description",
+    description="Set the description for the label",
+    short_desc="Describe a label",
+    flags=[
+        Flag(
+            "name",
+            short_desc="name",
+            description="The name of the label",
+        ),
+        Flag(
+            "desc",
+            short_desc="New description",
+            description="The new description of the label",
+        ),
+    ],
+)
+def label_redesc(args: argparse.Namespace) -> None:
+    """Change the description of a label.
+
+    :param args: argparse.Namespace (name, desc)
+    """
+    Label.get_by_name_or_raise(args.name).set_description(args.desc)
+    OutputManager().add_ok(f'Set description for label "{args.name}" to "{args.desc}"')

mreg_cli/commands/logging.py

@@ -0,0 +1,112 @@
+"""Logging commands for the CLI."""
+
+from __future__ import annotations
+
+import argparse
+import os
+from typing import Any
+
+from mreg_cli.commands.base import BaseCommand
+from mreg_cli.commands.registry import CommandRegistry
+from mreg_cli.config import MregCliConfig
+from mreg_cli.exceptions import InputFailure
+from mreg_cli.outputmanager import OutputManager
+from mreg_cli.types import Flag, LogLevel
+from mreg_cli.utilities.shared import sizeof_fmt
+
+command_registry = CommandRegistry()
+
+
+class LoggingCommmands(BaseCommand):
+    """Logging commands for the CLI."""
+
+    def __init__(self, cli: Any) -> None:
+        """Initialize the logging commands."""
+        super().__init__(
+            cli,
+            command_registry,
+            "logging",
+            "Logging commands for the CLI.",
+            "Manage logging.",
+        )
+
+
+@command_registry.register_command(
+    prog="start",
+    description="Start logging commands to a given file.",
+    short_desc="Start logging",
+    flags=[
+        Flag(
+            "filename",
+            description="The filename to log to.",
+            short_desc="Filename",
+            metavar="filename",
+        ),
+        Flag(
+            "level",
+            description="The logging level to use.",
+            short_desc="Level",
+            metavar="level",
+            choices=LogLevel.choices(),
+            default="INFO",
+        ),
+    ],
+)
+def start_logging(args: argparse.Namespace) -> None:
+    """Start logging to the given file."""
+    if not args.filename:
+        raise InputFailure("No filename given.")
+
+    config = MregCliConfig()
+    config.stop_logging()
+    config.start_logging(args.filename, args.level)
+
+
+@command_registry.register_command(
+    prog="stop",
+    description="Stop logging.",
+    short_desc="Stop logging",
+)
+def stop_logging(_: argparse.Namespace):
+    """Stop logging."""
+    MregCliConfig().stop_logging()
+
+
+@command_registry.register_command(
+    prog="level",
+    description="Set the logging level.",
+    short_desc="Set logging level",
+    flags=[
+        Flag(
+            "level",
+            description="The logging level to use.",
+            short_desc="Level",
+            metavar="level",
+            choices=["DEBUG", "INFO", "WARNING", "ERROR", "CRITICAL"],
+        ),
+    ],
+)
+def set_logging_level(args: argparse.Namespace) -> None:
+    """Set the logging level."""
+    config = MregCliConfig()
+    config.stop_logging()
+    config.start_logging(config.get_default_logfile(), args.level)
+
+
+@command_registry.register_command(
+    prog="status",
+    description="Print the logging status.",
+    short_desc="Logging status",
+)
+def logging_status(_: argparse.Namespace):
+    """Print the logging status."""
+    level = MregCliConfig().get_logging_level()
+    file = MregCliConfig().get_default_logfile()
+
+    lines_in_logfile = 0
+    with open(file, "r") as f:
+        lines_in_logfile = sum(1 for _ in f)
+
+    filesize = sizeof_fmt(os.path.getsize(file))
+
+    OutputManager().add_line(f"{level} > {file} ({lines_in_logfile} lines, {filesize})")