Merge pull request #193 from myheroyuki/paginate_line_break_options

Add option to specify line break character used in pagination
jazzband · Aug 21, 2022 · 3859de2 · 3859de2
2 parents 27017ef + d4867a2
commit 3859de2
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git a/src/prettytable/prettytable.py b/src/prettytable/prettytable.py
@@ -2004,7 +2004,7 @@ def _stringify_row(self, row, options, hrule):
 
         return "\n".join(bits)
 
-    def paginate(self, page_length=58, **kwargs):
+    def paginate(self, page_length=58, line_break="\f", **kwargs):
 
         pages = []
         kwargs["start"] = kwargs.get("start", 0)
@@ -2015,7 +2015,7 @@ def paginate(self, page_length=58, **kwargs):
             if kwargs["end"] == true_end:
                 break
             kwargs["start"] += page_length
-        return "\f".join(pages)
+        return line_break.join(pages)
 
     ##############################
     # CSV STRING METHODS         #

diff --git a/tests/test_prettytable.py b/tests/test_prettytable.py
@@ -1736,6 +1736,13 @@ def test_paginate():
     assert "\f" in paginated
     assert paginated.endswith(expected_page_2)
 
+    # Act
+    paginated = t.paginate(page_length=4, line_break="\n")
+
+    # Assert
+    assert "\f" not in paginated
+    assert "\n" in paginated
+
 
 def test_add_rows():
     """A table created with multiple add_row calls