###########################################################################

# 

#  Psyco class support module.

#   Copyright (C) 2001-2002  Armin Rigo et.al.

"""Psyco class support module.

'psyco.classes.psyobj' is an alternate Psyco-optimized root for classes.

Any class inheriting from it or using the metaclass '__metaclass__' might

get optimized specifically for Psyco. It is equivalent to call

psyco.bind() on the class object after its creation.

Importing everything from psyco.classes in a module will import the

'__metaclass__' name, so all classes defined after a

       from psyco.classes import *

will automatically use the Psyco-optimized metaclass.

"""

###########################################################################

__all__ = ['psyobj', 'psymetaclass', '__metaclass__']

from _psyco import compacttype

import core

from types import FunctionType

class psymetaclass(compacttype):

    "Psyco-optimized meta-class. Turns all methods into Psyco proxies."

    def __new__(cls, name, bases, dict):

        bindlist = dict.get('__psyco__bind__')

        if bindlist is None:

            bindlist = [key for key, value in dict.items()

                        if isinstance(value, FunctionType)]

        for attr in bindlist:

            dict[attr] = core.proxy(dict[attr])

        return super(psymetaclass, cls).__new__(cls, name, bases, dict)

psyobj = psymetaclass("psyobj", (), {})

__metaclass__ = psymetaclass