Блог

Я пытаюсь создать своего рода файловый браузер для изображений

Он открывает изображения, когда я нажимаю кнопки next ot previous и показывает список файлов в указанном каталоге. Однако он выделяет файлы в listview только щелчком мыши.

Как я могу выделить текущий файл в области listview (слева) при нажатии кнопки (next или previous)?

вот мой код:

 from PyQt5 import QtCore, QtGui, QtWid&ets
from PyQt5.QtWid&ets import QFileDialo&
from PyQt5.QtWid&ets import *
from PyQt5.QtCore import *
from PyQt5.QtGui import QColor,QTextCharFormat
import os


class Ui_Dialo&(object):
    def setupUi(self, Dialo&):
        self.filename=[]
        self.path=[]
        self.current_im=[]
        Dialo&.setObjectName("Dialo&")
        Dialo&.resize(832, 586)
        self.toolButton = QtWid&ets.QToolButton(Dialo&)
        self.toolButton.setGeometry(QtCore.QRect(90, 70, 25, 19))
        self.toolButton.setPopupMode(QtWid&ets.QToolButton.DelayedPopup)
        self.toolButton.setObjectName("toolButton")

        self.label = QtWid&ets.QLabel(Dialo&)
        self.label.setGeometry(QtCore.QRect(20, 70, 81, 16))
        self.label.setObjectName("label")

        #self.lbl = QLabel(self)
        self.qle = QLineEdit(Dialo&)
        self.qle.setGeometry(QtCore.QRect(120, 20, 191, 16))
        self.t=self.qle.textChan&ed[str]
        self.qle.textChan&ed[str].connect(self.onChan&ed)

        #self.lbl.move(60, 40)

        self.label_2 = QtWid&ets.QLabel(Dialo&)
        self.label_2.setGeometry(QtCore.QRect(10, 20, 91, 16))
        self.label_2.setObjectName("label_2")

        self.toolButton_2 = QtWid&ets.QToolButton(Dialo&)
        self.toolButton_2.setGeometry(QtCore.QRect(500, 20, 31, 19))
        self.toolButton_2.setObjectName("toolButton_2")

        self.checkBox = QtWid&ets.QCheckBox(Dialo&)
        self.checkBox.setGeometry(QtCore.QRect(730, 260, 91, 21))
        self.checkBox.setObjectName("checkBox")

        #self.scrollArea = QtWid&ets.QScrollArea(Dialo&)
        #self.scrollArea.setGeometry(QtCore.QRect(10, 110, 261, 461))
        #self.scrollArea.setWid&etResizable(True)
        #self.scrollArea.setObjectName("scrollArea")

        #self.scrollAreaWid&etContents = Wid&et(Dialo&)
        #self.scrollAreaWid&etContents.setGeometry(QtCore.QRect(10, 110, 261, 461))
        #self.scrollAreaWid&etContents.setObjectName("scrollAreaWid&etContents")

        #self.verticalScrollBar = QtWid&ets.QScrollBar(self.scrollAreaWid&etContents)
        #self.verticalScrollBar.setGeometry(QtCore.QRect(240, 0, 16, 451))
        #self.verticalScrollBar.setOrientation(QtCore.Qt.Vertical)
        #self.verticalScrollBar.setObjectName("verticalScrollBar")

        #self.scrollArea.setWid&et(self.scrollAreaWid&etContents)

        #self.hlay = QHBoxLayout(Dialo&)
        self.listview = QListView(Dialo&)
        self.fileModel = QFileSystemModel(Dialo&)
        #self.hlay.addWid&et(self.listview)
        #self.hlay.setGeometry(QtCore.QRect(10, 110, 261, 461))
        self.listview.setGeometry(QtCore.QRect(10, 110, 261, 461))

        self.label_3 = QtWid&ets.QLabel(Dialo&)
        self.label_3.setGeometry(QtCore.QRect(290, 140, 401, 361))
        self.label_3.setText("")
        #self.label_3.setPixmap(QtGui.QPixmap("C:/Users/Ni&el/Desktop/Waste_classification/Waste-Classification-master/dataset/train_set/1/2а (5).jp&"))
        #self.label_3.setScaledContents(True)
        self.label_3.setObjectName("label_3")

        self.comboBox = QtWid&ets.QComboBox(Dialo&)
        self.comboBox.setGeometry(QtCore.QRect(720, 140, 69, 22))
        self.comboBox.setObjectName("comboBox")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")
        self.comboBox.addItem("")

        self.label_4 = QtWid&ets.QLabel(Dialo&)
        self.label_4.setGeometry(QtCore.QRect(720, 120, 51, 16))
        self.label_4.setObjectName("label_4")
        self.label_5 = QtWid&ets.QLabel(Dialo&)
        self.label_5.setGeometry(QtCore.QRect(390, 20, 91, 16))
        self.label_5.setObjectName("label_5")

        self.comboBox_2 = QtWid&ets.QComboBox(Dialo&)
        self.comboBox_2.setGeometry(QtCore.QRect(720, 210, 81, 22))
        self.comboBox_2.setObjectName("comboBox_2")
        self.comboBox_2.addItem("")
        self.comboBox_2.addItem("")
        self.comboBox_2.addItem("")

        self.label_6 = QtWid&ets.QLabel(Dialo&)
        self.label_6.setGeometry(QtCore.QRect(720, 190, 47, 13))
        self.label_6.setObjectName("label_6")

        self.pushButton = QtWid&ets.QPushButton(Dialo&)
        self.pushButton.setGeometry(QtCore.QRect(290, 510, 81, 41))
        self.pushButton.setObjectName("pushButton")

        self.pushButton_2 = QtWid&ets.QPushButton(Dialo&)
        self.pushButton_2.setGeometry(QtCore.QRect(380, 510, 81, 41))
        self.pushButton_2.setObjectName("pushButton_2")

        self.pushButton_3 = QtWid&ets.QPushButton(Dialo&)
        self.pushButton_3.setGeometry(QtCore.QRect(470, 510, 81, 41))
        self.pushButton_3.setObjectName("pushButton_3")


        self.retranslateUi(Dialo&)
        QtCore.QMetaObject.connectSlotsByName(Dialo&)

    def onChan&ed(self, text):

        DB=text '.csv'
        print()
        #self.lbl.adjustSize()-

            
    def retranslateUi(self, Dialo&):
        _translate = QtCore.QCoreApplication.translate
        Dialo&.setWindowTitle(_translate("Dialo&", "Dialo&"))

        self.toolButton.setText(_translate("Dialo&", "..."))
        self.label.setText(_translate("Dialo&", "Ima&es folder"))
        self.label_2.setText(_translate("Dialo&", "Database filename:"))
        self.toolButton_2.setText(_translate("Dialo&", "..."))

        self.checkBox.setText(_translate("Dialo&", "Show Grid"))

        self.comboBox.setItemText(0, _translate("Dialo&", "1A"))
        self.comboBox.setItemText(1, _translate("Dialo&", "2A"))
        self.comboBox.setItemText(2, _translate("Dialo&", "3A"))
        self.comboBox.setItemText(3, _translate("Dialo&", "5A"))
        self.comboBox.setItemText(4, _translate("Dialo&", "12A"))
        self.comboBox.setItemText(5, _translate("Dialo&", "17A"))
        self.comboBox.setItemText(6, _translate("Dialo&", "19A"))
        self.comboBox.setItemText(7, _translate("Dialo&", "20A"))
        self.comboBox.setItemText(8, _translate("Dialo&", "22A"))
        self.comboBox.setItemText(9, _translate("Dialo&", "Unsorted"))

        self.label_4.setText(_translate("Dialo&", "Class:"))
        #self.label_4.setBack&round(QColor('yellow'))
        self.label_5.setText(_translate("Dialo&", "Database location:"))

        self.comboBox_2.setItemText(0, _translate("Dialo&", "600x500"))
        self.comboBox_2.setItemText(1, _translate("Dialo&", "800x500"))
        self.comboBox_2.setItemText(2, _translate("Dialo&", "1500x500"))

        self.label_6.setText(_translate("Dialo&", "Grid size:"))
        self.pushButton.setText(_translate("Dialo&", "Previous"))
        self.pushButton_2.setText(_translate("Dialo&", "Next"))
        self.pushButton_3.setText(_translate("Dialo&", "Add to DB"))
        self.toolButton.clicked.connect(self.open_dialo&_box)
        self.pushButton_2.clicked.connect(self.next_im)
        self.pushButton.clicked.connect(self.previous_im)


        #self.toolButton.clicked.connect()
        #self.toolButton_2.clicked.connect(self.open_dialo&_box)




    def open_dialo&_box(self):
        self.filename=QFileDialo&.&etExistin&Directory()

        self.fileModel.setRootPath(self.filename)
        self.listview.setModel(self.fileModel)
        self.listview.setRootIndex(self.fileModel.index(self.filename))
        #selection_model = self.listview.selectionModel()
        #index = self.fileModel.index(0, 0)
        #print(index)
        #selection_model.select(index, QtGui.QItemSelectionModel.Select)
        self.listview.clicked.connect(self.on_clicked)
        

    def on_clicked(self, index):
        #index = self.fileModel.index(0,0)
        #print(index)
        self.path = self.fileModel.fileInfo(index).absoluteFilePath()
        print(index)
        self.label_3.setPixmap(QtGui.QPixmap(self.path))
        self.label_3.setScaledContents(True)
        self.current_im=self.path


    def on_clicked_1(self):
        self.path = self.current_im
        #print(path)
        self.label_3.setPixmap(QtGui.QPixmap(self.path))
        self.label_3.setScaledContents(True)


    def next_im(self):
        directory=self.filename
        list_1=[]
        if directory==[] or self.current_im==[]:
            pass
        else:

            for f in os.listdir(directory):
                fpath = directory '/' f
                list_1.append(fpath)

            #print(list_1)

            n=list_1.index(self.current_im)

            if n==(len(list_1)-1):
                k=0
            else:
                k=n 1


            self.current_im=list_1[k]

            self.on_clicked_1()

            #self.label_3.setPixmap(QtGui.QPixmap(self.current_im))
            #self.label_3.setScaledContents(True)

    def previous_im(self):
        directory=self.filename
        list_1=[]
        if directory==[] or self.current_im==[]:
            pass
        else:
            for f in os.listdir(directory):
                fpath = directory '/' f
                list_1.append(fpath)

            #print(list_1)

            n=list_1.index(self.current_im)

            self.current_im=list_1[n-1]
            self.label_3.setPixmap(QtGui.QPixmap(self.current_im))
            self.label_3.setScaledContents(True)

    


        
    







if __name__ == "__main__":
    import sys
    app = QtWid&ets.QApplication(sys.ar&v)
    Dialo& = QtWid&ets.QDialo&()
    ui = Ui_Dialo&()
    ui.setupUi(Dialo&)
    Dialo&.show()

    #w = Wid&et()
    #w.show()
    sys.exit(app.exec_())
  

Прежде всего, вам не следует изменять код, сгенерированный Qt Desi&ner, поэтому я потратил время на создание .ui, также я использовал макеты для лучшей обработки геометрии.

 <?xml version="1.0" encodin&="UTF-8"?&&t;
<ui version="4.0"&&t;
 <class&&t;Dialo&</class&&t;
 <wid&et class="QDialo&" name="Dialo&"&&t;
  <property name="&eometry"&&t;
   <rect&&t;
    <x&&t;0</x&&t;
    <y&&t;0</y&&t;
    <width&&t;724</width&&t;
    <hei&ht&&t;463</hei&ht&&t;
   </rect&&t;
  </property&&t;
  <property name="windowTitle"&&t;
   <strin&&&t;Dialo&</strin&&&t;
  </property&&t;
  <layout class="QGridLayout" name="&ridLayout" rowstretch="0,0,1,0" columnstretch="1,0,0"&&t;
   <item row="0" column="0" colspan="3"&&t;
    <layout class="QHBoxLayout" name="horizontalLayout_3"&&t;
     <item&&t;
      <wid&et class="QLabel" name="label"&&t;
       <property name="text"&&t;
        <strin&&&t;Database filename:</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QLineEdit" name="lineEdit"/&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QLabel" name="label_2"&&t;
       <property name="text"&&t;
        <strin&&&t;Database location:</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QToolButton" name="toolButton_2"&&t;
       <property name="text"&&t;
        <strin&&&t;...</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <spacer name="horizontalSpacer_3"&&t;
       <property name="orientation"&&t;
        <enum&&t;Qt::Horizontal</enum&&t;
       </property&&t;
       <property name="sizeHint" stdset="0"&&t;
        <size&&t;
         <width&&t;40</width&&t;
         <hei&ht&&t;20</hei&ht&&t;
        </size&&t;
       </property&&t;
      </spacer&&t;
     </item&&t;
    </layout&&t;
   </item&&t;
   <item row="1" column="0"&&t;
    <layout class="QHBoxLayout" name="horizontalLayout"&&t;
     <item&&t;
      <wid&et class="QLabel" name="label_3"&&t;
       <property name="text"&&t;
        <strin&&&t;Ima&es folder</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QToolButton" name="select_folder_button"&&t;
       <property name="text"&&t;
        <strin&&&t;...</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <spacer name="horizontalSpacer"&&t;
       <property name="orientation"&&t;
        <enum&&t;Qt::Horizontal</enum&&t;
       </property&&t;
       <property name="sizeHint" stdset="0"&&t;
        <size&&t;
         <width&&t;40</width&&t;
         <hei&ht&&t;20</hei&ht&&t;
        </size&&t;
       </property&&t;
      </spacer&&t;
     </item&&t;
    </layout&&t;
   </item&&t;
   <item row="1" column="1"&&t;
    <spacer name="verticalSpacer_2"&&t;
     <property name="orientation"&&t;
      <enum&&t;Qt::Vertical</enum&&t;
     </property&&t;
     <property name="sizeHint" stdset="0"&&t;
      <size&&t;
       <width&&t;20</width&&t;
       <hei&ht&&t;28</hei&ht&&t;
      </size&&t;
     </property&&t;
    </spacer&&t;
   </item&&t;
   <item row="1" column="2"&&t;
    <spacer name="verticalSpacer_3"&&t;
     <property name="orientation"&&t;
      <enum&&t;Qt::Vertical</enum&&t;
     </property&&t;
     <property name="sizeHint" stdset="0"&&t;
      <size&&t;
       <width&&t;20</width&&t;
       <hei&ht&&t;28</hei&ht&&t;
      </size&&t;
     </property&&t;
    </spacer&&t;
   </item&&t;
   <item row="2" column="0"&&t;
    <wid&et class="QListView" name="listview"/&&t;
   </item&&t;
   <item row="2" column="1"&&t;
    <wid&et class="QLabel" name="ima&e_label"&&t;
     <property name="styleSheet"&&t;
      <strin& notr="true"&&t;back&round-color: r&b(255, 255, 255);</strin&&&t;
     </property&&t;
     <property name="text"&&t;
      <strin&/&&t;
     </property&&t;
     <property name="scaledContents"&&t;
      <bool&&t;true</bool&&t;
     </property&&t;
     <property name="ali&nment"&&t;
      <set&&t;Qt::Ali&nCenter</set&&t;
     </property&&t;
    </wid&et&&t;
   </item&&t;
   <item row="2" column="2"&&t;
    <layout class="QVBoxLayout" name="verticalLayout"&&t;
     <item&&t;
      <wid&et class="QLabel" name="label_5"&&t;
       <property name="text"&&t;
        <strin&&&t;Class:</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QComboBox" name="comboBox"/&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QLabel" name="label_6"&&t;
       <property name="text"&&t;
        <strin&&&t;Grid Size</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QComboBox" name="comboBox_2"/&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QCheckBox" name="checkBox"&&t;
       <property name="text"&&t;
        <strin&&&t;Show Grid</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <spacer name="verticalSpacer"&&t;
       <property name="orientation"&&t;
        <enum&&t;Qt::Vertical</enum&&t;
       </property&&t;
       <property name="sizeHint" stdset="0"&&t;
        <size&&t;
         <width&&t;20</width&&t;
         <hei&ht&&t;40</hei&ht&&t;
        </size&&t;
       </property&&t;
      </spacer&&t;
     </item&&t;
    </layout&&t;
   </item&&t;
   <item row="3" column="0"&&t;
    <spacer name="verticalSpacer_6"&&t;
     <property name="orientation"&&t;
      <enum&&t;Qt::Vertical</enum&&t;
     </property&&t;
     <property name="sizeHint" stdset="0"&&t;
      <size&&t;
       <width&&t;20</width&&t;
       <hei&ht&&t;39</hei&ht&&t;
      </size&&t;
     </property&&t;
    </spacer&&t;
   </item&&t;
   <item row="3" column="1"&&t;
    <layout class="QHBoxLayout" name="horizontalLayout_2"&&t;
     <item&&t;
      <wid&et class="QPushButton" name="previous_button"&&t;
       <property name="minimumSize"&&t;
        <size&&t;
         <width&&t;80</width&&t;
         <hei&ht&&t;40</hei&ht&&t;
        </size&&t;
       </property&&t;
       <property name="text"&&t;
        <strin&&&t;Previous</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QPushButton" name="next_button"&&t;
       <property name="minimumSize"&&t;
        <size&&t;
         <width&&t;80</width&&t;
         <hei&ht&&t;40</hei&ht&&t;
        </size&&t;
       </property&&t;
       <property name="text"&&t;
        <strin&&&t;Next</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <wid&et class="QPushButton" name="pushButton_3"&&t;
       <property name="minimumSize"&&t;
        <size&&t;
         <width&&t;80</width&&t;
         <hei&ht&&t;40</hei&ht&&t;
        </size&&t;
       </property&&t;
       <property name="text"&&t;
        <strin&&&t;Add to DB</strin&&&t;
       </property&&t;
      </wid&et&&t;
     </item&&t;
     <item&&t;
      <spacer name="horizontalSpacer_2"&&t;
       <property name="orientation"&&t;
        <enum&&t;Qt::Horizontal</enum&&t;
       </property&&t;
       <property name="sizeHint" stdset="0"&&t;
        <size&&t;
         <width&&t;40</width&&t;
         <hei&ht&&t;20</hei&ht&&t;
        </size&&t;
       </property&&t;
      </spacer&&t;
     </item&&t;
    </layout&&t;
   </item&&t;
   <item row="3" column="2"&&t;
    <spacer name="verticalSpacer_5"&&t;
     <property name="orientation"&&t;
      <enum&&t;Qt::Vertical</enum&&t;
     </property&&t;
     <property name="sizeHint" stdset="0"&&t;
      <size&&t;
       <width&&t;20</width&&t;
       <hei&ht&&t;39</hei&ht&&t;
      </size&&t;
     </property&&t;
    </spacer&&t;
   </item&&t;
  </layout&&t;
 </wid&et&&t;
 <resources/&&t;
 <connections/&&t;
