We got tables!

Author: PoshoDev

loaf/__init__.py

@@ -1,6 +1,8 @@
 # A Python module for effortless database usage.
 
 import pymysql, psycopg2, psycopg2.extras, sqlite3, datetime, socket, configparser
+from rich.table import Table
+from rich import print as rprint
 
 cursors = ["DEFAULT", "DICTIONARY"]
 modes = ["MySQL", "PostgreSQL", "SQLite"]
@@ -170,6 +172,40 @@ def call(self, procedure, args=[], rollback_on_error=None):
         # Return the result.
         return self.cursor.fetchall()
 
+    ### PRINT FUNCTIONS ###
+
+    # Prints the result of a query as a rich table. The 'data' argument can be a tuple or a dictionary. 
+    def print(self, data, title=None):
+        # If the data is a dictionary, convert it to a tuple.
+        if type(data) == dict:
+            data = tuple(data.items())
+        # If the data is a tuple, convert it to a list. This is done because the Rich library doesn't support tuples.
+        if type(data) == tuple:
+            data = list(data)
+        # If the data is a list, convert it to a table.
+        if type(data) == list:
+            table = Table(title=title)
+            # If the data is a list of tuples, convert it to a table.
+            if type(data[0]) == tuple:
+                for i in range(len(data[0])):
+                    table.add_column(data[0][i])
+                for i in range(1, len(data)):
+                    table.add_row(*data[i])
+            # If the data is a list of dictionaries, convert it to a table.
+            elif type(data[0]) == dict:
+                for i in data[0].keys():
+                    table.add_column(i)
+                for i in data:
+                    # Parse the values to strings. Then add the rows.
+                    values = []
+                    for j in i.values():
+                        values.append(parse(j))
+                    table.add_row(*values)
+            # Print the table.
+            rprint(table)
+        else:
+            raise Exception("Invalid data type.")
+
     ### EASY FUNCTIONS ###
 
     # A quick SELECT query.
@@ -285,6 +321,8 @@ def currentTimestamp(self):
 def parse(value):
     if value in [None, "", "NULL"]:
         return "NULL"
+    if isinstance(value, int):
+        return str(value)
     if isinstance(value, datetime.date):
         return value.strftime("%Y-%m-%d")
     return "'" + str(value).replace("'", "''") + "'"