Skip to content

Commit 0e40eee

Browse files
_internal.py: Replace page number validator with stricter type check
Assisted-by: Codex
1 parent fc720a3 commit 0e40eee

1 file changed

Lines changed: 1 addition & 17 deletions

File tree

src/pdfrest/models/_internal.py

Lines changed: 1 addition & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -627,21 +627,6 @@ def _validate_cmyk_values(value: Any) -> tuple[int, int, int, int] | None:
627627
return (channels[0], channels[1], channels[2], channels[3])
628628

629629

630-
def _validate_add_text_page(value: str | int) -> str | int:
631-
if isinstance(value, str):
632-
if value.lower() == "all":
633-
return "all"
634-
if value.isdigit():
635-
numeric_page = int(value)
636-
if numeric_page >= 1:
637-
return numeric_page
638-
else:
639-
if value >= 1:
640-
return value
641-
msg = 'page must be a positive integer or "all".'
642-
raise ValueError(msg)
643-
644-
645630
class PdfLiteralRedactionModel(BaseModel):
646631
type: Literal["literal"]
647632
value: Annotated[str, Field(min_length=1)]
@@ -1111,9 +1096,8 @@ class PdfAddTextObjectModel(BaseModel):
11111096
Field(serialization_alias="opacity", ge=0.0, le=1.0),
11121097
]
11131098
page: Annotated[
1114-
str | int,
1099+
Literal["all"] | Annotated[int, Field(ge=1)],
11151100
Field(serialization_alias="page"),
1116-
AfterValidator(_validate_add_text_page),
11171101
]
11181102
rotation: Annotated[float, Field(serialization_alias="rotation")]
11191103
text: Annotated[str, Field(min_length=1, serialization_alias="text")]

0 commit comments

Comments
 (0)