Source code for eye.widgets.misc

# this project is licensed under the WTFPLv2, see COPYING.txt for details

import logging
from weakref import ref

from PyQt5.QtCore import QEventLoop
from PyQt5.QtWidgets import QPlainTextEdit, QLabel, QWidget, QRubberBand, QApplication

from eye.app import qApp
from eye.qt import Slot, Signal
from eye.widgets.helpers import WidgetMixin

__all__ = ('LogWidget', 'PositionIndicator', 'WidgetPicker', 'interactiveWidgetPick')




class LogWidget(QPlainTextEdit):


	class LogHandler(logging.Handler):
		def __init__(self, widget):
			super(LogWidget.LogHandler, self).__init__()
			self.widget = widget
			self.setFormatter(logging.Formatter('%(asctime)s %(message)s'))



		def emit(self, record):
			self.widget.appendPlainText(self.format(record))



	def __init__(self, parent=None):
		super().__init__(parent=parent)
		self.handler = LogWidget.LogHandler(self)
		self.setReadOnly(True)



	def install(self):
		qApp().logger.addHandler(self.handler)




	def uninstall(self):
		qApp().logger.removeHandler(self.handler)






class PositionIndicator(QLabel, WidgetMixin):
	"""Widget indicating cursor position of currently focused editor

	When cursor position changes or focus goes to another editor widget, the text of this label is refreshed.
	"""

	format = '{percent:3.0f}% {line:5d}:{vcol:3d}'

	"""Text format of the label

	Uses PEP-3101 string formatting. Usable keys are `line`, `col`, `percent`, `offset`, `path`, `title` and `editor`.
	"""

	def __init__(self, format=None, **kwargs):
		super().__init__(**kwargs)
		if format is not None:
			self.format = format

		self.last_focus = lambda: None

		qApp().focusChanged.connect(self.focus_changed)



	@Slot('QWidget*', 'QWidget*')
	def focus_changed(self, _, new):
		if not hasattr(new, 'categories'):
			return
		if 'editor' not in new.categories():
			return
		if new.window() != self.window():
			return

		last_focus = self.last_focus()
		if last_focus:
			last_focus.cursorPositionChanged.disconnect(self.update_label)
			last_focus.linesChanged.disconnect(self.update_label)

		self.last_focus = ref(new)
		new.cursorPositionChanged.connect(self.update_label)
		new.linesChanged.connect(self.update_label)
		self.update_label()




	@Slot()
	def update_label(self):
		ed = self.last_focus()

		line, col = ed.cursor_position()
		offset = ed.cursor_offset()
		line, col = line + 1, col + 1
		lines = ed.lines()

		d = {
			'line': line,
			'col': col,
			'vcol': ed.cursor_visual_column() + 1,
			'percent': line * 100. / lines,
			'offset': offset,
			'path': ed.path,
			'title': ed.windowTitle(),
			'editor': ed,
		}
		self.setText(self.format.format(**d))






class WidgetPicker(QWidget):
	"""Widget for letting user point at another widget."""

	selected = Signal()

	def __init__(self):
		super().__init__()
		self.band = QRubberBand(QRubberBand.Rectangle)
		self.setMouseTracking(True)
		self.el = QEventLoop()



	def mousePressEvent(self, ev):
		self.el.quit()
		self.widget = QApplication.widgetAt(ev.globalPos())
		self.band.hide()




	def mouseMoveEvent(self, ev):
		widget = QApplication.widgetAt(ev.globalPos())
		if widget:
			rect = widget.frameGeometry()
			if widget.parent():
				rect.moveTo(widget.parent().mapToGlobal(rect.topLeft()))
			self.band.setGeometry(rect)
			self.band.show()
		else:
			self.band.hide()




	def run(self):
		self.grabMouse()
		try:
			self.el.exec_()
		finally:
			self.releaseMouse()
		return self.widget






def interactiveWidgetPick():
	"""Let user peek a widget by clicking on it.

	The user can point at open EYE widgets and click on one. Return the widget that was clicked
	by the user.
	"""
	w = WidgetPicker()
	return w.run()