Fix lint

Author: ezilber-akamai

linodecli/__init__.py

@@ -46,10 +46,7 @@
 
 # if any of these arguments are given, we don't need to prompt for configuration
 skip_config = (
-    any(
-        c in argv
-        for c in ["--skip-config", "--version", "completion"]
-    )
+    any(c in argv for c in ["--skip-config", "--version", "completion"])
     or TEST_MODE
 )
 
@@ -110,24 +107,34 @@ def main():  # pylint: disable=too-many-branches,too-many-statements
         # if not spec was found and we weren't baking, we're doomed
         sys.exit(ExitCodes.ARGUMENT_ERROR)
 
-    if parsed.command == "set-custom-alias" or parsed.command == "remove-custom-alias":
+    if parsed.command in ("set-custom-alias", "remove-custom-alias"):
         # Ensure both --command and --alias are provided
         # TODO: Can we use argparse for these
         if len(args) == 4:
             if args[0] != "--command" or args[2] != "--alias":
-                print("Usage: linode-cli set-custom-alias --command [COMMAND] --alias [ALIAS]", file=sys.stderr)
+                print(
+                    "Usage: linode-cli set-custom-alias --command "
+                    "[COMMAND] --alias [ALIAS]",
+                    file=sys.stderr,
+                )
                 sys.exit(ExitCodes.ARGUMENT_ERROR)
 
         # Get the indexes for --command and --alias
         try:
             command_index = args.index("--command") + 1
             alias_index = args.index("--alias") + 1
         except ValueError:
-            print("Both --command and --alias arguments are required.", file=sys.stderr)
+            print(
+                "Both --command and --alias arguments are required.",
+                file=sys.stderr,
+            )
             sys.exit(ExitCodes.ARGUMENT_ERROR)
 
         if command_index >= len(args) or alias_index >= len(args):
-            print("Both --command and --alias arguments must have valid values.", file=sys.stderr)
+            print(
+                "Both --command and --alias arguments must have valid values.",
+                file=sys.stderr,
+            )
             sys.exit(ExitCodes.ARGUMENT_ERROR)
 
         # Retrieve the command and alias from arguments
@@ -136,7 +143,9 @@ def main():  # pylint: disable=too-many-branches,too-many-statements
 
         # Check if the command is valid
         if command not in cli.ops:
-            print(f"Error: '{command}' is not a valid command.", file=sys.stderr)
+            print(
+                f"Error: '{command}' is not a valid command.", file=sys.stderr
+            )
             sys.exit(ExitCodes.ARGUMENT_ERROR)
 
         # Set the alias if it does not already exist
@@ -145,15 +154,19 @@ def main():  # pylint: disable=too-many-branches,too-many-statements
                 cli.config.set_custom_alias(alias, command)
                 print(f"Custom alias '{alias}' set for command '{command}'")
             else:
-                print(f"Custom alias '{alias}' already set for command '{command}'")
+                print(
+                    f"Custom alias '{alias}' already set for command '{command}'"
+                )
 
         # Remove the alias if it already exists
         if parsed.command == "remove-custom-alias":
             if (alias, command) in cli.config.get_custom_aliases().items():
                 cli.config.remove_custom_alias(alias, command)
                 print(f"Custom alias '{alias}' removed for command '{command}'")
             else:
-                print(f"Custom alias '{alias}' does not exist for command '{command}'")
+                print(
+                    f"Custom alias '{alias}' does not exist for command '{command}'"
+                )
 
         sys.exit(ExitCodes.SUCCESS)
 
@@ -271,20 +284,23 @@ def main():  # pylint: disable=too-many-branches,too-many-statements
         sys.exit(ExitCodes.UNRECOGNIZED_COMMAND)
 
     # handle a help for a command - either --help or no action triggers this
-    if (
-        parsed.command is not None
-        and parsed.action is None
-    ):
+    if parsed.command is not None and parsed.action is None:
         if parsed.command in cli.ops:
             print_help_command_actions(cli.ops, parsed.command)
             sys.exit(ExitCodes.SUCCESS)
         if parsed.command in cli.config.get_custom_aliases().keys():
-            print_help_command_actions(cli.ops, cli.config.get_custom_aliases()[parsed.command])
+            print_help_command_actions(
+                cli.ops, cli.config.get_custom_aliases()[parsed.command]
+            )
 
     if parsed.command is not None and parsed.action is not None:
         if parsed.help:
             if parsed.command in cli.config.get_custom_aliases().keys():
-                print_help_action(cli, cli.config.get_custom_aliases()[parsed.command], parsed.action)
+                print_help_action(
+                    cli,
+                    cli.config.get_custom_aliases()[parsed.command],
+                    parsed.action,
+                )
             else:
                 print_help_action(cli, parsed.command, parsed.action)
             sys.exit(ExitCodes.SUCCESS)

linodecli/configuration/config.py

@@ -619,17 +619,24 @@ def remove_custom_alias(self, alias: str, command: str):
         :param command: The command the alias is mapped to.
         """
         if not self.config.has_section("custom_aliases"):
-            print(f"Error: No custom aliases have been set.", file=sys.stderr)
+            print("Error: No custom aliases have been set.", file=sys.stderr)
             return
 
         if not self.config.has_option("custom_aliases", alias):
-            print(f"Error: Custom alias '{alias}' does not exist.", file=sys.stderr)
+            print(
+                f"Error: Custom alias '{alias}' does not exist.",
+                file=sys.stderr,
+            )
             return
 
         # Check if the alias maps to the given command
         existing_command = self.config.get("custom_aliases", alias)
         if existing_command != command:
-            print(f"Error: Custom alias '{alias}' is mapped to '{existing_command}', not '{command}'.", file=sys.stderr)
+            print(
+                f"Error: Custom alias '{alias}' is mapped to "
+                f"'{existing_command}', not '{command}'.",
+                file=sys.stderr,
+            )
             return
 
         # Remove the alias and update the config
@@ -642,4 +649,8 @@ def get_custom_aliases(self) -> Dict[str, str]:
 
         :return: A dictionary mapping custom alias names to their respective commands.
         """
-        return dict(self.config.items("custom_aliases")) if self.config.has_section("custom_aliases") else {}
+        return (
+            dict(self.config.items("custom_aliases"))
+            if (self.config.has_section("custom_aliases"))
+            else {}
+        )