pyegeria 0.7.45__py3-none-any.whl → 0.8.0__py3-none-any.whl

examples/widgets/ops/engine_actions.py

@@ -20,24 +20,32 @@ from pyegeria._exceptions import (
 )
 
 GERIA_METADATA_STORE = os.environ.get("EGERIA_METADATA_STORE", "active-metadata-store")
-EGERIA_KAFKA_ENDPOINT = os.environ.get('KAFKA_ENDPOINT', 'localhost:9092')
-EGERIA_PLATFORM_URL = os.environ.get('EGERIA_PLATFORM_URL', 'https://localhost:9443')
-EGERIA_VIEW_SERVER = os.environ.get('VIEW_SERVER', 'view-server')
-EGERIA_VIEW_SERVER_URL = os.environ.get('EGERIA_VIEW_SERVER_URL', 'https://localhost:9443')
-EGERIA_INTEGRATION_DAEMON = os.environ.get('INTEGRATION_DAEMON', 'integration-daemon')
-EGERIA_INTEGRATION_DAEMON_URL = os.environ.get('EGERIA_INTEGRATION_DAEMON_URL', 'https://localhost:9443')
-EGERIA_ADMIN_USER = os.environ.get('ADMIN_USER', 'garygeeke')
-EGERIA_ADMIN_PASSWORD = os.environ.get('ADMIN_PASSWORD', 'secret')
-EGERIA_USER = os.environ.get('EGERIA_USER', 'erinoverview')
-EGERIA_USER_PASSWORD = os.environ.get('EGERIA_USER_PASSWORD', 'secret')
+EGERIA_KAFKA_ENDPOINT = os.environ.get("KAFKA_ENDPOINT", "localhost:9092")
+EGERIA_PLATFORM_URL = os.environ.get("EGERIA_PLATFORM_URL", "https://localhost:9443")
+EGERIA_VIEW_SERVER = os.environ.get("VIEW_SERVER", "view-server")
+EGERIA_VIEW_SERVER_URL = os.environ.get(
+    "EGERIA_VIEW_SERVER_URL", "https://localhost:9443"
+)
+EGERIA_INTEGRATION_DAEMON = os.environ.get("INTEGRATION_DAEMON", "integration-daemon")
+EGERIA_INTEGRATION_DAEMON_URL = os.environ.get(
+    "EGERIA_INTEGRATION_DAEMON_URL", "https://localhost:9443"
+)
+EGERIA_ADMIN_USER = os.environ.get("ADMIN_USER", "garygeeke")
+EGERIA_ADMIN_PASSWORD = os.environ.get("ADMIN_PASSWORD", "secret")
+EGERIA_USER = os.environ.get("EGERIA_USER", "erinoverview")
+EGERIA_USER_PASSWORD = os.environ.get("EGERIA_USER_PASSWORD", "secret")
 
 
-@click.command('stop')
-@click.option('--server', default=EGERIA_VIEW_SERVER, help='Egeria metadata store to load')
-@click.option('--url', default=EGERIA_VIEW_SERVER_URL, help='URL of Egeria platform to connect to')
-@click.option('--userid', default=EGERIA_ADMIN_USER, help='Egeria admin user')
-@click.option('--password', default=EGERIA_ADMIN_PASSWORD, help='Egeria admin password')
-@click.option('--timeout', default=60, help='Number of seconds to wait')
+@click.command("stop")
+@click.option(
+    "--server", default=EGERIA_VIEW_SERVER, help="Egeria metadata store to load"
+)
+@click.option(
+    "--url", default=EGERIA_VIEW_SERVER_URL, help="URL of Egeria platform to connect to"
+)
+@click.option("--userid", default=EGERIA_ADMIN_USER, help="Egeria admin user")
+@click.option("--password", default=EGERIA_ADMIN_PASSWORD, help="Egeria admin password")
+@click.option("--timeout", default=60, help="Number of seconds to wait")
 def stop_daemon(file, server, url, userid, password, timeout):
     """Stop an engine-host daemon"""
     p_client = Platform(server, url, userid, password)
@@ -51,14 +59,18 @@ def stop_daemon(file, server, url, userid, password, timeout):
         p_client.close_session()
 
 
