nucliadb 6.8.0.post4925__py3-none-any.whl → 6.8.1.post4932__py3-none-any.whl

nucliadb/writer/tus/gcs.py

@@ -74,7 +74,7 @@ RETRIABLE_EXCEPTIONS = (
 
 
 class GCloudBlobStore(BlobStore):
-    session: Optional[aiohttp.ClientSession] = None
+    _session: Optional[aiohttp.ClientSession] = None
     loop = None
     upload_url: str
     object_base_url: str
@@ -84,6 +84,12 @@ class GCloudBlobStore(BlobStore):
     project: str
     executor = ThreadPoolExecutor(max_workers=5)
 
+    @property
+    def session(self) -> aiohttp.ClientSession:
+        if self._session is None:  # pragma: no cover
+            raise AttributeError("Session not initialized")
+        return self._session
+
     async def get_access_headers(self):
         if self._credentials is None:
             return {}
@@ -106,8 +112,9 @@ class GCloudBlobStore(BlobStore):
             return access_token.access_token
 
     async def finalize(self):
-        if self.session is not None:
-            await self.session.close()
+        if self._session is not None:
+            await self._session.close()
+            self._session = None
 
     async def initialize(
         self,
@@ -143,12 +150,9 @@ class GCloudBlobStore(BlobStore):
                 self._credentials = None
 
         loop = asyncio.get_event_loop()
-        self.session = aiohttp.ClientSession(loop=loop, timeout=TIMEOUT)
+        self._session = aiohttp.ClientSession(loop=loop, timeout=TIMEOUT)
 
     async def check_exists(self, bucket_name: str):
-        if self.session is None:
-            raise AttributeError()
-
         headers = await self.get_access_headers()
         # Using object access url instead of bucket access to avoid
         # giving admin permission to the SA, needed to GET a bucket
@@ -163,8 +167,6 @@ class GCloudBlobStore(BlobStore):
         return False
 
     async def create_bucket(self, bucket_name: str):
-        if self.session is None:
-            raise AttributeError()
         headers = await self.get_access_headers()
         url = f"{self.object_base_url}?project={self.project}"
 
@@ -199,10 +201,6 @@ class GCloudFileStorageManager(FileStorageManager):
         _resumable_uri : uri to resumable upload
         _uri : finished uploaded image
         """
-
-        if self.storage.session is None:
-            raise AttributeError()
-
         upload_file_id = dm.get("upload_file_id")
         if upload_file_id is not None:
             await self.delete_upload(upload_file_id, kbid)
@@ -287,8 +285,6 @@ class GCloudFileStorageManager(FileStorageManager):
 
     @backoff.on_exception(backoff.expo, RETRIABLE_EXCEPTIONS, jitter=backoff.random_jitter, max_tries=4)
     async def _append(self, dm: FileDataManager, data, offset):
-        if self.storage.session is None:
-            raise AttributeError()
         if dm.size:
             size = str(dm.size)
         else:
@@ -353,8 +349,6 @@ class GCloudFileStorageManager(FileStorageManager):
     @backoff.on_exception(backoff.expo, RETRIABLE_EXCEPTIONS, jitter=backoff.random_jitter, max_tries=4)
     async def finish(self, dm: FileDataManager):
         if dm.size == 0:
-            if self.storage.session is None:
-                raise AttributeError()
             # In case of empty file, we need to send a PUT request with empty body
             # and Content-Range header set to "bytes */0"
             headers = {

{nucliadb-6.8.0.post4925.dist-info → nucliadb-6.8.1.post4932.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: nucliadb
-Version: 6.8.0.post4925
+Version: 6.8.1.post4932
 Summary: NucliaDB
 Author-email: Nuclia <nucliadb@nuclia.com>
 License-Expression: AGPL-3.0-or-later
@@ -19,11 +19,11 @@ Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3 :: Only
 Requires-Python: <4,>=3.9
 Description-Content-Type: text/markdown
-Requires-Dist: nucliadb-telemetry[all]>=6.8.0.post4925
-Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.8.0.post4925
-Requires-Dist: nucliadb-protos>=6.8.0.post4925
-Requires-Dist: nucliadb-models>=6.8.0.post4925
-Requires-Dist: nidx-protos>=6.8.0.post4925
+Requires-Dist: nucliadb-telemetry[all]>=6.8.1.post4932
+Requires-Dist: nucliadb-utils[cache,fastapi,storages]>=6.8.1.post4932
+Requires-Dist: nucliadb-protos>=6.8.1.post4932
+Requires-Dist: nucliadb-models>=6.8.1.post4932
+Requires-Dist: nidx-protos>=6.8.1.post4932
 Requires-Dist: nucliadb-admin-assets>=1.0.0.post1224
 Requires-Dist: nuclia-models>=0.47.0
 Requires-Dist: uvicorn[standard]

{nucliadb-6.8.0.post4925.dist-info → nucliadb-6.8.1.post4932.dist-info}/RECORD

@@ -374,13 +374,13 @@ nucliadb/writer/tus/__init__.py,sha256=Kera0BtxoDX0ngPftXiMjNgjrhtQ3l2XFc5nJqSBO
 nucliadb/writer/tus/azure.py,sha256=XhWAlWTM0vmXcXtuEPYjjeEhuZjiZXZu8q9WsJ7omFE,4107
 nucliadb/writer/tus/dm.py,sha256=bVoXqt_dpNvTjpffPYhj1JfqK6gfLoPr0hdkknUCZ9E,5488
 nucliadb/writer/tus/exceptions.py,sha256=WfZSSjsHfoy63wUFlH3QoHx7FMoCNA1oKJmWpZZDnCo,2156
-nucliadb/writer/tus/gcs.py,sha256=iTo2Xd8w8oazBjxHpRICJh317AQcyhnC1n2i97Lsooo,14003
+nucliadb/writer/tus/gcs.py,sha256=pzJDJSXZKh7VrFYw-wiRSCCXfXklfGwDZSMAnjoh_48,13871
 nucliadb/writer/tus/local.py,sha256=7jYa_w9b-N90jWgN2sQKkNcomqn6JMVBOVeDOVYJHto,5193
 nucliadb/writer/tus/s3.py,sha256=vu1BGg4VqJ_x2P1u2BxqPKlSfw5orT_a3R-Ln5oPUpU,8483
 nucliadb/writer/tus/storage.py,sha256=ToqwjoYnjI4oIcwzkhha_MPxi-k4Jk3Lt55zRwaC1SM,2903
 nucliadb/writer/tus/utils.py,sha256=MSdVbRsRSZVdkaum69_0wku7X3p5wlZf4nr6E0GMKbw,2556
-nucliadb-6.8.0.post4925.dist-info/METADATA,sha256=Of26ErA1L2l3SsuBlrfG_O1ywcdnjx4K_eXiJvqSbZs,4158
-nucliadb-6.8.0.post4925.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-nucliadb-6.8.0.post4925.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
-nucliadb-6.8.0.post4925.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
-nucliadb-6.8.0.post4925.dist-info/RECORD,,
+nucliadb-6.8.1.post4932.dist-info/METADATA,sha256=j-QmNX1P8xGjploIJ2vMzm015CCHBQ0ZqD2Tv3AwI5I,4158
+nucliadb-6.8.1.post4932.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+nucliadb-6.8.1.post4932.dist-info/entry_points.txt,sha256=XqGfgFDuY3zXQc8ewXM2TRVjTModIq851zOsgrmaXx4,1268
+nucliadb-6.8.1.post4932.dist-info/top_level.txt,sha256=hwYhTVnX7jkQ9gJCkVrbqEG1M4lT2F_iPQND1fCzF80,20
+nucliadb-6.8.1.post4932.dist-info/RECORD,,