#!/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 setTab(self, url=QtCore.QUrl("")): self.tab = Tab(url, self) self.setCentralWidget(self.tab) return self.tab 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.wb.page().setLinkDelegationPolicy(QtWebKit.QWebPage.DelegateExternalLinks) self.wb.linkClicked.connect(lambda url: QtGui.QDesktopServices.openUrl(url)) 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)) wb.setTab(QtCore.QUrl.fromUserInput(url)) if wb.tabs.count() == 0: #wb.addTab(QtCore.QUrl('http://devicenzo.googlecode.com')) wb.setTab(QtCore.QUrl('http://facebook.com')) wb.show() sys.exit(app.exec_())