nucliadb-utils 6.3.1.post3524__py3-none-any.whl → 6.3.1.post3526__py3-none-any.whl

nucliadb_utils/audit/stream.py

@@ -57,7 +57,7 @@ from nucliadb_utils.nuclia_usage.utils.kb_usage_report import KbUsageReportUtili
 
 
 class RequestContext:
-    def __init__(self):
+    def __init__(self: "RequestContext"):
         self.audit_request: AuditRequest = AuditRequest()
         self.start_time: float = time.monotonic()
         self.path: str = ""

nucliadb_utils/nats.py

@@ -47,8 +47,9 @@ def get_traced_jetstream(
 
     if tracer_provider is not None and jetstream is not None:  # pragma: no cover
         logger.info(f"Configuring {service_name} jetstream with telemetry")
-        jetstream = JetStreamContextTelemetry(jetstream, service_name, tracer_provider)
-    return jetstream
+        return JetStreamContextTelemetry(jetstream, service_name, tracer_provider)
+    else:
+        return jetstream
 
 
 class MessageProgressUpdater:

nucliadb_utils/signals.py

@@ -56,9 +56,9 @@ class Signal:
 
     async def dispatch(self, payload: Any):
         """Send signal to all registered callbacks by they priority order."""
-        assert isinstance(
-            payload, self.payload_model_type
-        ), "Can't dispatch a signal with an invalid model"
+        assert isinstance(payload, self.payload_model_type), (
+            "Can't dispatch a signal with an invalid model"
+        )
 
         awaitables = [
             cb(payload=payload)

nucliadb_utils/storages/exceptions.py

@@ -39,7 +39,7 @@ class InvalidOffset(Exception):
 class ResumableUploadGone(Exception):
     def __init__(self, text: str):
         self.text = text
-        super().__init__("Resumable upload is no longer available " "Google: \n " f"{text}")
+        super().__init__(f"Resumable upload is no longer available Google: \n {text}")
 
 
 class CouldNotCopyNotFound(Exception):

nucliadb_utils/storages/gcs.py

@@ -371,7 +371,7 @@ class GCSStorageField(StorageField):
                 await self.storage.delete_upload(self.field.old_uri, self.field.bucket_name)
             except GoogleCloudException as e:
                 logger.warning(
-                    f"Could not delete existing google cloud file " f"with uri: {self.field.uri}: {e}"
+                    f"Could not delete existing google cloud file with uri: {self.field.uri}: {e}"
                 )
         if self.field.upload_uri != self.key:
             await self.move(self.field.upload_uri, self.key, self.field.bucket_name, self.bucket)
@@ -551,7 +551,7 @@ class GCSStorage(Storage):
                     data = {"text": text}
                 if resp.status == 404:
                     logger.warning(
-                        f"Attempt to delete not found gcloud: {data}, " f"status: {resp.status}",
+                        f"Attempt to delete not found gcloud: {data}, status: {resp.status}",
                         exc_info=True,
                     )
                 elif resp.status not in (200, 204):
@@ -658,7 +658,7 @@ class GCSStorage(Storage):
                     logger.error("Not implemented")
                 elif resp.status == 404:
                     logger.error(
-                        f"Attempt to delete not found gcloud: {data}, " f"status: {resp.status}",
+                        f"Attempt to delete not found gcloud: {data}, status: {resp.status}",
                         exc_info=True,
                     )
                 else:

nucliadb_utils/storages/s3.py

@@ -392,7 +392,7 @@ class S3Storage(Storage):
             self._session = get_session()
         return self._session
 
-    async def initialize(self):
+    async def initialize(self: "S3Storage") -> None:
         session = AioSession()
         self._s3aioclient: AioBaseClient = await self._exit_stack.enter_async_context(
             session.create_client("s3", **self.opts)

nucliadb_utils/tests/azure.py

@@ -19,7 +19,7 @@
 #
 from contextlib import ExitStack
 from dataclasses import dataclass
-from typing import Any, Generator
+from typing import Any, Generator, Iterator
 from unittest.mock import patch
 
 import pytest
@@ -112,7 +112,7 @@ def azurite() -> Generator[AzuriteFixture, None, None]:
 
 
 @pytest.fixture(scope="function")
-def azure_storage_settings(azurite: AzuriteFixture) -> dict[str, Any]:
+def azure_storage_settings(azurite: AzuriteFixture) -> Iterator[dict[str, Any]]:
     settings = {
         "file_backend": FileBackendConfig.AZURE,
         "azure_account_url": azurite.account_url,

nucliadb_utils/tests/fixtures.py

@@ -18,7 +18,7 @@
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 #
 import os
-from typing import Any, Type
+from typing import Any, Iterator, Type
 from unittest.mock import Mock
 
 import pytest
@@ -75,7 +75,7 @@ async def storage(request):
 
 
 @pytest.fixture(scope="function")
-def storage_settings(request, storage) -> dict[str, Any]:
+def storage_settings(request, storage) -> Iterator[dict[str, Any]]:
     """Useful fixture that returns the settings used in the generic `storage`
     fixture.
 

nucliadb_utils/tests/gcs.py

@@ -20,7 +20,7 @@
 import re
 from concurrent.futures.thread import ThreadPoolExecutor
 from contextlib import ExitStack
-from typing import Any, Optional
+from typing import Any, Iterator, Optional
 from unittest.mock import patch
 
 import docker  # type: ignore  # type: ignore
@@ -91,7 +91,7 @@ def running_in_mac_os() -> bool:
 
 
 @pytest.fixture(scope="function")
-def gcs_storage_settings(gcs) -> dict[str, Any]:
+def gcs_storage_settings(gcs) -> Iterator[dict[str, Any]]:
     settings = {
         "file_backend": FileBackendConfig.GCS,
         "gcs_endpoint_url": gcs,

nucliadb_utils/tests/local.py

@@ -19,7 +19,7 @@
 #
 from contextlib import ExitStack
 from pathlib import Path
-from typing import Any
+from typing import Any, Iterator
 from unittest.mock import patch
 
 import pytest
@@ -29,7 +29,7 @@ from nucliadb_utils.storages.local import LocalStorage
 
 
 @pytest.fixture(scope="function")
-def local_storage_settings(tmp_path: Path) -> dict[str, Any]:
+def local_storage_settings(tmp_path: Path) -> Iterator[dict[str, Any]]:
     settings = {
         "file_backend": FileBackendConfig.LOCAL,
         "local_files": str((tmp_path / "blob").absolute()),

nucliadb_utils/tests/s3.py

@@ -18,7 +18,7 @@
 # along with this program. If not, see <http://www.gnu.org/licenses/>.
 #
 from contextlib import ExitStack
-from typing import Any
+from typing import Any, AsyncIterator
 from unittest.mock import patch
 
 import pytest
@@ -62,7 +62,7 @@ def s3():
 
 
 @pytest.fixture(scope="function")
-async def s3_storage_settings(s3) -> dict[str, Any]:
+async def s3_storage_settings(s3) -> AsyncIterator[dict[str, Any]]:
     settings = {
         "file_backend": FileBackendConfig.S3,
         "s3_endpoint": s3,

nucliadb_utils/utilities.py

@@ -181,7 +181,7 @@ async def _create_storage(gcs_scopes: Optional[List[str]] = None) -> Storage:
         raise ConfigurationError("Invalid storage settings, please configure FILE_BACKEND")
 
 
-async def teardown_storage():
+async def teardown_storage() -> None:
     storage: Optional[Storage] = get_utility(Utility.STORAGE)
     if storage is None:
         return

{nucliadb_utils-6.3.1.post3524.dist-info → nucliadb_utils-6.3.1.post3526.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: nucliadb_utils
-Version: 6.3.1.post3524
+Version: 6.3.1.post3526
 Summary: NucliaDB util library
 Author-email: Nuclia <nucliadb@nuclia.com>
 License: AGPL
@@ -27,8 +27,8 @@ Requires-Dist: nats-py[nkeys]>=2.6.0
 Requires-Dist: PyNaCl
 Requires-Dist: pyjwt>=2.4.0
 Requires-Dist: mrflagly>=0.2.9
-Requires-Dist: nucliadb-protos>=6.3.1.post3524
-Requires-Dist: nucliadb-telemetry>=6.3.1.post3524
+Requires-Dist: nucliadb-protos>=6.3.1.post3526
+Requires-Dist: nucliadb-telemetry>=6.3.1.post3526
 Provides-Extra: cache
 Requires-Dist: redis>=4.3.4; extra == "cache"
 Requires-Dist: orjson>=3.6.7; extra == "cache"

{nucliadb_utils-6.3.1.post3524.dist-info → nucliadb_utils-6.3.1.post3526.dist-info}/RECORD

@@ -7,15 +7,15 @@ nucliadb_utils/exceptions.py,sha256=y_3wk77WLVUtdo-5FtbBsdSkCtK_DsJkdWb5BoPn3qo,
 nucliadb_utils/featureflagging.py,sha256=YxDiXzWuiDlHtqgeTVeyakmbAdzBePzJpgJv53ELbmI,2259
 nucliadb_utils/grpc.py,sha256=apu0uePnkGHCAT7GRQ9YZfRYyFj26kJ440i8jitbM3U,3314
 nucliadb_utils/helpers.py,sha256=nPw8yod3hP-pxq80VF8QC36s7ygSg0dBUdfI-LatvCs,1600
-nucliadb_utils/nats.py,sha256=99YJkK1QMfWDxKJK8u0acK0lLYb8yThalzulGETiT80,15266
+nucliadb_utils/nats.py,sha256=-ehKk0u5ZhiJEyg67yV26YysBGEGp4TDSQKq3aydJn0,15275
 nucliadb_utils/partition.py,sha256=jBgy4Hu5Iwn4gjbPPcthSykwf-qNx-GcLAIwbzPd1d0,1157
 nucliadb_utils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 nucliadb_utils/run.py,sha256=Es0_Bu5Yc-LWczvwL6gzWqSwC85RjDCk-0oFQAJi9g4,1827
 nucliadb_utils/settings.py,sha256=RnGhEUvwv6faNqALiqDCivvzNOyyXVBflYh_37uNkow,8193
-nucliadb_utils/signals.py,sha256=JRNv2y9zLtBjOANBf7krGfDGfOc9qcoXZ6N1nKWS2FE,2674
+nucliadb_utils/signals.py,sha256=lo_Mk12NIX5Au--3H3WObvDOXq_OMurql2qiC2TnAao,2676
 nucliadb_utils/store.py,sha256=kQ35HemE0v4_Qg6xVqNIJi8vSFAYQtwI3rDtMsNy62Y,890
 nucliadb_utils/transaction.py,sha256=YYnTpxCDs56lo0tS6ErABjk9WjDuieUc4f7r63Q_OP8,7864
-nucliadb_utils/utilities.py,sha256=usyqgkfinZeDoUSFpIugmKKJZpoi8wppFL_lAhhlkBI,15371
+nucliadb_utils/utilities.py,sha256=Vc4zLpDf-FQh9bs0ZyAfGAjzlbbMTMWf3VWt2Ao5V3k,15379
 nucliadb_utils/aiopynecone/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb_utils/aiopynecone/client.py,sha256=MPyHnDXwhukJr7U3CJh7BpsekfSuOkyM4g5b9LLtzc8,22941
 nucliadb_utils/aiopynecone/exceptions.py,sha256=fUErx3ceKQK1MUbOnYcZhIzpNe8UVAptZE9JIRDLXDE,4000
@@ -23,7 +23,7 @@ nucliadb_utils/aiopynecone/models.py,sha256=XkNIZx4bxdbVo9zYVn8IRp70q4DWUMWN79yb
 nucliadb_utils/audit/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZVV_MJn4bIXMa20ks,835
 nucliadb_utils/audit/audit.py,sha256=VDUqcSToEoz5m7iG-4omjTlsBSEJGcp0_xl51idnU3Q,3384
 nucliadb_utils/audit/basic.py,sha256=pVf1oKp7T_oTDCyt8QbSsfC_uWjF66ooF6rqY-SWSrc,4032
-nucliadb_utils/audit/stream.py,sha256=AT4OuLTtQ3mID8Geasn6d7qRK8D7MIMUi7GemzU_7mk,16017
+nucliadb_utils/audit/stream.py,sha256=SovPj5iTzXka2cg4NnwAZwnOYljhsT6ynhDC2OsRVPc,16035
 nucliadb_utils/cache/__init__.py,sha256=itSI7dtTwFP55YMX4iK7JzdMHS5CQVUiB1XzQu4UBh8,833
 nucliadb_utils/cache/exceptions.py,sha256=Zu-O_-0-yctOEgoDGI92gPzWfBMRrpiAyESA62ld6MA,975
 nucliadb_utils/cache/nats.py,sha256=-AjCfkFgKVdJUlGR0hT9JDSNkPVFg4S6w9eW-ZIcXPM,7037
@@ -40,24 +40,24 @@ nucliadb_utils/nuclia_usage/utils/__init__.py,sha256=cp15ZcFnHvpcu_5-aK2A4uUyvuZ
 nucliadb_utils/nuclia_usage/utils/kb_usage_report.py,sha256=6lLuxCCPQVn3dOuZNL5ThPjl2yws-1TJ_7duhQSWkPU,3934
 nucliadb_utils/storages/__init__.py,sha256=5Qc8AUWiJv9_JbGCBpAn88AIJhwDlm0OPQpg2ZdRL4U,872
 nucliadb_utils/storages/azure.py,sha256=FK4c_v9AUAwagScm_F1uDmJeQQq7P4jZswiD2trwb4A,16394
-nucliadb_utils/storages/exceptions.py,sha256=_xUDbfr3XkmUc0xXYZu3DpG1C8TtfmEweRwj9mbRvMU,2507
-nucliadb_utils/storages/gcs.py,sha256=R35XQmf_M8Ba_eIuQumjAWSgU_lSouoTp-TTua-VgOM,28911
+nucliadb_utils/storages/exceptions.py,sha256=GOPKH-F3dPTfHEkwGNfVkSfF70eWJJXjI83yccw9WpA,2501
+nucliadb_utils/storages/gcs.py,sha256=5QODikIj26tfjz4KqhNgezPqqmqrG1yvUbarKAQ6sS0,28899
 nucliadb_utils/storages/local.py,sha256=JxlWNtu49JJ04dq6o7bBAqbpbeYpVyvvBM5jq1sGJ-4,11003
 nucliadb_utils/storages/nuclia.py,sha256=vEv94xAT7QM2g80S25QyrOw2pzvP2BAX-ADgZLtuCVc,2097
 nucliadb_utils/storages/object_store.py,sha256=HtKjIKhErSBvuqx1SuCOnL0SkiHqgfyekNMP8o2piZU,4492
-nucliadb_utils/storages/s3.py,sha256=pgakBqBLfDAFbdjwwX6u19QTv7gQYZ82Pjw67Yi6ob8,20895
+nucliadb_utils/storages/s3.py,sha256=isw4v9L4tlwLklNotdF-l7mU4coshPzO5wj_tbuSNOc,20916
 nucliadb_utils/storages/settings.py,sha256=ugCPy1zxBOmA2KosT-4tsjpvP002kg5iQyi42yCGCJA,1285
 nucliadb_utils/storages/storage.py,sha256=1LSaZKQ4xSoBP85OVATh7zBZdBQ369Wl1uYBymZPfPw,21135
 nucliadb_utils/storages/utils.py,sha256=8g2rIwJeYIumQLOB47Yw1rx3twlhRB_cJxer65QfZmk,1479
 nucliadb_utils/tests/__init__.py,sha256=Oo9CAE7B0eW5VHn8sHd6o30SQzOWUhktLPRXdlDOleA,1456
 nucliadb_utils/tests/asyncbenchmark.py,sha256=vrX_x9ifCXi18PfNShc23w9x_VUiB_Ph-2nuolh9z3Q,10707
-nucliadb_utils/tests/azure.py,sha256=Dg-Eb4KVScG-O6P9y-bVQZTAKTNUMQ0i-CKEd9IdrWw,4474
-nucliadb_utils/tests/fixtures.py,sha256=i0sqPqe5a5JlKGFdaIvOlHYkZ3pHZ2hTIgTsaIB3vSM,3472
-nucliadb_utils/tests/gcs.py,sha256=KW_DLet1WRlssSW55eI-IQ-0d94Jo2Oh7Di4xGv4JCc,4685
-nucliadb_utils/tests/local.py,sha256=fXIBasrvdaFJM-sw2wk1_oiFzBcm9O10iCyC-OiXwY8,1914
+nucliadb_utils/tests/azure.py,sha256=NvMrPG6gfbpDE0m_aZgaa7eorbmA1r9rhAsAANhMlJk,4494
+nucliadb_utils/tests/fixtures.py,sha256=uNiZ8qaOz2bSmSn-2ZMXxhQcHVh3tCyr5A-e0XaoRr8,3492
+nucliadb_utils/tests/gcs.py,sha256=MBMzn_UHU5SU6iILuCsB5zU4umhNcaCw_MKrxZhwvOc,4705
+nucliadb_utils/tests/local.py,sha256=cxIfPrKuqs5Ef0nbrVYQQAH2mwc4E0iD9bC2sWegS-c,1934
 nucliadb_utils/tests/nats.py,sha256=RWHjwqq5esuO7OFbP24yYX1cXnpPLcWJwDUdmwCpH28,1897
-nucliadb_utils/tests/s3.py,sha256=pl-RJFjA4MH6iXkqhsh5g8gDuEhrYu1nPZ-laxlrMlE,3704
-nucliadb_utils-6.3.1.post3524.dist-info/METADATA,sha256=tMP8Wx4jEKAM8zESbKBv1QQKqJe7jvbxoW0md9DHIDc,2209
-nucliadb_utils-6.3.1.post3524.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
-nucliadb_utils-6.3.1.post3524.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
-nucliadb_utils-6.3.1.post3524.dist-info/RECORD,,
+nucliadb_utils/tests/s3.py,sha256=DACUh3HvgH3BchKFZ9R7RFUzsrg3v9A-cxTcXx4nmvA,3734
+nucliadb_utils-6.3.1.post3526.dist-info/METADATA,sha256=aZS3f5jsY6xY837mfcDJxkqUsUksTbzTzpld0e5IRdg,2209
+nucliadb_utils-6.3.1.post3526.dist-info/WHEEL,sha256=jB7zZ3N9hIM9adW7qlTAyycLYW9npaWKLRzaoVcLKcM,91
+nucliadb_utils-6.3.1.post3526.dist-info/top_level.txt,sha256=fE3vJtALTfgh7bcAWcNhcfXkNPp_eVVpbKK-2IYua3E,15
+nucliadb_utils-6.3.1.post3526.dist-info/RECORD,,