sparrowhawk/cdist
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

remove _read_stdin and open stdin file in non-binary mode

Browse source 
Signed-off-by: Nico Schottelius <nico@brief.schottelius.org>
This commit is contained in:
Nico Schottelius 2012-11-15 09:32:44 +01:00
parent 03ec09c771
commit a498330bd3
1 changed files with 3 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

10
cdist/emulator.py
View file

@ -149,8 +149,6 @@ class Emulator(object):
# Record / Append source
self.cdist_object.source.append(self.object_source)
def _read_stdin(self):
return self.stdin.read()
def save_stdin(self):
"""If something is written to stdin, save it in the object as
$__object/stdin so it can be accessed in manifest and gencode-*
@ -160,12 +158,10 @@ class Emulator(object):
try:
# go directly to file instead of using CdistObject's api
# as that does not support streaming
# FIXME: no streaming needed anymore
path = os.path.join(self.cdist_object.absolute_path, 'stdin')
with open(path, 'wb') as fd:
chunk = self._read_stdin()
while chunk:
fd.write(chunk)
chunk = self._read_stdin()
with open(path, 'w') as fd:
fd.write(self.stdin.read())
except EnvironmentError as e:
raise cdist.Error('Failed to read from stdin: %s' % e)