Mercurial > hg
changeset 40316:09540a5f0a15
py3: reinvent print() function for contrib/hgclient.py
author | Yuya Nishihara <yuya@tcha.org> |
---|---|
date | Tue, 16 Oct 2018 07:04:07 +0200 |
parents | 431a831342d2 |
children | 6958eb9bdcd6 |
files | contrib/hgclient.py |
diffstat | 1 files changed, 9 insertions(+), 3 deletions(-) [+] |
line wrap: on
line diff
--- a/contrib/hgclient.py Tue Oct 16 07:08:12 2018 +0200 +++ b/contrib/hgclient.py Tue Oct 16 07:04:07 2018 +0200 @@ -2,6 +2,7 @@ from __future__ import absolute_import, print_function import os +import re import signal import socket import struct @@ -19,9 +20,14 @@ if sys.version_info[0] >= 3: stdout = sys.stdout.buffer stderr = sys.stderr.buffer + def bprint(*args): + # remove b'' as well for ease of test migration + pargs = [re.sub(br'''\bb(['"])''', br'\1', b'%s' % a) for a in args] + stdout.write(b' '.join(pargs) + b'\n') else: stdout = sys.stdout stderr = sys.stderr + bprint = print def connectpipe(path=None): cmdline = [b'hg', b'serve', b'--cmdserver', b'pipe'] @@ -90,7 +96,7 @@ def runcommand(server, args, output=stdout, error=stderr, input=None, outfilter=lambda x: x): - print(b'*** runcommand', b' '.join(args)) + bprint(b'*** runcommand', b' '.join(args)) stdout.flush() server.stdin.write(b'runcommand\n') writeblock(server, b'\0'.join(args)) @@ -113,10 +119,10 @@ elif ch == b'r': ret, = struct.unpack('>i', data) if ret != 0: - print(b' [%d]' % ret) + bprint(b' [%d]' % ret) return ret else: - print(b"unexpected channel %c: %r" % (ch, data)) + bprint(b"unexpected channel %c: %r" % (ch, data)) if ch.isupper(): return