Merge pull request #80 from tubular/fix/sparkly/disable-spark3-kafka-…

…read

FIX disable kafka reading for spark 3

sparkly/reader.py

@@ -22,8 +22,6 @@
 except ImportError:
     from urlparse import urlparse, parse_qsl
 
-from pyspark.streaming.kafka import KafkaUtils, OffsetRange
-
 from sparkly.utils import parse_schema
 
 
@@ -247,6 +245,11 @@ def kafka(self,
         Raises:
             InvalidArgumentError
         """
+        try:
+            from pyspark.streaming.kafka import KafkaUtils, OffsetRange
+        except ImportError:
+            raise NotImplementedError('Reading from kafka not supported')
+
         if not key_deserializer or not value_deserializer or not schema:
             raise InvalidArgumentError('You should specify all of parameters:'
                                        '`key_deserializer`, `value_deserializer` and `schema`')