octoleo
/
pdflayers
mirror of https://github.com/Llewellynvdm/pdflayers.git
1
0
Fork 0
Code Issues Releases Activity

Merge pull request #4 from SimonSegerblomRex/fix-2 

Avoid accessing keys in /Usage
This commit is contained in:
Simon Segerblom Rex 2019-05-16 22:24:11 +02:00 committed by GitHub
parent 79bbdaab05 26e1f6a543
commit 48f9730d0f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pdflayers.py
View File

@ -31,14 +31,12 @@ def set_layer_visibility(pdf, layers_to_show):
sys.exit(1)
ocgs_on = []
ocgs_off = []
for ocg in ocgs:
if ocg.Name in layers_to_show:
logging.info("Layer %s will be visible.", ocg.Name)
ocgs_on.append(ocg)
else:
logging.info("Layer %s will be hidden.", ocg.Name)
ocgs_off.append(ocg)
ocgs_config = pikepdf.Dictionary(
BaseState=pikepdf.Name('/OFF'),
@ -52,12 +50,11 @@ def set_layer_visibility(pdf, layers_to_show):
)
# Needed for the PDF viwer in google-chrome (at least):
# TODO: Set PrintState and ExportState too..? Exception+ handling...
# Check PDF reference. Maybe better to just delete ViewState..?
for ocg in ocgs_on:
ocg.Usage.View.ViewState = pikepdf.Name('/ON')
for ocg in ocgs_off:
ocg.Usage.View.ViewState = pikepdf.Name('/OFF')
for ocg in ocgs:
if '/View' in ocg.Usage:
del ocg.Usage.View
if '/Print' in ocg.Usage:
del ocg.Usage.Print
def main():