Merge pull request #32 from jonathanrocher/feature/add_step_5.1

Feature: add step 5.1

Author: jonathanrocher

stage5.1_fuller_application/README.md

@@ -0,0 +1,3 @@
+# Second real version of the pycasa ETS pyface application!
+Building on the application state 4.2, this version simply adds an editor and a 
+view for a folder to be visualized.

stage5.1_fuller_application/pycasa/__init__.py

@@ -0,0 +1,2 @@
+
+__version__ = "0.0.1"

stage5.1_fuller_application/pycasa/app/__init__.py

@@ -0,0 +1,28 @@
+# coding=utf-8
+""" TaskApplication object for the Pycasa app.
+"""
+import logging
+
+from pyface.tasks.api import TasksApplication, TaskFactory
+from ..ui.tasks.pycasa_task import PycasaTask
+
+logger = logging.getLogger(__name__)
+
+
+class PycasaApplication(TasksApplication):
+    """ An application to say hello.
+    """
+    id = "pycasa_application"
+
+    name = "Pycasa"
+
+    description = "An example Tasks application that explores image files."
+
+    def _task_factories_default(self):
+        return [
+            TaskFactory(
+                id='pycasa.pycasa_task_factory',
+                name="Main Pycasa Task Factory",
+                factory=PycasaTask
+            )
+        ]

stage5.1_fuller_application/pycasa/app/app.py

@@ -0,0 +1,11 @@
+
+from pycasa.app.app import PycasaApplication
+
+
+def main():
+    app = PycasaApplication()
+    app.run()
+
+
+if __name__ == '__main__':
+    main()

stage5.1_fuller_application/pycasa/app/main.py

@@ -0,0 +1,9 @@
+from os.path import expanduser
+from traits.api import Directory, Event, HasStrictTraits
+
+
+class FileBrowser(HasStrictTraits):
+    root = Directory(expanduser("~"))
+
+    #: Item last double-clicked on in the tree view
+    requested_item = Event

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

@@ -0,0 +1,41 @@
+# General imports
+import os
+import PIL.Image
+from PIL.ExifTags import TAGS
+import numpy as np
+
+# ETS imports
+from traits.api import cached_property, Dict, File, HasStrictTraits, Property
+
+SUPPORTED_FORMATS = [".png", ".jpg", ".jpeg"]
+
+
+class ImageFile(HasStrictTraits):
+    """ Model to hold an image file.
+    """
+    filepath = File
+
+    metadata = Property(Dict, depends_on="filepath")
+
+    def to_array(self):
+        file_ext = os.path.splitext(self.filepath)[1].lower()
+        if not self.filepath or file_ext not in SUPPORTED_FORMATS:
+            return np.array([])
+
+        with PIL.Image.open(self.filepath) as img:
+            return np.asarray(img)
+
+    @cached_property
+    def _get_metadata(self):
+        file_ext = os.path.splitext(self.filepath)[1].lower()
+        if not self.filepath or file_ext not in SUPPORTED_FORMATS:
+            return {}
+
+        with PIL.Image.open(self.filepath) as img:
+            exif = img._getexif()
+
+        if exif:
+            return {TAGS[k]: v for k, v in exif.items()
+                    if k in TAGS}
+        else:
+            return {}

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

@@ -0,0 +1,45 @@
+
+# General imports
+import glob
+from os.path import expanduser, isdir, split
+
+import pandas as pd
+
+# ETS imports
+from traits.api import (
+    Directory, HasStrictTraits, Instance, List, observe,
+)
+
+# Local imports
+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))
+
+    def __init__(self, **traits):
+        super(ImageFolder, self).__init__(**traits)
+        if not isdir(self.directory):
+            msg = f"The provided directory isn't a real directory: " \
+                  f"{self.directory}"
+            raise ValueError(msg)
+
+    @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]
+        )