-@click.command('start')
-@click.option('--server', default=EGERIA_VIEW_SERVER, help='Egeria metadata store to load')
-@click.option('--url', default=EGERIA_VIEW_SERVER_URL, help='URL of Egeria platform to connect to')
-@click.option('--userid', default=EGERIA_ADMIN_USER, help='Egeria admin user')
-@click.option('--password', default=EGERIA_ADMIN_PASSWORD, help='Egeria admin password')
-@click.option('--timeout', default=60, help='Number of seconds to wait')
+@click.command("start")
+@click.option(
+    "--server", default=EGERIA_VIEW_SERVER, help="Egeria metadata store to load"
+)
+@click.option(
+    "--url", default=EGERIA_VIEW_SERVER_URL, help="URL of Egeria platform to connect to"
+)
+@click.option("--userid", default=EGERIA_ADMIN_USER, help="Egeria admin user")
+@click.option("--password", default=EGERIA_ADMIN_PASSWORD, help="Egeria admin password")
+@click.option("--timeout", default=60, help="Number of seconds to wait")
 def start_daemon(file, server, url, userid, password, timeout):
-    """Start or restart an engine-host from its known configuration """
+    """Start or restart an engine-host from its known configuration"""
     p_client = Platform(server, url, userid, password)
     try:
         p_client.activate_server_stored_config()

examples/widgets/ops/integration_daemon_actions.py

@@ -22,79 +22,97 @@ from pyegeria._exceptions import (
 )
 
 
-@click.command('add-target')
-@click.argument('integration-connector')
-@click.argument('metadata-element-guid')
-@click.argument('catalog-target-name')
+@click.command("add-target")
+@click.argument("integration-connector")
+@click.argument("metadata-element-guid")
+@click.argument("catalog-target-name")
 @click.pass_context
-def add_catalog_target(ctx, integration_connector: str, metadata_element_guid:str, catalog_target_name:str)-> str:
+def add_catalog_target(
+    ctx,
+    integration_connector: str,
+    metadata_element_guid: str,
+    catalog_target_name: str,
+) -> str:
     """Add catalog targets to the specified integration connector"""
     try:
         if integration_connector not in INTEGRATION_GUIDS.keys():
-            click.echo('Integration connector is not known')
+            click.echo("Integration connector is not known")
 
         c = ctx.obj
-        a_client = AutomatedCuration(c.view_server, c.view_server_url, c.userid, c.password)
+        a_client = AutomatedCuration(
+            c.view_server, c.view_server_url, c.userid, c.password
+        )
         token = a_client.create_egeria_bearer_token()
 
-        guid = a_client.add_catalog_target(INTEGRATION_GUIDS[integration_connector], metadata_element_guid,
-                                           catalog_target_name)
-
-        click.echo(f"Added catalog target to {integration_connector} with a return guid of {guid}")
+        guid = a_client.add_catalog_target(
+            INTEGRATION_GUIDS[integration_connector],
+            metadata_element_guid,
+            catalog_target_name,
+        )
 
+        click.echo(
+            f"Added catalog target to {integration_connector} with a return guid of {guid}"
+        )
 
     except (InvalidParameterException, PropertyServerException) as e:
         print_exception_response(e)
 
 
-
-@click.command('remove-target')
-@click.argument('relationship-guid')
+@click.command("remove-target")
+@click.argument("relationship-guid")
 @click.pass_context
 def remove_catalog_target(ctx, relationship_guid: str):
     """Remove the catalog target specified by the relationship guidr"""
     try:
         c = ctx.obj
-        a_client = AutomatedCuration(c.view_server, c.view_server_url, c.userid, c.password)
+        a_client = AutomatedCuration(
+            c.view_server, c.view_server_url, c.userid, c.password
+        )
         token = a_client.create_egeria_bearer_token()
 
         a_client.remove_catalog_target(relationship_guid)
 
-        click.echo(f"Removed catalog target with relationship guid of {relationship_guid}")
-
+        click.echo(
+            f"Removed catalog target with relationship guid of {relationship_guid}"
+        )
 
     except (InvalidParameterException, PropertyServerException) as e:
         print_exception_response(e)
 
 
