-
Notifications
You must be signed in to change notification settings - Fork 0
/
auth.py
107 lines (94 loc) · 3.84 KB
/
auth.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
from http.cookiejar import CookieJar
from html.parser import HTMLParser
from urllib.parse import urlencode, parse_qs
from urllib.request import Request, build_opener, HTTPRedirectHandler, \
HTTPCookieProcessor # HTTPRedirectHandler?
from getpass import getpass
def getAuthURL(client_id, scope):
params = {'client_id': client_id,
'scope': scope,
'display': 'mobile',
'response_type': 'token',
'v': '5.37',
'redirect_uri': 'https://oauth.vk.com/blank.html'}
return 'https://oauth.vk.com/authorize?'+urlencode(params)
class AuthFormParser(HTMLParser):
def __init__(self):
HTMLParser.__init__(self)
self.url = None
self.params = {}
self.in_form = False
self.form_parsed = False
self.method = "GET"
def handle_starttag(self, tag, attrs):
tag = tag.lower()
if tag == "form":
if self.form_parsed:
raise RuntimeError("Second form on page")
if self.in_form:
raise RuntimeError("Already in form")
self.in_form = True
if not self.in_form:
return
attrs = dict((name.lower(), value) for name, value in attrs)
if tag == "form":
self.url = attrs["action"]
if "method" in attrs:
self.method = attrs["method"]
elif tag == "input" and "type" in attrs and "name" in attrs:
if attrs["type"] in ["hidden", "text", "password"]:
self.params[attrs["name"]] = \
attrs["value"] if "value" in attrs else ""
def handle_endtag(self, tag):
tag = tag.lower()
if tag == "form":
if not self.in_form:
raise RuntimeError("Unexpected end of <form>")
self.in_form = False
self.form_parsed = True
def clear(self):
self.url = None
self.params = {}
self.in_form = False
self.form_parsed = False
self.method = "GET"
def auth(client_id, scope, email, password,
user_agent='Mozilla/5.0 (Windows NT 6.3; WOW64; rv:39.0)'
' Gecko/20100101 Firefox/39.0'):
"""Portable and quick emulator of browser for Vk API authorisation."""
user_agent = ('User-Agent', user_agent)
opener = build_opener(HTTPCookieProcessor(CookieJar()))
parser = AuthFormParser()
request = Request(getAuthURL(client_id, scope))
request.add_header(*user_agent)
parser.feed(opener.open(request).read().decode('utf8'))
request = Request(parser.url)
request.add_header(*user_agent)
if 'email' in parser.params and "pass" in parser.params:
parser.params["email"] = email
parser.params["pass"] = password
data = urlencode(parser.params).encode('ascii')
parser.clear()
response = opener.open(request, data)
if 'access_token' in response.geturl():
final_url = response.geturl()
else:
parser.feed(response.read().decode('utf8'))
request = Request(parser.url)
request.add_header(*user_agent)
data = urlencode(parser.params).encode('ascii')
final_url = opener.open(request, data).geturl()
return parse_qs(final_url.split('#')[1])
else:
raise Exception('Wrong page')
if __name__ == '__main__':
# Добавить обработку параметров командной строки
data = auth(input('client_id> '),
input('scope> '),
input('email> '),
getpass('pass> '))
for key in data:
print('{0: <{2}}: {1}'.format(key, data[key],
max(len(k) for k in data)))