sascha/workload-automation
1
0
Fork 0
mirror of https://github.com/ARM-software/workload-automation.git synced 2025-02-22 21:08:51 +00:00
Code Issues Releases Wiki Activity

Merge pull request #224 from setrofim/master

Browse Source 
ssh: fixing rasing of CalledProcessErrorWithStderr
This commit is contained in:
Sebastian Goscik 2016-08-23 17:26:07 +01:00 committed by GitHub
commit b1ae5a5465
2 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
wlauto/utils/misc.py
View File

@ -85,6 +85,7 @@ class TimeoutError(Exception):
class CalledProcessErrorWithStderr(CalledProcessError): class CalledProcessErrorWithStderr(CalledProcessError):
def __init__(self, *args, **kwargs): def __init__(self, *args, **kwargs):
self.output = kwargs.pop("output")
self.error = kwargs.pop("error") self.error = kwargs.pop("error")
super(CalledProcessErrorWithStderr, self).__init__(*args, **kwargs) super(CalledProcessErrorWithStderr, self).__init__(*args, **kwargs)

3
wlauto/utils/ssh.py
View File

@ -245,7 +245,8 @@ class SshShell(object):
except subprocess.CalledProcessError as e: except subprocess.CalledProcessError as e:
raise CalledProcessErrorWithStderr(e.returncode, raise CalledProcessErrorWithStderr(e.returncode,
e.cmd.replace(pass_string, ''), e.cmd.replace(pass_string, ''),
e.output, getattr(e, 'error', '')) output=e.output,
error=getattr(e, 'error', ''))
except TimeoutError as e: except TimeoutError as e:
raise TimeoutError(e.command.replace(pass_string, ''), e.output) raise TimeoutError(e.command.replace(pass_string, ''), e.output)