rahix/pdfnavigation
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Show TOC entries

Browse source
This commit is contained in:
Rahix 2024-08-19 02:34:33 +02:00
parent b774b90504
commit 0d9dbd7cb4
1 changed files with 26 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

35
src/main.py
View file

@ -7,29 +7,46 @@ from PIL import Image, ImageTk
CURRENT_PAGE = 0
def main():
def main() -> None:
parser = argparse.ArgumentParser()
parser.add_argument("file")
args = parser.parse_args()
doc: pymupdf.Document
with pymupdf.open(args.file) as doc:
toc = doc.get_toc()
def toc_entry_for_page(page_num: int) -> str:
last_title = ""
for lvl, t, p in toc:
if (p - 1) > page_num:
return last_title
if (p - 1) == page_num:
return t
last_title = t
root = tkinter.Tk()
frm = ttk.Frame(root, padding=10)
frm.grid()
label = ttk.Label(frm)
label.grid(column=1, row=0)
page_title = ttk.Label(frm)
page_title.grid(column=1, row=0)
def load_page(i: int):
page = doc[i]
page_view = ttk.Label(frm)
page_view.grid(column=1, row=1)
def load_page(page_num: int):
page = doc[page_num]
pix = page.get_pixmap()
mode = "RGBA" if pix.alpha else "RGB"
img = Image.frombytes(mode, [pix.width, pix.height], pix.samples)
img = ImageTk.PhotoImage(img)
label.configure(image=img)
label.image = img
page_view.configure(image=img)
page_view.image = img
page_title.configure(text=toc_entry_for_page(page_num))
def next_page():
global CURRENT_PAGE
@ -41,8 +58,8 @@ def main():
CURRENT_PAGE -= 1
load_page(CURRENT_PAGE)
ttk.Button(frm, text="Previous", command=previous_page).grid(column=0, row=0)
ttk.Button(frm, text="Next", command=next_page).grid(column=2, row=0)
ttk.Button(frm, text="Previous", command=previous_page).grid(column=0, row=1)
ttk.Button(frm, text="Next", command=next_page).grid(column=2, row=1)
load_page(CURRENT_PAGE)