Skip to content
Snippets Groups Projects

fix uploader

Merged Anton Teplyakov requested to merge bugfix/fix-file-uploader into master
2 files
+ 7
8
Compare changes
  • Side-by-side
  • Inline
Files
2
+ 6
7
@@ -20,7 +20,6 @@ async def read_chunks(
@@ -20,7 +20,6 @@ async def read_chunks(
if not data:
if not data:
break
break
yield data
yield data
file_object.seek(0)
class FileUploader:
class FileUploader:
@@ -30,8 +29,8 @@ class FileUploader:
@@ -30,8 +29,8 @@ class FileUploader:
data_provider: PerxisDataProvider,
data_provider: PerxisDataProvider,
chunk_size: int = DEFAULT_CHUNK_SIZE,
chunk_size: int = DEFAULT_CHUNK_SIZE,
) -> None:
) -> None:
self._file_provider = file_provider
self.file_provider = file_provider
self._data_provider = data_provider
self.data_provider = data_provider
self.chunk_size = chunk_size
self.chunk_size = chunk_size
async def __put_chunks(
async def __put_chunks(
@@ -60,7 +59,7 @@ class FileUploader:
@@ -60,7 +59,7 @@ class FileUploader:
file_size: int,
file_size: int,
collection_id: str
collection_id: str
) -> items_pb2.CreateResponse:
) -> items_pb2.CreateResponse:
message = await self._file_provider.start_upload(
message = await self.file_provider.start_upload(
file_name=file_name,
file_name=file_name,
file_size=file_size,
file_size=file_size,
)
)
@@ -70,7 +69,7 @@ class FileUploader:
@@ -70,7 +69,7 @@ class FileUploader:
part_urls=message.upload.part_urls,
part_urls=message.upload.part_urls,
)
)
await self._file_provider.complete_upload(
await self.file_provider.complete_upload(
file_id=message.upload.file.id,
file_id=message.upload.file.id,
upload_id=message.upload.upload_id,
upload_id=message.upload.upload_id,
parts=parts,
parts=parts,
@@ -86,7 +85,7 @@ class FileUploader:
@@ -86,7 +85,7 @@ class FileUploader:
"file": file,
"file": file,
})
})
message = await self._data_provider.create(
message = await self.data_provider.create(
data=data,
data=data,
collection_id=collection_id,
collection_id=collection_id,
)
)
@@ -105,7 +104,7 @@ class FileUploader:
@@ -105,7 +104,7 @@ class FileUploader:
file_size=file_size,
file_size=file_size,
collection_id=collection_id,
collection_id=collection_id,
)
)
await self._data_provider.publish(
await self.data_provider.publish(
item_id=message.created.id,
item_id=message.created.id,
collection_id=collection_id,
collection_id=collection_id,
)
)
Loading