Python3 compatibility fixes

enable/abstract_window.py

@@ -1,5 +1,6 @@
 
 # Major library imports
+import six.moves as sm
 from numpy import dot
 
 # Enthought library imports
@@ -134,7 +135,7 @@ def _init_gc(self):
             gc.clear(self.bgcolor_)
         else:
             # Fixme: should use clip_to_rects
-            update_union = reduce(union_bounds, self._update_region)
+            update_union = sm.reduce(union_bounds, self._update_region)
             gc.clip_to_rect(*update_union)
         return
 
@@ -264,7 +265,7 @@ def _handle_key_event(self, event_type, event):
             if history is not None and len(history) > 0:
                 # Assemble all the transforms
                 transforms = [c.get_event_transform() for c in history]
-                total_transform = reduce(dot, transforms[::-1])
+                total_transform = sm.reduce(dot, transforms[::-1])
                 key_event.push_transform(total_transform)
             elif self.mouse_owner_transform is not None:
                 key_event.push_transform(self.mouse_owner_transform)
@@ -313,7 +314,7 @@ def _handle_mouse_event(self, event_name, event, set_focus=False):
             if history is not None and len(history) > 0:
                 # Assemble all the transforms
                 transforms = [c.get_event_transform() for c in history]
-                total_transform = reduce(dot, transforms[::-1])
+                total_transform = sm.reduce(dot, transforms[::-1])
                 mouse_event.push_transform(total_transform)
             elif self.mouse_owner_transform is not None:
                 mouse_event.push_transform(self.mouse_owner_transform)

enable/base.py

@@ -24,6 +24,7 @@
 from __future__ import generators
 
 # Major library imports
+import six.moves as sm
 
 # Enthought library imports
 from traits.api import TraitError
@@ -82,11 +83,11 @@ def str_to_font ( object, name, value ):
         face_name  = []
         for word in value.split():
             lword = word.lower()
-            if font_families.has_key( lword ):
+            if lword in font_families:
                family = font_families[ lword ]
-            elif font_styles.has_key( lword ):
+            elif lword in font_styles:
                style = font_styles[ lword ]
-            elif font_weights.has_key( lword ):
+            elif lword in font_weights:
                weight = font_weights[ lword ]
             elif lword == 'underline':
                underline = 1
@@ -103,8 +104,8 @@ def str_to_font ( object, name, value ):
                     underline = underline)
     except:
         pass
-    raise TraitError, ( object, name, 'a font descriptor string',
-                        repr( value ) )
+    raise TraitError(( object, name, 'a font descriptor string',
+                        repr( value ) ))
 
 str_to_font.info = ( "a string describing a font (e.g. '12 pt bold italic " +
                      "swiss family Arial' or 'default 12')" )
@@ -239,7 +240,7 @@ def send_event_to ( components, event_name, event ):
         setattr( component, pre_event_name, event )
         if event.handled:
             return len( components )
-    for i in xrange( len( components ) - 1, -1, -1 ):
+    for i in sm.range( len( components ) - 1, -1, -1 ):
         setattr( components[i], event_name, event )
         if event.handled:
             return i

enable/canvas.py

@@ -81,7 +81,7 @@ def remove(self, *components):
                 component.container = None
                 self._components.remove(component)
             else:
-                raise RuntimeError, "Unable to remove component from container."
+                raise RuntimeError("Unable to remove component from container.")
 
             # Check to see if we need to compact.
             x, y, x2, y2 = self._bounding_box

enable/compass.py

@@ -1,6 +1,8 @@
 
 from __future__ import with_statement
 
+import six
+
 from numpy import array, pi
 
 # Enthought library imports
@@ -101,7 +103,7 @@ def normal_left_down(self, event):
                   "w": array((-far, -half_width, -near, half_width)) }
         if self.enable_center:
             rects["c"] = array((-near, -near, near, near))
-        for direction, rect in rects.items():
+        for direction, rect in six.iteritems(rects):
             if (rect[0] <= x <= rect[2]) and (rect[1] <= y <= rect[3]):
                 self.event_state = "clicked"
                 self.clicked = direction

enable/component.py

@@ -376,21 +376,24 @@ def __init__(self, **traits):
         # may override the bulk default.
         padding = traits.pop('padding', None)
         padding_traits = {}
-        for name in traits.keys():
-            # Use .keys() so we can modify the dict during iteration safely.
-            if name in ['padding_top', 'padding_bottom', 'padding_left',
-                'padding_right']:
+        for name in ['padding_top',
+                     'padding_bottom',
+                     'padding_left',
+                     'padding_right']:
+            try:
                 padding_traits[name] = traits.pop(name)
+            except KeyError:
+                pass
 
-        if traits.has_key("container"):
+        if "container" in traits:
             # After the component is otherwise configured, make sure our
             # container gets notified of our being added to it.
             container = traits.pop("container")
-            super(Component,self).__init__(**traits)
+            super(Component, self).__init__(**traits)
             self._set_padding_traits(padding, padding_traits)
             container.add(self)
         else:
-            super(Component,self).__init__(**traits)
+            super(Component, self).__init__(**traits)
             self._set_padding_traits(padding, padding_traits)
         return
 

enable/constraints_container.py

@@ -4,6 +4,8 @@
 #------------------------------------------------------------------------------
 from collections import deque
 
+import six
+
 # traits imports
 from traits.api import Any, Bool, Callable, Dict, Either, Instance, List, \
     Property
@@ -243,7 +245,7 @@ def __components_changed(self, new):
         """ Make sure components that are added can be used with constraints.
         """
         # Clear the component maps
-        for key, item in self._component_map.iteritems():
+        for key, item in six.iteritems(self._component_map):
             item.on_trait_change(self._component_size_hint_changed,
                                  'layout_size_hint', remove=True)
         self._component_map = {}
