diff options
Diffstat (limited to 'bsie/extractor/preview.py')
| -rw-r--r-- | bsie/extractor/preview.py | 10 |
1 files changed, 5 insertions, 5 deletions
diff --git a/bsie/extractor/preview.py b/bsie/extractor/preview.py index 145a01a..fcda71c 100644 --- a/bsie/extractor/preview.py +++ b/bsie/extractor/preview.py @@ -7,7 +7,8 @@ import typing import PIL.Image # bsie imports -from bsie.utils import bsfs, node, ns +from bsie.matcher import nodes +from bsie.utils import bsfs, ns # inner-module imports from . import base @@ -67,10 +68,10 @@ class Preview(base.Extractor): def extract( self, - subject: node.Node, + subject: nodes.Entity, content: typing.Callable[[int], PIL.Image.Image], principals: typing.Iterable[bsfs.schema.Predicate], - ) -> typing.Iterator[typing.Tuple[node.Node, bsfs.schema.Predicate, typing.Any]]: + ) -> typing.Iterator[typing.Tuple[nodes.Node, bsfs.schema.Predicate, typing.Any]]: # check principals if self.schema.predicate(ns.bse.preview) not in principals: return @@ -82,10 +83,9 @@ class Preview(base.Extractor): buffer = io.BytesIO() img.save(buffer, format='jpeg') # create a preview node - preview = node.Node(ns.bsn.Preview, + preview = nodes.Preview( ucid=bsfs.uuid.UCID.from_bytes(buffer.getvalue()), size=max_side, - source=subject, ) # yield triples yield subject, self.schema.predicate(ns.bse.preview), preview |