-
Notifications
You must be signed in to change notification settings - Fork 0
/
set-toc.py
63 lines (31 loc) · 829 Bytes
/
set-toc.py
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
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
import pymupdf
import session
import datetime
print("Insert TOC")
filename = input("Filename: ")
doc = pymupdf.open(filename)
#instext = input('Text: ')
def tocItem():
print("Insert TOC item:")
lvl = int(input('Level: '))
title = input("Title: ")
pgnum = int(input("Page (1-based) :"))
last = input("Last one? (y/n) ")
if(last == "y" or last == "Y"):
last = True
else:
last = False
return ([lvl,title,pgnum,None],last)
tocdata = []
def tocExtend():
newitem = tocItem()
tocdata.append(newitem[0])
if(newitem[1] == False):
tocExtend()
tocExtend()
ins = doc.set_toc(tocdata)
newfile = input("New file: ")
doc.save(newfile)
print("TOC inserted",ins)
doc.close()
session.add([datetime.datetime.now(),'set_toc',ins,filename,newfile])