Merge pull request #385 from tatuylonen/padright

xxyzz · web-flow · commit daf64d07fd92 · 2025-07-28T15:10:00.000+08:00
Also apply changes in parser function `padleft` to `padright`
diff --git a/src/wikitextprocessor/parserfns.py b/src/wikitextprocessor/parserfns.py
@@ -1062,8 +1062,7 @@ def padright_fn(
     """Implements the padright parser function."""
     v = expander(args[0]) if args else ""
     cntstr = expander(args[1]).strip() if len(args) >= 2 else "0"
-    arg2 = expander(args[2]) if len(args) >= 3 and args[2] else "0"
-    pad = arg2 if len(args) >= 3 and arg2 else "0"
+    pad = expander(args[2]) if len(args) >= 3 else "0"
     if not cntstr.isdigit():
         cnt = 0
         if cntstr.startswith("-") and cntstr[1:].isdigit():
@@ -1075,7 +1074,7 @@ def padright_fn(
             )
     else:
         cnt = int(cntstr)
-    if cnt - len(v) > len(pad):
+    if cnt - len(v) > len(pad) and len(pad) > 0:
         pad = pad * ((cnt - len(v)) // len(pad))
     if len(v) < cnt:
         v = v + pad[: cnt - len(v)]
diff --git a/tests/test_parserfns.py b/tests/test_parserfns.py
@@ -214,3 +214,8 @@ def test_padleft_zero_division(self):
         self.wtp.start_page("land")
         self.assertEqual(self.wtp.expand("{{padleft:|1|}}"), "")
         self.assertEqual(self.wtp.expand("{{padleft:|1}}"), "0")
+
+    def test_padright_zero_division(self):
+        self.wtp.start_page("land")
+        self.assertEqual(self.wtp.expand("{{padright:|1|}}"), "")
+        self.assertEqual(self.wtp.expand("{{padright:|1}}"), "0")
