aboutsummaryrefslogtreecommitdiffstats
path: root/bsie/extractor/image/iptc.py
diff options
context:
space:
mode:
Diffstat (limited to 'bsie/extractor/image/iptc.py')
-rw-r--r--bsie/extractor/image/iptc.py70
1 files changed, 70 insertions, 0 deletions
diff --git a/bsie/extractor/image/iptc.py b/bsie/extractor/image/iptc.py
new file mode 100644
index 0000000..195eff7
--- /dev/null
+++ b/bsie/extractor/image/iptc.py
@@ -0,0 +1,70 @@
+
+# standard imports
+import typing
+
+# bsie imports
+from bsie.utils import bsfs, node, ns
+
+# inner-module imports
+from .. import base
+
+# exports
+__all__: typing.Sequence[str] = (
+ 'Iptc',
+ )
+
+
+## code ##
+
+class Iptc(base.Extractor):
+ """Turn IPTC keywords into tags."""
+
+ CONTENT_READER = 'bsie.reader.exif.Iptc'
+
+ def __init__(self):
+ super().__init__(bsfs.schema.from_string(base.SCHEMA_PREAMBLE + '''
+ bsn:Tag rdfs:subClassOf bsfs:Node .
+
+ bse:tag rdfs:subClassOf bsfs:Predicate ;
+ rdfs:domain bsn:Entity ;
+ rdfs:range bsn:Tag .
+
+ <https://schema.bsfs.io/ie/Node/Tag#label> rdfs:subClassOf bsfs:Predicate ;
+ rdfs:domain bsn:Tag ;
+ rdfs:range xsd:string ;
+ bsfs:unique "true"^^xsd:boolean .
+
+ '''))
+ self._callmap = {
+ self.schema.predicate(ns.bse.tag): self._keywords,
+ }
+
+ def extract(
+ self,
+ subject: node.Node,
+ content: dict,
+ principals: typing.Iterable[bsfs.schema.Predicate],
+ ) -> typing.Iterator[typing.Tuple[node.Node, bsfs.schema.Predicate, typing.Any]]:
+ for pred in principals:
+ # find callback
+ clbk = self._callmap.get(pred)
+ if clbk is None:
+ continue
+ # produce triples
+ yield from clbk(subject, content)
+
+ def _keywords(
+ self,
+ subject: node.Node,
+ content: dict,
+ ) -> typing.Iterator[typing.Tuple[node.Node, bsfs.schema.Predicate, typing.Any]]:
+ if 'Iptc.Application2.Keywords' not in content:
+ return
+ for keyword in content['Iptc.Application2.Keywords']:
+ tag = node.Node(ns.bsn.Tag, label=keyword)
+ yield subject, self.schema.predicate(ns.bse.tag), tag
+ yield tag, self.schema.predicate(ns.bst.label), keyword
+
+
+
+## EOF ##