Make it possible to set more writer settings via writer class constru…

…ctors
thombashi · May 4, 2021 · b7047e1 · b7047e1
1 parent 5b1e0f5
commit b7047e1
Show file tree

Hide file tree

Showing 9 changed files with 13 additions and 10 deletions.
diff --git a/pytablewriter/writer/_table_writer.py b/pytablewriter/writer/_table_writer.py
@@ -276,7 +276,10 @@ def __init__(self, **kwargs) -> None:
         self._iter_count = None  # type: Optional[int]
 
         self.__default_style = kwargs.get("default_style", Style())
-        self.__col_style_list = kwargs.get("column_styles", [])  # type: List[Optional[Style]]
+
+        self.__col_style_list = []  # type: List[Optional[Style]]
+        self.column_styles = kwargs.get("column_styles", [])
+
         self._style_filters = []  # type: List[StyleFilterFunc]
         self._styler = self._create_styler(self)
         self.style_filter_kwargs = kwargs.get("style_filter_kwargs", {})  # type: Dict[str, Any]

diff --git a/pytablewriter/writer/text/_borderless.py b/pytablewriter/writer/text/_borderless.py
@@ -29,7 +29,7 @@ def __init__(self, **kwargs) -> None:
         self.char_left_side_row = ""
         self.char_right_side_row = ""
 
-        self.indent_string = "    "
+        self.indent_string = kwargs.get("indent_string", "    ")
         self.is_write_header_separator_row = False
         self.is_write_value_separator_row = False
         self.is_write_opening_row = False

diff --git a/pytablewriter/writer/text/_css.py b/pytablewriter/writer/text/_css.py
@@ -28,7 +28,7 @@ def __init__(self, **kwargs) -> None:
         super().__init__(**kwargs)
 
         self.is_padding = False
-        self.indent_string = "    "
+        self.indent_string = kwargs.get("indent_string", "    ")
 
         self._dp_extractor.preprocessor.is_escape_html_tag = False
         self._quoting_flags = copy.deepcopy(NOT_QUOTING_FLAGS)

diff --git a/pytablewriter/writer/text/_csv.py b/pytablewriter/writer/text/_csv.py
@@ -38,7 +38,7 @@ def __init__(self, **kwargs) -> None:
 
         self._set_chars("")
         self.indent_string = ""
-        self.column_delimiter = ","
+        self.column_delimiter = kwargs.get("column_delimiter", ",")
         self._margin = 0
 
         self.is_padding = False

diff --git a/pytablewriter/writer/text/_html.py b/pytablewriter/writer/text/_html.py
@@ -49,7 +49,7 @@ def __init__(self, **kwargs) -> None:
         super().__init__(**kwargs)
 
         self.is_padding = False
-        self.indent_string = "    "
+        self.indent_string = kwargs.get("indent_string", "    ")
 
         self._dp_extractor.preprocessor.line_break_repl = "<br>"
         self._dp_extractor.preprocessor.is_escape_html_tag = False

diff --git a/pytablewriter/writer/text/_latex.py b/pytablewriter/writer/text/_latex.py
@@ -30,7 +30,7 @@ def __init__(self, **kwargs) -> None:
 
         self.is_write_opening_row = True
         self.is_write_closing_row = True
-        self.indent_string = "    "
+        self.indent_string = kwargs.get("indent_string", "    ")
         self.column_delimiter = " & "
         self.char_right_side_row = r" \\"
 

diff --git a/pytablewriter/writer/text/_rst.py b/pytablewriter/writer/text/_rst.py
@@ -34,7 +34,7 @@ def __init__(self, **kwargs) -> None:
         self.char_opening_row_cross_point = "+"
         self.char_closing_row_cross_point = "+"
 
-        self.indent_string = "    "
+        self.indent_string = kwargs.get("indent_string", "    ")
         self.is_write_header_separator_row = True
         self.is_write_value_separator_row = True
         self.is_write_opening_row = True

diff --git a/pytablewriter/writer/text/_unicode.py b/pytablewriter/writer/text/_unicode.py
@@ -50,7 +50,7 @@ def __init__(self, **kwargs) -> None:
         self.char_closing_row = "─"
         self.char_closing_row_cross_point = "┴"
 
-        self.indent_string = "    "
+        self.indent_string = kwargs.get("indent_string", "    ")
         self.is_write_header_separator_row = True
         self.is_write_value_separator_row = True
         self.is_write_opening_row = True
@@ -108,7 +108,7 @@ def __init__(self, **kwargs) -> None:
         self.char_closing_row = "━"
         self.char_closing_row_cross_point = "┻"
 
-        self.indent_string = "    "
+        self.indent_string = kwargs.get("indent_string", "    ")
         self.is_write_header_separator_row = True
         self.is_write_value_separator_row = True
         self.is_write_opening_row = True

diff --git a/pytablewriter/writer/text/sourcecode/_sourcecode.py b/pytablewriter/writer/text/sourcecode/_sourcecode.py
@@ -45,7 +45,7 @@ def margin(self, value: int) -> None:
     def __init__(self, **kwargs) -> None:
         super().__init__(**kwargs)
 
-        self.indent_string = "    "
+        self.indent_string = kwargs.get("indent_string", "    ")
         self.column_delimiter = ", "
         self._margin = 0