-@click.command('update-target')
-@click.argument('relationship-guid')
-@click.argument('catalog-target-name')
+@click.command("update-target")
+@click.argument("relationship-guid")
+@click.argument("catalog-target-name")
 @click.pass_context
-def update_catalog_target(ctx, relationship_guid: str, catalog_target_name:str):
-    """Update the catalog target specified by the relationship guid """
+def update_catalog_target(ctx, relationship_guid: str, catalog_target_name: str):
+    """Update the catalog target specified by the relationship guid"""
     try:
         c = ctx.obj
-        a_client = AutomatedCuration(c.view_server, c.view_server_url, c.userid, c.password)
+        a_client = AutomatedCuration(
+            c.view_server, c.view_server_url, c.userid, c.password
+        )
         token = a_client.create_egeria_bearer_token()
 
         guid = a_client.update_catalog_target(relationship_guid, catalog_target_name)
 
-        click.echo(f"Update catalog target with relationship guid of  {relationship_guid} to a catalog target name of "
-                   f"{catalog_target_name} with a return guid of {guid}")
-
+        click.echo(
+            f"Update catalog target with relationship guid of  {relationship_guid} to a catalog target name of "
+            f"{catalog_target_name} with a return guid of {guid}"
+        )
 
     except (InvalidParameterException, PropertyServerException) as e:
         print_exception_response(e)
 
 
-@click.command('stop')
+@click.command("stop")
 @click.pass_context
 def stop_server(ctx):
     """Stop an integration daemon"""
     try:
         c = ctx.obj
-        p_client = Platform(c.integration_daemon, c.integration_daemon_url, c.userid, c.password)
+        p_client = Platform(
+            c.integration_daemon, c.integration_daemon_url, c.userid, c.password
+        )
 
         p_client.shutdown_server()
 
@@ -103,13 +121,15 @@ def stop_server(ctx):
         print_exception_response(e)
 
 
-@click.command('start')
+@click.command("start")
 @click.pass_context
 def start_server(ctx):
-    """Start or restart an integration daemon from its known configuration """
+    """Start or restart an integration daemon from its known configuration"""
     try:
         c = ctx.obj
-        p_client = Platform(c.integration_daemon, c.integration_daemon_url, c.userid, c.password)
+        p_client = Platform(
+            c.integration_daemon, c.integration_daemon_url, c.userid, c.password
+        )
 
         p_client.activate_server_stored_config()
 
@@ -117,4 +137,3 @@ def start_server(ctx):
 
     except (InvalidParameterException, PropertyServerException) as e:
         print_exception_response(e)
-

examples/widgets/tech/get_element_info.py

@@ -25,66 +25,90 @@ from pyegeria import (
     InvalidParameterException,
     PropertyServerException,
     UserNotAuthorizedException,
-    Client, ClassificationManager
+    Client,
+    ClassificationManager,
 )
 
 
 EGERIA_METADATA_STORE = os.environ.get("EGERIA_METADATA_STORE", "active-metadata-store")
