PYQT5 不同tablewidget的右键菜单

我有两个tablewidget

想要两个tablewidget的右键都能生成不同的右键菜单

但目前我已经设置了第一个tablewidget的右键菜单,但第二个怎么设置也没有反应

请问大神们是怎么设置的

if __name__=='__main__':

    app=QApplication(sys.argv)
    mainWindow=mainWindow()
    mainWindow.show()

    #这一段正常显示
    def generateMenu(pos):
        menu = QMenu()
        item1 = menu.addAction('菜单1')
        item2 = menu.addAction('菜单2')

        action = menu.exec_(mainWindow.tableWidget.mapToGlobal(pos))

       if action == item1:
            print('菜单1')
       elif action == item2:
            print('菜单2')

    mainWindow.tableWidget.customContextMenuRequested.connect(generateMenu)  

    #这一段无法显示
    def generateMenu_2(pos):
        menu = QMenu()
        item1 = menu.addAction('菜单3')
        item2 = menu.addAction('菜单4')

        action = menu.exec_(mainWindow.tableWidget_2.mapToGlobal(pos))

       if action == item1:
            print('菜单3')
       elif action == item2:
            print('菜单4')

    mainWindow.tableWidget_2.customContextMenuRequested.connect(generateMenu_2) 
    

    sys.exit(app.exec_())

已经解决了,忘记了创建右键菜单的动作

mainWindow.tableWidget.setContextMenuPolicy(Qt.CustomContextMenu)
mainWindow.tableWidget_2.setContextMenuPolicy(Qt.CustomContextMenu)