browser.py 749 B

12345678910111213141516171819202122
  1. import lib
  2. try:
  3. from youtube_upload.auth import webkit_qt
  4. WEBKIT_BACKEND = "qt"
  5. except ImportError:
  6. from youtube_upload.auth import webkit_gtk
  7. WEBKIT_BACKEND = "gtk"
  8. except ImportError:
  9. WEBKIT_BACKEND = None
  10. def get_code(url, size=(640, 480), title="Google authentication"):
  11. if WEBKIT_BACKEND == "qt":
  12. lib.debug("Using webkit backend: QT")
  13. with lib.default_sigint():
  14. return webkit_qt.get_code(url, size=size, title=title)
  15. elif WEBKIT_BACKEND == "gtk":
  16. lib.debug("Using webkit backend: GTK")
  17. with lib.default_sigint():
  18. return webkit_gtk.get_code(url, size=size, title=title)
  19. else:
  20. raise NotImplementedError("GUI auth requires pywebkitgtk or qtwebkit")