Add a model and model-view for image folder

Author: siddhantwahal

stage3.0_first_views/pycasa/model/image_folder.py

@@ -0,0 +1,34 @@
+import glob
+from os.path import expanduser, split
+
+import pandas as pd
+
+from traits.api import (
+    Directory, HasStrictTraits, Instance, List, observe,
+)
+
+from pycasa.model.image_file import ImageFile
+
+SUPPORTED_FORMATS = [".png", ".jpg", ".jpeg", ".PNG", ".JPG", ".JPEG"]
+
+
+class ImageFolder(HasStrictTraits):
+    """ Model for a folder of images.
+    """
+    directory = Directory(expanduser("~"))
+
+    images = List(Instance(ImageFile))
+
+    @observe("directory")
+    def _get_images(self, event):
+        self.images = [
+            ImageFile(filepath=file)
+            for fmt in SUPPORTED_FORMATS
+            for file in glob.glob(f"{self.directory}/*{fmt}")
+        ]
+
+    def create_metadata_df(self):
+        return pd.DataFrame(
+            [img.metadata for img in self.images],
+            index=[split(img.filepath)[1] for img in self.images]
+        )

stage3.0_first_views/pycasa/ui/image_folder_view.py

@@ -0,0 +1,50 @@
+from traits.api import Instance, Property
+from traitsui.api import Item, ModelView, View
+from traitsui.ui_editors.data_frame_editor import DataFrameEditor
+
+from pycasa.model.image_folder import ImageFolder
+
+
+class ImageFolderView(ModelView):
+    """ ModelView for a folder of images.
+    """
+    model = Instance(ImageFolder)
+
+    metadata_df = Property(depends_on="model.images.items")
+
+    view = View(
+        Item('model.directory', style="readonly", show_label=False),
+        Item(
+            'metadata_df',
+            editor=DataFrameEditor(
+                columns=[
+                    'ApertureValue',
+                    'ExifVersion',
+                    'Model',
+                    'Make',
+                    'LensModel',
+                    'DateTime',
+                    'ShutterSpeedValue',
+                    'XResolution',
+                    'YResolution'
+                ]
+            ),
+            show_label=False
+        ),
+        resizable=True
+    )
+
+    def _get_metadata_df(self):
+        return self.model.create_metadata_df()
+
+
+if __name__ == '__main__':
+    from os.path import dirname, join
+    import ets_tutorial
+
+    TUTORIAL_DIR = dirname(ets_tutorial.__file__)
+    SAMPLE_IMG_DIR = join(TUTORIAL_DIR, "..", "sample_images")
+
+    image_file = ImageFolder(directory=SAMPLE_IMG_DIR)
+    view = ImageFolderView(model=image_file)
+    view.configure_traits()

stage3.0_first_views/pycasa/ui/test_image_folder_view.py

@@ -0,0 +1,20 @@
+import unittest
+from os.path import dirname, join
+
+from traitsui.testing.api import UITester, IsVisible
+
+import ets_tutorial
+from pycasa.model.image_folder import ImageFolder
+from image_folder_view import ImageFolderView
+
+TUTORIAL_DIR = dirname(ets_tutorial.__file__)
+SAMPLE_IMG_DIR = join(TUTORIAL_DIR, "..", "sample_images")
+
+
+class TestImageFolderView(unittest.TestCase):
+    def test_smoke(self):
+        view = ImageFolderView(model=ImageFolder(directory=SAMPLE_IMG_DIR))
+        tester = UITester()
+        with tester.create_ui(view) as ui:
+            df = tester.find_by_name(ui, "metadata_df")
+            df.inspect(IsVisible())