forked from chazeon/PDF-Watermark-Remover
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathremove.py
More file actions
31 lines (19 loc) · 724 Bytes
/
remove.py
File metadata and controls
31 lines (19 loc) · 724 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
from pypdf import PdfReader, PdfWriter
def removeWatermark(input_fname: str, output_fname: str):
with open(input_fname, "rb") as input_file:
reader = PdfReader(input_file)
writer = PdfWriter()
for n in range(len(reader.pages)):
page = reader.pages[n]
del page["/Contents"][-1]
writer.add_page(page)
with open(output_fname, "wb") as output_file:
writer.write(output_file)
if __name__ == "__main__":
import sys
if len(sys.argv) == 2:
removeWatermark(sys.argv[1], sys.argv[1])
elif len(sys.argv) == 3:
removeWatermark(sys.argv[1], sys.argv[2])
else:
raise RuntimeError("Wrong number of arguments!")