mercurial-scm/hg-stable: mercurial/util.py comparison

py3: wrap file object to write patch in native eol preserving byte-ness

comparison

equal deleted inserted replaced

-:a00c38b33047
+:472c68cda3f8
 (10, 1 << 10, _('%.1f KB')),
 (1, 1 << 10, _('%.2f KB')),
 (1, 1, _('%.0f bytes')),
 )
+class transformingwriter(object):
+"""Writable file wrapper to transform data by function"""
+def __init__(self, fp, encode):
+self._fp = fp
+self._encode = encode
+def close(self):
+self._fp.close()
+def flush(self):
+self._fp.flush()
+def write(self, data):
+return self._fp.write(self._encode(data))
 # Matches a single EOL which can either be a CRLF where repeated CR
 # are removed or a LF. We do not care about old Macintosh files, so a
 # stray CR is an error.
 _eolre = remod.compile(br'\r*\n')
 return _eolre.sub('\n', s)
 def tocrlf(s):
 return _eolre.sub('\r\n', s)
+def _crlfwriter(fp):
+return transformingwriter(fp, tocrlf)
 if pycompat.oslinesep == '\r\n':
 tonativeeol = tocrlf
 fromnativeeol = tolf
+nativeeolwriter = _crlfwriter
 else:
 tonativeeol = pycompat.identity
 fromnativeeol = pycompat.identity
+nativeeolwriter = pycompat.identity
 def escapestr(s):
 # call underlying function of s.encode('string_escape') directly for
 # Python 3 compatibility
 return codecs.escape_encode(s)[0]

Mercurial > public > mercurial-scm > hg-stable