def openDataDir(self):
dialog = QFileDialog(self)
dialog.setFileMode(QFileDialog.Directory)
dialog.setOption(QFileDialog.ReadOnly)
if dialog.exec():
datadir = dialog.selectedFiles()[0]
classesdir = YOSO.classesDir(datadir)
classesdir_it = QDirIterator(classesdir, YOSO.IMAGE_FILE_TEMPLATES,
QDir.Files)
classes = []
while classesdir_it.hasNext():
class_image = classesdir_it.next()
match = YOSO.CLASSES_RE.match(basename(class_image))
if match != None:
class_num = int(match.group('cls'))
class_name = match.group('name')
class_object = Class(class_num, class_name,
QPixmap(class_image))
classes.append(class_object)
classes.sort(key=lambda c: c.number)
classes_model = ClassListModel(classes)
self._classes_view.setModel(classes_model)
self._workspace.setModel(classes_model)
self._classes_view.setEnabled(len(classes) > 0)
selMod = self._classes_view.selectionModel()
selMod.currentChanged.connect(self._workspace.setDefaultClass)
selMod.setCurrentIndex(classes_model.index(0, 0),
QItemSelectionModel.Select)
self._top_images_dir = YOSO.imagesDir(datadir)
self._top_labels_dir = YOSO.labelsDir(datadir)
imagedir_it = QDirIterator(self._top_images_dir,
QDir.AllDirs | QDir.NoDotAndDotDot,
QDirIterator.Subdirectories)
self._image_dirs_combo_box.clear()
self._image_dirs_combo_box.addItem(self._top_images_dir)
while imagedir_it.hasNext():
img_dir = imagedir_it.next()
self._image_dirs_combo_box.addItem(img_dir)
lbl_dir = img_dir.replace(self._top_images_dir,
self._top_labels_dir)
try:
makedirs(lbl_dir)
except OSError as ex:
if ex.errno != errno.EEXIST:
raise
def find(self):
self.filesTable.setRowCount(0)
self.path = self.directoryComboBox.currentText()
fileName = self.filterComboBox.currentText()
files = []
if not fileName:
fileName = "*"
fileName = fileName.split(", ")
# not used, kept for full text search
text = self.textComboBox.currentText()
self.updateComboBox(self.directoryComboBox)
self.updateComboBox(self.filterComboBox)
self.updateComboBox(self.textComboBox)
self.currentDir = QDir(self.path)
self.it = QDirIterator(self.path, fileName, self.dirFilters,
self.searchRecursivity)
while self.it.hasNext():
files.append(self.it.next())
# For full text search, not used
if text:
files = self.findFiles(files, text)
self.showFiles(files)
def getDir(self):
"""list all files and subdirectories for a given dir with regex."""
msg = self.asyncRead()
if msg:
path = msg["path"]
dirselect = False
if "dirSelect" in msg:
dirselect = msg["dirSelect"]
if path == "":
path = "/home"
path = os.path.normpath(path)
filters = ["*"]
if "filter" in msg:
filters = msg["filter"]
dirIter = QDirIterator(path)
data = list()
while dirIter.hasNext():
dirIter.next()
if dirIter.fileName() in [".", ".."]:
continue
if not dirIter.fileInfo().isDir():
if dirselect:
continue
match = False
for fi in filters:
reg = QRegExp(fi)
reg.setPatternSyntax(QRegExp.Wildcard)
if reg.exactMatch(dirIter.fileName()):
match = True
break
if not match:
continue
elem = dict()
elem["name"] = dirIter.fileName()
elem["path"] = dirIter.filePath()
inf = dirIter.fileInfo() # type: QFileInfo
elem["isDir"] = inf.isDir()
last = inf.lastModified() # type: QDateTime
elem["lastChange"] = last.toString("yyyy.MM.dd - HH:MM")
if inf.isDir():
elem["fileSize"] = ""
else:
elem["fileSize"] = self._sizeHumanReadableStr(inf.size())
data.append(elem)
data = sorted(data,
key=lambda di: (not di["isDir"], di["name"].lower()))
msg["data"] = data
logging.debug("There is/are %i element/s in %s", len(data), path)
self.send(msg)
def run(self):
file_iter = QDirIterator(self._path, QDirIterator.Subdirectories)
while file_iter.hasNext():
file_name = file_iter.next()
if Path(file_name).is_dir():
continue
condition = [
magic.from_file(file_name).lower().startswith(ext)
for ext in self._filter
]
if any(condition):
new_resource = Resource(path=file_name,
image=QPixmap(file_name),
thumbnail=QPixmap(file_name).scaled(
100, 100, Qt.KeepAspectRatio,
Qt.SmoothTransformation))
self.resource_loaded.emit(new_resource)
self.finished.emit()
def folderIterator(self):
folderChosen = QFileDialog.getExistingDirectory(self, '打开音频文件夹', '.')
if folderChosen != None:
it = QDirIterator(folderChosen)
it.next()
while it.hasNext():
if it.fileInfo().isDir() == False and it.filePath() != '.':
fInfo = it.fileInfo()
print(it.filePath(), fInfo.suffix())
if fInfo.suffix() in ('mp3', 'ogg', 'wav', 'm4a'):
print('added file', fInfo.fileName())
self.listWid.addItem(fInfo.fileName())
self.playlist.addMedia(QMediaContent(QUrl.fromLocalFile(it.filePath())))
it.next()
if it.fileInfo().isDir() == False and it.filePath() != '.':
fInfo = it.fileInfo()
print(it.filePath(), fInfo.suffix())
if fInfo.suffix() in ('mp3', 'ogg', 'wav', 'm4a'):
print('added file', fInfo.fileName())
self.listWid.addItem(fInfo.fileName())
self.playlist.addMedia(QMediaContent(QUrl.fromLocalFile(it.filePath())))
# 设置ListItem高度
for x in range(self.listWid.count()):
self.listWid.item(x).setSizeHint(QSize(100, 30))
def dropEvent(self, event):
if event.mimeData().hasUrls():
event.setDropAction(Qt.CopyAction)
event.accept()
icon = QIcon("loaded.png")
for url in event.mimeData().urls():
if url.isLocalFile():
fi = QFileInfo(url.toLocalFile())
if fi.isDir():
it = QDirIterator(
fi.fileName(),
QDir.Files,
QDirIterator.Subdirectories,
)
while it.hasNext():
item = QListWidgetItem()
item.setData(PathRole, it.next())
self.addItem(item)
elif fi.isFile():
item = QListWidgetItem()
item.setData(PathRole, url.toLocalFile())
self.addItem(item)
else:
event.ignore()
def dir_choosing(self):
self.index = 0
self.new_playlist = True
directory = QFileDialog.getExistingDirectory(None, "Open Directory", "/", QFileDialog.ShowDirsOnly)
if directory:
os.chdir(directory)
self.tableWidget.setRowCount(0)
self.playlist = []
filters = QDir.Files
nameFilters = ["*.mp3", "*.MP3", "*.Mp3", "*.mP3"]
qDirIterator = QDirIterator(directory, nameFilters, filters, QDirIterator.Subdirectories)
while qDirIterator.hasNext():
qDirIterator.next()
fileInfo = qDirIterator.fileInfo()
fdir = fileInfo.absoluteDir().absolutePath()
song = qDirIterator.filePath()
self.playlist.append(song)
try:
song_tags = TinyTag.get(song)
except:
print("Corrupted file:\n", len(self.playlist),
song, "\nCan't get the tags")
self.add_corrupted_files(song)
else:
self.add_items_to_list(song_tags, song)
self.playlist_action()
def __init__(self, size, parent=None):
super(FlippablePad, self).__init__(self.boundsFromSize(size),
QColor(226, 255, 92, 64), parent)
numIcons = size.width() * size.height()
pixmaps = []
it = QDirIterator(":/images", ["*.png"])
while it.hasNext() and len(pixmaps) < numIcons:
pixmaps.append(it.next())
iconRect = QRectF(-54, -54, 108, 108)
iconColor = QColor(214, 240, 110, 128)
self.iconGrid = []
n = 0
for y in range(size.height()):
row = []
for x in range(size.width()):
rect = RoundRectItem(iconRect, iconColor, self)
rect.setZValue(1)
rect.setPos(self.posForLocation(x, y, size))
rect.setPixmap(pixmaps[n % len(pixmaps)])
n += 1
row.append(rect)
self.iconGrid.append(row)
def startSearch(self, query):
print("search started..", query)
filters = QDir.Files
nameFilters = ["*.cpp"]
iterator = QDirIterator("/home/alexanderb", nameFilters, filters,
QDirIterator.Subdirectories)
while (iterator.hasNext()):
filePath = iterator.next()
fileInfo = QFileInfo(filePath)
currentFile = QFile(filePath)
currentFile.open(QFile.ReadOnly | QFile.Text)
fileContents = currentFile.readAll().data().decode('utf8',
errors='ignore')
if (fileContents.find(query) != -1):
qtwItem = QTreeWidgetItem()
qtwItem.setText(0, fileInfo.fileName())
qtwItem.setText(1, fileInfo.suffix())
qtwItem.setText(2, str(fileInfo.size() / 1024))
qtwItem.setText(3,
fileInfo.lastModified().toString("MM/dd/yyyy"))
qtwItem.setText(4, fileInfo.created().toString("MM/dd/yyyy"))
qtwItem.setText(5, str("...here is the content..."))
qtwItem.setText(6, filePath)
self.qtwItems.append(qtwItem)
self.match_found.emit(qtwItem)
self.finished.emit()
def loadStamps(self):
prefs = preferences.Preferences.instance()
stampsDirectory = prefs.stampsDirectory()
stampsDir = QDir(stampsDirectory)
iterator = QDirIterator(stampsDirectory, ["*.stamp"],
QDir.Files | QDir.Readable)
while (iterator.hasNext()):
stampFileName = iterator.next()
stampFile = QFile(stampFileName)
if (not stampFile.open(QIODevice.ReadOnly)):
continue
data = stampFile.readAll()
document = QJsonDocument.fromBinaryData(data)
if (document.isNull()):
# document not valid binary data, maybe it's an JSON text file
error = QJsonParseError()
document = QJsonDocument.fromJson(data, error)
if (error.error != QJsonParseError.NoError):
qDebug("Failed to parse stamp file:" + error.errorString())
continue
stamp = TileStamp.fromJson(document.object(), stampsDir)
if (stamp.isEmpty()):
continue
stamp.setFileName(iterator.fileInfo().fileName())
self.mTileStampModel.addStamp(stamp)
index = stamp.quickStampIndex()
if (index >= 0 and index < self.mQuickStamps.size()):
self.mQuickStamps[index] = stamp
def selectFolder(self, folder=None):
debugMsg("SelectFolder")
if folder is None:
folder = self.treeView.currentIndex()
self.listWidget.clear()
self.currentFolder = self.filesystemmodel.filePath(
self.treeView.currentIndex())
print("current folder: ", self.currentFolder)
if (self.showRecursive):
flag = QDirIterator.Subdirectories
else:
flag = QDirIterator.NoIteratorFlags
file_it = QDirIterator(self.currentFolder, [
"*.jpg", "*.jpeg", "*.tiff", "*.png", "*.JPG", "*.JPEG", ".PNG",
"*.TIFF"
], QDir.Files, flag)
self._current_files = []
while file_it.hasNext():
filename = file_it.next()
#print ("filename: ",filename)
#print (" - replacing", self.currentFolder+os.path.sep)
item = filename.replace(self.currentFolder, ".")
#print (" -->item: ",item)
self._current_files.append(item)
self.listWidget.addItem(item)
self.listWidget.clear()
self._current_files.sort()
self.listWidget.addItems(self._current_files)
return
def configure(self, machineType, machineConfig, machinePort, machineStart):
it = QDirIterator(self.dataConfigDir, ['*.cfg.json'])
while it.hasNext():
fileInfo = QFileInfo(it.next())
basename = fileInfo.baseName()
self.machineConfigCB.addItem(basename)
self.reloadSettings()
def import_emojis():
import emoji
emojis = []
iterator = QDirIterator(":/emoji/", QDirIterator.Subdirectories)
while iterator.hasNext():
emoji = iterator.next()
emojis.append(emoji[8:13])
return emojis
def hasChildren(self, parent):
"""Overridden to only show the expand arrow when there are subfolders."""
if parent.flags() & Qt.ItemFlag.ItemNeverHasChildren:
return False
elif (self.filePath(parent)):
return QDirIterator(
self.filePath(parent),
self.filter() | QDir.Filter.NoDotAndDotDot,
QDirIterator.IteratorFlag.NoIteratorFlags).hasNext()
return QFileSystemModel.hasChildren(self, parent)
def loadAvailableLanguages(self):
self.mLanguages.clear()
nameFilters = QStringList()
nameFilters.append("tiled_*.qm")
iterator = QDirIterator(self.mTranslationsDir, nameFilters,
QDir.Files | QDir.Readable)
while (iterator.hasNext()):
iterator.next()
baseName = iterator.fileInfo().completeBaseName()
# Cut off "tiled_" from the start
self.mLanguages.append(baseName[6:])
def start_search(self, query, search_directory):
self.keep_searching = True
print("search started..", query)
filters = QDir.Files
nameFilters = [
"*.cpp", "*.txt", "*.pdf", "*.doc", "*.docx", "*.xlsx", "*.xls",
"*.ppt", "*.pptx"
]
iterator = QDirIterator(search_directory, nameFilters, filters,
QDirIterator.Subdirectories)
while (iterator.hasNext()):
QApplication.processEvents()
if (self.keep_searching):
file_path = iterator.next()
if (os.access(file_path, os.R_OK)):
try:
file_info = QFileInfo(file_path)
file_contents = parser.from_file(file_path)
except:
continue
if (file_contents['status'] == 200
and 'content' in file_contents.keys()
and file_contents['content'] is not None):
found_index = file_contents['content'].find(query)
if (found_index != -1):
snippet = file_contents['content'].strip().replace(
'\n', ' ').replace('\r', '')
snippet_index = snippet.find(query)
qtw_item = QTreeWidgetItem()
qtw_item.setText(0, file_info.fileName())
qtw_item.setText(1, file_info.suffix())
qtw_item.setText(2, str(file_info.size() / 1024))
qtw_item.setText(
3,
file_info.lastModified().toString(
"MM/dd/yyyy"))
qtw_item.setText(
4,
file_info.created().toString("MM/dd/yyyy"))
qtw_item.setText(
5,
str(snippet)[snippet_index - 5:snippet_index +
10])
qtw_item.setText(6, file_path)
self.qtw_items.append(qtw_item)
self.match_found.emit(qtw_item)
self.finished.emit()
def run(self):
if not os.path.exists(self.folder_location):
return None
filter = QDir.Dirs | QDir.Files | QDir.Hidden
iterator = QDirIterator(self.folder_location, filter, QDirIterator.Subdirectories)
while iterator.hasNext():
path = iterator.next()
if os.path.isfile(path):
file_name = os.path.split(path)
self.scanning_signal.emit("[SCANNING]: "+file_name[1])
self.scan_for_yara(path=path)
self.scan_for_tlsh(path=path)
self.scanning_signal.emit(None)
def folderIterator(self):
folderChosen = QFileDialog.getExistingDirectory(self, 'Open Music Folder', '~')
if folderChosen != None:
it = QDirIterator(folderChosen)
it.next()
while it.hasNext():
if it.fileInfo().isDir() == False and it.filePath() != '.':
fInfo = it.fileInfo()
if fInfo.suffix() in ('mp3', 'ogg', 'wav', 'm4a'):
self.playlist.addMedia(QMediaContent(QUrl.fromLocalFile(it.filePath())))
it.next()
if it.fileInfo().isDir() == False and it.filePath() != '.':
fInfo = it.fileInfo()
if fInfo.suffix() in ('mp3', 'ogg', 'wav', 'm4a'):
self.playlist.addMedia(QMediaContent(QUrl.fromLocalFile(it.filePath())))
def openImages(self, directory):
image_it = QDirIterator(directory, labeler.IMAGE_FILE_TEMPLATES,
QDir.Files, QDirIterator.Subdirectories)
self._current_images = []
while image_it.hasNext():
self._current_images.append(image_it.next())
self._current_images.sort()
self._image_spinner.setRange(0, 0)
self._image_spinner.setEnabled(False)
self._image_spinner.setValue(0)
image_total = len(self._current_images)
if image_total > 0:
self._progress_bar.setRange(0, image_total)
self._image_spinner.setRange(1, image_total)
self._image_spinner.setEnabled(True)
self._image_spinner.setValue(1)
def load_playlist(self, path):
'''load new playlist'''
debug('load playlist')
self.playlist.clear()
# Note: not actually sure these formats are all supported ...
files = QDirIterator(path, ['*.mp3', '*.ogg', '*.wav', '*.flac'], flags=QDirIterator.Subdirectories)
while files.hasNext():
filename = files.next()
debug('+ {}'.format(filename))
url = QUrl.fromLocalFile(filename)
if not self.playlist.addMedia(QMediaContent(url)):
debug('addMedia() => False')
self.player.setPlaylist(self.playlist)
def folder_iterator(self):
folder_chosen = QFileDialog.getExistingDirectory(self, '打开文件夹', '~')
if folder_chosen is not None:
it = QDirIterator(folder_chosen)
it.next()
while it.hasNext():
if (not it.fileInfo().isDir()) and it.filePath() != '.':
f_info = it.fileInfo()
if f_info.suffix() in ('mp3', 'ogg', 'wav', 'm4a'):
self.playlist.addMedia(
QMediaContent(QUrl.fromLocalFile(it.filePath())))
it.next()
if (not it.fileInfo().isDir()) and it.filePath() != '.':
f_info = it.fileInfo()
if f_info.suffix() in ('mp3', 'ogg', 'wav', 'm4a'):
self.playlist.addMedia(
QMediaContent(QUrl.fromLocalFile(it.filePath())))
def folderIterator(self):
folderChosen = QFileDialog.getOpenFileName(None, 'Open Music Folder', '~',
'All Files(*.*)')
if folderChosen != None:
it = QDirIterator(folderChosen)
it.next()
while it.hasNext():
if it.fileInfo().isDir() == False and it.filePath() != '.':
fInfo = it.fileInfo()
if fInfo.suffix() in ('mp3', 'mid', 'wav', 'flac'):
self.playlist.append(
QMediaContent(QUrl.fromLocalFile(it.filePath())))
it.next()
if it.fileInfo().isDir() == False and it.filePath() != '.':
fInfo = it.fileInfo()
if fInfo.suffix() in ('mp3', 'mid', 'wav', 'flac'):
self.playlist.append(
QMediaContent(QUrl.fromLocalFile(it.filePath())))
def countFiles(self, folder):
files = []
if self.subfolderLevel > 0:
try:
path = glob.glob('%s/%s' % (folder, '*/' * self.subfolderLevel))[0]
self.pathToDisplay = path
except:
return 0
else:
path = folder
it = QDirIterator(path, (QDir.Files | QDir.NoSymLinks),
self.countRecursivity)
while it.hasNext():
files.append(it.next())
count = len(files)
return count
def add_files(self):
folder_Chosen = QFileDialog\
.getExistingDirectory(self,
'Open Music Folder',
expanduser('~'))
if folder_Chosen is not None:
it = QDirIterator(folder_Chosen)
it.next()
while it.hasNext():
if it.fileInfo().isDir() == False\
and it.filePath() != '.':
fInfo = it.fileInfo()
print(it.filePath(), fInfo.suffix())
if fInfo.suffix() in ('mp3', 'ogg', 'wav'):
print('added file ', fInfo.fileName())
self.currentPlaylist. \
addMedia(QMediaContent(
QUrl.fromLocalFile(it.filePath())))
it.next()
def main():
mainView = "qrc:///main.qml"
application = QGuiApplication(sys.argv)
qmlRegisterType(ElHamalEncryptor, 'elhamal', 1, 0, 'Encryptor')
qmlRegisterType(ElHamalDecryptor, 'elhamal', 1, 0, 'Decryptor')
qmlRegisterType(ElHamalKeysGenerator, 'elhamal', 1, 0, 'KeysGenerator')
qInstallMessageHandler(handleStatusChange)
it = QDirIterator(":", QDirIterator.Subdirectories)
while it.hasNext():
print(it.next())
view = QQuickView()
view.setResizeMode(QQuickView.SizeViewToRootObject)
view.setSource(QUrl(mainView))
sys.exit(application.exec_())
def UItranslations(self):
""" Get list of available ui translations
"""
# iterate over resource file to find available translations
fltr = QDir.Dirs | QDir.Files | QDir.Hidden
iterator = QDirIterator(':', fltr, QDirIterator.Subdirectories)
while iterator.hasNext():
filePath = iterator.next()
if '/translations/ts/' in filePath:
fileName = os.path.basename(str(filePath[1:]))
locale = fileName.replace('lector_', '').replace('.qm', '')
if locale:
self.ui.cbLang.addItem(locale)
locale = settings.get('ui:lang')
if not locale:
locale = QLocale.system().name()
currentIndex = self.ui.cbLang.findText(locale)
if currentIndex <= -1:
currentIndex = self.ui.cbLang.findText('en_GB')
self.ui.cbLang.setCurrentIndex(currentIndex)
def scanFolder(self, path):
self._tempSongs = {}
filters = QDir.Files
nameFilters = [
"*.wav", "*.wma", "*.mp2", "*.mp3", "*.mp4", "*.m4a", "*.flac",
"*.ogg"
]
qDirIterator = QDirIterator(path, nameFilters, filters,
QDirIterator.Subdirectories)
while qDirIterator.hasNext():
qDirIterator.next()
fileInfo = qDirIterator.fileInfo()
fdir = fileInfo.absoluteDir().absolutePath()
fpath = qDirIterator.filePath()
fsize = fileInfo.size() / (1024 * 1024)
time.sleep(0.05)
if fsize >= 1:
self.scanfileChanged.emit(fpath)
self.tipMessageChanged.emit(fpath)
self.tipMessageChanged.emit('')
self.scanfileFinished.emit()
def addSongToPlaylist(self, piosenka):
if self.folder != None:
it = QDirIterator(self.folder)
it.next()
while it.hasNext():
if it.fileInfo().isDir() == False and it.filePath() != '.':
fInfo = it.fileInfo()
if fInfo.suffix() in ('mp3', 'ogg', 'wav', 'm4a'):
if it.fileName() == piosenka:
self.playlist.addMedia(
QMediaContent(QUrl.fromLocalFile(
it.filePath())))
self.listView.addItem(it.fileName())
it.next()
if it.fileInfo().isDir() == False and it.filePath() != '.':
fInfo = it.fileInfo()
if fInfo.suffix() in ('mp3', 'ogg', 'wav', 'm4a'):
if it.fileName() == piosenka:
self.playlist.addMedia(
QMediaContent(QUrl.fromLocalFile(it.filePath())))
self.listView.addItem(it.fileName())
# Parsec Cloud (https://parsec.cloud) Copyright (c) AGPLv3 2016-2021 Scille SAS
try:
from parsec.core.gui._resources_rc import * # noqa
except ImportError as exc:
raise ModuleNotFoundError(
"""PyQt resources bundle hasn't been generated.
You must install the parsec package or run `python setup.py generate_pyqt_resources_bundle`
"""
) from exc
if __name__ == "__main__":
from PyQt5.QtCore import QDirIterator
it = QDirIterator(":", QDirIterator.Subdirectories)
while it.hasNext():
print(it.next())
def on_execute_clicked(self):
self.Log.clear()
it = QDirIterator(self.currentDir, QDirIterator.NoIteratorFlags)
while it.hasNext():
self.Log.addItem(it.next())
self.Path.setText(self.currentDir.absolutePath())
