1 files changed, 30 insertions, 11 deletions
diff --git a/eos_downloader/cli/debug/commands.py b/eos_downloader/cli/debug/commands.py
index 107b8a0..5a0d7f8 100644
--- a/eos_downloader/cli/debug/commands.py
+++ b/eos_downloader/cli/debug/commands.py
@@ -22,32 +22,51 @@ import eos_downloader.eos
 
 @click.command()
 @click.pass_context
-@click.option('--output', default=str('arista.xml'), help='Path to save XML file', type=click.Path(), show_default=True)
-@click.option('--log-level', '--log', help='Logging level of the command', default=None, type=click.Choice(['debug', 'info', 'warning', 'error', 'critical'], case_sensitive=False))
+@click.option(
+    "--output",
+    default=str("arista.xml"),
+    help="Path to save XML file",
+    type=click.Path(),
+    show_default=True,
+)
+@click.option(
+    "--log-level",
+    "--log",
+    help="Logging level of the command",
+    default=None,
+    type=click.Choice(
+        ["debug", "info", "warning", "error", "critical"], case_sensitive=False
+    ),
+)
 def xml(ctx: click.Context, output: str, log_level: str) -> None:
     # sourcery skip: remove-unnecessary-cast
     """Extract XML directory structure"""
     console = Console()
     # Get from Context
-    token = ctx.obj['token']
+    token = ctx.obj["token"]
 
     logger.remove()
     if log_level is not None:
         logger.add("eos-downloader.log", rotation="10 MB", level=log_level.upper())
 
     my_download = eos_downloader.eos.EOSDownloader(
-        image='unset',
-        software='EOS',
-        version='unset',
+        image="unset",
+        software="EOS",
+        version="unset",
         token=token,
-        hash_method='sha512sum')
+        hash_method="sha512sum",
+    )
 
     my_download.authenticate()
-    xml_object: ET.ElementTree = my_download._get_folder_tree()  # pylint: disable=protected-access
+    xml_object: ET.ElementTree = (
+        my_download.get_folder_tree()
+    )  # pylint: disable=protected-access
     xml_content = xml_object.getroot()
 
-    xmlstr = minidom.parseString(ET.tostring(xml_content)).toprettyxml(indent="    ", newl='')
-    with open(output, "w", encoding='utf-8') as f:
+    xmlstr = minidom.parseString(ET.tostring(xml_content)).toprettyxml(
+        indent="    ", newl=""
+    )
+    with open(output, "w", encoding="utf-8") as f:
         f.write(str(xmlstr))
 
-    console.print(f'XML file saved in: { output }')
+    console.print(f"XML file saved in: { output }")