Update data.py

GeospatialPython · GeospatialPython · commit 727cfa4c6afc · 2015-11-03T22:20:38.000-06:00
diff --git a/srtm/data.py b/srtm/data.py
@@ -156,8 +156,12 @@ def get_image(self, size, latitude_interval, longitude_interval, max_elevation,
         """
         Returns a PIL image.
         """
-        import Image as mod_image
-        import ImageDraw as mod_imagedraw
+        try:
+            import Image as mod_image
+            import ImageDraw as mod_imagedraw
+        except:
+            from PIL import Image as mod_image
+            from PIL import Image as mod_image
 
         if not size or len(size) != 2:
             raise Exception('Invalid size %s' % size)
