Как описать функцию смены цвета кнопки по условию в PyQT?

Question

[Владислав]

Есть фронт сделанный на виджетах PyQT

В этом коде описывется запуск браузера и функции эмуляции кликов внутри страницы, вызываемые нажатиями кнопок. Для упрощения тут порезал JS скрипты и количество кнопок

from PyQt5 import QtWidgets
        from interface import Ui_MainWindow
        from selenium import webdriver
        from selenium.webdriver.common.keys import Keys
        from selenium.webdriver.common.by import By
        from selenium.webdriver.common.action_chains import ActionChains
        from selenium.webdriver.support import expected_conditions
        from selenium.webdriver.support.wait import WebDriverWait
        
        driver = webdriver.Chrome(executable_path='C:\chromedriver.exe')
        url = "https://www.google.com/"
        driver.get(url) 
        
        class ExampleApp(QtWidgets.QMainWindow, Ui_MainWindow):
            def __init__(self):  #связываем функции с UI кнопками
                super().__init__()
                self.setupUi(self)
                self.pushButton.clicked.connect(self.NoAdress)
                self.pushButton_2.clicked.connect(self.SpecTransp)
                self.pushButton_3.clicked.connect(self.NoSvetofor)
        
            def NoAdress(self):          #функция для кнопки 1
                driver.execute_script('''some js code 1''')
                         
            def SpecTransp(self): #функция для кнопки 2
                driver.execute_script('''some js code 2''')
        
            def NoSvetofor(self): #функция для кнопки 3
                driver.execute_script('''some js code 3''')
            
        if __name__ == '__main__':
            app = QtWidgets.QApplication([])
            window = ExampleApp()
            window.show()
            app.exec_()

Втыкание этого кода в одну из функций кнопок меняет ее цвет

self.pushButton.setStyleSheet('QPushButton {background-color: red; color: white;}')

Как описать функцию, которая будет менять цвет одной из кнопок в зависимости от заданного условия? Т.е. допустим если x == 1: то button_1 выделяется красным, если x = 2: то кнопка 2 выделяется красным, x == 3: button_3 и т.д.