def test_filter(self):
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0")
ip = ImagePicker(model_path=sample_path)
# hacky, but: set a filter that should only match one image in the
# test set and verify only one image exists afterward
ip.setFilterTags(["hardway"])
self.assertEqual(ip.model().rowCount(), 1)
def test_filter(self):
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0")
ip = ImagePicker(model_path=sample_path)
# hacky, but: set a filter that should only match one image in the
# test set and verify only one image exists afterward
ip.setFilterTags(["hardway"])
self.assertEqual(ip.model().rowCount(), 1)
def test_sort(self):
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0")
ip = ImagePicker(model_path=sample_path)
model = ip.model()
# filter so that two images match, but one image matches more tags,
# then make sure the one that matches more is at the top of the list
ip.setFilterTags(["book", "gg"])
self.assertEqual(ip.model().rowCount(), 2)
ip.sort()
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole),
"http://doodle.town")
def test_sort(self):
file_dir = os.path.dirname(os.path.realpath(__file__))
sample_path = os.path.join(file_dir, "data", "0")
ip = ImagePicker(model_path=sample_path)
model = ip.model()
# filter so that two images match, but one image matches more tags,
# then make sure the one that matches more is at the top of the list
ip.setFilterTags(["book", "gg"])
self.assertEqual(ip.model().rowCount(), 2)
ip.sort()
self.assertEqual(model.data(model.index(0, 0), Qt.DisplayRole), "http://doodle.town")
def __init__(self, parent=None):
'''
Define widgets, their layout, and signal/slot interactions
'''
QMainWindow.__init__(self, parent)
self.resize(800, 600)
centralWidget = QWidget(self)
self.setCentralWidget(centralWidget)
# Define widgets
self.tagEdit = QLineEdit(centralWidget)
self.tagEdit.setPlaceholderText("Current Image Tags")
self.urlEdit = QLineEdit(centralWidget)
self.urlEdit.setPlaceholderText("Current Image URL")
self.urlEdit.setDisabled(True)
saveButton = QPushButton("Save Selection", centralWidget)
deleteButton = QPushButton("Delete Selection", centralWidget)
self.searchEdit = QLineEdit(centralWidget)
self.searchEdit.setPlaceholderText("Image Tag Search")
self.liveCheckBox = ClearableCheckBox("WordLogger", centralWidget)
self.clearSearchButton = QPushButton("Clear Filter", centralWidget)
self.imagePicker = ImagePicker(centralWidget)
self.getUrlEdit = QLineEdit(centralWidget)
self.getUrlEdit.setPlaceholderText("Download Image URL")
self.getUrlButton = QPushButton("Download URL", centralWidget)
self.pasteUrlButton = QPushButton("Paste for Download", centralWidget)
# Set layout
layout = QGridLayout(centralWidget)
layout.addWidget(self.imagePicker, 0, 0, 3, 5)
layout.addWidget(QLabel("Selection:"), 5, 0)
layout.addWidget(self.urlEdit, 5, 1)
layout.addWidget(self.tagEdit, 5, 2)
layout.addWidget(saveButton, 5, 3)
layout.addWidget(deleteButton, 5, 4)
layout.addWidget(QLabel("Search:"), 6, 0)
layout.addWidget(self.searchEdit, 6, 1, 1, 2)
layout.addWidget(self.liveCheckBox, 6, 3, 1, 1)
layout.addWidget(self.clearSearchButton, 6, 4, 1, 1)
layout.addWidget(QLabel("Download:"), 7, 0)
layout.addWidget(self.getUrlEdit, 7, 1, 1, 2)
layout.addWidget(self.getUrlButton, 7, 3, 1, 1)
layout.addWidget(self.pasteUrlButton, 7, 4, 1, 1)
# Let user search by tag
self.searchEdit.textChanged.connect(
self.imagePicker.setFilterTagsString)
self.clearSearchButton.clicked.connect(self.clearSearch)
# Reflect selection in tags and URL boxes
self.imagePicker.selectedTagsStringChanged.connect(
self.tagEdit.setText)
self.imagePicker.selectedUrlChanged.connect(self.urlEdit.setText)
# Enabling saving changes of tags to model
saveButton.clicked.connect(self.saveTags)
self.tagEdit.returnPressed.connect(saveButton.clicked)
# Enable deleting selections from model
deleteButton.clicked.connect(self.imagePicker.deleteSelected)
self.imagePicker.preDelete.connect(self.tagEdit.clear)
self.imagePicker.preDelete.connect(self.urlEdit.clear)
self.imagePicker.preDelete.connect(self.showDeletedMessage)
# Set up image downloading
self.getUrlButton.clicked.connect(self.downloadImage)
self.pasteUrlButton.clicked.connect(self.pasteUrl)
# Enable wordlogging support
self.wordlogger = get_wordlogger()
if self.wordlogger:
self.liveCheckBox.stateChanged.connect(self.toggleWordLogging)
else:
self.liveCheckBox.setEnabled(False)
# Enable fast URL copying (this might be annoying)
self.imagePicker.clicked.connect(self.copyUrl)
self.statusBar().showMessage("Click an image to copy its URL")
class MainWindow(QMainWindow):
def __init__(self, parent=None):
'''
Define widgets, their layout, and signal/slot interactions
'''
QMainWindow.__init__(self, parent)
self.resize(800, 600)
centralWidget = QWidget(self)
self.setCentralWidget(centralWidget)
# Define widgets
self.tagEdit = QLineEdit(centralWidget)
self.tagEdit.setPlaceholderText("Current Image Tags")
self.urlEdit = QLineEdit(centralWidget)
self.urlEdit.setPlaceholderText("Current Image URL")
self.urlEdit.setDisabled(True)
saveButton = QPushButton("Save Selection", centralWidget)
deleteButton = QPushButton("Delete Selection", centralWidget)
self.searchEdit = QLineEdit(centralWidget)
self.searchEdit.setPlaceholderText("Image Tag Search")
self.liveCheckBox = ClearableCheckBox("WordLogger", centralWidget)
self.clearSearchButton = QPushButton("Clear Filter", centralWidget)
self.imagePicker = ImagePicker(centralWidget)
self.getUrlEdit = QLineEdit(centralWidget)
self.getUrlEdit.setPlaceholderText("Download Image URL")
self.getUrlButton = QPushButton("Download URL", centralWidget)
self.pasteUrlButton = QPushButton("Paste for Download", centralWidget)
# Set layout
layout = QGridLayout(centralWidget)
layout.addWidget(self.imagePicker, 0, 0, 3, 5)
layout.addWidget(QLabel("Selection:"), 5, 0)
layout.addWidget(self.urlEdit, 5, 1)
layout.addWidget(self.tagEdit, 5, 2)
layout.addWidget(saveButton, 5, 3)
layout.addWidget(deleteButton, 5, 4)
layout.addWidget(QLabel("Search:"), 6, 0)
layout.addWidget(self.searchEdit, 6, 1, 1, 2)
layout.addWidget(self.liveCheckBox, 6, 3, 1, 1)
layout.addWidget(self.clearSearchButton, 6, 4, 1, 1)
layout.addWidget(QLabel("Download:"), 7, 0)
layout.addWidget(self.getUrlEdit, 7, 1, 1, 2)
layout.addWidget(self.getUrlButton, 7, 3, 1, 1)
layout.addWidget(self.pasteUrlButton, 7, 4, 1, 1)
# Let user search by tag
self.searchEdit.textChanged.connect(
self.imagePicker.setFilterTagsString)
self.clearSearchButton.clicked.connect(self.clearSearch)
# Reflect selection in tags and URL boxes
self.imagePicker.selectedTagsStringChanged.connect(
self.tagEdit.setText)
self.imagePicker.selectedUrlChanged.connect(self.urlEdit.setText)
# Enabling saving changes of tags to model
saveButton.clicked.connect(self.saveTags)
self.tagEdit.returnPressed.connect(saveButton.clicked)
# Enable deleting selections from model
deleteButton.clicked.connect(self.imagePicker.deleteSelected)
self.imagePicker.preDelete.connect(self.tagEdit.clear)
self.imagePicker.preDelete.connect(self.urlEdit.clear)
self.imagePicker.preDelete.connect(self.showDeletedMessage)
# Set up image downloading
self.getUrlButton.clicked.connect(self.downloadImage)
self.pasteUrlButton.clicked.connect(self.pasteUrl)
# Enable wordlogging support
self.wordlogger = get_wordlogger()
if self.wordlogger:
self.liveCheckBox.stateChanged.connect(self.toggleWordLogging)
else:
self.liveCheckBox.setEnabled(False)
# Enable fast URL copying (this might be annoying)
self.imagePicker.clicked.connect(self.copyUrl)
self.statusBar().showMessage("Click an image to copy its URL")
def showDeletedMessage(self):
self.statusBar().showMessage("Deleted image")
def clearSearch(self):
self.searchEdit.setText("")
self.statusBar().showMessage("Cleared search")
def copyUrl(self):
if self.imagePicker.selectedIndexes():
url = self.urlEdit.text()
QApplication.clipboard().setText(url)
print "copied URL to clipboard:", url
self.statusBar().showMessage("Copied '%s'" % url)
def pasteUrl(self):
data = QApplication.clipboard().mimeData()
if data.hasText():
self.getUrlEdit.setText(data.text())
else:
self.statusBar().showMessage("Paste URL: No text on clipboard")
print "Paste URL: No text on clipboard"
self.statusBar().showMessage("Pasted URL")
def toggleWordLogging(self, new_state):
if (new_state == Qt.Checked) and not self.wordlogger.is_active():
# clicking into UI elements disables live search
self.searchEdit.textEdited.connect(self.liveCheckBox.clear)
self.urlEdit.textEdited.connect(self.liveCheckBox.clear)
self.tagEdit.textEdited.connect(self.liveCheckBox.clear)
self.getUrlEdit.textEdited.connect(self.liveCheckBox.clear)
self.getUrlButton.clicked.connect(self.liveCheckBox.clear)
self.pasteUrlButton.clicked.connect(self.liveCheckBox.clear)
self.imagePicker.clicked.connect(self.liveCheckBox.clear)
# start word logger thread
self.wordlogger.start()
self.wordlogger.wordEntered.connect(self.searchEdit.setText)
else:
# stop word logger thread
self.wordlogger.stop()
self.wordlogger.wordEntered.disconnect(self.searchEdit.setText)
# disable connections to UI elements
self.searchEdit.textEdited.disconnect(self.liveCheckBox.clear)
self.urlEdit.textEdited.disconnect(self.liveCheckBox.clear)
self.tagEdit.textEdited.disconnect(self.liveCheckBox.clear)
self.getUrlEdit.textEdited.disconnect(self.liveCheckBox.clear)
self.getUrlButton.clicked.disconnect(self.liveCheckBox.clear)
self.pasteUrlButton.clicked.disconnect(self.liveCheckBox.clear)
self.imagePicker.clicked.disconnect(self.liveCheckBox.clear)
self.statusBar().showMessage("Toggled word logger")
def downloadImage(self):
downloader = ImageDownloader(self)
downloader.failure.connect(self.showDownloadError)
downloader.success.connect(self.imagePicker.addImage)
downloader.success.connect(self.searchEdit.clear)
url = self.getUrlEdit.text()
downloader.get(url, self.imagePicker.getLocalFilepath(url))
self.statusBar().showMessage("Downloading %s" % url)
def showDownloadError(self, message):
mb = QMessageBox()
mb.setText(message)
mb.exec_()
def saveTags(self):
self.imagePicker.setTagsString(self.tagEdit.text())
self.statusBar().showMessage("Saved tags")
def __init__(self, parent=None):
'''
Define widgets, their layout, and signal/slot interactions
'''
QMainWindow.__init__(self, parent)
self.resize(800,600)
centralWidget = QWidget(self)
self.setCentralWidget(centralWidget)
# Define widgets
self.tagEdit = QLineEdit(centralWidget)
self.tagEdit.setPlaceholderText("Current Image Tags")
self.urlEdit = QLineEdit(centralWidget)
self.urlEdit.setPlaceholderText("Current Image URL")
self.urlEdit.setDisabled(True)
saveButton = QPushButton("Save Selection", centralWidget)
deleteButton = QPushButton("Delete Selection", centralWidget)
self.searchEdit = QLineEdit(centralWidget)
self.searchEdit.setPlaceholderText("Image Tag Search")
self.liveCheckBox = ClearableCheckBox("WordLogger", centralWidget)
self.clearSearchButton = QPushButton("Clear Filter", centralWidget)
self.imagePicker = ImagePicker(centralWidget)
self.getUrlEdit = QLineEdit(centralWidget)
self.getUrlEdit.setPlaceholderText("Download Image URL")
self.getUrlButton = QPushButton("Download URL", centralWidget)
self.pasteUrlButton = QPushButton("Paste for Download", centralWidget)
# Set layout
layout = QGridLayout(centralWidget)
layout.addWidget(self.imagePicker, 0, 0, 3, 5)
layout.addWidget(QLabel("Selection:"), 5, 0)
layout.addWidget(self.urlEdit, 5, 1)
layout.addWidget(self.tagEdit, 5, 2)
layout.addWidget(saveButton, 5, 3)
layout.addWidget(deleteButton, 5, 4)
layout.addWidget(QLabel("Search:"), 6, 0)
layout.addWidget(self.searchEdit, 6, 1, 1, 2)
layout.addWidget(self.liveCheckBox, 6, 3, 1, 1)
layout.addWidget(self.clearSearchButton, 6, 4, 1, 1)
layout.addWidget(QLabel("Download:"), 7, 0)
layout.addWidget(self.getUrlEdit, 7, 1, 1, 2)
layout.addWidget(self.getUrlButton, 7, 3, 1, 1)
layout.addWidget(self.pasteUrlButton, 7, 4, 1, 1)
# Let user search by tag
self.searchEdit.textChanged.connect(self.imagePicker.setFilterTagsString)
self.clearSearchButton.clicked.connect(self.clearSearch)
# Reflect selection in tags and URL boxes
self.imagePicker.selectedTagsStringChanged.connect(self.tagEdit.setText)
self.imagePicker.selectedUrlChanged.connect(self.urlEdit.setText)
# Enabling saving changes of tags to model
saveButton.clicked.connect(self.saveTags)
self.tagEdit.returnPressed.connect(saveButton.clicked)
# Enable deleting selections from model
deleteButton.clicked.connect(self.imagePicker.deleteSelected)
self.imagePicker.preDelete.connect(self.tagEdit.clear)
self.imagePicker.preDelete.connect(self.urlEdit.clear)
self.imagePicker.preDelete.connect(self.showDeletedMessage)
# Set up image downloading
self.getUrlButton.clicked.connect(self.downloadImage)
self.pasteUrlButton.clicked.connect(self.pasteUrl)
# Enable wordlogging support
self.wordlogger = get_wordlogger()
if self.wordlogger:
self.liveCheckBox.stateChanged.connect(self.toggleWordLogging)
else:
self.liveCheckBox.setEnabled(False)
# Enable fast URL copying (this might be annoying)
self.imagePicker.clicked.connect(self.copyUrl)
self.statusBar().showMessage("Click an image to copy its URL")
class MainWindow(QMainWindow):
def __init__(self, parent=None):
'''
Define widgets, their layout, and signal/slot interactions
'''
QMainWindow.__init__(self, parent)
self.resize(800,600)
centralWidget = QWidget(self)
self.setCentralWidget(centralWidget)
# Define widgets
self.tagEdit = QLineEdit(centralWidget)
self.tagEdit.setPlaceholderText("Current Image Tags")
self.urlEdit = QLineEdit(centralWidget)
self.urlEdit.setPlaceholderText("Current Image URL")
self.urlEdit.setDisabled(True)
saveButton = QPushButton("Save Selection", centralWidget)
deleteButton = QPushButton("Delete Selection", centralWidget)
self.searchEdit = QLineEdit(centralWidget)
self.searchEdit.setPlaceholderText("Image Tag Search")
self.liveCheckBox = ClearableCheckBox("WordLogger", centralWidget)
self.clearSearchButton = QPushButton("Clear Filter", centralWidget)
self.imagePicker = ImagePicker(centralWidget)
self.getUrlEdit = QLineEdit(centralWidget)
self.getUrlEdit.setPlaceholderText("Download Image URL")
self.getUrlButton = QPushButton("Download URL", centralWidget)
self.pasteUrlButton = QPushButton("Paste for Download", centralWidget)
# Set layout
layout = QGridLayout(centralWidget)
layout.addWidget(self.imagePicker, 0, 0, 3, 5)
layout.addWidget(QLabel("Selection:"), 5, 0)
layout.addWidget(self.urlEdit, 5, 1)
layout.addWidget(self.tagEdit, 5, 2)
layout.addWidget(saveButton, 5, 3)
layout.addWidget(deleteButton, 5, 4)
layout.addWidget(QLabel("Search:"), 6, 0)
layout.addWidget(self.searchEdit, 6, 1, 1, 2)
layout.addWidget(self.liveCheckBox, 6, 3, 1, 1)
layout.addWidget(self.clearSearchButton, 6, 4, 1, 1)
layout.addWidget(QLabel("Download:"), 7, 0)
layout.addWidget(self.getUrlEdit, 7, 1, 1, 2)
layout.addWidget(self.getUrlButton, 7, 3, 1, 1)
layout.addWidget(self.pasteUrlButton, 7, 4, 1, 1)
# Let user search by tag
self.searchEdit.textChanged.connect(self.imagePicker.setFilterTagsString)
self.clearSearchButton.clicked.connect(self.clearSearch)
# Reflect selection in tags and URL boxes
self.imagePicker.selectedTagsStringChanged.connect(self.tagEdit.setText)
self.imagePicker.selectedUrlChanged.connect(self.urlEdit.setText)
# Enabling saving changes of tags to model
saveButton.clicked.connect(self.saveTags)
self.tagEdit.returnPressed.connect(saveButton.clicked)
# Enable deleting selections from model
deleteButton.clicked.connect(self.imagePicker.deleteSelected)
self.imagePicker.preDelete.connect(self.tagEdit.clear)
self.imagePicker.preDelete.connect(self.urlEdit.clear)
self.imagePicker.preDelete.connect(self.showDeletedMessage)
# Set up image downloading
self.getUrlButton.clicked.connect(self.downloadImage)
self.pasteUrlButton.clicked.connect(self.pasteUrl)
# Enable wordlogging support
self.wordlogger = get_wordlogger()
if self.wordlogger:
self.liveCheckBox.stateChanged.connect(self.toggleWordLogging)
else:
self.liveCheckBox.setEnabled(False)
# Enable fast URL copying (this might be annoying)
self.imagePicker.clicked.connect(self.copyUrl)
self.statusBar().showMessage("Click an image to copy its URL")
def showDeletedMessage(self):
self.statusBar().showMessage("Deleted image")
def clearSearch(self):
self.searchEdit.setText("")
self.statusBar().showMessage("Cleared search")
def copyUrl(self):
if self.imagePicker.selectedIndexes():
url = self.urlEdit.text()
QApplication.clipboard().setText(url)
print "copied URL to clipboard:", url
self.statusBar().showMessage("Copied '%s'" % url)
def pasteUrl(self):
data = QApplication.clipboard().mimeData()
if data.hasText():
self.getUrlEdit.setText(data.text())
else:
self.statusBar().showMessage("Paste URL: No text on clipboard")
print "Paste URL: No text on clipboard"
self.statusBar().showMessage("Pasted URL")
def toggleWordLogging(self, new_state):
if (new_state == Qt.Checked) and not self.wordlogger.is_active():
# clicking into UI elements disables live search
self.searchEdit.textEdited.connect(self.liveCheckBox.clear)
self.urlEdit.textEdited.connect(self.liveCheckBox.clear)
self.tagEdit.textEdited.connect(self.liveCheckBox.clear)
self.getUrlEdit.textEdited.connect(self.liveCheckBox.clear)
self.getUrlButton.clicked.connect(self.liveCheckBox.clear)
self.pasteUrlButton.clicked.connect(self.liveCheckBox.clear)
self.imagePicker.clicked.connect(self.liveCheckBox.clear)
# start word logger thread
self.wordlogger.start()
self.wordlogger.wordEntered.connect(self.searchEdit.setText)
else:
# stop word logger thread
self.wordlogger.stop()
self.wordlogger.wordEntered.disconnect(self.searchEdit.setText)
# disable connections to UI elements
self.searchEdit.textEdited.disconnect(self.liveCheckBox.clear)
self.urlEdit.textEdited.disconnect(self.liveCheckBox.clear)
self.tagEdit.textEdited.disconnect(self.liveCheckBox.clear)
self.getUrlEdit.textEdited.disconnect(self.liveCheckBox.clear)
self.getUrlButton.clicked.disconnect(self.liveCheckBox.clear)
self.pasteUrlButton.clicked.disconnect(self.liveCheckBox.clear)
self.imagePicker.clicked.disconnect(self.liveCheckBox.clear)
self.statusBar().showMessage("Toggled word logger")
def downloadImage(self):
downloader = ImageDownloader(self)
downloader.failure.connect(self.showDownloadError)
downloader.success.connect(self.imagePicker.addImage)
downloader.success.connect(self.searchEdit.clear)
url = self.getUrlEdit.text()
downloader.get(url, self.imagePicker.getLocalFilepath(url))
self.statusBar().showMessage("Downloading %s" % url)
def showDownloadError(self, message):
mb = QMessageBox()
mb.setText(message)
mb.exec_()
def saveTags(self):
self.imagePicker.setTagsString(self.tagEdit.text())
self.statusBar().showMessage("Saved tags")
def test_instantiation(self):
i = ImageSearcher()
i = ImagePicker()