diff options
Diffstat (limited to 'sdv/docker/sdvstate/core')
-rw-r--r-- | sdv/docker/sdvstate/core/__init__.py | 21 | ||||
-rw-r--r-- | sdv/docker/sdvstate/core/load_pdf.py | 40 |
2 files changed, 61 insertions, 0 deletions
diff --git a/sdv/docker/sdvstate/core/__init__.py b/sdv/docker/sdvstate/core/__init__.py new file mode 100644 index 0000000..ed33752 --- /dev/null +++ b/sdv/docker/sdvstate/core/__init__.py @@ -0,0 +1,21 @@ +# Copyright 2020 University Of Delhi. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + + +""" +Core package +contains all program specific dependencies +""" + +from .load_pdf import load_pdf diff --git a/sdv/docker/sdvstate/core/load_pdf.py b/sdv/docker/sdvstate/core/load_pdf.py new file mode 100644 index 0000000..6ce22f0 --- /dev/null +++ b/sdv/docker/sdvstate/core/load_pdf.py @@ -0,0 +1,40 @@ +# Copyright 2020 University Of Delhi. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. +# See the License for the specific language governing permissions and +# limitations under the License. + + +"""Loads PDF file into settings +""" + +import json +import yaml + +from tools.conf import settings + +def load_pdf(): + """ + Updates settings with PDF data + """ + filename = settings.getValue('pdf_file') + with open(filename) as handle: + data = handle.read() + + try: + pdf = json.loads(data) + except json.decoder.JSONDecodeError: + try: + pdf = yaml.safe_load(data) + except yaml.parser.ParserError: + raise Exception(f"Invalid PDF file: {filename}") + + settings.setValue('pdf_file', pdf) |