pyatspi/accessible.py

   1 '''
   2 Creates functions at import time that are mixed into the
   3 Accessibility.Accessible base class to make it more Pythonic.
   4
   5 Based on public domain code originally posted at
   6 U{http://wwwx.cs.unc.edu/~parente/cgi-bin/RuntimeClassMixins}.
   7
   8 @var _ACCESSIBLE_CACHE: Pairs hash values for accessible objects to
   9   L{_PropertyCache} bags. We do not store actual accessibles in the dictionary
  10   because that would +1 their ref counts and cause __del__ to never be called
  11   which is the method we rely on to properly invalidate cache entries.
  12 @type _ACCESSIBLE_CACHE: dictionary
  13 @var _CACHE_LEVEL: Current level of caching enabled. Checked dynamically by
  14   L{_AccessibleMixin}
  15 @type _CACHE_LEVEL: integer
  16
  17 @author: Peter Parente
  18 @organization: IBM Corporation
  19 @copyright: Copyright (c) 2005, 2007 IBM Corporation
  20 @license: LGPL
  21
  22 This library is free software; you can redistribute it and/or
  23 modify it under the terms of the GNU Library General Public
  24 License as published by the Free Software Foundation; either
  25 version 2 of the License, or (at your option) any later version.
  26
  27 This library is distributed in the hope that it will be useful,
  28 but WITHOUT ANY WARRANTY; without even the implied warranty of
  29 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  30 Library General Public License for more details.
  31
  32 You should have received a copy of the GNU Library General Public
  33 License along with this library; if not, write to the
  34 Free Software Foundation, Inc., 59 Temple Place - Suite 330,
  35 Boston, MA 02111-1307, USA.
  36
  37 Portions of this code originally licensed and copyright (c) 2005, 2007
  38 IBM Corporation under the BSD license, available at
  39 U{http://www.opensource.org/licenses/bsd-license.php}
  40 '''
  41 import new
  42 import types
  43 import ORBit
  44 import Accessibility
  45 import constants
  46 import utils
  47 import registry
  48
  49 _ACCESSIBLE_CACHE = {}
  50 _CACHE_LEVEL = None
  51
  52 class _PropertyCache(object):
  53   '''Fixed-size bag class for holding cached values.'''
  54   __slots__ = ('name', 'description', 'rolename')
  55
  56 def getCacheLevel():
  57   '''
  58   Gets the current level of caching.
  59
  60   @return: None indicating no caching is in effect.
  61     L{constants.CACHE_INTERFACES} indicating all interface query results are
  62     cached. L{constants.CACHE_PROPERTIES} indicating all basic accessible
  63     properties are cached.
  64   @rtype: integer
  65   '''
  66   return _CACHE_LEVEL
  67
  68 def setCacheLevel(val):
  69   '''
  70   Sets the desired level of caching for all accessible objects created after
  71   this function is invoked. Immediately clears the current accessible cache.
  72
  73   @param val: None indicating no caching is in effect.
  74     L{constants.CACHE_INTERFACES} indicating all interface query results are
  75     cached. L{constants.CACHE_PROPERTIES} indicating all basic accessible
  76     properties are cached plus all interfaces.
  77   @type val: integer
  78   '''
  79   global _CACHE_LEVEL
  80   if _CACHE_LEVEL != val:
  81     # empty our accessible cache
  82     _ACCESSIBLE_CACHE.clear()
  83     # need to register/unregister for listeners depending on caching level
  84     if val == constants.CACHE_PROPERTIES:
  85       r = registry.Registry()
  86       r.registerEventListener(_updateCache, *constants.CACHE_EVENTS)
  87     else:
  88       r = registry.Registry()
  89       r.deregisterEventListener(_updateCache, *constants.CACHE_EVENTS)
  90   _CACHE_LEVEL = val
  91
  92 def clearCache():
  93   '''Forces a clear of the entire cache.'''
  94   _ACCESSIBLE_CACHE.clear()
  95
  96 def printCache(template='%s'):
  97   '''
  98   Prints the contents of the cache.
  99
 100   @param template: Format string to use when printing
 101   @type template: string
 102   '''
 103   print template % _ACCESSIBLE_CACHE
 104
 105 def _updateCache(event):
 106   '''
 107   Invalidates an entry in the cache when the hash value of a source of an event
 108   matches an entry in the cache.
 109
 110   @param event: One of the L{constants.CACHE_EVENTS} event types
 111   @type event: L{event.Event}
 112   '''
 113   try:
 114     del _ACCESSIBLE_CACHE[hash(event.source)]
 115   except KeyError:
 116     return
 117
 118 def _makeQuery(iid):
 119   '''
 120   Builds a function querying to a specific interface and returns it.
 121
 122   @param iid: Interface identifier to use when querying
 123   @type iid: string
 124   @return: Function querying to the given interface
 125   @rtype: function
 126   '''
 127   def _inner(self):
 128     '''
 129     Queries an object for another interface.
 130
 131     @return: An object with the desired interface
 132     @rtype: object
 133     @raise NotImplementedError: When the desired interface is not supported
 134     '''
 135     try:
 136       i = self._icache[iid]
 137     except KeyError:
 138       # interface not cached
 139       caching = True
 140     except TypeError:
 141       # determine if we're caching
 142       caching = _CACHE_LEVEL is not None
 143       if caching:
 144         # initialize the cache
 145         self._icache = {}
 146     else:
 147       # check if our cached result was an interface, or an indicator that the
 148       # interface is not supported
 149       if i is None:
 150         raise NotImplementedError
 151       else:
 152         return i
 153
 154     try:
 155       # do the query remotely
 156       i = self.queryInterface(iid)
 157     except Exception, e:
 158       raise LookupError(e)
 159     if i is None:
 160       # cache that the interface is not supported
 161       if caching:
 162         self._icache[iid] = None
 163       raise NotImplementedError
 164
 165     # not needed according to ORBit2 spec, but makes Java queries work
 166     # more reliably according to Orca experience
 167     i = i._narrow(i.__class__)
 168     if caching:
 169       # cache the narrow'ed result, but only if we're caching for this object
 170       self._icache[iid] = i
 171     return i
 172
 173   return _inner
 174
 175 def _makeExceptionHandler(func):
 176   '''
 177   Builds a function calling the one it wraps in try/except statements catching
 178   CORBA exceptions.
 179
 180   @return: Function calling the method being wrapped
 181   @rtype: function
 182   '''
 183   def _inner(self, *args, **kwargs):
 184     try:
 185       # try calling the original func
 186       return func(self, *args, **kwargs)
 187     except ORBit.CORBA.NO_IMPLEMENT, e:
 188       # raise Python exception
 189       raise NotImplementedError(e)
 190     except ORBit.CORBA.Exception, e:
 191       # raise Python exception
 192       raise LookupError(e)
 193   return _inner
 194
 195 def _mixInterfaces(cls, interfaces):
 196   '''
 197   Add methods for querying to interfaces other than the base accessible to
 198   the given class.
 199
 200   @param cls: Class to mix interface methods into
 201   @type cls: class
 202   @param interfaces: Classes representing AT-SPI interfaces
 203   @type interfaces: list of class
 204   '''
 205   # create functions in this module for all interfaces listed in constants
 206   for interface in interfaces:
 207     # build name of converter from the name of the interface
 208     name = 'query%s' % utils.getInterfaceName(interface)
 209     # build a function that queries to the given interface
 210     func = _makeQuery(utils.getInterfaceIID(interface))
 211     # build a new method that is a clone of the original function
 212     method = new.function(func.func_code, func.func_globals, name,
 213                           func.func_defaults, func.func_closure)
 214     # add the method to the given class
 215     setattr(cls, name, method)
 216
 217 def _mixExceptions(cls):
 218   '''
 219   Wraps all methods and properties in a class with handlers for CORBA
 220   exceptions.
 221
 222   @param cls: Class to mix interface methods into
 223   @type cls: class
 224   '''
 225   # get a method type as a reference from a known method
 226   method_type = Accessibility.Accessible.getRole.__class__
 227   # loop over all names in the new class
 228   for name in cls.__dict__.keys():
 229     obj = cls.__dict__[name]
 230     # check if we're on a protected or private method
 231     if name.startswith('_'):
 232       continue
 233     # check if we're on a method
 234     elif isinstance(obj, method_type):
 235       # wrap the function in an exception handler
 236       method = _makeExceptionHandler(obj)
 237       # add the wrapped function to the class
 238       setattr(cls, name, method)
 239     # check if we're on a property
 240     elif isinstance(obj, property):
 241       # wrap the getters and setters
 242       if obj.fget:
 243         func = getattr(cls, obj.fget.__name__)
 244         getter = _makeExceptionHandler(func)
 245       else:
 246         getter = None
 247       if obj.fset:
 248         func = getattr(cls, obj.fset.__name__)
 249         setter = _makeExceptionHandler(func)
 250       else:
 251         setter = None
 252       setattr(cls, name, property(getter, setter))
 253
 254 def _mixClass(cls, new_cls, ignore=[]):
 255   '''
 256   Adds the methods in new_cls to cls. After mixing, all instances of cls will
 257   have the new methods. If there is a method name clash, the method already in
 258   cls will be prefixed with '_mix_' before the new method of the same name is
 259   mixed in.
 260
 261   @note: _ is not the prefix because if you wind up with __ in front of a
 262   variable, it becomes private and mangled when an instance is created.
 263   Difficult to invoke from the mixin class.
 264
 265   @param cls: Existing class to mix features into
 266   @type cls: class
 267   @param new_cls: Class containing features to add
 268   @type new_cls: class
 269   @param ignore: Ignore these methods from the mixin
 270   @type ignore: iterable
 271   '''
 272   # loop over all names in the new class
 273   for name, func in new_cls.__dict__.items():
 274     if name in ignore:
 275       continue
 276     if isinstance(func, types.FunctionType):
 277       # build a new function that is a clone of the one from new_cls
 278       method = new.function(func.func_code, func.func_globals, name,
 279                             func.func_defaults, func.func_closure)
 280       try:
 281         # check if a method of the same name already exists in the target
 282         old_method = getattr(cls, name)
 283       except AttributeError:
 284         pass
 285       else:
 286         # rename the old method so we can still call it if need be
 287         setattr(cls, '_mix_'+name, old_method)
 288       # add the clone to cls
 289       setattr(cls, name, method)
 290     elif isinstance(func, staticmethod):
 291       try:
 292         # check if a method of the same name already exists in the target
 293         old_method = getattr(cls, name)
 294       except AttributeError:
 295         pass
 296       else:
 297         # rename the old method so we can still call it if need be
 298         setattr(cls, '_mix_'+name, old_method)
 299       setattr(cls, name, func)
 300     elif isinstance(func, property):
 301       try:
 302         # check if a method of the same name already exists in the target
 303         old_prop = getattr(cls, name)
 304       except AttributeError:
 305         pass
 306       else:
 307         # IMPORTANT: We save the old property before overwriting it, even
 308         # though we never end up calling the old prop from our mixin class.
 309         # If we don't save the old one, we seem to introduce a Python ref count
 310         # problem where the property get/set methods disappear before we can
 311         # use them at a later time. This is a minor waste of memory because
 312         # a property is a class object and we only overwrite a few of them.
 313         setattr(cls, '_mix_'+name, old_prop)
 314       setattr(cls, name, func)
 315
 316 class _AccessibleMixin(object):
 317   '''
 318   Defines methods to be added to the Accessibility.Accessible class. The
 319   features defined here will be added to the Accessible class at run time so
 320   that all instances of Accessible have them (i.e. there is no need to
 321   explicitly wrap an Accessible in this class or derive a new class from it.)
 322
 323   @cvar SLOTTED_CLASSES: Mapping from raw Accessibility class to a new class
 324     having the slots defined by L{SLOTS}
 325   @type SLOTTED_CLASSES: dictionary
 326   @cvar SLOTS: All slots to create
 327   @type SLOTS: tuple
 328   '''
 329   SLOTTED_CLASSES = {}
 330   SLOTS = ('_icache',)
 331
 332   def __new__(cls):
 333     '''
 334     Creates a new class mimicking the one requested, but with an extra _cache
 335     attribute set in the __slots__ tuple. This field can be set to a dictionary
 336     or other object to allow caching to occur.
 337
 338     Note that we can't simply mix __slots__ into this class because __slots__
 339     has an effect only at class creation time.
 340
 341     @param cls: Accessibility object class
 342     @type cls: class
 343     @return: Instance of the new class
 344     @rtype: object
 345     '''
 346     try:
 347       # check if we've already created a new version of the class
 348       new_cls = _AccessibleMixin.SLOTTED_CLASSES[cls]
 349     except KeyError:
 350       # create the new class if not
 351       new_cls = type(cls.__name__, (cls,),
 352                      {'__module__' : cls.__module__,
 353                       '__slots__' : _AccessibleMixin.SLOTS})
 354       _AccessibleMixin.SLOTTED_CLASSES[cls] = new_cls
 355     obj = cls._mix___new__(new_cls)
 356     # don't create the interface cache until we need it
 357     obj._icache = None
 358     return obj
 359
 360   def __del__(self):
 361     '''
 362     Decrements the reference count on the accessible object when there are no
 363     Python references to this object. This provides automatic reference
 364     counting for AT-SPI objects. Also removes this object from the cache if
 365     we're caching properties.
 366     '''
 367     try:
 368       del _ACCESSIBLE_CACHE[hash(self)]
 369     except KeyError:
 370       pass
 371     try:
 372       self.unref()
 373     except Exception:
 374       pass
 375
 376   def __iter__(self):
 377     '''
 378     Iterator that yields one accessible child per iteration. If an exception is
 379     encountered, None is yielded instead.
 380
 381     @return: A child accessible
 382     @rtype: Accessibility.Accessible
 383     '''
 384     for i in xrange(self.childCount):
 385       try:
 386         yield self.getChildAtIndex(i)
 387       except LookupError:
 388         yield None
 389
 390   def __str__(self):
 391     '''
 392     Gets a human readable representation of the accessible.
 393
 394     @return: Role and name information for the accessible
 395     @rtype: string
 396     '''
 397     try:
 398       return '[%s | %s]' % (self.getRoleName(), self.name)
 399     except Exception:
 400       return '[DEAD]'
 401
 402   def __nonzero__(self):
 403     '''
 404     @return: True, always
 405     @rtype: boolean
 406     '''
 407     return True
 408
 409   def __getitem__(self, index):
 410     '''
 411     Thin wrapper around getChildAtIndex.
 412
 413     @param index: Index of desired child
 414     @type index: integer
 415     @return: Accessible child
 416     @rtype: Accessibility.Accessible
 417     '''
 418     n = self.childCount
 419     if index >= n:
 420       raise IndexError
 421     elif index < -n:
 422       raise IndexError:
 423     elif index < 0:
 424       index += n
 425     return self.getChildAtIndex(index)
 426
 427   def __len__(self):
 428     '''
 429     Thin wrapper around childCount.
 430
 431     @return: Number of child accessibles
 432     @rtype: integer
 433     '''
 434     return self.childCount
 435
 436   def _get_name(self):
 437     '''
 438     Gets the name of the accessible from the cache if it is available,
 439     otherwise, fetches it remotely.
 440
 441     @return: Name of the accessible
 442     @rtype: string
 443     '''
 444     if _CACHE_LEVEL != constants.CACHE_PROPERTIES:
 445       return self._get_name()
 446
 447     cache = _ACCESSIBLE_CACHE
 448     h = hash(self)
 449     try:
 450       return cache[h].name
 451     except KeyError:
 452       # no cached info for this object yet
 453       name = self._get_name()
 454       pc = _PropertyCache()
 455       pc.name = name
 456       cache[h] = pc
 457       return name
 458     except AttributeError:
 459       # no cached name for this object yet
 460       name = self._get_name()
 461       cache[h].name = name
 462       return name
 463
 464   name = property(_get_name, Accessibility.Accessible._set_name)
 465
 466   def getRoleName(self):
 467     '''
 468     Gets the unlocalized role name of the accessible from the cache if it is
 469     available, otherwise, fetches it remotely.
 470
 471     @return: Role name of the accessible
 472     @rtype: string
 473     '''
 474     if _CACHE_LEVEL != constants.CACHE_PROPERTIES:
 475       return self._mix_getRoleName()
 476
 477     cache = _ACCESSIBLE_CACHE
 478     h = hash(self)
 479     try:
 480       return cache[h].rolename
 481     except KeyError, e:
 482       # no cached info for this object yet
 483       rolename = self._mix_getRoleName()
 484       pc = _PropertyCache()
 485       pc.rolename = rolename
 486       cache[h] = pc
 487       return rolename
 488     except AttributeError, e:
 489       # no cached name for this object yet
 490       rolename = self._mix_getRoleName()
 491       cache[h].rolename = rolename
 492       return rolename
 493
 494   def _get_description(self):
 495     '''
 496     Gets the description of the accessible from the cache if it is available,
 497     otherwise, fetches it remotely.
 498
 499     @return: Description of the accessible
 500     @rtype: string
 501     '''
 502     if _CACHE_LEVEL != constants.CACHE_PROPERTIES:
 503       return self._get_description()
 504
 505     cache = _ACCESSIBLE_CACHE
 506     h = hash(self)
 507     try:
 508       return cache[h].description
 509     except KeyError:
 510       # no cached info for this object yet
 511       description = self._get_description()
 512       pc = _PropertyCache()
 513       pc.description = description
 514       cache[h] = pc
 515       return description
 516     except AttributeError:
 517       # no cached name for this object yet
 518       description = self._get_description()
 519       cache[h].description = description
 520       return description
 521
 522   description = property(_get_description,
 523                          Accessibility.Accessible._set_description)
 524
 525   def getIndexInParent(self):
 526     '''
 527     Gets the index of this accessible in its parent. Uses the implementation of
 528     this method provided by the Accessibility.Accessible object, but checks the
 529     bound of the value to ensure it is not outside the range of childCount
 530     reported by this accessible's parent.
 531
 532     @return: Index of this accessible in its parent
 533     @rtype: integer
 534     '''
 535     i = self._mix_getIndexInParent()
 536     try:
 537       # correct for out-of-bounds index reporting
 538       return min(self.parent.childCount-1, i)
 539     except AttributeError:
 540       # return sentinel if there is no parent
 541       return -1
 542
 543   def getApplication(self):
 544     '''
 545     Gets the most-parent accessible (the application) of this accessible. Tries
 546     using the getApplication method introduced in AT-SPI 1.7.0 first before
 547     resorting to traversing parent links.
 548
 549     @warning: Cycles involving more than the previously traversed accessible
 550       are not detected by this code.
 551     @return: Application object
 552     @rtype: Accessibility.Application
 553     '''
 554     try:
 555       return self._mix_getApplication()
 556     except AttributeError:
 557       pass
 558     curr = self
 559     try:
 560       while curr.parent is not None and (not curr.parent == curr):
 561         curr = curr.parent
 562       return curr
 563     except Exception:
 564       pass
 565     # return None if the application isn't reachable for any reason
 566     return None
 567
 568 # 1. mix the exception handlers into all queryable interfaces
 569 map(_mixExceptions, constants.ALL_INTERFACES)
 570 # 2. mix the exception handlers into other Accessibility objects
 571 map(_mixExceptions, [Accessibility.StateSet])
 572 # 3. mix the new functions
 573 _mixClass(Accessibility.Accessible, _AccessibleMixin,
 574           ['_get_name', '_get_description'])
 575 # 4. mix queryInterface convenience methods
 576 _mixInterfaces(Accessibility.Accessible, constants.ALL_INTERFACES)