pyegeria 0.2.4__py3-none-any.whl → 5.3.0.dev3__py3-none-any.whl

pyegeria/utils.py

@@ -6,81 +6,71 @@ General utility functions in support of the Egeria Python Client package.
 
 """
 import json
-import textwrap
-
-import pandas as pd
-from tabulate import tabulate
-
-
-def wrap_text(df: pd.DataFrame, wrap_len: int = 30) -> pd.DataFrame:
-    """ Wrap the text in a dataframe
-    Parameters
-    ----------
-    df : pandas.DataFrame
-        The DataFrame to wrap the text in.
-    wrap_len : int, optional
-        The maximum length of each cell's contents to wrap. Defaults to 30.
-
-    Returns
-    -------
-    pandas.DataFrame
-        A new DataFrame with the wrapped text.
-
-    """
-
-    # Helper function to wrap text in a particular cell
-    def wrap_cell_contents(cell_contents: str, wrap: int = 30) -> str:
-        """ Wrap the text in a cell
-        Parameters
-        ----------
-        cell_contents : str or any
-            The contents of a cell in a dataframe.
-        wrap : int, optional
-            The maximum width at which to wrap the cell contents. Default is 30.
-
-        Returns
-        -------
-        str or any
-            If the cell_contents is a string and its length is greater than wrap_len,
-            the contents are wrapped at wrap_len width using textwrap.
-            Otherwise, the original cell_contents are returned as is.
-        """
-        if isinstance(cell_contents, str) and len(cell_contents) > wrap:
-            return textwrap.fill(cell_contents, width=wrap)
-        else:
-            return cell_contents
-
-    # Apply the helper function to each element in the DataFrame
-    return df.map(lambda x: wrap_cell_contents(x, wrap_len))
-
-
-def print_nice_table(df, wrap_len: int = 20, tablefmt: str = "grid") -> None:
-    """ Print a nice table from the data frame"""
-    print(tabulate(wrap_text(df, wrap_len), headers="keys", tablefmt=tablefmt))
-
-
-def get_json_as_table(input_json, wrap_len: int = 30, tablefmt: str = "grid") -> str:
-    """ return the input json as a table"""
-    data = json.loads(json.dumps(input_json))
-    df = pd.json_normalize(data)
-    return tabulate(wrap_text(df, wrap_len), headers="keys", tablefmt=tablefmt)
-
-
-def print_json_list_as_table(input_json, wrap_len: int = 30, tablefmt: str = "grid") -> None:
-    """ print a json list as a table"""
-    data = json.loads(json.dumps(input_json))
-    df = pd.json_normalize(data)
-    print(tabulate(wrap_text(df, wrap_len), headers="keys", tablefmt=tablefmt))
-
-
-def print_rest_request(url):
-    """
-
-    Args:
-        url:
-    """
-    print(" ")
-    print(url)
+from rich import print, print_json
+from datetime import datetime
+from rich.console import Console
+
+console = Console(width=200)
+
+# def wrap_text(df: pd.DataFrame, wrap_len: int = 30) -> pd.DataFrame:
+#     """ Wrap the text in a dataframe
+#     Parameters
+#     ----------
+#     df : pandas.DataFrame
+#         The DataFrame to wrap the text in.
+#     wrap_len : int, optional
+#         The maximum length of each cell's contents to wrap. Defaults to 30.
+#
+#     Returns
+#     -------
+#     pandas.DataFrame
+#         A new DataFrame with the wrapped text.
+#
+#     """
+#
+#     # Helper function to wrap text in a particular cell
+#     def wrap_cell_contents(cell_contents: str, wrap: int = 30) -> str:
+#         """ Wrap the text in a cell
+#         Parameters
+#         ----------
+#         cell_contents : str or any
+#             The contents of a cell in a dataframe.
+#         wrap : int, optional
+#             The maximum width at which to wrap the cell contents. Default is 30.
+#
+#         Returns
+#         -------
+#         str or any
+#             If the cell_contents is a string and its length is greater than wrap_len,
+#             the contents are wrapped at wrap_len width using textwrap.
+#             Otherwise, the original cell_contents are returned as is.
+#         """
+#         if isinstance(cell_contents, str) and len(cell_contents) > wrap:
+#             return textwrap.fill(cell_contents, width=wrap)
+#         else:
+#             return cell_contents
+#
+#     # Apply the helper function to each element in the DataFrame
+#     return df.map(lambda x: wrap_cell_contents(x, wrap_len))
+#
+#
+# def print_nice_table(df, wrap_len: int = 20, tablefmt: str = "grid") -> None:
+#     """ Print a nice table from the data frame"""
+#     print(tabulate(wrap_text(df, wrap_len), headers="keys", tablefmt=tablefmt))
+#
+#
+# def get_json_as_table(input_json, wrap_len: int = 30, tablefmt: str = "grid") -> str:
+#     """ return the input json as a table"""
+#     data = json.loads(json.dumps(input_json))
+#     df = pd.json_normalize(data)
+#     return tabulate(wrap_text(df, wrap_len), headers="keys", tablefmt=tablefmt)
+#
+#
+# def print_json_list_as_table(input_json, wrap_len: int = 30, tablefmt: str = "grid") -> None:
+#     """ print a json list as a table"""
+#     data = json.loads(json.dumps(input_json))
+#     df = pd.json_normalize(data)
+#     print(tabulate(wrap_text(df, wrap_len), headers="keys", tablefmt=tablefmt))
 
 
 def print_rest_request_body(body):
@@ -90,8 +80,7 @@ def print_rest_request_body(body):
         body:
     """
     pretty_body = json.dumps(body, indent=4)
-    print(pretty_body)
-    print(" ")
+    print_json(pretty_body, indent=4, sort_keys=True)
 
 
 def print_rest_response(response):
@@ -101,9 +90,8 @@ def print_rest_response(response):
         response:
     """
     print("Returns:")
-    pretty_response = json.dumps(response, indent=4)
-    print(pretty_response)
-    print(" ")
+    pretty_body = json.dumps(response, indent=4)
+    print_json(pretty_body, indent=4, sort_keys=True)
 
 
 def print_guid_list(guids):
@@ -112,7 +100,7 @@ def print_guid_list(guids):
         print("No assets created")
     else:
         pretty_guids = json.dumps(guids, indent=4)
-        print(pretty_guids)
+        print_json(pretty_guids, indent=4, sort_keys=True)
 
 
 #
@@ -137,9 +125,7 @@ def print_response(response):
     print(" ")
 
 
-def print_unexpected_response(
-        server_name, platform_name, platform_url, response
-):
+def print_unexpected_response(server_name, platform_name, platform_url, response):
     """
 
     Args:
@@ -190,7 +176,7 @@ def get_last_guid(guids):
 
 
 def body_slimmer(body: dict) -> dict:
-    """ body_slimmer is a little function to remove unused keys from a dict
+    """body_slimmer is a little function to remove unused keys from a dict
 
     Parameters
     ----------
@@ -203,3 +189,7 @@ def body_slimmer(body: dict) -> dict:
     """
     slimmed = {key: value for key, value in body.items() if value}
     return slimmed
+
+
+if __name__ == "__main__":
+    print("Main-Utils")