-EGERIA_KAFKA_ENDPOINT = os.environ.get('KAFKA_ENDPOINT', 'localhost:9092')
-EGERIA_PLATFORM_URL = os.environ.get('EGERIA_PLATFORM_URL', 'https://localhost:9443')
-EGERIA_VIEW_SERVER = os.environ.get('VIEW_SERVER', 'view-server')
-EGERIA_VIEW_SERVER_URL = os.environ.get('EGERIA_VIEW_SERVER_URL', 'https://localhost:9443')
-EGERIA_INTEGRATION_DAEMON = os.environ.get('INTEGRATION_DAEMON', 'integration-daemon')
-EGERIA_ADMIN_USER = os.environ.get('ADMIN_USER', 'garygeeke')
-EGERIA_ADMIN_PASSWORD = os.environ.get('ADMIN_PASSWORD', 'secret')
-EGERIA_USER = os.environ.get('EGERIA_USER', 'erinoverview')
-EGERIA_USER_PASSWORD = os.environ.get('EGERIA_USER_PASSWORD', 'secret')
-EGERIA_JUPYTER = bool(os.environ.get('EGERIA_JUPYTER', 'False'))
-EGERIA_WIDTH = int(os.environ.get('EGERIA_WIDTH', '200'))
-
-
-def display_elements(om_type: str, server: str, url: str, username: str, user_password: str,
-                 jupyter: bool = EGERIA_JUPYTER, width: int = EGERIA_WIDTH
-                 ):
+EGERIA_KAFKA_ENDPOINT = os.environ.get("KAFKA_ENDPOINT", "localhost:9092")
+EGERIA_PLATFORM_URL = os.environ.get("EGERIA_PLATFORM_URL", "https://localhost:9443")
+EGERIA_VIEW_SERVER = os.environ.get("VIEW_SERVER", "view-server")
+EGERIA_VIEW_SERVER_URL = os.environ.get(
+    "EGERIA_VIEW_SERVER_URL", "https://localhost:9443"
+)
+EGERIA_INTEGRATION_DAEMON = os.environ.get("INTEGRATION_DAEMON", "integration-daemon")
+EGERIA_ADMIN_USER = os.environ.get("ADMIN_USER", "garygeeke")
+EGERIA_ADMIN_PASSWORD = os.environ.get("ADMIN_PASSWORD", "secret")
+EGERIA_USER = os.environ.get("EGERIA_USER", "erinoverview")
+EGERIA_USER_PASSWORD = os.environ.get("EGERIA_USER_PASSWORD", "secret")
+EGERIA_JUPYTER = bool(os.environ.get("EGERIA_JUPYTER", "False"))
+EGERIA_WIDTH = int(os.environ.get("EGERIA_WIDTH", "200"))
+
+
+def display_elements(
+    om_type: str,
+    server: str,
+    url: str,
+    username: str,
+    user_password: str,
+    jupyter: bool = EGERIA_JUPYTER,
+    width: int = EGERIA_WIDTH,
+):
     c = ClassificationManager(server, url, user_id=username, user_pwd=user_password)
 
     bearer_token = c.create_egeria_bearer_token(username, user_password)
 
     try:
-        console = Console(width=width, force_terminal=not jupyter, style="bold white on black")
+        console = Console(
+            width=width, force_terminal=not jupyter, style="bold white on black"
+        )
         r = c.get_elements(om_type)
         if type(r) is not list:
             print(f"\n\n\tno elements found: {r}")
             sys.exit(1)
 
-        tree = Tree(f"Elements for Open Metadata Type:{om_type}\n* There are {len(r)} elements", style="bold bright_white on black",
-                    guide_style="bold bright_blue")
+        tree = Tree(
+            f"Elements for Open Metadata Type:{om_type}\n* There are {len(r)} elements",
+            style="bold bright_white on black",
+            guide_style="bold bright_blue",
+        )
         t = tree.add(f"Elements for {om_type}", style="bold bright_white on black")
         for element in r:
-            header = element['elementHeader']
-            el_type = header["type"]['typeName']
-            el_home = header['origin']['homeMetadataCollectionName']
-            el_create_time = header['versions']['createTime']
-            el_guid = header['guid']
-
-            el_md = (f"#### Element Basics\n"
-                     f"* **Type**: {el_type}\n"
-                     f"* **Home**: {el_home}\n"
-                     f"* **Created**: {el_create_time}\n"
-                     f"* **GUID**: {el_guid}\n ---\n")
-            for prop in element['properties'].keys():
+            header = element["elementHeader"]
+            el_type = header["type"]["typeName"]
+            el_home = header["origin"]["homeMetadataCollectionName"]
+            el_create_time = header["versions"]["createTime"]
+            el_guid = header["guid"]
+
+            el_md = (
+                f"#### Element Basics\n"
+                f"* **Type**: {el_type}\n"
+                f"* **Home**: {el_home}\n"
+                f"* **Created**: {el_create_time}\n"
+                f"* **GUID**: {el_guid}\n ---\n"
+            )
+            for prop in element["properties"].keys():
                 el_md += f"* **{prop}**: {element['properties'][prop]}\n"
 
             el_out = Markdown(el_md)
