You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

55 lines
2.0 KiB

4 years ago
  1. """Robust apply mechanism
  2. Provides a function "call", which can sort out
  3. what arguments a given callable object can take,
  4. and subset the given arguments to match only
  5. those which are acceptable.
  6. """
  7. import sys
  8. if sys.hexversion >= 0x3000000:
  9. im_func = '__func__'
  10. im_self = '__self__'
  11. im_code = '__code__'
  12. func_code = '__code__'
  13. else:
  14. im_func = 'im_func'
  15. im_self = 'im_self'
  16. im_code = 'im_code'
  17. func_code = 'func_code'
  18. def function( receiver ):
  19. """Get function-like callable object for given receiver
  20. returns (function_or_method, codeObject, fromMethod)
  21. If fromMethod is true, then the callable already
  22. has its first argument bound
  23. """
  24. if hasattr(receiver, '__call__'):
  25. # Reassign receiver to the actual method that will be called.
  26. if hasattr( receiver.__call__, im_func) or hasattr( receiver.__call__, im_code):
  27. receiver = receiver.__call__
  28. if hasattr( receiver, im_func ):
  29. # an instance-method...
  30. return receiver, getattr(getattr(receiver, im_func), func_code), 1
  31. elif not hasattr( receiver, func_code):
  32. raise ValueError('unknown reciever type %s %s'%(receiver, type(receiver)))
  33. return receiver, getattr(receiver,func_code), 0
  34. def robustApply(receiver, *arguments, **named):
  35. """Call receiver with arguments and an appropriate subset of named
  36. """
  37. receiver, codeObject, startIndex = function( receiver )
  38. acceptable = codeObject.co_varnames[startIndex+len(arguments):codeObject.co_argcount]
  39. for name in codeObject.co_varnames[startIndex:startIndex+len(arguments)]:
  40. if name in named:
  41. raise TypeError(
  42. """Argument %r specified both positionally and as a keyword for calling %r"""% (
  43. name, receiver,
  44. )
  45. )
  46. if not (codeObject.co_flags & 8):
  47. # fc does not have a **kwds type parameter, therefore
  48. # remove unacceptable arguments.
  49. named = dict([(k,v) for k,v in named.items() if k in acceptable])
  50. return receiver(*arguments, **named)