Add option to specify line break character used in pagination

jazzband · Aug 17, 2022 · f75f6ad · f75f6ad
1 parent 27017ef
commit f75f6ad
Show file tree

Hide file tree

Showing 2 changed files with 5 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
@@ -1735,6 +1735,9 @@ def test_paginate():
     assert paginated.startswith(expected_page_1)
     assert "\f" in paginated
     assert paginated.endswith(expected_page_2)
+    paginated = t.paginate(page_length=4, line_break="\n")
+    assert "\f" not in paginated
+    assert "\n" in paginated
 
 
 def test_add_rows():