-            p = Panel.fit(el_out, title=element['properties']['qualifiedName'],
-                  style = "bold white on black")
+            p = Panel.fit(
+                el_out,
+                title=element["properties"]["qualifiedName"],
+                style="bold white on black",
+            )
             t = tree.add(p)
 
         print(tree)
 
         c.close_session()
 
-    except (InvalidParameterException, PropertyServerException, UserNotAuthorizedException, ValueError) as e:
+    except (
+        InvalidParameterException,
+        PropertyServerException,
+        UserNotAuthorizedException,
+        ValueError,
+    ) as e:
         if type(e) is str:
             console.print_exception()
         else:
@@ -92,7 +116,6 @@ def display_elements(om_type: str, server: str, url: str, username: str, user_pa
             console.print(f"\n ===> Looks like the type {om_type} isn't known...\n")
 
 
-
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--server", help="Name of the server to display status for")
@@ -109,9 +132,11 @@ def main():
     user_pass = args.password if args.password is not None else EGERIA_USER_PASSWORD
 
     try:
-        om_type = Prompt.ask("Enter the Open Metadata Type to find Elements for", default=None)
+        om_type = Prompt.ask(
+            "Enter the Open Metadata Type to find Elements for", default=None
+        )
         display_elements(om_type, server, url, userid, user_pass)
-    except (KeyboardInterrupt):
+    except KeyboardInterrupt:
         pass
 
 

examples/widgets/tech/get_guid_info.py

@@ -24,49 +24,68 @@ from pyegeria import (
     InvalidParameterException,
     PropertyServerException,
     UserNotAuthorizedException,
-    Client, ClassificationManager
+    Client,
+    ClassificationManager,
 )
 
 
 EGERIA_METADATA_STORE = os.environ.get("EGERIA_METADATA_STORE", "active-metadata-store")
-EGERIA_KAFKA_ENDPOINT = os.environ.get('KAFKA_ENDPOINT', 'localhost:9092')
-EGERIA_PLATFORM_URL = os.environ.get('EGERIA_PLATFORM_URL', 'https://localhost:9443')
-EGERIA_VIEW_SERVER = os.environ.get('VIEW_SERVER', 'view-server')
-EGERIA_VIEW_SERVER_URL = os.environ.get('EGERIA_VIEW_SERVER_URL', 'https://localhost:9443')
-EGERIA_INTEGRATION_DAEMON = os.environ.get('INTEGRATION_DAEMON', 'integration-daemon')
-EGERIA_ADMIN_USER = os.environ.get('ADMIN_USER', 'garygeeke')
-EGERIA_ADMIN_PASSWORD = os.environ.get('ADMIN_PASSWORD', 'secret')
-EGERIA_USER = os.environ.get('EGERIA_USER', 'erinoverview')
-EGERIA_USER_PASSWORD = os.environ.get('EGERIA_USER_PASSWORD', 'secret')
-EGERIA_JUPYTER = bool(os.environ.get('EGERIA_JUPYTER', 'False'))
-EGERIA_WIDTH = int(os.environ.get('EGERIA_WIDTH', '200'))
-
-
-def display_guid(guid: str, server: str, url: str, username: str, user_password: str,
-                 jupyter: bool = EGERIA_JUPYTER, width: int = EGERIA_WIDTH
-                 ):
+EGERIA_KAFKA_ENDPOINT = os.environ.get("KAFKA_ENDPOINT", "localhost:9092")
+EGERIA_PLATFORM_URL = os.environ.get("EGERIA_PLATFORM_URL", "https://localhost:9443")
+EGERIA_VIEW_SERVER = os.environ.get("VIEW_SERVER", "view-server")
+EGERIA_VIEW_SERVER_URL = os.environ.get(
+    "EGERIA_VIEW_SERVER_URL", "https://localhost:9443"
+)
+EGERIA_INTEGRATION_DAEMON = os.environ.get("INTEGRATION_DAEMON", "integration-daemon")
+EGERIA_ADMIN_USER = os.environ.get("ADMIN_USER", "garygeeke")
+EGERIA_ADMIN_PASSWORD = os.environ.get("ADMIN_PASSWORD", "secret")
+EGERIA_USER = os.environ.get("EGERIA_USER", "erinoverview")
+EGERIA_USER_PASSWORD = os.environ.get("EGERIA_USER_PASSWORD", "secret")
+EGERIA_JUPYTER = bool(os.environ.get("EGERIA_JUPYTER", "False"))
+EGERIA_WIDTH = int(os.environ.get("EGERIA_WIDTH", "200"))
+
+
+def display_guid(
+    guid: str,
+    server: str,
+    url: str,
+    username: str,
+    user_password: str,
+    jupyter: bool = EGERIA_JUPYTER,
+    width: int = EGERIA_WIDTH,
+):
     c = Client(server, url, user_id=username)
-    url = (f"{url}/servers/{server}/open-metadata/repository-services/users/{username}/"
-           f"instances/entity/{guid}")
+    url = (
+        f"{url}/servers/{server}/open-metadata/repository-services/users/{username}/"
+        f"instances/entity/{guid}"
+    )
     # c =  ClassificationManager(server, url)
 
     bearer_token = c.create_egeria_bearer_token(username, user_password)
 
     try:
-        console = Console(width=width, force_terminal=not jupyter, style="bold white on black")
+        console = Console(
+            width=width, force_terminal=not jupyter, style="bold white on black"
+        )
         r = c.make_request("GET", url)
         if r.status_code == 200:
             pass
         # r = c.retrieve_instance_for_guid(guid)
-        e = r.json()['entity']
-        p = e['properties']['instanceProperties']
+        e = r.json()["entity"]
+        p = e["properties"]["instanceProperties"]
 
         type_name = Text(f"Type is: {e['type']['typeDefName']}")
-        metadataCollection = Text(f"Metadadata Collection: {e['metadataCollectionName']}")
+        metadataCollection = Text(
+            f"Metadadata Collection: {e['metadataCollectionName']}"
+        )
         created = Text(f"Created at: {e['createTime']}")
         details = Text(f"Details: {json.dumps(p, indent=2)}")
 
-        tree = Tree(f"{guid}", style="bold bright_white on black", guide_style="bold bright_blue")
+        tree = Tree(
+            f"{guid}",
+            style="bold bright_white on black",
+            guide_style="bold bright_blue",
+        )
 
         tree = tree.add(type_name)
         tree.add(metadataCollection)
@@ -76,7 +95,12 @@ def display_guid(guid: str, server: str, url: str, username: str, user_password:
 
         c.close_session()
 
-    except (InvalidParameterException, PropertyServerException, UserNotAuthorizedException, ValueError) as e:
+    except (
+        InvalidParameterException,
+        PropertyServerException,
+        UserNotAuthorizedException,
+        ValueError,
+    ) as e:
         if type(e) is str:
             console.print_exception()
         else:
@@ -84,7 +108,6 @@ def display_guid(guid: str, server: str, url: str, username: str, user_password:
             console.print(f"\n ===> Looks like the GUID isn't known...\n")
 
 
-
 def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--server", help="Name of the server to display status for")
@@ -103,7 +126,7 @@ def main():
     try:
         guid = Prompt.ask("Enter the GUID to retrieve", default=None)
         display_guid(guid, server, url, userid, user_pass)
-    except (KeyboardInterrupt):
+    except KeyboardInterrupt:
         pass
 
 

examples/widgets/tech/list_asset_types.py

@@ -21,26 +21,33 @@ from pyegeria import (
     PropertyServerException,
     UserNotAuthorizedException,
     print_exception_response,
-    RegisteredInfo
+    RegisteredInfo,
 )
 
 EGERIA_METADATA_STORE = os.environ.get("EGERIA_METADATA_STORE", "active-metadata-store")
-EGERIA_KAFKA_ENDPOINT = os.environ.get('KAFKA_ENDPOINT', 'localhost:9092')
-EGERIA_PLATFORM_URL = os.environ.get('EGERIA_PLATFORM_URL', 'https://localhost:9443')
-EGERIA_VIEW_SERVER = os.environ.get('VIEW_SERVER', 'view-server')
-EGERIA_VIEW_SERVER_URL = os.environ.get('EGERIA_VIEW_SERVER_URL', 'https://localhost:9443')
-EGERIA_INTEGRATION_DAEMON = os.environ.get('INTEGRATION_DAEMON', 'integration-daemon')
-EGERIA_ADMIN_USER = os.environ.get('ADMIN_USER', 'garygeeke')
-EGERIA_ADMIN_PASSWORD = os.environ.get('ADMIN_PASSWORD', 'secret')
-EGERIA_USER = os.environ.get('EGERIA_USER', 'erinoverview')
-EGERIA_USER_PASSWORD = os.environ.get('EGERIA_USER_PASSWORD', 'secret')
-EGERIA_JUPYTER = bool(os.environ.get('EGERIA_JUPYTER', 'False'))
-EGERIA_WIDTH = int(os.environ.get('EGERIA_WIDTH', '200'))
-
-
-def display_asset_types(server: str, url: str, username: str, user_password: str,
-                        jupyter: bool = EGERIA_JUPYTER, width: int = EGERIA_WIDTH
-                        ):
+EGERIA_KAFKA_ENDPOINT = os.environ.get("KAFKA_ENDPOINT", "localhost:9092")
+EGERIA_PLATFORM_URL = os.environ.get("EGERIA_PLATFORM_URL", "https://localhost:9443")
+EGERIA_VIEW_SERVER = os.environ.get("VIEW_SERVER", "view-server")
+EGERIA_VIEW_SERVER_URL = os.environ.get(
+    "EGERIA_VIEW_SERVER_URL", "https://localhost:9443"
+)
+EGERIA_INTEGRATION_DAEMON = os.environ.get("INTEGRATION_DAEMON", "integration-daemon")
+EGERIA_ADMIN_USER = os.environ.get("ADMIN_USER", "garygeeke")
+EGERIA_ADMIN_PASSWORD = os.environ.get("ADMIN_PASSWORD", "secret")
+EGERIA_USER = os.environ.get("EGERIA_USER", "erinoverview")
+EGERIA_USER_PASSWORD = os.environ.get("EGERIA_USER_PASSWORD", "secret")
+EGERIA_JUPYTER = bool(os.environ.get("EGERIA_JUPYTER", "False"))
+EGERIA_WIDTH = int(os.environ.get("EGERIA_WIDTH", "200"))
+
+
+def display_asset_types(
+    server: str,
+    url: str,
+    username: str,
+    user_password: str,
+    jupyter: bool = EGERIA_JUPYTER,
+    width: int = EGERIA_WIDTH,
+):
     r_client = RegisteredInfo(server, url, username)
     token = r_client.create_egeria_bearer_token(username, user_password)
     asset_types = r_client.list_asset_types()
@@ -56,7 +63,7 @@ def display_asset_types(server: str, url: str, username: str, user_password: str
             show_lines=True,
             box=box.ROUNDED,
             caption=f"Asset Types from Server '{server}' @ Platform - {url}",
-            expand=True
+            expand=True,
         )
 
         table.add_column("Name")
@@ -74,9 +81,7 @@ def display_asset_types(server: str, url: str, username: str, user_password: str
                 description = a_type.get("description", "none")
                 version = a_type.get("version", " ")
                 super_type = a_type.get("superType", "none")
-                table.add_row(
-                    name, description, super_type, str(version)
-                )
+                table.add_row(name, description, super_type, str(version))
         return table
 
     try:
@@ -84,7 +89,11 @@ def display_asset_types(server: str, url: str, username: str, user_password: str
         with console.pager(styles=True):
             console.print(generate_table())
 
-    except (InvalidParameterException, PropertyServerException, UserNotAuthorizedException) as e:
+    except (
+        InvalidParameterException,
+        PropertyServerException,
+        UserNotAuthorizedException,
+    ) as e:
         print_exception_response(e)
         assert e.related_http_code != "200", "Invalid parameters"
     finally:
@@ -109,5 +118,6 @@ def main():
 
     display_asset_types(server, url, userid, user_pass)
 
+
 if __name__ == "__main__":
-    main()
+    main()