Update the import error

Author: BenConstable9

text_2_sql/text_2_sql_core/src/text_2_sql_core/data_dictionary/cli.py

@@ -6,6 +6,7 @@
 import typer
 from rich import print as rich_print
 from tenacity import RetryError
+import traceback
 
 logging.basicConfig(level=logging.INFO)
 
@@ -103,11 +104,13 @@ def create(
             rich_print(f"Database Engine {engine.value} is not supported.")
 
             raise typer.Exit(code=1)
-    except ImportError:
+    except Exception as e:
+        logging.error(e)
         detailed_error = f"""Failed to import {
-            engine.value} Data Dictionary Creator. Check you have installed the optional dependencies for this database engine."""
+            engine.value} Data Dictionary Creator. Check you have installed the optional dependencies for this database engine and have configured all the environmental variables."""
         rich_print("Text2SQL Data Dictionary Creator Failed ❌")
         rich_print(detailed_error)
+        rich_print(f"Error Messages: {traceback.format_exc()}")
 
         raise typer.Exit(code=1)
 
@@ -120,14 +123,15 @@ def create(
         rich_print("Text2SQL Data Dictionary Creator Failed ❌")
 
         rich_print(f"Error Messages: {e}")
+        rich_print(traceback.format_exc())
 
         raise typer.Exit(code=1)
     except Exception as e:
         logging.error(e)
 
         rich_print("Text2SQL Data Dictionary Creator Failed ❌")
 
-        rich_print(f"Error Messages: {e}")
+        rich_print(f"Error Messages: {traceback.format_exc()}")
 
         raise typer.Exit(code=1)
     else:

text_2_sql/text_2_sql_core/src/text_2_sql_core/data_dictionary/postgresql_data_dictionary_creator.py

@@ -4,6 +4,7 @@
     DataDictionaryCreator,
     EntityItem,
 )
+import asyncio
 import os
 
 from text_2_sql_core.utils.database import DatabaseEngine
@@ -112,3 +113,8 @@ def extract_entity_relationships_sql_query(self) -> str:
             pg_attribute pk_col ON fk.confrelid = pk_col.attrelid AND fk.confkey[1] = pk_col.attnum
         ORDER BY
             "EntitySchema", "Entity", "ForeignEntitySchema", "ForeignEntity";"""
+
+
+if __name__ == "__main__":
+    data_dictionary_creator = PostgresqlDataDictionaryCreator()
+    asyncio.run(data_dictionary_creator.create_data_dictionary())