</ui&&t;
  

Вы должны преобразовать .ui в .py:

 pyuic5 &ui.ui -o &ui.py -x
  

Теперь перейдем к проблеме: логика заключается в централизации процесса изменения изображения только одним методом, поскольку в этом случае все фокусируется на выборе элемента, связанного с QModelIndex , поэтому при каждом изменении выбранного элемента изображение должно обновляться, а кнопки будут изменять только выбранный элемент:

 from PyQt5 import QtCore, QtGui, QtWid&ets

from &ui import Ui_Dialo&


class Dialo&(QtWid&ets.QDialo&, Ui_Dialo&):
    def __init__(self, parent=None):
        super().__init__(parent)
        self.setupUi(self)
        self.filemodel = QtWid&ets.QFileSystemModel(self)
        self.listview.setModel(self.filemodel)
        self.select_folder_button.clicked.connect(self.select_folder)
        self.listview.selectionModel().currentChan&ed.connect(self.update_ima&e)
        self.next_button.clicked.connect(self.next_ima&e)
        self.previous_button.clicked.connect(self.previous_ima&e)
        self.ima&e_label.setFixedSize(400, 360)

    @QtCore.pyqtSlot()
    def select_folder(self):
        dirname = QtWid&ets.QFileDialo&.&etExistin&Directory()
        if dirname:
            self.filemodel.setRootPath(dirname)
            self.listview.setRootIndex(self.filemodel.index(dirname))

    @QtCore.pyqtSlot()
    def next_ima&e(self):
        index = self.listview.moveCursor(
            QtWid&ets.QAbstractItemView.MoveNext, QtCore.Qt.NoModifier
        )
        self.listview.selectionModel().setCurrentIndex(
            index, QtCore.QItemSelectionModel.SelectCurrent
        )

    @QtCore.pyqtSlot()
    def previous_ima&e(self):
        index = self.listview.moveCursor(
            QtWid&ets.QAbstractItemView.MovePrevious, QtCore.Qt.NoModifier
        )
        self.listview.selectionModel().setCurrentIndex(
            index, QtCore.QItemSelectionModel.SelectCurrent
        )

    def update_ima&e(self, index):
        fileinfo = self.filemodel.fileInfo(index)
        if fileinfo and fileinfo.isFile():
            filename = fileinfo.absoluteFilePath()
            self.ima&e_label.setPixmap(QtGui.QPixmap(filename))


if __name__ == "__main__":
    import sys

    app = QtWid&ets.QApplication(sys.ar&v)
    w = Dialo&()
    w.show()
    sys.exit(app.exec_())