Bump black to fix vulnerability (#2046)

code-requirements.txt

@@ -1,4 +1,4 @@
 flake8==4.0.1
-black==23.3.0
+black==24.3.0
 isort==5.10.1
 pre-commit==2.20.0

nucliadb/migrations/0002_rollover_shards.py

@@ -20,8 +20,7 @@
 from nucliadb.migrator.context import ExecutionContext
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0003_allfields_key.py

@@ -26,8 +26,7 @@
 from nucliadb.migrator.migrator import logger
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0004_rollover_shards.py

@@ -20,8 +20,7 @@
 from nucliadb.migrator.context import ExecutionContext
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0005_rollover_shards.py

@@ -20,8 +20,7 @@
 from nucliadb.migrator.context import ExecutionContext
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0006_rollover_shards.py

@@ -20,9 +20,7 @@
 from nucliadb.migrator.context import ExecutionContext
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
-async def migrate_kb(context: ExecutionContext, kbid: str) -> None:
-    ...
+async def migrate_kb(context: ExecutionContext, kbid: str) -> None: ...

nucliadb/migrations/0008_cleanup_leftover_rollover_metadata.py

@@ -28,8 +28,7 @@
 from nucliadb.migrator.context import ExecutionContext
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0009_upgrade_relations_and_texts_to_v2.py

@@ -30,8 +30,7 @@
 from nucliadb.migrator.context import ExecutionContext
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0010_fix_corrupt_indexes.py

@@ -40,8 +40,7 @@
 ]
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0011_materialize_labelset_ids.py

@@ -32,8 +32,7 @@
 logger = logging.getLogger(__name__)
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0012_rollover_shards.py

@@ -38,8 +38,7 @@
 ]
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0013_rollover_shards.py

@@ -20,9 +20,7 @@
 from nucliadb.migrator.context import ExecutionContext
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
-async def migrate_kb(context: ExecutionContext, kbid: str) -> None:
-    ...
+async def migrate_kb(context: ExecutionContext, kbid: str) -> None: ...

nucliadb/migrations/0014_rollover_shards.py

@@ -31,8 +31,7 @@
 logger = logging.getLogger(__name__)
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0015_targeted_rollover.py

@@ -39,8 +39,7 @@
 ]
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0016_upgrade_to_paragraphs_v2.py

@@ -34,12 +34,10 @@
 logger = logging.getLogger(__name__)
 
 
-class ShardsObjectNotFound(Exception):
-    ...
+class ShardsObjectNotFound(Exception): ...
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/migrations/0017_multiple_writable_shards.py

@@ -37,8 +37,7 @@
 logger = logging.getLogger(__name__)
 
 
-async def migrate(context: ExecutionContext) -> None:
-    ...
+async def migrate(context: ExecutionContext) -> None: ...
 
 
 async def migrate_kb(context: ExecutionContext, kbid: str) -> None:

nucliadb/nucliadb/common/cluster/discovery/k8s.py

@@ -259,11 +259,11 @@ async def update_node_data_cache(self) -> None:
                             continue
                         existing = self.node_id_cache[pod_name]
                         try:
-                            self.node_id_cache[
-                                pod_name
-                            ] = await self._query_node_metadata(
-                                existing.address,
-                                read_replica=existing.primary_id is not None,
+                            self.node_id_cache[pod_name] = (
+                                await self._query_node_metadata(
+                                    existing.address,
+                                    read_replica=existing.primary_id is not None,
+                                )
                             )
                         except NodeConnectionError:  # pragma: no cover
                             self._maybe_remove_stale_node(pod_name)

nucliadb/nucliadb/common/cluster/discovery/single.py

@@ -48,5 +48,4 @@ async def initialize(self) -> None:
             ]
         )
 
-    async def finalize(self) -> None:
-        ...
+    async def finalize(self) -> None: ...

nucliadb/nucliadb/common/http_clients/auth.py

@@ -49,9 +49,9 @@ def __init__(self):
         )
         self.headers = {}
         if nuclia_settings.nuclia_service_account is not None:
-            self.headers[
-                "X-NUCLIA-NUAKEY"
-            ] = f"Bearer {nuclia_settings.nuclia_service_account}"
+            self.headers["X-NUCLIA-NUAKEY"] = (
+                f"Bearer {nuclia_settings.nuclia_service_account}"
+            )
 
     async def __aenter__(self):
         return self

nucliadb/nucliadb/common/http_clients/processing.py

@@ -159,9 +159,9 @@ def __init__(self):
         self.base_url = get_processing_api_url()
         self.headers = {}
         if nuclia_settings.nuclia_service_account is not None:
-            self.headers[
-                "X-STF-NUAKEY"
-            ] = f"Bearer {nuclia_settings.nuclia_service_account}"
+            self.headers["X-STF-NUAKEY"] = (
+                f"Bearer {nuclia_settings.nuclia_service_account}"
+            )
 
     async def __aenter__(self):
         return self

nucliadb/nucliadb/common/locking.py

