Skip to content

Commit 6450417

Browse files
committed
Resolve PR review comments
1 parent 184e528 commit 6450417

File tree

3 files changed

+11
-9
lines changed

3 files changed

+11
-9
lines changed

pdfly/cat.py

+7-7
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ def main(
5757
fn_pgrgs: List[str],
5858
output: Path,
5959
verbose: bool,
60-
use_complements: bool = False,
60+
inverted_page_selection: bool = False,
6161
) -> None:
6262
filename_page_ranges = parse_filepaths_and_pagerange_args(
6363
filename, fn_pgrgs
@@ -91,14 +91,14 @@ def main(
9191
f"WARNING: Page range {page_range} is out of bounds",
9292
file=sys.stderr,
9393
)
94-
if not use_complements:
95-
for page_num in range(*page_range.indices(len(reader.pages))):
96-
writer.add_page(reader.pages[page_num])
97-
else:
94+
if inverted_page_selection:
9895
all_page_nums = set(range(len(reader.pages)))
9996
page_nums = set(range(*page_range.indices(len(reader.pages))))
100-
compl_page_nums = all_page_nums - page_nums
101-
for page_num in compl_page_nums:
97+
inverted_page_nums = all_page_nums - page_nums
98+
for page_num in inverted_page_nums:
99+
writer.add_page(reader.pages[page_num])
100+
else:
101+
for page_num in range(*page_range.indices(len(reader.pages))):
102102
writer.add_page(reader.pages[page_num])
103103
writer.write(output_fh)
104104
except Exception:

pdfly/cli.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -124,7 +124,9 @@ def rm(
124124
False, help="show page ranges as they are being read"
125125
),
126126
) -> None:
127-
pdfly.cat.main(filename, fn_pgrgs, output, verbose, use_complements=True)
127+
pdfly.cat.main(
128+
filename, fn_pgrgs, output, verbose, inverted_page_selection=True
129+
)
128130

129131

130132
@entry_point.command(name="meta") # type: ignore[misc]

tests/test_rm.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,7 @@ def test_rm_subset_warn_on_missing_pages(capsys, tmp_path):
7070
assert "WARN" in captured.out
7171

7272

73-
@pytest.mark.xfail() # There is currently a bug there
73+
@pytest.mark.xfail()
7474
def test_rm_subset_ensure_reduced_size(tmp_path, two_pages_pdf_filepath):
7575
exit_code = run_cli(
7676
[

0 commit comments

Comments
 (0)