Add python2 support (#2)

Author: vfdev-5

.travis.yml

@@ -3,16 +3,21 @@ language: python
 matrix:
     include:
         - python: 3.5
+        - python: 2.7
 
 install:
   - pip install numpy Pillow tqdm
+  - if [[ "$TRAVIS_PYTHON_VERSION" == "2.7" ]]; then
+      pip install backports.tempfile;
+    fi
   - python setup.py install
   - pip install flake8 coveralls pytest-cov
 
 # command to run tests
 script:
-  - py.test --cov image_dataset_viz --cov-report term-missing
   - flake8
+  - py.test --cov image_dataset_viz --cov-report term-missing
+
 
 after_success:
   - coveralls

README.md

@@ -5,8 +5,6 @@
 
 Observe dataset of images and targets in few shots
 
-**Python 3 only**
-
 ![VEDAI example](examples/vedai_example.png)
 
 ## Descriptions
@@ -18,11 +16,11 @@ in few shots.
 ## Installation 
 
 ```bash
-python3 setup.py install
+python setup.py install
 ```
 or 
 ```bash
-pip3 install git+https://github.com/vfdev-5/ImageDatasetViz.git
+pip install git+https://github.com/vfdev-5/ImageDatasetViz.git
 ```
 
 ## Usage
@@ -79,6 +77,7 @@ def read_img_fn(img_filepath):
 ```
 and let's say the annotations are just lines with points and a label, e.g. `12 23 34 45 56 67 car`
 ```python
+from pathlib import Path
 import numpy as np
 
 def read_target_fn(target_filepath):

image_dataset_viz/__init__.py

@@ -1,4 +1,4 @@
 
 __version__ = '0.1'
 
-from .dataset_exporter import DatasetExporter, render_datapoint
+from image_dataset_viz.dataset_exporter import DatasetExporter, render_datapoint

image_dataset_viz/dataset_exporter.py

@@ -1,5 +1,10 @@
+from __future__ import division
 import sys
-from pathlib import Path
+
+try:
+    from pathlib import Path
+except ImportError:
+    from pathlib2 import Path
 
 import numpy as np
 from PIL import ImageDraw, ImageFont, Image

setup.py

@@ -1,16 +1,30 @@
-from codecs import open as codecs_open
+import os
+import io
+import re
 from setuptools import setup, find_packages
-from image_dataset_viz import __version__
 
 
-# Get the long description from the relevant file
-with codecs_open('README.md', encoding='utf-8') as f:
-    long_description = f.read()
+def read(*names, **kwargs):
+    with io.open(os.path.join(os.path.dirname(__file__), *names),
+                 encoding=kwargs.get("encoding", "utf8")) as fp:
+        return fp.read()
+
+
+def find_version(*file_paths):
+    version_file = read(*file_paths)
+    version_match = re.search(r"^__version__ = ['\"]([^'\"]*)['\"]", version_file, re.M)
+    if version_match:
+        return version_match.group(1)
+    raise RuntimeError("Unable to find version string.")
+
+
+long_description = read("README.md")
+version = find_version('image_dataset_viz', '__init__.py')
 
 
 setup(
     name="image_dataset_viz",
-    version=__version__,
+    version=version,
     description=u"Observe dataset of images and targets in few shots",
     long_description=long_description,
     author="vfdev-5",
@@ -19,7 +33,8 @@
     install_requires=[
         'numpy',
         'Pillow',
-        'tqdm'
+        'tqdm',
+        'pathlib2;python_version<"3"'
     ],
     license='MIT',
     test_suite="tests",

tests/test_dataset_exporter.py

@@ -1,6 +1,12 @@
 
-import tempfile
-from pathlib import Path
+import sys
+
+if sys.version_info[0] < 3:
+    from backports import tempfile
+    from pathlib2 import Path
+else:
+    import tempfile
+    from pathlib import Path
 
 from unittest import TestCase, main