fix: reduce extra space left and right of plain (#53)

DenverCoder1 · web-flow · commit 009d5ab87537 · 2022-10-30T18:18:58.000-06:00
diff --git a/table2ascii/preset_style.py b/table2ascii/preset_style.py
@@ -46,4 +46,4 @@ class PresetStyle:
     ascii_rounded = TableStyle.from_string(r"/===\|| |=|=||-|-|\|=/")
     ascii_rounded_box = TableStyle.from_string(r"/===\||||=||||-|||\||/")
     markdown = TableStyle.from_string("     ||||-|||         ")
-    plain = TableStyle.from_string("                      ")
+    plain = TableStyle.from_string("                      ").set(left_and_right_edge="")
diff --git a/table2ascii/table_style.py b/table2ascii/table_style.py
@@ -75,3 +75,21 @@ def from_string(cls, string: str) -> "TableStyle":
             TableStyle.from_string("╔═╦═╗║║ ╟─╫─╢     ╚╩═╝")
         """
         return cls(*string)
+
+    def set(self, **kwargs) -> "TableStyle":
+        """
+        Set attributes of the TableStyle
+
+        Args:
+            kwargs: The attributes to set
+
+        Returns:
+            A TableStyle object with the attributes set
+
+        Example::
+
+            TableStyle().set(top_left_corner="╔", top_and_bottom_edge="═")
+        """
+        for key, value in kwargs.items():
+            setattr(self, key, value)
+        return self
diff --git a/tests/test_styles.py b/tests/test_styles.py
@@ -661,9 +661,9 @@ def test_plain():
         style=PresetStyle.plain,
     )
     expected = (
-        "   #     G     H     R     S   \n"
-        "   1    30    40    35    30   \n"
-        "   2    30    40    35    30   \n"
-        "  SUM   130   140   135   130  "
+        "  #     G     H     R     S  \n"
+        "  1    30    40    35    30  \n"
+        "  2    30    40    35    30  \n"
+        " SUM   130   140   135   130 "
     )
     assert text == expected

Original file line number	Diff line number	Diff line change
`@@ -661,9 +661,9 @@ def test_plain():`
`661`	`661`	`style=PresetStyle.plain,`
`662`	`662`	`)`
`663`	`663`	`expected = (`
`664`		`- " # G H R S \n"`
`665`		`- " 1 30 40 35 30 \n"`
`666`		`- " 2 30 40 35 30 \n"`
`667`		`- " SUM 130 140 135 130 "`
	`664`	`+ " # G H R S \n"`
	`665`	`+ " 1 30 40 35 30 \n"`
	`666`	`+ " 2 30 40 35 30 \n"`
	`667`	`+ " SUM 130 140 135 130 "`
`668`	`668`	`)`
`669`	`669`	`assert text == expected`