refactor: use explicit kwargs over **options, type fixes

Author: DenverCoder1

table2ascii/options.py

@@ -2,16 +2,15 @@
 from typing import List, Optional
 
 from .alignment import Alignment
-from .preset_style import PresetStyle
 from .table_style import TableStyle
 
 
 @dataclass
 class Options:
     """Class for storing options that the user sets"""
 
-    first_col_heading: bool = False
-    last_col_heading: bool = False
-    column_widths: Optional[List[int]] = None
-    alignments: Optional[List[Alignment]] = None
-    style: TableStyle = PresetStyle.double_thin_compact
+    first_col_heading: bool
+    last_col_heading: bool
+    column_widths: Optional[List[int]]
+    alignments: Optional[List[Alignment]]
+    style: TableStyle

table2ascii/table_to_ascii.py

@@ -3,19 +3,27 @@
 
 from .alignment import Alignment
 from .options import Options
+from .preset_style import PresetStyle
+from .table_style import TableStyle
 
 
 class TableToAscii:
     """Class used to convert a 2D Python table to ASCII text"""
 
-    def __init__(self, header: List, body: List[List], footer: List, options: Options):
+    def __init__(
+        self,
+        header: Optional[List[Any]],
+        body: Optional[List[List[Any]]],
+        footer: Optional[List[Any]],
+        options: Options,
+    ):
         """
         Validate arguments and initialize fields
 
         Args:
-            header (List): The values in the header of the table
-            body (List[List]): The rows of values in the body of the table
-            footer (List): The values in the footer of the table
+            header (Optional[List[Any]]): The values in the header of the table
+            body (Optional[List[List[Any]]]): The rows of values in the body of the table
+            footer (Optional[List[Any]]): The values in the footer of the table
             options (Options): The options for the table
         """
         # initialize fields
@@ -31,9 +39,7 @@ def __init__(self, header: List, body: List[List], footer: List, options: Option
 
         # check if footer has a different number of columns
         if footer and len(footer) != self.__columns:
-            raise ValueError(
-                "Footer must have the same number of columns as the other rows"
-            )
+            raise ValueError("Footer must have the same number of columns as the other rows")
         # check if any rows in body have a different number of columns
         if body and any(len(row) != self.__columns for row in body):
             raise ValueError(
@@ -45,9 +51,7 @@ def __init__(self, header: List, body: List[List], footer: List, options: Option
         if options.column_widths:
             # check that the right number of columns were specified
             if len(options.column_widths) != self.__columns:
-                raise ValueError(
-                    "Length of `column_widths` list must equal the number of columns"
-                )
+                raise ValueError("Length of `column_widths` list must equal the number of columns")
             # check that each column is at least as large as the minimum size
             for i in range(len(options.column_widths)):
                 option = options.column_widths[i]
@@ -62,9 +66,7 @@ def __init__(self, header: List, body: List[List], footer: List, options: Option
 
         # check if alignments specified have a different number of columns
         if options.alignments and len(options.alignments) != self.__columns:
-            raise ValueError(
-                "Length of `alignments` list must equal the number of columns"
-            )
+            raise ValueError("Length of `alignments` list must equal the number of columns")
 
     def __count_columns(self) -> int:
         """
@@ -101,9 +103,7 @@ def longest_line(text: str) -> int:
             # number of characters in column of i of header, each body row, and footer
             header_size = longest_line(str(self.__header[i])) if self.__header else 0
             body_size = (
-                map(lambda row, i=i: longest_line(str(row[i])), self.__body)
-                if self.__body
-                else [0]
+                map(lambda row, i=i: longest_line(str(row[i])), self.__body) if self.__body else [0]
             )
             footer_size = longest_line(str(self.__footer[i])) if self.__footer else 0
             # get the max and add 2 for padding each side with a space
@@ -256,7 +256,7 @@ def __heading_sep_to_ascii(self) -> str:
             filler=self.__style.heading_row_sep,
         )
 
-    def __body_to_ascii(self) -> str:
+    def __body_to_ascii(self, body: List[List[Any]]) -> str:
         """
         Assembles the body of the ascii table
 
@@ -278,7 +278,7 @@ def __body_to_ascii(self) -> str:
                 right_edge=self.__style.left_and_right_edge,
                 filler=row,
             )
-            for row in self.__body
+            for row in body
         )
 
     def to_ascii(self) -> str:
@@ -296,7 +296,7 @@ def to_ascii(self) -> str:
             table += self.__heading_sep_to_ascii()
         # add table body
         if self.__body:
-            table += self.__body_to_ascii()
+            table += self.__body_to_ascii(self.__body)
         # add table footer
         if self.__footer:
             table += self.__heading_sep_to_ascii()
@@ -308,25 +308,41 @@ def to_ascii(self) -> str:
 
 
 def table2ascii(
-    header: Optional[List] = None,
-    body: Optional[List[List]] = None,
-    footer: Optional[List] = None,
-    **options,
+    header: Optional[List[Any]] = None,
+    body: Optional[List[List[Any]]] = None,
+    footer: Optional[List[Any]] = None,
+    *,
+    first_col_heading: bool = False,
+    last_col_heading: bool = False,
+    column_widths: Optional[List[int]] = None,
+    alignments: Optional[List[Alignment]] = None,
+    style: TableStyle = PresetStyle.double_thin_compact,
 ) -> str:
     """
     Convert a 2D Python table to ASCII text
 
     Args:
-        header (:class:`Optional[List]`): List of column values in the table's header row
-        body (:class:`Optional[List[List]]`): 2-dimensional list of values in the table's body
-        footer (:class:`Optional[List]`): List of column values in the table's footer row
-        style (:class:`TableStyle`): Table style to use for styling (preset styles can be imported)
-        column_widths (:class:`List[int]`): List of widths in characters for each column (defaults to auto-sizing)
-        alignments (:class:`List[Alignment]`): List of alignments (ex. `[Alignment.LEFT, Alignment.CENTER, Alignment.RIGHT]`)
+        header (:class:`Optional[List[Any]]`): List of column values in the table's header row
+        body (:class:`Optional[List[List[Any]]]`): 2-dimensional list of values in the table's body
+        footer (:class:`Optional[List[Any]]`): List of column values in the table's footer row
         first_col_heading (:class:`bool`): Whether to add a header column separator after the first column
         last_col_heading (:class:`bool`): Whether to add a header column separator before the last column
+        column_widths (:class:`List[int]`): List of widths in characters for each column (defaults to auto-sizing)
+        alignments (:class:`List[Alignment]`): List of alignments (ex. `[Alignment.LEFT, Alignment.CENTER, Alignment.RIGHT]`)
+        style (:class:`TableStyle`): Table style to use for styling (preset styles can be imported)
 
     Returns:
         str: The generated ASCII table
     """
-    return TableToAscii(header, body, footer, Options(**options)).to_ascii()
+    return TableToAscii(
+        header,
+        body,
+        footer,
+        Options(
+            first_col_heading=first_col_heading,
+            last_col_heading=last_col_heading,
+            column_widths=column_widths,
+            alignments=alignments,
+            style=style,
+        ),
+    ).to_ascii()