From dd10f6db78db480f83e57332b0f5b5c4a4d0a67d Mon Sep 17 00:00:00 2001
From: Steve Nyemba <nyemba@gmail.com>
Date: Mon, 24 Feb 2025 09:35:36 -0600
Subject: [PATCH] bug fix: version & cli

---
 bin/transport         | 14 +++++++++++---
 transport/__init__.py |  2 +-
 2 files changed, 12 insertions(+), 4 deletions(-)

diff --git a/bin/transport b/bin/transport
index eb8b17a..6ca01bc 100755
--- a/bin/transport
+++ b/bin/transport
@@ -103,14 +103,22 @@ def supported (format:Annotated[str,typer.Argument(help="format of the output, s
     else:
          print (_df)
     print ()
-
+@app_i.command(name="version")
+def version ():
+    """
+    This function will return the version of the data-transport
+    """
+    print()
+    print (f'[bold] {transport.__app_name__} ,[blue] {transport.__edition__} edition [/blue], version {transport.__version__}[/bold]')
+    print ()
+ 
 @app_i.command(name="license")
 def info():
     """
     This function will display version and license information
     """
-
-    print (f'[bold] {transport.__app_name__} ,version {transport.__version__}[/bold]')
+    print()
+    print (f'[bold] {transport.__app_name__} ,{transport.__edition__}, version {transport.__version__}[/bold]')
     print ()
     print (transport.__license__)
 
diff --git a/transport/__init__.py b/transport/__init__.py
index 33a3261..c3bb901 100644
--- a/transport/__init__.py
+++ b/transport/__init__.py
@@ -22,7 +22,7 @@ from transport import sql, nosql, cloud, other, warehouse
 import pandas as pd
 import json
 import os
-from info import __version__,__author__,__email__,__license__,__app_name__,__whatsnew__
+from info import __version__,__author__,__email__,__license__,__app_name__,__whatsnew__,__edition__
 from transport.iowrapper import IWriter, IReader, IETL
 from transport.plugins import PluginLoader
 from transport import providers