mongodb/mongo-python-driver
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Remove redundant branch in GridFS (#2064)

Browse Source
This commit is contained in:
Shane Harvey 2025-02-27 15:07:50 -08:00 committed by GitHub
parent c9a85ad321
commit e52965eea4
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 14 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
gridfs/asynchronous/grid_file.py
View File

@ -1301,11 +1301,8 @@ class AsyncGridIn:
raise ValueError("cannot write to a closed file")
try:
if isinstance(data, AsyncGridOut):
read = data.read
else:
# file-like
read = data.read
# file-like
read = data.read
except AttributeError:
# string
if not isinstance(data, (str, bytes)):
@ -1317,7 +1314,7 @@ class AsyncGridIn:
raise TypeError(
"must specify an encoding for file in order to write str"
) from None
read = io.BytesIO(data).read # type: ignore[assignment]
read = io.BytesIO(data).read
if inspect.iscoroutinefunction(read):
await self._write_async(read)
@ -1331,15 +1328,15 @@ class AsyncGridIn:
except BaseException:
await self.abort()
raise
self._buffer.write(to_write) # type: ignore
if len(to_write) < space: # type: ignore
self._buffer.write(to_write)
if len(to_write) < space:
return # EOF or incomplete
await self._flush_buffer()
to_write = read(self.chunk_size)
while to_write and len(to_write) == self.chunk_size: # type: ignore
while to_write and len(to_write) == self.chunk_size:
await self._flush_data(to_write)
to_write = read(self.chunk_size)
self._buffer.write(to_write) # type: ignore
self._buffer.write(to_write)
async def _write_async(self, read: Any) -> None:
if self._buffer.tell() > 0:

17
gridfs/synchronous/grid_file.py
View File

@ -1291,11 +1291,8 @@ class GridIn:
raise ValueError("cannot write to a closed file")
try:
if isinstance(data, GridOut):
read = data.read
else:
# file-like
read = data.read
# file-like
read = data.read
except AttributeError:
# string
if not isinstance(data, (str, bytes)):
@ -1307,7 +1304,7 @@ class GridIn:
raise TypeError(
"must specify an encoding for file in order to write str"
) from None
read = io.BytesIO(data).read # type: ignore[assignment]
read = io.BytesIO(data).read
if inspect.iscoroutinefunction(read):
self._write_async(read)
@ -1321,15 +1318,15 @@ class GridIn:
except BaseException:
self.abort()
raise
self._buffer.write(to_write) # type: ignore
if len(to_write) < space: # type: ignore
self._buffer.write(to_write)
if len(to_write) < space:
return # EOF or incomplete
self._flush_buffer()
to_write = read(self.chunk_size)
while to_write and len(to_write) == self.chunk_size: # type: ignore
while to_write and len(to_write) == self.chunk_size:
self._flush_data(to_write)
to_write = read(self.chunk_size)
self._buffer.write(to_write) # type: ignore
self._buffer.write(to_write)
def _write_async(self, read: Any) -> None:
if self._buffer.tell() > 0: