-
Notifications
You must be signed in to change notification settings - Fork 20
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #27 from pedohorse/dev
python3 support implemented
- Loading branch information
Showing
41 changed files
with
4,921 additions
and
8 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
|
@@ -3,3 +3,4 @@ | |
**/*.pyc | ||
*token.tok | ||
venv | ||
venv3 |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,194 @@ | ||
import re | ||
import json | ||
from urllib import request | ||
from .nethelper import urlopen_nt | ||
try: | ||
from PySide2.QtWidgets import QDialog, QVBoxLayout, QLabel, QSizePolicy, QPushButton, QMessageBox | ||
from PySide2.QtWebEngineWidgets import QWebEngineView, QWebEngineProfile, QWebEnginePage | ||
from PySide2.QtCore import Slot, Qt, QUrl | ||
except ImportError: | ||
raise NotImplementedError('web auth implemented only for QT5. Sorry, people who still use houdini 16.5. You will have to create access token manually. contact me to ask how.') | ||
|
||
import time | ||
|
||
|
||
class QGithubDeviceAuthDialog(QDialog): | ||
def __init__(self, client_id='42e8e8e9d844e45c2d05', hint_username=None, parent=None): | ||
super(QGithubDeviceAuthDialog, self).__init__(parent=parent) | ||
self.setWindowTitle('Log into your GitHub account and enter this code') | ||
|
||
self.__webview = QWebEngineView(parent=self) | ||
self.__webview.setSizePolicy(QSizePolicy.Expanding, QSizePolicy.Expanding) | ||
self.__webview.urlChanged.connect(self.on_url_changed) | ||
|
||
self.__infolaabel = QLabel('<p style="font-size:14px">' | ||
'<br>You need to allow hpaste to modify your gists on github. for that you need to log in to your account and authorize hpaste.\n' | ||
'<br>You can do it in <b>any</b> browser, not only in this window. Just go to <a href="https://github.com/login/device">https://github.com/login/device</a> and enter the code below.\n' | ||
'<br>close this window when you are done' | ||
'</p>', parent=self) | ||
self.__infolaabel.setTextFormat(Qt.RichText) | ||
self.__infolaabel.setTextInteractionFlags(Qt.TextBrowserInteraction) | ||
self.__infolaabel.setOpenExternalLinks(True) | ||
self.__devidlabel = QLabel(parent=self) | ||
self.__devidlabel.setTextInteractionFlags(Qt.TextBrowserInteraction) | ||
ff = self.__devidlabel.font() | ||
ff.setPointSize(64) | ||
self.__devidlabel.setFont(ff) | ||
|
||
self.__reload_button = QPushButton('log in to a different account', parent=self) | ||
self.__reload_button.clicked.connect(self.reload_button_clicked) | ||
|
||
self.__layout = QVBoxLayout() | ||
self.setLayout(self.__layout) | ||
self.__layout.addWidget(self.__infolaabel) | ||
self.__layout.addWidget(self.__devidlabel) | ||
self.__layout.addWidget(self.__reload_button) | ||
self.__layout.addWidget(self.__webview) | ||
self.__result = None | ||
|
||
# self.setGeometry(512, 256, 1024, 768) | ||
self.resize(1024, 860) | ||
|
||
# init auth process | ||
self.__client_id = client_id | ||
self.__headers = {'User-Agent': 'HPaste', 'Accept': 'application/json', 'charset': 'utf-8', 'Content-Type': 'application/json'} | ||
self.__hint_username = hint_username | ||
|
||
self.__webprofile = None | ||
self.__webpage = None | ||
|
||
self.__device_code = None | ||
self.__interval = None | ||
self.__await_login_redirect = False | ||
self.reinit_code() | ||
|
||
def reinit_code(self): | ||
reqdata = {'client_id': self.__client_id, | ||
'scope': 'gist'} | ||
req = request.Request('https://github.com/login/device/code', data=json.dumps(reqdata).encode('UTF-8'), headers=self.__headers) | ||
req.get_method = lambda: 'POST' | ||
code, ret = urlopen_nt(req) | ||
if code != 200: | ||
raise RuntimeError('code %d when trying to register device' % code) | ||
init_data = json.loads(ret.read().decode('UTF-8')) | ||
print(init_data) | ||
self.__device_code = init_data['device_code'] | ||
self.__interval = init_data.get('interval', 5) | ||
url = init_data['verification_uri'] | ||
|
||
self.__await_login_redirect = True | ||
self.__webprofile = QWebEngineProfile(parent=self.__webview) | ||
self.__webpage = QWebEnginePage(self.__webprofile, parent=self.__webview) # just to be sure they are deleted in proper order | ||
self.__webview.setPage(self.__webpage) | ||
self.__webview.load(QUrl(url)) | ||
self.__devidlabel.setText('code: %s' % (init_data['user_code'],)) | ||
|
||
def get_result(self): | ||
return self.__result | ||
|
||
def closeEvent(self, event): | ||
# here we assume user has done his part, so lets get checking | ||
for attempt in range(5): | ||
reqdata = {'client_id': self.__client_id, | ||
'device_code': self.__device_code, | ||
'grant_type': 'urn:ietf:params:oauth:grant-type:device_code'} | ||
req = request.Request('https://github.com/login/oauth/access_token', data=json.dumps(reqdata).encode('UTF-8'), headers=self.__headers) | ||
req.get_method = lambda: 'POST' | ||
code, ret = urlopen_nt(req) | ||
if code == 200: | ||
rep = json.loads(ret.read().decode('UTF-8')) | ||
print(rep) | ||
if 'error' not in rep: # a SUCC | ||
headers = {'User-Agent': 'HPaste', | ||
'Authorization': 'Token %s' % rep['access_token'], | ||
'Accept': 'application/vnd.github.v3+json'} | ||
req = request.Request(r'https://api.github.com/user', headers=headers) | ||
usercode, userrep = urlopen_nt(req) | ||
if usercode != 200: | ||
raise RuntimeError('could not probe! %d' % (usercode,)) | ||
userdata = json.loads(userrep.read().decode('UTF-8')) | ||
print(userdata) | ||
|
||
self.__result = {'token': rep['access_token'], | ||
'user': userdata['login']} | ||
break | ||
|
||
# NO SUCC | ||
errcode = rep['error'] | ||
if errcode == 'authorization_pending': | ||
# note that this error will happen if user just closes down the window | ||
break | ||
elif errcode == 'slow_down': | ||
self.__interval = rep.get('interval', self.__interval + 5) | ||
time.sleep(self.__interval) | ||
continue | ||
elif errcode == 'expired_token': | ||
if QMessageBox.warning(self, 'device code expired', 'it took you too long to enter the code, now it\'s expired.\nWant to retry?', buttons=QMessageBox.Yes | QMessageBox.No) == QMessageBox.Yes: | ||
event.reject() | ||
self.reinit_code() | ||
return | ||
break | ||
elif errcode == 'unsupported_grant_type': | ||
raise RuntimeError('unsupported grant type. probably github changed API. need to update the plugin') | ||
elif errcode == 'incorrect_client_credentials': | ||
raise RuntimeError('incorect client id. probably pedohorse changed hpaste id for some reason. update the plugin') | ||
elif errcode == 'incorrect_device_code': | ||
if QMessageBox.warning(self, 'bad device code', 'server reported wrong device code\nWant to retry?', buttons=QMessageBox.Yes | QMessageBox.No) == QMessageBox.Yes: | ||
event.reject() | ||
self.reinit_code() | ||
return | ||
break | ||
elif errcode == 'access_denied': | ||
# means user denied the request | ||
break | ||
else: | ||
raise RuntimeError('unexpected error: %s' % (json.dumps(rep),)) | ||
else: | ||
raise RuntimeError('bad return code. server reported with bad return code %d' % code) | ||
else: | ||
if QMessageBox.warning(self, 'unknown error', 'could not manage to check authorization in reasonable amount of attempts\nWant to retry?', buttons=QMessageBox.Yes | QMessageBox.No) == QMessageBox.Yes: | ||
event.reject() | ||
self.reinit_code() | ||
return | ||
|
||
return super(QGithubDeviceAuthDialog, self).closeEvent(event) | ||
|
||
@Slot(object) | ||
def on_url_changed(self, qurl): | ||
url = qurl.toString() | ||
print(url) | ||
if not self.__await_login_redirect: | ||
return | ||
if qurl.path() != '/login': | ||
return | ||
self.__await_login_redirect = False | ||
if self.__hint_username is not None: | ||
if qurl.hasQuery(): | ||
qurl.setQuery('&'.join((qurl.query(), 'login=%s' % self.__hint_username))) | ||
else: | ||
qurl.setQuery('login=%s' % self.__hint_username) | ||
print('redir to %s' % qurl.toString) | ||
self.__webview.load(qurl) | ||
|
||
@Slot() | ||
def reload_button_clicked(self): | ||
self.reinit_code() | ||
|
||
|
||
if __name__ == '__main__': # testing | ||
import sys | ||
import string | ||
import random | ||
from PySide2.QtWidgets import QApplication | ||
qapp = QApplication(sys.argv) | ||
# w = QWebAuthDialog('https://www.google.com', r'https://www.google.com/search\?(.*)') | ||
webauthstate = ''.join(random.choice(string.ascii_letters) for _ in range(32)) | ||
webauthparms = {'client_id': '42e8e8e9d844e45c2d05', | ||
'redirect_uri': 'https://github.com/login/oauth/success', | ||
'scope': 'gist', | ||
'state': webauthstate} | ||
w = QGithubDeviceAuthDialog(client_id='42e8e8e9d844e45c2d05', hint_username='ololovich', parent=None) | ||
res = w.exec_() | ||
print(res == QGithubDeviceAuthDialog.Accepted) | ||
print(w.get_result()) | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,63 @@ | ||
raise NotImplementedError() | ||
# TODO: Ran into temporary deadend: This requires exposing client_secret for github authentication... so no point using it now | ||
# TODO: need to implement something like device flow, but it is in beta currently | ||
# TODO: or enter personal access token manually | ||
|
||
import re | ||
try: | ||
from PySide2.QtWidgets import QDialog, QVBoxLayout | ||
from PySide2.QtWebEngineWidgets import QWebEngineView | ||
from PySide2.QtCore import QUrl | ||
except ImportError: | ||
raise NotImplementedError('web auth implemented only for QT5. Sorry, people who still use houdini 16.5') | ||
|
||
|
||
class QWebAuthDialog(QDialog): | ||
def __init__(self, url, success_re, parent=None): | ||
super(QWebAuthDialog, self).__init__(parent=parent) | ||
|
||
self.__webview = QWebEngineView(parent=self) | ||
self.__webview.load(QUrl(url)) | ||
self.__succre = re.compile(success_re) | ||
self.__webview.urlChanged.connect(self.on_url_changed) | ||
|
||
self.__layout = QVBoxLayout() | ||
self.setLayout(self.__layout) | ||
self.__layout.addWidget(self.__webview) | ||
self.__result = None | ||
|
||
def get_result(self): | ||
return self.__result | ||
|
||
def on_url_changed(self, qurl): | ||
url = qurl.toString() | ||
match = self.__succre.match(url) | ||
print(url, match) | ||
if match: | ||
self.__result = match | ||
self.accept() | ||
|
||
|
||
if __name__ == '__main__': # testing | ||
import sys | ||
import string | ||
import random | ||
from PySide2.QtWidgets import QApplication | ||
qapp = QApplication(sys.argv) | ||
# w = QWebAuthDialog('https://www.google.com', r'https://www.google.com/search\?(.*)') | ||
webauthstate = ''.join(random.choice(string.ascii_letters) for _ in range(32)) | ||
webauthparms = {'client_id': '42e8e8e9d844e45c2d05', | ||
'redirect_uri': 'https://github.com/login/oauth/success', | ||
'scope': 'gist', | ||
'state': webauthstate} | ||
w = QWebAuthDialog(url='https://github.com/login/oauth/authorize?' + | ||
'&'.join('%s=%s' % (k, v) for k, v in webauthparms.iteritems()), | ||
success_re=r'https://github.com/login/oauth/success\?(.*)', | ||
parent=None) | ||
w.setGeometry(512, 256, 1024, 768) | ||
res = w.exec_() | ||
print(res == QWebAuthDialog.Accepted) | ||
print(w.get_result()) | ||
if res == QWebAuthDialog.Accepted: | ||
print(w.get_result().groups()) | ||
# qapp.exec_() |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,5 @@ | ||
try: | ||
from .hpaste import * | ||
except: | ||
print("couldnt init hpaste") | ||
from .hpasteweb import * |
Oops, something went wrong.