cleanup (#63)

pysparkling/streaming/context.py

@@ -149,22 +149,6 @@ def remember(self, duration):
         """Provided for compatibility, but does nothing here."""
         pass
 
-    def start(self):
-        """Start processing streams."""
-
-        def cb():
-            time_ = time.time()
-            log.debug('Step {}'.format(time_))
-
-            # run a step on all streams
-            for d in self._dstreams:
-                d._step(time_)
-
-        self._pcb = PeriodicCallback(cb, self.batch_duration * 1000.0)
-        self._pcb.start()
-        self._on_stop_cb.append(self._pcb.stop)
-        StreamingContext._activeContext = self
-
     def socketBinaryStream(self, hostname, port, length):
         """Create a TCP socket server for binary input.
 
@@ -203,6 +187,22 @@ def socketTextStream(self, hostname, port):
         self._on_stop_cb.append(tcp_text_stream.stop)
         return DStream(tcp_text_stream, self, deserializer)
 
+    def start(self):
+        """Start processing streams."""
+
+        def cb():
+            time_ = time.time()
+            log.debug('Step {}'.format(time_))
+
+            # run a step on all streams
+            for d in self._dstreams:
+                d._step(time_)
+
+        self._pcb = PeriodicCallback(cb, self.batch_duration * 1000.0)
+        self._pcb.start()
+        self._on_stop_cb.append(self._pcb.stop)
+        StreamingContext._activeContext = self
+
     def stop(self, stopSparkContext=True, stopGraceFully=False):
         """Stop processing streams.
 

pysparkling/streaming/tcpstream.py

@@ -24,9 +24,6 @@ def __call__(self, data):
 
 class TCPTextStream(TCPServer):
     def __init__(self, delimiter=b'\n'):
-        if TCPServer is False:
-            log.error('Run \'pip install tornado\' to use TCPStream.')
-
         super(TCPTextStream, self).__init__()
         self.delimiter = delimiter
         self.buffer = []
@@ -58,9 +55,6 @@ class TCPBinaryStream(TCPServer):
     """
 
     def __init__(self, length=None):
-        if TCPServer is False:
-            log.error('Run \'pip install tornado\' to use TCPStream.')
-
         super(TCPBinaryStream, self).__init__()
         self.length = length
         self.buffer = []