#!/usr/bin/env python2
"A web browser that will never exceed 128 lines of code. (not counting blanks)"
import sys, os, json, tempfile
from PyQt4 import QtGui, QtCore, QtWebKit, QtNetwork
settings = QtCore.QSettings("ralsina", "devicenzo")
class MainWindow(QtGui.QMainWindow):
def __init__(self):
QtGui.QMainWindow.__init__(self)
self.tabs = QtGui.QTabWidget(self, tabsClosable=True, movable=True, currentChanged=self.currentTabChanged, elideMode=QtCore.Qt.ElideRight, tabCloseRequested=lambda idx: self.tabs.widget(idx).deleteLater())
self.setCentralWidget(self.tabs)
self.bars = {}
self.star = QtGui.QAction(QtGui.QIcon.fromTheme("emblem-favorite"), "Bookmark", self, checkable=True, triggered=self.bookmarkPage, shortcut="Ctrl+d")
self.tabs.setCornerWidget(QtGui.QToolButton(self, text="New Tab", icon=QtGui.QIcon.fromTheme("document-new"), clicked=lambda: self.addTab().url.setFocus(), shortcut="Ctrl+t"))
self.addAction(QtGui.QAction("Full Screen", self, checkable=True, toggled=lambda v: self.showFullScreen() if v else self.showNormal(), shortcut="F11"))
self.bookmarks = self.get("bookmarks", {})
self.bookmarkPage() # Load the bookmarks menu
self.history = self.get("history", []) + self.bookmarks.keys()
self.completer = QtGui.QCompleter(QtCore.QStringList([QtCore.QString(u) for u in self.history]))
# Use a app-wide, persistent cookiejar
self.cookies = QtNetwork.QNetworkCookieJar(QtCore.QCoreApplication.instance())
self.cookies.setAllCookies([QtNetwork.QNetworkCookie.parseCookies(c)[0] for c in self.get("cookiejar", [])])
# Downloads bar at the bottom of the window
self.downloads = QtGui.QToolBar("Downloads")
self.addToolBar(QtCore.Qt.BottomToolBarArea, self.downloads)
# Proxy support
proxy_url = QtCore.QUrl(os.environ.get(‘http_proxy‘, ‘‘))
QtNetwork.QNetworkProxy.setApplicationProxy(QtNetwork.QNetworkProxy(QtNetwork.QNetworkProxy.HttpProxy if unicode(proxy_url.scheme()).startswith(‘http‘) else QtNetwork.QNetworkProxy.Socks5Proxy, proxy_url.host(), proxy_url.port(), proxy_url.userName(), proxy_url.password())) if ‘http_proxy‘ in os.environ else None
[self.addTab(QtCore.QUrl(u)) for u in self.get("tabs", [])]
def fetch(self, reply):
destination = QtGui.QFileDialog.getSaveFileName(self, "Save File", os.path.expanduser(os.path.join(‘~‘, unicode(reply.url().path()).split(‘/‘)[-1])))
if destination:
bar = QtGui.QProgressBar(format=‘%p% - ‘ + os.path.basename(unicode(destination)))
cancel = QtGui.QToolButton(bar, icon=QtGui.QIcon.fromTheme("process-stop"), clicked=reply.abort)
self.downloads.addWidget(bar)
reply.downloadProgress.connect(self.progress)
reply.finished.connect(self.finished)
self.bars[unicode(reply.url().toString())] = [bar, reply, unicode(destination), cancel]
def finished(self):
url = unicode(self.sender().url().toString())
bar, reply, fname, cancel = self.bars[url]
redirURL = unicode(reply.attribute(QtNetwork.QNetworkRequest.RedirectionTargetAttribute).toString())
del self.bars[url]
bar.deleteLater()
cancel.deleteLater()
if redirURL and redirURL != url:
return self.fetch(redirURL, fname)
with open(fname, ‘wb‘) as f:
f.write(str(reply.readAll()))
progress = lambda self, received, total: self.bars[unicode(self.sender().url().toString())][0].setValue(100. * received / total)
def closeEvent(self, ev):
self.put("history", self.history)
self.put("cookiejar", [str(c.toRawForm()) for c in self.cookies.allCookies()])
self.put("tabs", [unicode(self.tabs.widget(i).url.text()) for i in range(self.tabs.count())])
return QtGui.QMainWindow.closeEvent(self, ev)
def put(self, key, value):
"Persist an object somewhere under a given key"
settings.setValue(key, json.dumps(value))
settings.sync()
def get(self, key, default=None):
"Get the object stored under ‘key‘ in persistent storage, or the default value"
v = settings.value(key)
return json.loads(unicode(v.toString())) if v.isValid() else default
def addTab(self, url=QtCore.QUrl("")):
self.tabs.setCurrentIndex(self.tabs.addTab(Tab(url, self), ""))
return self.tabs.currentWidget()
def currentTabChanged(self, idx):
if self.tabs.widget(idx) is None:
return self.close()
self.setWindowTitle(self.tabs.widget(idx).wb.title() or "De Vicenzo")
def bookmarkPage(self, v=None):
if v and v is not None:
self.bookmarks[unicode(self.tabs.currentWidget().url.text())] = unicode(self.tabs.currentWidget().wb.title())
elif v is not None:
del (self.bookmarks[unicode(self.tabs.currentWidget().url.text())])
self.star.setMenu(QtGui.QMenu())
[self.star.menu().addAction(QtGui.QAction(title, self, triggered=lambda u=QtCore.QUrl(url): self.tabs.currentWidget().load(u))) for url, title in self.bookmarks.items()]
self.put(‘bookmarks‘, self.bookmarks)
def addToHistory(self, url):
self.history.append(url)
self.completer.setModel(QtGui.QStringListModel(list(set(self.bookmarks.keys() + self.history))))
class Tab(QtGui.QWidget):
def __init__(self, url, container):
self.container = container
QtGui.QWidget.__init__(self)
self.pbar = QtGui.QProgressBar(maximumWidth=120, visible=False)
self.wb = QtWebKit.QWebView(loadProgress=lambda v: (self.pbar.show(), self.pbar.setValue(v)) if self.amCurrent() else None, loadFinished=self.pbar.hide, loadStarted=lambda: self.pbar.show() if self.amCurrent() else None, titleChanged=lambda t: container.tabs.setTabText(container.tabs.indexOf(self), t) or (container.setWindowTitle(t) if self.amCurrent() else None), iconChanged=lambda: container.tabs.setTabIcon(container.tabs.indexOf(self), self.wb.icon()))
self.wb.page().networkAccessManager().setCookieJar(container.cookies)
self.wb.page().setForwardUnsupportedContent(True)
self.wb.page().unsupportedContent.connect(container.fetch)
self.wb.page().downloadRequested.connect(lambda req: container.fetch(self.page().networkAccessManager().get(req)))
self.setLayout(QtGui.QVBoxLayout(spacing=0))
self.layout().setContentsMargins(0, 0, 0, 0)
self.tb = QtGui.QToolBar("Main Toolbar", self)
self.layout().addWidget(self.tb)
self.layout().addWidget(self.wb)
for a, sc in [[QtWebKit.QWebPage.Back, "Alt+Left"], [QtWebKit.QWebPage.Forward, "Alt+Right"], [QtWebKit.QWebPage.Reload, "Ctrl+r"]]:
self.tb.addAction(self.wb.pageAction(a))
self.wb.pageAction(a).setShortcut(sc)
self.url = QtGui.QLineEdit(returnPressed=lambda: self.wb.setUrl(QtCore.QUrl.fromUserInput(self.url.text())))
self.url.setCompleter(container.completer)
self.tb.addWidget(self.url)
self.tb.addAction(container.star)
# FIXME: if I was seriously golfing, all of these can go in a single lambda
self.wb.urlChanged.connect(lambda u: self.url.setText(u.toString()))
self.wb.urlChanged.connect(lambda u: container.addToHistory(unicode(u.toString())))
self.wb.urlChanged.connect(lambda u: container.star.setChecked(unicode(u.toString()) in container.bookmarks) if self.amCurrent() else None)
# FIXME: do this using a tooltip
#self.wb.page().linkHovered.connect(lambda l: container.statusBar().showMessage(l, 3000))
self.search = QtGui.QLineEdit(visible=False, maximumWidth=200, returnPressed=lambda: self.wb.findText(self.search.text()), textChanged=lambda: self.wb.findText(self.search.text()))
self.showSearch = QtGui.QShortcut("Ctrl+F", self, activated=lambda: self.search.show() or self.search.setFocus())
self.hideSearch = QtGui.QShortcut("Esc", self, activated=lambda: (self.search.hide(), self.setFocus()))
self.wb.setLayout(QtGui.QVBoxLayout(spacing=0))
self.wb.layout().addWidget(self.search, 0, QtCore.Qt.AlignRight)
self.wb.layout().addStretch()
self.wb.layout().addWidget(self.pbar, 0, QtCore.Qt.AlignRight)
self.wb.layout().setContentsMargins(3, 3, 25, 3)
self.do_close = QtGui.QShortcut("Ctrl+W", self, activated=lambda: container.tabs.removeTab(container.tabs.indexOf(self)))
self.do_quit = QtGui.QShortcut("Ctrl+q", self, activated=lambda: container.close())
self.zoomIn = QtGui.QShortcut("Ctrl++", self, activated=lambda: self.wb.setZoomFactor(self.wb.zoomFactor() + 0.2))
self.zoomOut = QtGui.QShortcut("Ctrl+-", self, activated=lambda: self.wb.setZoomFactor(self.wb.zoomFactor() - 0.2))
self.zoomOne = QtGui.QShortcut("Ctrl+0", self, activated=lambda: self.wb.setZoomFactor(1))
self.urlFocus = QtGui.QShortcut("Ctrl+l", self, activated=self.url.setFocus)
self.previewer = QtGui.QPrintPreviewDialog(paintRequested=self.wb.print_)
self.do_print = QtGui.QShortcut("Ctrl+p", self, activated=self.previewer.exec_)
self.wb.settings().setAttribute(QtWebKit.QWebSettings.PluginsEnabled, True)
self.wb.settings().setIconDatabasePath(tempfile.mkdtemp())
self.wb.load(url)
amCurrent = lambda self: self.container.tabs.currentWidget() == self
createWindow = lambda self, windowType: self.container.addTab()
if __name__ == "__main__":
app = QtGui.QApplication(sys.argv)
wb = MainWindow()
for url in sys.argv[1:]:
wb.addTab(QtCore.QUrl.fromUserInput(url))
if wb.tabs.count() == 0:
wb.addTab(QtCore.QUrl(‘http://devicenzo.googlecode.com‘))
wb.show()
sys.exit(app.exec_())
