Make save_stream pass through kwargs (#396)

* make save_stream pass through kwargs

* make base save_stream ok with extra [kw]args

* make key argument explicit
This commit is contained in:
JP Smith 2017-07-17 14:06:42 -04:00 committed by GitHub
parent ddc3b12848
commit 47a5d9752f

View File

@ -94,7 +94,7 @@ class Store(object):
return s.read()
@contextmanager
def save_stream(self, key):
def save_stream(self, key, *rest, **kwargs):
"""
Return a managed file-like object into which the calling code can write
arbitrary data.
@ -437,8 +437,8 @@ class ManticoreOutput(object):
self.save_fds(state)
self._store.save_state(state, self._named_key('pkl'))
def save_stream(self, *rest):
return self._store.save_stream(*rest)
def save_stream(self, key, *rest, **kwargs):
return self._store.save_stream(key, *rest, **kwargs)
@contextmanager
def _named_stream(self, name):