Mercurial > public > mercurial-scm > hg

comparison mercurial/utils/compression.py @ 52643:5cc8deb96b48

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
pyupgrade: modernize calls to superclass methods This is the `legacy` fixer in `pyupgrade`, with the loop yielding the offset of `yield` statements commented out.
author Matt Harbison <matt_harbison@yahoo.com>
date Sun, 05 Jan 2025 22:23:31 -0500
parents f4733654f144
children
comparison
equal deleted inserted replaced
52642:73ab542565e0 52643:5cc8deb96b48
377 return b''.join(buf) 377 return b''.join(buf)
378 378
379 379
380 class _GzipCompressedStreamReader(_CompressedStreamReader): 380 class _GzipCompressedStreamReader(_CompressedStreamReader):
381 def __init__(self, fh): 381 def __init__(self, fh):
382 super(_GzipCompressedStreamReader, self).__init__(fh) 382 super().__init__(fh)
383 self._decompobj = zlib.decompressobj() 383 self._decompobj = zlib.decompressobj()
384 384
385 def _decompress(self, chunk): 385 def _decompress(self, chunk):
386 newbuf = self._decompobj.decompress(chunk) 386 newbuf = self._decompobj.decompress(chunk)
387 if newbuf: 387 if newbuf:
396 pass 396 pass
397 397
398 398
399 class _BZ2CompressedStreamReader(_CompressedStreamReader): 399 class _BZ2CompressedStreamReader(_CompressedStreamReader):
400 def __init__(self, fh): 400 def __init__(self, fh):
401 super(_BZ2CompressedStreamReader, self).__init__(fh) 401 super().__init__(fh)
402 self._decompobj = bz2.BZ2Decompressor() 402 self._decompobj = bz2.BZ2Decompressor()
403 403
404 def _decompress(self, chunk): 404 def _decompress(self, chunk):
405 newbuf = self._decompobj.decompress(chunk) 405 newbuf = self._decompobj.decompress(chunk)
406 if newbuf: 406 if newbuf:
416 self._eof = True 416 self._eof = True
417 417
418 418
419 class _TruncatedBZ2CompressedStreamReader(_BZ2CompressedStreamReader): 419 class _TruncatedBZ2CompressedStreamReader(_BZ2CompressedStreamReader):
420 def __init__(self, fh): 420 def __init__(self, fh):
421 super(_TruncatedBZ2CompressedStreamReader, self).__init__(fh) 421 super().__init__(fh)
422 newbuf = self._decompobj.decompress(b'BZ') 422 newbuf = self._decompobj.decompress(b'BZ')
423 if newbuf: 423 if newbuf:
424 self._pending.append(newbuf) 424 self._pending.append(newbuf)
425 425
426 426
427 class _ZstdCompressedStreamReader(_CompressedStreamReader): 427 class _ZstdCompressedStreamReader(_CompressedStreamReader):
428 def __init__(self, fh, zstd): 428 def __init__(self, fh, zstd):
429 super(_ZstdCompressedStreamReader, self).__init__(fh) 429 super().__init__(fh)
430 self._zstd = zstd 430 self._zstd = zstd
431 self._decompobj = zstd.ZstdDecompressor().decompressobj() 431 self._decompobj = zstd.ZstdDecompressor().decompressobj()
432 432
433 def _decompress(self, chunk): 433 def _decompress(self, chunk):
434 newbuf = self._decompobj.decompress(chunk) 434 newbuf = self._decompobj.decompress(chunk)