@@ -38,8 +38,7 @@
 KB_SHARDS_LOCK = "shards-kb-{kbid}"
 
 
-class ResourceLocked(Exception):
-    ...
+class ResourceLocked(Exception): ...
 
 
 @dataclass

nucliadb/nucliadb/common/maindb/exceptions.py

@@ -17,13 +17,10 @@
 # You should have received a copy of the GNU Affero General Public License
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 #
-class ConflictError(Exception):
-    ...
+class ConflictError(Exception): ...
 
 
-class NotFoundError(Exception):
-    ...
+class NotFoundError(Exception): ...
 
 
-class UnsetUtility(Exception):
-    ...
+class UnsetUtility(Exception): ...

nucliadb/nucliadb/export_import/models.py

@@ -65,8 +65,7 @@ class ExportMetadata(Metadata):
     exported_resources: list[str] = list()
 
 
-class ImportMetadata(Metadata):
-    ...
+class ImportMetadata(Metadata): ...
 
 
 class NatsTaskMessage(BaseModel):

nucliadb/nucliadb/export_import/utils.py

@@ -253,8 +253,7 @@ async def get_labels(context: ApplicationContext, kbid: str) -> kb_pb2.Labels:
         return await datamanagers.labels.get_labels(txn, kbid=kbid)
 
 
-class EndOfStream(Exception):
-    ...
+class EndOfStream(Exception): ...
 
 
 class ExportStream:

nucliadb/nucliadb/ingest/consumer/auditing.py

@@ -106,9 +106,9 @@ async def handle_message(self, raw_data: bytes) -> None:
     @metrics.handler_histo.wrap({"type": "audit_counter"})
     async def process_kb(self, kbid: str) -> None:
         try:
-            shard_groups: list[
-                writer_pb2.ShardObject
-            ] = await self.shard_manager.get_shards_by_kbid(kbid)
+            shard_groups: list[writer_pb2.ShardObject] = (
+                await self.shard_manager.get_shards_by_kbid(kbid)
+            )
         except ShardsNotFound:
             logger.warning(f"No shards found for kbid {kbid}, skipping")
             return

nucliadb/nucliadb/ingest/consumer/consumer.py

@@ -173,9 +173,11 @@ async def subscription_worker(self, msg: Msg):
                 try:
                     with consumer_observer(
                         {
-                            "source": "writer"
-                            if pb.source == pb.MessageSource.WRITER
-                            else "processor"
+                            "source": (
+                                "writer"
+                                if pb.source == pb.MessageSource.WRITER
+                                else "processor"
+                            )
                         }
                     ):
                         await self._process(pb, seqid)

nucliadb/nucliadb/ingest/fields/base.py

@@ -399,9 +399,9 @@ async def set_field_metadata(
     ) -> tuple[FieldComputedMetadata, list[str], dict[str, list[str]]]:
         if self.type in SUBFIELDFIELDS:
             try:
-                actual_payload: Optional[
-                    FieldComputedMetadata
-                ] = await self.get_field_metadata(force=True)
+                actual_payload: Optional[FieldComputedMetadata] = (
+                    await self.get_field_metadata(force=True)
+                )
             except KeyError:
                 actual_payload = None
         else:
@@ -471,9 +471,9 @@ async def get_field_metadata_cf(self) -> Optional[CloudFile]:
     async def set_large_field_metadata(self, payload: LargeComputedMetadataWrapper):
         if self.type in SUBFIELDFIELDS:
             try:
-                actual_payload: Optional[
-                    LargeComputedMetadata
-                ] = await self.get_large_field_metadata(force=True)
+                actual_payload: Optional[LargeComputedMetadata] = (
+                    await self.get_large_field_metadata(force=True)
+                )
             except KeyError:
                 actual_payload = None
         else:

nucliadb/nucliadb/ingest/orm/resource.py

@@ -1159,9 +1159,9 @@ async def iterate_sentences(
                 for fieldmetadata in self.basic.fieldmetadata:
                     field_id = self.generate_field_id(fieldmetadata.field)
                     for annotationparagraph in fieldmetadata.paragraphs:
-                        userdefinedparagraphclass[
-                            annotationparagraph.key
-                        ] = annotationparagraph
+                        userdefinedparagraphclass[annotationparagraph.key] = (
+                            annotationparagraph
+                        )
 
         for (type_id, field_id), field in fields.items():
             fieldid = FieldID(field_type=type_id, field=field_id)  # type: ignore
@@ -1273,9 +1273,9 @@ async def iterate_paragraphs(
                 for fieldmetadata in self.basic.fieldmetadata:
                     field_id = self.generate_field_id(fieldmetadata.field)
                     for annotationparagraph in fieldmetadata.paragraphs:
-                        userdefinedparagraphclass[
-                            annotationparagraph.key
-                        ] = annotationparagraph
+                        userdefinedparagraphclass[annotationparagraph.key] = (
+                            annotationparagraph
+                        )
 
         for (type_id, field_id), field in fields.items():
             fieldid = FieldID(field_type=type_id, field=field_id)  # type: ignore