ereuse/pyvckit
8
0
Fork 0
Code Issues Pull requests Packages Projects Releases Wiki Activity

fix documentloader

Browse source
This commit is contained in:
Cayo Puigdefabregas 2024-06-03 17:59:09 +02:00
parent d0d4ac8d6c
commit 305f95865c
1 changed files with 6 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
pyvckit/document_loader.py
View file

@ -9,9 +9,11 @@ Remote document loader using Requests.
.. moduleauthor:: Tim McNamara <tim.mcnamara@okfn.org> .. moduleauthor:: Tim McNamara <tim.mcnamara@okfn.org>
.. moduleauthor:: Olaf Conradi <olaf@conradi.org> .. moduleauthor:: Olaf Conradi <olaf@conradi.org>
""" """
import re
import string import string
import json import json
import urllib.parse as urllib_parse import urllib.parse as urllib_parse
from pyld import jsonld
from pyld.jsonld import (JsonLdError, parse_link_header, LINK_HEADER_REL) from pyld.jsonld import (JsonLdError, parse_link_header, LINK_HEADER_REL)
@ -20,7 +22,6 @@ def get_cache():
with open("cache_context.json") as f: with open("cache_context.json") as f:
doc_str = f.read() doc_str = f.read()
if doc_str: if doc_str:
# import pdb; pdb.set_trace()
return json.loads(doc_str) return json.loads(doc_str)
return {} return {}
@ -48,7 +49,6 @@ def requests_document_loader(secure=False, **kwargs):
:return: the RemoteDocument. :return: the RemoteDocument.
""" """
# import pdb; pdb.set_trace()
cache = get_cache() cache = get_cache()
if cache.get(url): if cache.get(url):
return cache[url] return cache[url]
@ -109,11 +109,11 @@ def requests_document_loader(secure=False, **kwargs):
not re.match(r'^application\/(\w*\+)?json$', content_type)): not re.match(r'^application\/(\w*\+)?json$', content_type)):
doc['contentType'] = 'application/ld+json' doc['contentType'] = 'application/ld+json'
doc['documentUrl'] = jsonld.prepend_base(url, linked_alternate['target']) doc['documentUrl'] = jsonld.prepend_base(url, linked_alternate['target'])
# import pdb; pdb.set_trace()
cache[url] = doc cache[url] = doc
f = open("cache_context.json", "w") with open("cache_context.json", "w") as f:
f.write(json.dumps(cache)) f.write(json.dumps(cache))
f.close()
return doc return doc
except JsonLdError as e: except JsonLdError as e:
raise e raise e