Come estendere una classe in Python?

In Python come puoi estendere una classe? Ad esempio, se ho

color.py

class Color:
    def __init__(self, color):
        self.color = color
    def getcolor(self):
        return self.color

color_extended.py

import Color

class Color:
    def getcolor(self):
        return self.color + " extended!"

Ma questo non funziona ... Mi aspetto che se ci lavoro color_extended.py, quando creo un oggetto a colori e uso la getcolorfunzione, restituirà l'oggetto con la stringa "extended!" alla fine. Inoltre dovrebbe avere gotton l'init dall'importazione.

Supponiamo che Python 3.1

Grazie

Uso:

import color

class Color(color.Color):
    ...

Se questo fosse Python 2.x, dovresti anche derivare color.Colorda object, per renderlo una classe di nuovo stile :

class Color(object):
    ...

Questo non è necessario in Python 3.x.

Un altro modo per estendere (in particolare significa, aggiungere nuovi metodi, non modificare quelli esistenti) le classi, anche quelle incorporate, è usare un preprocessore che aggiunge la capacità di estendersi al di fuori / al di sopra dell'ambito di Python stesso, convertendo l'estensione in normale sintassi di Python prima che Python riesca effettivamente a vederla.

L'ho fatto per estendere la str()classe di Python 2 , ad esempio. str()è un target particolarmente interessante a causa del collegamento implicito a dati citati come 'this'e 'that'.

Ecco un po 'di codice di estensione, in cui l'unica sintassi non Python aggiunta è il extend:testDottedQuadbit:

extend:testDottedQuad
def testDottedQuad(strObject):
    if not isinstance(strObject, basestring): return False
    listStrings = strObject.split('.')
    if len(listStrings) != 4: return False
    for strNum in listStrings:
        try:    val = int(strNum)
        except: return False
        if val < 0: return False
        if val > 255: return False
    return True

Dopodiché posso scrivere nel codice fornito al preprocessore:

if '192.168.1.100'.testDottedQuad():
    doSomething()

dq = '216.126.621.5'
if not dq.testDottedQuad():
    throwWarning();

dqt = ''.join(['127','.','0','.','0','.','1']).testDottedQuad()
if dqt:
    print 'well, that was fun'

Il preprocessore lo mangia, sputa il normale Python senza monkeypatching e Python fa quello che volevo che facesse.

Proprio come un preprocessore ac aggiunge funzionalità a c, così anche un preprocessore Python può aggiungere funzionalità a Python.

L'implementazione del mio preprocessore è troppo grande per una risposta di overflow dello stack, ma per coloro che potrebbero essere interessati, è qui su GitHub.