andry81/tacklelib

# python module for commands with extension modules usage: tacklelib, plumbum

tkl_source_module(CMDOPLIB_ROOT, 'cmdoplib.std.xsh')

import sys, plumbum

def call_svn(args_list,
             stdin = None, stdout = None, stderr = None,
             max_stdout_lines = 9, ignore_warnings = False,
             **kwargs):
  # retarget stdout/stderr to original handles to avoid it's interception and so the blocking
  if max_stdout_lines < 0:
    if not stdout:
      stdout = sys.stdout
    if not stderr:
      stderr = sys.stderr

  try:
    ret = call('${SVN}', args_list,
      stdin = stdin, stdout = stdout, stderr = stderr,
      **kwargs)

  except plumbum.ProcessExecutionError as proc_err:
    if len(proc_err.stdout) > 0:
      print(proc_err.stdout.rstrip())
    if len(proc_err.stderr) > 0:
      print(proc_err.stderr.rstrip())
    if stdout is None or not hasattr(stdout, 'name') or stdout.name != '<null>':
      print('<') # end of a command output
    raise

  else:
    # cut out the middle of the stdout
    stdout_lines = ret[1].rstrip()
    stderr_lines = ret[2].rstrip()

    tkl.print_max(stdout_lines, max_lines = max_stdout_lines)
    if len(stderr_lines) > 0:
      print(stderr_lines)
      if not ignore_warnings:
        stderr_warning_match = re.match('warning:\\s[^+-].*', stderr_lines, flags = re.MULTILINE)
        if stderr_warning_match:
          raise Exception('specific warnings from the `svn ...` command is treated as errors')

    if stdout is None or not hasattr(stdout, 'name') or stdout.name != '<null>':
      print('<') # end of a command output

  return ret

def call_svn_no_except(args_list, **kwargs):
  return call_svn(
    args_list,
    no_except = True,
    **kwargs)