| Alphare |
| hg-reviewers |
This was required for Python 2 support, which we no longer need to support.
| Automatic diff as part of commit; lint not applicable. |
| Automatic diff as part of commit; unit tests not applicable. |
| Path | Packages | |||
|---|---|---|---|---|
| M | mercurial/archival.py (35 lines) |
| return out.getvalue() | return out.getvalue() | ||||
| class tarit(object): | class tarit(object): | ||||
| """write archive to tar file or stream. can write uncompressed, | """write archive to tar file or stream. can write uncompressed, | ||||
| or compress with gzip or bzip2.""" | or compress with gzip or bzip2.""" | ||||
| if pycompat.ispy3: | |||||
| GzipFileWithTime = gzip.GzipFile # camelcase-required | |||||
| else: | |||||
| class GzipFileWithTime(gzip.GzipFile): | |||||
| def __init__(self, *args, **kw): | |||||
| timestamp = None | |||||
| if 'mtime' in kw: | |||||
| timestamp = kw.pop('mtime') | |||||
| if timestamp is None: | |||||
| self.timestamp = time.time() | |||||
| else: | |||||
| self.timestamp = timestamp | |||||
| gzip.GzipFile.__init__(self, *args, **kw) | |||||
| def _write_gzip_header(self): | |||||
| self.fileobj.write(b'\037\213') # magic header | |||||
| self.fileobj.write(b'\010') # compression method | |||||
| fname = self.name | |||||
| if fname and fname.endswith(b'.gz'): | |||||
| fname = fname[:-3] | |||||
| flags = 0 | |||||
| if fname: | |||||
| flags = gzip.FNAME # pytype: disable=module-attr | |||||
| self.fileobj.write(pycompat.bytechr(flags)) | |||||
| gzip.write32u( # pytype: disable=module-attr | |||||
| self.fileobj, int(self.timestamp) | |||||
| ) | |||||
| self.fileobj.write(b'\002') | |||||
| self.fileobj.write(b'\377') | |||||
| if fname: | |||||
| self.fileobj.write(fname + b'\000') | |||||
| def __init__(self, dest, mtime, kind=b''): | def __init__(self, dest, mtime, kind=b''): | ||||
| self.mtime = mtime | self.mtime = mtime | ||||
| self.fileobj = None | self.fileobj = None | ||||
| def taropen(mode, name=b'', fileobj=None): | def taropen(mode, name=b'', fileobj=None): | ||||
| if kind == b'gz': | if kind == b'gz': | ||||
| mode = mode[0:1] | mode = mode[0:1] | ||||
| if not fileobj: | if not fileobj: | ||||
| fileobj = open(name, mode + b'b') | fileobj = open(name, mode + b'b') | ||||
| gzfileobj = self.GzipFileWithTime( | gzfileobj = gzip.GzipFile( | ||||
| name, | name, | ||||
| pycompat.sysstr(mode + b'b'), | pycompat.sysstr(mode + b'b'), | ||||
| zlib.Z_BEST_COMPRESSION, | zlib.Z_BEST_COMPRESSION, | ||||
| fileobj, | fileobj, | ||||
| mtime=mtime, | mtime=mtime, | ||||
| ) | ) | ||||
| self.fileobj = gzfileobj | self.fileobj = gzfileobj | ||||
| return ( | return ( | ||||