class newGroup(QWidget): def __init__(self, ref): super(newGroup, self).__init__() self.ui = Ui_newGroup() self.ui.setupUi(self) self.ref = ref self.ui.createButton.clicked.connect(self.createGroup) self.ui.cancelButton.clicked.connect(self.cancel) # Populate the library names. self.library_list = [] for file in os.listdir('libraries'): if file.endswith('.db'): self.library_list.append(file[:-3]) for library in self.library_list: self.ui.librarySelector.addItem(library) def createGroup(self): if self.ui.nameEdit.text() == "": errorMessage("Group name required!") return if (self.ui.nameEdit.text() + ".group") in os.listdir("groups/"): errorMessage("Group of that name already exists!") return filename = 'groups/' + self.ui.nameEdit.text() + ".group" with open(filename, 'w') as file: file.write("library:" + self.ui.librarySelector.currentText() + '\n\n') self.ref.openGroup(filename) self.close() def cancel(self): self.close() self.ref.show()
def __init__(self, ref): super(newGroup, self).__init__() self.ui = Ui_newGroup() self.ui.setupUi(self) self.ref = ref self.ui.createButton.clicked.connect(self.createGroup) self.ui.cancelButton.clicked.connect(self.cancel) # Populate the library names. self.library_list = [] for file in os.listdir('libraries'): if file.endswith('.db'): self.library_list.append(file[:-3]) for library in self.library_list: self.ui.librarySelector.addItem(library)