#!/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_())

pyqt练习x3.21