initial commit

facebook

@@ -0,0 +1,179 @@
+#!/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_())