95 lines
2.7 KiB
Python
95 lines
2.7 KiB
Python
import os
|
|
import sys
|
|
import time
|
|
import json
|
|
import threading
|
|
from urllib.parse import urlparse, parse_qs, unquote
|
|
|
|
import webview
|
|
|
|
|
|
def extract_token(url_string):
|
|
token = None
|
|
expires_in = 3600
|
|
parsed = urlparse(url_string)
|
|
|
|
if parsed.fragment:
|
|
params = parse_qs(parsed.fragment)
|
|
else:
|
|
params = parse_qs(parsed.query)
|
|
|
|
if 'access_token' in params:
|
|
token = params['access_token'][0]
|
|
if 'expires_in' in params:
|
|
try:
|
|
expires_in = int(params['expires_in'][0])
|
|
except ValueError:
|
|
pass
|
|
|
|
if not token:
|
|
start_marker = "access_token%253D"
|
|
end_marker = "%25"
|
|
|
|
start_index = url_string.find(start_marker)
|
|
if start_index != -1:
|
|
token_start_index = start_index + len(start_marker)
|
|
remaining_url = url_string[token_start_index:]
|
|
end_index = remaining_url.find(end_marker)
|
|
|
|
if end_index != -1:
|
|
raw_token = remaining_url[:end_index]
|
|
else:
|
|
amp_index = remaining_url.find('&')
|
|
if amp_index != -1:
|
|
raw_token = remaining_url[:amp_index]
|
|
else:
|
|
raw_token = remaining_url
|
|
token = unquote(raw_token)
|
|
|
|
return token, expires_in
|
|
|
|
|
|
def main_auth(auth_url, output_path):
|
|
def poll_url():
|
|
try:
|
|
url = window.get_current_url()
|
|
except Exception:
|
|
url = None
|
|
if url:
|
|
token, expires_in = extract_token(url)
|
|
if token:
|
|
data = {"token": token, "expires_in": expires_in}
|
|
with open(output_path, "w", encoding="utf-8") as f:
|
|
json.dump(data, f)
|
|
window.destroy()
|
|
return
|
|
threading.Timer(0.5, poll_url).start()
|
|
|
|
def on_loaded():
|
|
threading.Timer(0.5, poll_url).start()
|
|
|
|
window = webview.create_window("VK Авторизация", auth_url)
|
|
window.events.loaded += on_loaded
|
|
storage_path = os.path.join(os.path.dirname(output_path), "webview_profile")
|
|
webview.start(private_mode=False, storage_path=storage_path)
|
|
|
|
|
|
def main():
|
|
# Supports both: `python auth_webview.py <auth_url> <output_path>`
|
|
# and: `python auth_webview.py --auth <auth_url> <output_path>`
|
|
args = sys.argv[1:]
|
|
if len(args) == 3 and args[0] == "--auth":
|
|
auth_url, output_path = args[1], args[2]
|
|
elif len(args) == 2:
|
|
auth_url, output_path = args[0], args[1]
|
|
else:
|
|
print("Usage: auth_webview.py [--auth] <auth_url> <output_path>")
|
|
return 1
|
|
|
|
main_auth(auth_url, output_path)
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|