nucliadb 4.0.0.post542__py3-none-any.whl → 6.2.1.post2777__py3-none-any.whl

nucliadb/ingest/fields/link.py

@@ -19,15 +19,14 @@
 #
 from typing import Any, Optional
 
-from nucliadb_protos.resources_pb2 import CloudFile, FieldLink, LinkExtractedData
-
 from nucliadb.ingest.fields.base import Field
+from nucliadb_protos.resources_pb2 import CloudFile, FieldLink, LinkExtractedData
 from nucliadb_utils.storages.storage import StorageField
 
 LINK_METADATA = "link_metadata"
 
 
-class Link(Field):
+class Link(Field[FieldLink]):
     pbklass = FieldLink
     value: FieldLink
     type: str = "u"
@@ -46,7 +45,7 @@ class Link(Field):
     async def set_value(self, payload: FieldLink):
         await self.db_set_value(payload)
 
-    async def get_value(self) -> FieldLink:
+    async def get_value(self) -> Optional[FieldLink]:
         return await self.db_get_value()
 
     async def set_link_extracted_data(self, link_extracted_data: LinkExtractedData):
@@ -80,9 +79,7 @@ class Link(Field):
             sf_generated: StorageField = self.storage.file_extracted(
                 self.kbid, self.uuid, self.type, self.id, f"generated/{fileid}"
             )
-            cf_generated: CloudFile = await self.storage.normalize_binary(
-                origincf, sf_generated
-            )
+            cf_generated: CloudFile = await self.storage.normalize_binary(origincf, sf_generated)
             link_extracted_data.file_generated[fileid].CopyFrom(cf_generated)
 
         sf: StorageField = self.storage.file_extracted(
@@ -96,7 +93,5 @@ class Link(Field):
             sf: StorageField = self.storage.file_extracted(
                 self.kbid, self.uuid, self.type, self.id, LINK_METADATA
             )
-            self.link_extracted_data = await self.storage.download_pb(
-                sf, LinkExtractedData
-            )
+            self.link_extracted_data = await self.storage.download_pb(sf, LinkExtractedData)
         return self.link_extracted_data

nucliadb/ingest/fields/text.py

@@ -19,21 +19,26 @@
 #
 
 import hashlib
-
-from nucliadb_protos.resources_pb2 import FieldText
+from typing import Optional
 
 from nucliadb.ingest.fields.base import Field
+from nucliadb_protos.resources_pb2 import FieldAuthor, FieldText
 
 
-class Text(Field):
+class Text(Field[FieldText]):
     pbklass = FieldText
     value: FieldText
     type: str = "t"
 
+    async def generated_by(self) -> FieldAuthor:
+        value = await self.get_value()
+        assert value is not None, "Can't know who generated the field if it has no value!"
+        return value.generated_by
+
     async def set_value(self, payload: FieldText):
         if payload.md5 == "":
             payload.md5 = hashlib.md5(payload.body.encode()).hexdigest()
         await self.db_set_value(payload)
 
-    async def get_value(self) -> FieldText:
+    async def get_value(self) -> Optional[FieldText]:
         return await self.db_get_value()