from types import MethodType
import sys

if hasattr(sys, 'version_info') and sys.version_info[0]>=2:
    def MixIn(pyClass, mixInClass, makeAncestor=0):
        """
        Mixes in the attributes of the mixInClass into the pyClass. These attributes are typically methods (but don't have to be). Note that private attributes, denoted by a double underscore, are not mixed in. Collisions are resolved by the mixInClass' attribute overwriting the pyClass'. This gives mix-ins the power to override the behavior of the pyClass.

        After using MixIn(), instances of the pyClass will respond to the messages of the mixInClass.

        An assertion fails if you try to mix in a class with itself.

        The pyClass will be given a new attribute mixInsForCLASSNAME which is a list of all mixInClass' that have ever been installed, in the order they were installed. You may find this useful for inspection and debugging.

        You are advised to install your mix-ins at the start up of your program, prior to the creation of any objects. This approach will result in less headaches. But like most things in Python, you're free to do whatever you're willing to live with.  :-)

        There is a bitchin' article in the Linux Journal, April 2001, "Using Mix-ins with Python" by Chuck Esterbrook which gives a thorough treatment of this topic.

        An example, that resides in Webware, is MiddleKit.Core.ModelUser.py, which install mix-ins for SQL adapters. Search for "MixIn(".

        If makeAncestor is 1, then a different technique is employed: the mixInClass is made the first base class of the pyClass. You probably don't need to use this and if you do, be aware that your mix-in can no longer override attributes/methods in pyClass.

        This function only exists if you are using Python 2.0 or later. Python 1.5.2 has a problem where functions (as in aMethod.im_func) are tied to their class, when in fact, they should be totally generic with only the methods being tied to their class. Apparently this was fixed in Py 2.0.
        """
        assert mixInClass is not pyClass, 'mixInClass = %r, pyClass = %r' % (mixInClass, pyClass)
        if makeAncestor:
            if mixInClass not in pyClass.__bases__:
                pyClass.__bases__ = (mixInClass,) + pyClass.__bases__
        else:
            # Recursively traverse the mix-in ancestor classes in order
            # to support inheritance
            baseClasses = list(mixInClass.__bases__)
            baseClasses.reverse()
            for baseClass in baseClasses:
                MixIn(pyClass, baseClass)

            # Track the mix-ins made for a particular class
            attrName = 'mixInsFor'+pyClass.__name__
            mixIns = getattr(pyClass, attrName, None)
            if mixIns is None:
                mixIns = []
                setattr(pyClass, attrName, mixIns)

            # Make sure we haven't done this before
            # Er, woops. Turns out we like to mix-in more than once sometimes.
            #assert not mixInClass in mixIns, 'pyClass = %r, mixInClass = %r, mixIns = %r' % (pyClass, mixInClass, mixIns)

            # Record our deed for future inspection
            mixIns.append(mixInClass)

            # Install the mix-in methods into the class
            for name in dir(mixInClass):
                if not name.startswith('__'): # skip private members
                    member = getattr(mixInClass, name)
                    if type(member) is MethodType:
                        member = member.im_func
                    setattr(pyClass, name, member)