diff options
author | Victor Stinner <victor.stinner@gmail.com> | 2015-01-15 21:53:27 +0100 |
---|---|---|
committer | Victor Stinner <victor.stinner@gmail.com> | 2015-01-15 21:53:27 +0100 |
commit | 7d342ae6c4ae5a8e5456352589787f6be1dc3a62 (patch) | |
tree | 03207ccf58880a748188836155ceed72f03d65d6 | |
parent | d0b9679f9198df2bfef2e9faaed70d68f9067271 (diff) | |
download | trollius-7d342ae6c4ae5a8e5456352589787f6be1dc3a62.tar.gz |
Backout changeset 3a06020af8cf
StreamWriter: close() now clears the reference to the transport
StreamWriter now raises an exception if it is closed: write(), writelines(),
write_eof(), can_write_eof(), get_extra_info(), drain().
-rw-r--r-- | asyncio/streams.py | 25 |
1 files changed, 4 insertions, 21 deletions
diff --git a/asyncio/streams.py b/asyncio/streams.py index 12ab1c5..7ff16a4 100644 --- a/asyncio/streams.py +++ b/asyncio/streams.py @@ -258,22 +258,8 @@ class StreamWriter: self._reader = reader self._loop = loop - def close(self): - if self._transport is None: - return - self._transport.close() - self._transport = None - - def _check_closed(self): - if self._transport is None: - raise RuntimeError('StreamWriter is closed') - def __repr__(self): - info = [self.__class__.__name__] - if self._transport is not None: - info.append('transport=%r' % self._transport) - else: - info.append('closed') + info = [self.__class__.__name__, 'transport=%r' % self._transport] if self._reader is not None: info.append('reader=%r' % self._reader) return '<%s>' % ' '.join(info) @@ -283,23 +269,21 @@ class StreamWriter: return self._transport def write(self, data): - self._check_closed() self._transport.write(data) def writelines(self, data): - self._check_closed() self._transport.writelines(data) def write_eof(self): - self._check_closed() return self._transport.write_eof() def can_write_eof(self): - self._check_closed() return self._transport.can_write_eof() + def close(self): + return self._transport.close() + def get_extra_info(self, name, default=None): - self._check_closed() return self._transport.get_extra_info(name, default) @coroutine @@ -311,7 +295,6 @@ class StreamWriter: w.write(data) yield from w.drain() """ - self._check_closed() if self._reader is not None: exc = self._reader.exception() if exc is not None: |