From 43161742a9ba3f56b69ad326a25ab4aada9ba7a8 Mon Sep 17 00:00:00 2001
From: Pedro Ferreira <pedro@dete.st>
Date: Sun, 4 Feb 2018 22:19:15 +0100
Subject: [PATCH] Add is_connected flag

---
 secret_handshake/network.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/secret_handshake/network.py b/secret_handshake/network.py
index b510263..5416957 100644
--- a/secret_handshake/network.py
+++ b/secret_handshake/network.py
@@ -35,6 +35,7 @@ class SHSDuplexStream(object):
     def __init__(self):
         self.write_stream = None
         self.read_stream = None
+        self.is_connected = False
 
     def write(self, data):
         self.write_stream.write(data)
@@ -45,6 +46,7 @@ class SHSDuplexStream(object):
     def close(self):
         self.write_stream.close()
         self.read_stream.close()
+        self.is_connected = False
 
     @async_generator
     async def __aiter__(self):
@@ -148,6 +150,7 @@ class SHSClient(SHSDuplexStream, SHSEndpoint):
 
         self.read_stream, self.write_stream = get_stream_pair(reader, writer, **keys)
         self.writer = writer
+        self.is_connected = True
         if self._on_connect:
             await self._on_connect()