youtube-upload/youtube_upload/auth/webkit_qt.py

CHECK_AUTH_JS = """
    var code = document.getElementById("code");
    var access_denied = document.getElementById("access_denied");
    var result;
    
    if (code) {
        result = {authorized: true, code: code.value};
    } else if (access_denied) {
        result = {authorized: false, message: access_denied.innerText};
    } else {
        result = {};
    }
    result;
"""

def _on_qt_page_load_finished(dialog, webview):
    to_s = lambda x: (str(x.toUtf8()) if hasattr(x,'toUtf8') else x)
    frame = webview.page().currentFrame()
    try: #PySide does not QStrings
        from QtCore import QString
        jscode = QString(CHECK_AUTH_JS)
    except ImportError:
        jscode = CHECK_AUTH_JS
    res = frame.evaluateJavaScript(jscode)
    try:
        authorization = dict((to_s(k), to_s(v)) for (k, v) in res.toPyObject().items())
    except AttributeError: #PySide returns the result in pure Python
        authorization = dict((to_s(k), to_s(v)) for (k, v) in res.items())
    if "authorized" in authorization:
        dialog.authorization_code = authorization.get("code")
        dialog.close()
   
def get_code(url, size=(640, 480), title="Google authentication"):
    """Open a QT webkit window and return the access code."""
    try:
        from PyQt4 import QtCore, QtGui, QtWebKit
    except ImportError:
        from PySide import QtCore, QtGui, QtWebKit
    app = QtGui.QApplication([])
    dialog = QtGui.QDialog()
    dialog.setWindowTitle(title)
    dialog.resize(*size)
    webview = QtWebKit.QWebView()
    webpage = QtWebKit.QWebPage()
    webview.setPage(webpage)           
    webpage.loadFinished.connect(lambda: _on_qt_page_load_finished(dialog, webview))
    webview.setUrl(QtCore.QUrl.fromEncoded(url))
    layout = QtGui.QGridLayout()
    layout.addWidget(webview)
    dialog.setLayout(layout)
    dialog.authorization_code = None
    dialog.show()
    app.exec_()
    return dialog.authorization_code