Eu criei a figura de um matplotlib em umQMainWindow, usando PyQt e estou tentando adicionar um botão à barra de ferramentas do matplotlib no meu código. Isto é oNavigationToolbar que eu criei:

Eu adicionei esses botões usando oaddWidget método. Mas, o que eu preciso é criar um ícone e colocá-lo na barra de ferramentas. Esta é uma parte do meu código:

class A(QMainWindow):
  def __init__(self):
    QMainWindow.__init__(self)

    self.mainWidget = QWidget()
    self.setCentralWidget(self.mainWidget)
    layout = QVBoxLayout()
    self.mainWidget.setLayout(layout)

    self.figure_canvas = FigureCanvas(Figure())
    layout.addWidget(self.figure_canvas, 10)

    self.axes = self.figure_canvas.figure.add_subplot(111)

    self.navigation_toolbar = NavigationToolbar2(self.figure_canvas, self)
    self.addToolBar(Qt.TopToolBarArea, self.navigation_toolbar) 

    self.btn_selection_tool3 = QPushButton(, "Connect")
    self.navigation_toolbar.addWidget(self.btn_selection_tool3)

    self.btn_selection_tool2 = QPushButton()
    self.navigation_toolbar.addWidget(self.btn_selection_tool2)

    self.btn_showgrid = QPushButton("Show Grid")
    self.navigation_toolbar.addWidget(self.btn_showgrid)

    self.btn_hidegrid = QPushButton("Hide Grid")
    self.navigation_toolbar.addWidget(self.btn_hidegrid)

app = QApplication(sys.argv)
window = A()
window.show()
sys.exit(app.exec_())

Vi alguns códigos e perguntas, e os encontrei, mas não consegui realizar o que precisava. Estes são os links que eu li:

dale lane - barra de ferramentas de navegação personalizada

Modifique a barra de ferramentas

Os links me disseram apenas como excluir alguns deles, e um funciona com o wx.

Como posso adicionar esses botões na barra de ferramentas, sem usarQPushbutton ouaddWidget métodos?. Espero que possa me ajudar

------ EDIT ------

Com base no comentário de @three_pineapples, tentei adicionar esta classe ao meu código:

class MyToolbar(NavigationToolbar2):
  def __init__(self):
    NavigationToolbar2.__init__(self)
    self.iconDir = os.path.join(os.path.dirname(os.path.abspath(__file__)),
        "..", "images", "icons", "")

    self.a = self.addAction(QIcon(iconDir + "BYE2.ico"),
        "Bye", self.bye)
    self.a.setToolTip("GoodBye")

  def bye(self):
    print "See you next time")

E instanciei fazendo:

self.navigation_toolbar = MyToolbar(), ao invés de:

self.navigation_toolbar = NavigationToolbar2(self.figure_canvas, self)

Mas, estou recebendo este erro:

TypeError: __init__() takes at least 3 arguments (1 given)

Eu tentei adicionar*args ekwargs, mas não sei o que estou perdendo aqui.

É assim que se adiciona um botão à barra de ferramentas do matplotlib? Espero que possa me ajudar.