Exemplo n.º 1
0
 def saveFilesToDB(self):
     selWindow = DBSelectorWindow(self)
     if not selWindow.exec_(): return
     dbName = selWindow.getDBName()
     if not dbName:
         messageBox(QMessageBox.Critical, "Fehler", "Der Name darf nicht leer sein!",  self, QMessageBox.Ok)
         return
     self.currentReader = Reader()
     
     self._progresses.setEnabled(True)
     self.currentReader.ready.connect(self.readerReady)
     self.currentReader.folderStatusUpdated.connect(self.updateFolderStatus)
     self.currentReader.fileStatusUpdated.connect(self.updateFileStatus)
     self.currentReader.save_folder(str(self.currentFolder), "%s.sqlite" %str(dbName))
Exemplo n.º 2
0
class AnalyzerTab(myWidget):
    def __init__(self, parent = None, *args):
        super(AnalyzerTab, self).__init__(parent, *args)
        self.currentFolder = QDir.currentPath()
        self.addFileRow()
        self.addListWidget()
        self.addReadRow()
        self.addProgressBars()
                
    def addFileRow(self):
        self._openBtn = QPushButton(QString.fromUtf8("Ordner auswählen"), self)
        self._openTextField = QLineEdit(self)
        
        self._openBtn.clicked.connect(self.openFolder)
        self.myLayout.addRow(self._openTextField, self._openBtn)

    def addListWidget(self):
        self._listWidget = QListWidget(self)
        self.myLayout.addRow(self._listWidget)
    
    def addReadRow(self):
        self._readBtn = QPushButton(QString.fromUtf8("Inhalt speichern"), self)
        self.myLayout.addRow(self._readBtn)
        self._readBtn.clicked.connect(self.saveFilesToDB)
    
    def addProgressBars(self):
        self._progresses = myWidget(self)
        self._progressFiles = QProgressBar(self._progresses)
        self._progressParagraphs = QProgressBar(self._progresses)
        self._progresses.myLayout.addRow("Dateien", self._progressFiles)
        self._progresses.myLayout.addRow("Paragraphen", self._progressParagraphs)
        self._progresses.setEnabled(False)
        self.myLayout.addRow(self._progresses)
        
    
    def openFolder(self):
        newFolder = str(QFileDialog.getExistingDirectory(parent=self, caption=QString.fromUtf8("Ordner auswählen"), directory = self.currentFolder))
        if not newFolder: 
            print "HIER: ", newFolder
            return            
        self._openTextField.setText(newFolder)
        self.listFolderFiles(newFolder)
        self.currentFolder = newFolder
    
    
    def listFolderFiles(self, folderPath):
        if not os.path.isdir(folderPath):
            print "%s is not a folder!" %folderPath
            return 
        self._listWidget.clear()
        for fName in os.listdir(folderPath):
            if fName.endswith(".doc") and not fName.startswith("~"): self._listWidget.addItem(fName)
    
    
    def saveFilesToDB(self):
        selWindow = DBSelectorWindow(self)
        if not selWindow.exec_(): return
        dbName = selWindow.getDBName()
        if not dbName:
            messageBox(QMessageBox.Critical, "Fehler", "Der Name darf nicht leer sein!",  self, QMessageBox.Ok)
            return
        self.currentReader = Reader()
        
        self._progresses.setEnabled(True)
        self.currentReader.ready.connect(self.readerReady)
        self.currentReader.folderStatusUpdated.connect(self.updateFolderStatus)
        self.currentReader.fileStatusUpdated.connect(self.updateFileStatus)
        self.currentReader.save_folder(str(self.currentFolder), "%s.sqlite" %str(dbName))
    
    def readerReady(self):
        self._progresses.setEnabled(False)
    
    def updateFolderStatus(self, ready, fileCount):
        self._progressFiles.setMaximum(100)
        self._progressFiles.setValue(self._calc_status(ready, fileCount))
    
    def updateFileStatus(self, ready, paragraphCount):
        self._progressParagraphs.setMaximum(100)
        self._progressParagraphs.setValue(self._calc_status(ready, paragraphCount))
    
    def _calc_status(self, done, toDo):
        r = done * 100 / toDo
        if r == 99:
            return 100
        else:
            return r
Exemplo n.º 3
0
            #Check if ant variable of this file is in the random selection, if so add them to the list
            random_variable_declarations = list(
                set(random_sample) & set(variable_declarations))
            if len(random_variable_declarations) > 0:
                files_and_docstrings[file_name] = random_variable_declarations
        # if verbose:
        #     for key in files_and_docstrings.keys():
        #         print key
        #         for docstring in files_and_docstrings[key]:
        #             print docstring
        print(files_and_docstrings)
        if output_file_name:
            with open(output_file_name, "w") as output_file:
                for key in files_and_docstrings.keys():
                    output_file.write(key + "\n")
                    for docstring in files_and_docstrings[key]:
                        output_file.write(docstring + "\n")
                    output_file.write("=" * 20 + "\n")


if __name__ == "__main__":
    retriever = Retriever()
    reader = Reader("(Parameters|Returns).*[^ ]*( : )")
    selector = Selector(retriever, reader)

    if (len(sys.argv) < 2):
        raise Exception(
            "Usage: python fileSelector.py <path\\to\\root\\directory>")
    root_folder = sys.argv[1]
    selector.retrieve_random_sample_file_names(root_folder, 'test.txt', False)