Domanda Python

Il mio modulo brt.py contiene questa classe:

class TreeNode:
def __init__(self, key, val, parent, left=None, right=None):
self.key = key
self.payload = val
self.leftChild = left
self.rightChild = right
self.parent = parent


Quando apro il python succede questo:

>>> import brt //importa senza problemi, quindi la classe e' definita
>>> nodo = TreeNode(1, 2)

Traceback (most recent call last):
File "<pyshell#2>", line 1, in <module>
nodo = TreeNode(1, 2)
NameError: name 'TreeNode' is not defined



Lo fa anche se definisco un oggetto di un'altra classe. Sicuramente e' una bischerata ma non riesco a vederlo. Ty.

mi pare che non gli passi il campo "parent"

Non e' quello il problema, avevo gia' provato, ho fatto copy/paste sul pezzo sbagliato:D

Ho risolto, i metodi in un modulo sono attributi dell'oggetto modulo, dopo l'import dovevo fare nodo = brt.TreeNode(1, 2, 3):rain: