--- a/lib/twisted_sse_demo/sse_client.py	Fri Apr 19 01:08:01 2019 -0700
+++ b/lib/twisted_sse_demo/sse_client.py	Fri Apr 19 04:18:44 2019 -0700
@@ -2,13 +2,15 @@
 
 
 class EventSourceProtocol(LineReceiver):
-    def __init__(self):
+    def __init__(self, onConnectionLost):
+        self.onConnectionLost = onConnectionLost
+        self.delimiter = b'\n'
         self.MAX_LENGTH = 1 << 20
         self.callbacks = {}
         self.finished = None
         # Initialize the event and data buffers
-        self.event = 'message'
-        self.data = ''
+        self.event = b'message'
+        self.data = b''
 
     def lineLengthExceeded(self, line):
         raise NotImplementedError('line too long')
@@ -20,47 +22,46 @@
         self.callbacks[event] = func
         
     def lineReceived(self, line):
-        if line == '':
+        if line == b'':
             # Dispatch event
             self.dispatchEvent()
         else:
             try:
-                field, value = line.split(':', 1)
+                field, value = line.split(b':', 1)
                 # If value starts with a space, strip it.
                 value = lstrip(value)
             except ValueError:
                 # We got a line with no colon, treat it as a field(ignore)
                 return
 
-            if field == '':
+            if field == b'':
                 # This is a comment; ignore
                 pass
-            elif field == 'data':
-                self.data += value + '\n'
-            elif field == 'event':
+            elif field == b'data':
+                self.data += value + b'\n'
+            elif field == b'event':
                 self.event = value
-            elif field == 'id':
+            elif field == b'id':
                 # Not implemented
                 pass
-            elif field == 'retry':
+            elif field == b'retry':
                 # Not implemented
                 pass
 
     def connectionLost(self, reason):
-        if self.finished:
-            self.finished.callback(None)
+        self.onConnectionLost(reason)
 
     def dispatchEvent(self):
         """
         Dispatch the event
         """
         # If last character is LF, strip it.
-        if self.data.endswith('\n'):
+        if self.data.endswith(b'\n'):
             self.data = self.data[:-1]
         if self.event in self.callbacks:
             self.callbacks[self.event](self.data)
-        self.data = ''
-        self.event = 'message'
+        self.data = b''
+        self.event = b'message'
 
 def lstrip(value):
-    return value[1:] if value.startswith(' ') else value
+    return value[1:] if value.startswith(b' ') else value