Creé una figura de matplotlib en unQMainWindow, usando PyQt e intento agregar un botón a la barra de herramientas de matplotlib en mi código. Este es elNavigationToolbar que creé:

Agregué esos botones usando eladdWidget método. Pero, lo que necesito es crear un icono y ponerlo en la barra de herramientas. Esta es una parte de mi 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 algunos códigos y preguntas hechas, y encontré estos, pero no pude lograr lo que necesitaba. Estos son los enlaces que leí:

dale lane - personalizar barra de herramientas de navegación

Modificar la barra de herramientas

Los enlaces solo me dijeron cómo eliminar algunos de ellos, y uno funciona con wx.

¿Cómo puedo agregar estos botones en la barra de herramientas, sin usarQPushbutton oaddWidget métodos ?. espero que me pueda ayudar

------ EDITAR ------

Según el comentario de @three_pineapples, he intentado agregar esta clase a mi 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")

Y creé una instancia haciendo:

self.navigation_toolbar = MyToolbar(), en vez de:

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

Pero recibo este error:

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

He intentado agregar*args ykwargs, pero no sé lo que me estoy perdiendo aquí.

¿Es esta la forma de agregar un botón a la barra de herramientas de matplotlib? Espero que me pueda ayudar.