Как фильтровать файлы по их расширению на Python?

Question

[Maxim045]

У меня есть код, который отображает в списке файлы, находящиеся в выбранной папке. Пожалуйста, подскажите, как осуществить их фильтрацию, чтобы в списке отображались файлы только с определенным расширением(MP3, WAV)?

import sys, os, design
from PyQt5 import QtWidgets

class Browse(QtWidgets.QMainWindow, design.Ui_MainWindow):
    def __init__(self):
        super().__init__()
        self.setupUi(self)
        self.ButtonBrowse.clicked.connect(self.browse_folder)

    def browse_folder(self):
        self.listWidget.clear()
        directory = QtWidgets.QFileDialog.getExistingDirectory(self, "Выберите папку")

        if directory:
            for file_name in os.listdir(directory):
                self.listWidget.addItem(file_name)
def main():
    app = QtWidgets.QApplication(sys.argv)
    window = Browse()
    window.show()
    app.exec_()

if __name__ == '__main__':
    main()

Answer 1

[Dimonchik]

вот сюда

if directory:
            for file_name in os.listdir(directory):
                self.listWidget.addItem(file_name)

вставьте условие
if filename.endswith('.mp3')

Comments

[Maxim045]

Спасибо!