utils/ui.py - master - vstor/DroidTool

utils/ui.py
85 строк3 KB
vstor
refactor and add logging
07 янв 2025, 13:31
07 янв 2025, 13:3197d1c62
import utils.fastboot as fastboot
from res.gui import Ui_MainWindow
from PyQt6 import QtWidgets
import sys
# import os
# import subprocess
import utils.adb as adb
from utils.log import logging

class MainWindow(QtWidgets.QMainWindow):
    def check_FB(self): #Поиск усройств в режиме fastboot
        if len(fastboot.check()):
            self.ui.StatusConn_FB.setText(fastboot.check()[0])
            self.ui.RebootBtn_FB.setEnabled(True)
            self.ui.SelectorTypeReboot_FB.setEnabled(True)
            self.ui.Part.setEnabled(True)
        else:
            self.ui.StatusConn_FB.setText("ОТКЛЮЧЕНО")
            self.ui.RebootBtn_FB.setEnabled(False)
            self.ui.SelectorTypeReboot_FB.setEnabled(False)
            self.ui.Part.setEnabled(False)
        logging.debug(f"{fastboot.check()}")

    def reboot_FB(self): #перезапуск устройства в режиме fastboot
        fastboot.reboot(self.ui.SelectorTypeReboot_FB.currentText())

    def reboot(self): #перезапуск устройства
        adb.reboot(self.ui.SelectorTypeReboot.currentText())
    def install(self): #установка приложения на устройство
        path = QtWidgets.QFileDialog.getOpenFileName()[0]
        adb.install(path)
    def remove(self): #удаления приложения
        appr = self.ui.RemoveSelector.currentText()
        adb.remove(appr)

    def check(self): #поиск усройств в режиме adb
        if len(adb.check()) == 4:
            self.ui.StatusConn.setText("ОТКЛЮЧЕНО")
            self.ui.RebootBtn.setEnabled(False)
            self.ui.SelectorTypeReboot.setEnabled(False)
            self.ui.InstallBtn.setEnabled(False)
        else:
            self.ui.StatusConn.setText(adb.check()[4])
            self.ui.RebootBtn.setEnabled(True)
            self.ui.SelectorTypeReboot.setEnabled(True)
            self.ui.InstallBtn.setEnabled(True)
            self.ui.RemoveSelector.setEnabled(True)
            self.ui.DelButton.setEnabled(True)
            for i in adb.list_apps():
                self.ui.RemoveSelector.addItem(i.split(":")[1])
        logging.debug(adb.check())
    def flash(self):
        path = QtWidgets.QFileDialog.getOpenFileName()[0]
        part = self.ui.Part.text()
        if part != "":
            fastboot.flash(path)
        else:
            pass
    def allowflash(self):
        if self.ui.Part.text() != "":
            self.ui.Flash.setEnabled(True)
        else:
            self.ui.Flash.setEnabled(False)
            
        
    def __init__(self):
        super(MainWindow,self).__init__()
        self.ui = Ui_MainWindow()
        self.ui.setupUi(self)

        self.ui.SearchBtn.clicked.connect(self.check)
        self.ui.InstallBtn.clicked.connect(self.install)
        self.ui.RebootBtn.clicked.connect(self.reboot)
        self.ui.SearchBtn_FB.clicked.connect(self.check_FB)
        self.ui.RebootBtn_FB.clicked.connect(self.reboot_FB)
        self.ui.DelButton.clicked.connect(self.remove)
        self.ui.Flash.clicked.connect(self.flash)
        self.ui.Part.textEdited.connect(self.allowflash)
        logging.info("ui init")
def runui():
    logging.info("ui started succes")
    app = QtWidgets.QApplication([])
    application = MainWindow()
    application.show()
    sys.exit(app.exec())