Reuse folder model and view from stage 3.

Author: jonathanrocher

stage5.1_fuller_application/pycasa/model/image_folder.py

@@ -1,48 +1,34 @@
-# General imports
-import os
+import glob
+from os.path import expanduser, split
 
 import pandas as pd
-import numpy as np
 
-# ETS imports
-from traits.api import Directory, HasStrictTraits, Instance
+from traits.api import (
+    Directory, HasStrictTraits, Instance, List, observe,
+)
 
-# Local imports
-from .image_file import ImageFile, SUPPORTED_FORMATS
+from pycasa.model.image_file import ImageFile
+
+SUPPORTED_FORMATS = [".png", ".jpg", ".jpeg", ".PNG", ".JPG", ".JPEG"]
 
 
 class ImageFolder(HasStrictTraits):
-    """ Model to hold an image folder.
+    """ Model for a folder of images.
     """
-    path = Directory
-
-    data = Instance(pd.DataFrame)
-
-    def __init__(self, **traits):
-        # Don't forget this!
-        super(ImageFolder, self).__init__(**traits)
-        if not os.path.isdir(self.path):
-            msg = f"Unable to create an ImageFolder from {self.path} since" \
-                  f" it is not a valid directory."
-            raise ValueError(msg)
-
-        self.data = self.to_dataframe()
-
-    def to_dataframe(self):
-        if not self.path:
-            return pd.DataFrame({"filename": [], "Num. faces": []})
-
-        data = []
-        for filename in os.listdir(self.path):
-            file_ext = os.path.splitext(filename)[1].lower()
-            if file_ext in SUPPORTED_FORMATS:
-                filepath = os.path.join(self.path, filename)
-                img_file = ImageFile(filepath=filepath)
-                file_data = {"filename": filename, "Num. faces": np.nan}
-                try:
-                    file_data.update(img_file.metadata)
-                except Exception:
-                    pass
-                data.append(file_data)
-
-        return pd.DataFrame(data)
+    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]
+        )

stage5.1_fuller_application/pycasa/ui/image_folder_editor.py

@@ -30,7 +30,7 @@ def create(self, parent):
     # -------------------------------------------------------------------------
 
     def _get_name(self):
-        return self.obj.path[:25]
+        return self.obj.directory[:25]
 
     def _get_tooltip(self):
-        return self.obj.path
+        return self.obj.directory

stage5.1_fuller_application/pycasa/ui/image_folder_view.py

@@ -1,20 +1,45 @@
-# General imports
-
-# ETS imports
-from traits.api import Instance
+from traits.api import cached_property, Instance, Property
 from traitsui.api import Item, ModelView, View
 from traitsui.ui_editors.data_frame_editor import DataFrameEditor
-# Local imports
-from ..model.image_folder import ImageFolder
+
+from pycasa.model.image_folder import ImageFolder
+
+DISPLAYED_COLUMNS = [
+    'ApertureValue', 'ExifVersion', 'Model', 'Make', 'LensModel', 'DateTime',
+    'ShutterSpeedValue', 'XResolution', 'YResolution'
+]
 
 
 class ImageFolderView(ModelView):
-    """ ModelView for an image folder object.
+    """ ModelView for a folder of images.
     """
     model = Instance(ImageFolder)
 
+    metadata_df = Property(depends_on="model.images.items")
+
     view = View(
-        Item("model.path", style="readonly", show_label=False),
-        Item("model.data", editor=DataFrameEditor(), style="readonly",
-             show_label=False)
+        Item('model.directory', style="readonly", show_label=False),
+        Item(
+            'metadata_df',
+            editor=DataFrameEditor(columns=DISPLAYED_COLUMNS),
+            show_label=False,
+            width=1200,
+        ),
+        resizable=True
     )
+
+    @cached_property
+    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()

stage5.1_fuller_application/pycasa/ui/tasks/pycasa_task.py

@@ -52,7 +52,7 @@ def open_in_central_pane(self, filepath):
             obj = ImageFile(filepath=filepath)
             self.central_pane.edit(obj, factory=ImageFileEditor)
         elif file_ext == "":
-            obj = ImageFolder(path=filepath)
+            obj = ImageFolder(directory=filepath)
             self.central_pane.edit(obj, factory=ImageFolderEditor)
         else:
             print("Unsupported file format: {}".format(file_ext))