@@ -293,7 +295,7 @@ def _build_layout_table(self):
         zero_offset = (0, 0)
         offset_table = [zero_offset]
         layout_table = []
-        queue = deque((0, child) for child in self._component_map.itervalues())
+        queue = deque((0, child) for child in six.itervalues(self._component_map))
 
         # Micro-optimization: pre-fetch bound methods and store globals
         # as locals. This method is not on the code path of a resize
@@ -321,7 +323,7 @@ def _build_layout_table(self):
                 running_index += 1
                 if isinst(item, Container_):
                     if item.transfer_layout_ownership(self):
-                        for child in item._component_map.itervalues():
+                        for child in six.itervalues(item._component_map):
                             push((running_index, child))
 
         return offset_table, layout_table

enable/container.py

@@ -101,7 +101,7 @@ def __init__(self, *components, **traits):
         Component.__init__(self, **traits)
         for component in components:
             self.add(component)
-        if "bounds" in traits.keys() and "auto_size" not in traits.keys():
+        if "bounds" in traits and "auto_size" not in traits:
             self.auto_size = False
 
         if 'intercept_events' in traits:
@@ -130,7 +130,7 @@ def remove(self, *components):
                 component.container = None
                 self._components.remove(component)
             else:
-                raise RuntimeError, "Unable to remove component from container."
+                raise RuntimeError("Unable to remove component from container.")
 
             # Check to see if we need to compact.
             if self.auto_size:

enable/coordinate_box.py

@@ -1,4 +1,6 @@
 
+import six
+
 # Enthought library imports
 from traits.api import HasTraits, Enum, Instance, Property, Tuple
 
@@ -162,7 +164,7 @@ def _get_width(self):
 
     def _set_width(self, val):
 
-        if isinstance(val, basestring):
+        if isinstance(val, six.string_types):
             try:
                 val = float(val)
             except:
@@ -177,7 +179,7 @@ def _get_height(self):
         return self.bounds[1]
 
     def _set_height(self, val):
-        if isinstance(val, basestring):
+        if isinstance(val, six.string_types):
             try:
                 val = float(val)
             except:

enable/enable_traits.py

@@ -2,6 +2,8 @@
 Define the base Enable object traits
 """
 
+import six
+
 # Major library imports
 from numpy import arange, array
 
@@ -75,7 +77,7 @@
     'dot':       array( [ 2.0, 2.0 ] ),
     'long dash': array( [ 9.0, 5.0 ] )
 }
-__line_style_trait_map_keys = __line_style_trait_values.keys()
+__line_style_trait_map_keys = list(six.iterkeys(__line_style_trait_values))
 LineStyleEditor = EnumEditor( values=__line_style_trait_map_keys)
 
 def __line_style_trait( value='solid', **metadata ):

enable/events.py

@@ -3,6 +3,7 @@
 
 For a list of all the possible event suffixes, see interactor.py.
 """
+import six.moves as sm
 
 # Major library imports
 from numpy import array, dot
@@ -103,7 +104,7 @@ def net_transform(self):
         if len(self._transform_stack) == 0:
             return affine.affine_identity()
         else:
-            return reduce(dot, self._transform_stack[::-1])
+            return sm.reduce(dot, self._transform_stack[::-1])
 
     def current_pointer_position(self):
         """

enable/example_support.py

@@ -14,7 +14,7 @@
 if not ETSConfig.toolkit:
     for toolkit, toolkit_module in (('wx', 'wx'), ('qt4', 'PyQt4')):
         try:
-            exec "import " + toolkit_module
+            exec("import " + toolkit_module)
             ETSConfig.toolkit = toolkit
             break
         except ImportError:

enable/gadgets/vu_meter.py

@@ -1,6 +1,9 @@
 
 import math
 
+import six
+from six.moves as sm
+
 from traits.api import Float, Property, List, Str, Range
 from enable.api import Component
 from kiva.trait_defs.kiva_font_trait import KivaFont
@@ -37,7 +40,7 @@ class VUMeter(Component):
 
     # Values of the percentage-based ticks; these are drawn and labeled along
     # the bottom of the curve axis.
-    percent_ticks = List(range(0, 101, 20))
+    percent_ticks = List(list(sm.range(0, 101, 20)))
 
     # Text to write in the middle of the VU Meter.
     text = Str("VU")

enable/graphics_context.py

@@ -27,7 +27,7 @@ class EnableGCMixin(object):
     window = None  #Instance(AbstractWindow)
 
     def __init__(self, *args, **kwargs):
-        if kwargs.has_key("window"):
+        if "window" in kwargs:
             self.window = kwargs.pop("window")
         super(EnableGCMixin, self).__init__(*args, **kwargs)
         return
@@ -59,8 +59,8 @@ def clear_clip_region(self, color, update_region):
         return
 
     def alpha(self, alpha):
-        raise NotImplementedError, \
-            "The alpha() method is not compatible with DisplayPDF; use clear() instead."
+        raise NotImplementedError(
+            "The alpha() method is not compatible with DisplayPDF; use clear() instead.")
 
     def stretch_draw(self, image, x, y, dx, dy):
         "Draws an image 'stretched' to fit a specified area"

enable/layout/ab_constrainable.py

@@ -4,7 +4,10 @@
 #------------------------------------------------------------------------------
 from abc import ABCMeta
 
+import six
 
+
+@six.add_metaclass(ABCMeta)
 class ABConstrainable(object):
     """ An abstract base class for objects that can be laid out using
     layout helpers.
@@ -14,5 +17,3 @@ class ABConstrainable(object):
     which are `LinearSymbolic` instances.
 
     """
-    __metaclass__ = ABCMeta
-