import requests
from bs4 import BeautifulSoup
[docs]class Error(Exception):
"""Base class for exceptions in this module."""
pass
[docs]class MaintenanceError(Error):
pass
[docs]class Passbuy:
nif_jar = None
bp_jar = None
person_id = None
def __init__(self, username, password, realm='minidrett', verify=True):
self.username = username
self.password = password
self.realm = realm
self.person_id = None
if realm in ['mi', 'minidrett']:
self.realm = 'minidrett'
self.login_page = 'Login'
elif realm in ['ka', 'klubbadmin']:
self.realm = 'ka'
self.login_page = 'Home/Login'
elif realm in ['sa', 'sportsadmin']:
self.realm = 'sa'
self.login_page = ''
else:
self.login_page = ''
# Simple logic to verify not maintance
if verify:
if self.is_maintanance():
raise MaintenanceError('{} is down for maintance'.format(self.realm))
[docs] def is_maintanance(self):
"""Check if maintanance mode
NIF web (KA/MI/SA) gives normal http 200, need to check title
Also: re.search('(?<=<title>).+?(?=</title>)', mytext, re.DOTALL).group().strip()
:returns boolean is_maintanance:
"""
r = requests.get('https://{}.nif.no'.format(self.realm))
if r.status_code == 503:
return True
elif r.status_code == 200:
html = BeautifulSoup(r.text, 'lxml')
if html.title.text.strip() == 'Vedlikehold':
return True
elif html.title.text.strip().startswith('Release'):
return True
return False
[docs] def login(self):
return self.minidrett()
[docs] def nif_realm(self):
r = requests.get('https://{}.nif.no/'.format(self.realm))
self.nif_jar = r.cookies
resp = requests.get('https://{}.nif.no/{}'.format(self.realm, self.login_page),
cookies=self.nif_jar,
allow_redirects=False)
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, resp.cookies)
if resp.status_code == 302:
r1 = requests.get(resp.headers.get('Location', ''),
cookies=self.nif_jar,
allow_redirects=False)
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, r1.cookies)
if r1.status_code == 302:
r2 = requests.get(r1.headers.get('Location', ''),
cookies=self.nif_jar,
allow_redirects=False)
cookiejar = requests.cookies.merge_cookies(self.nif_jar, r2.cookies)
if r2.status_code == 302:
r3 = requests.get('https://id.nif.no{}'.format(r2.headers.get('Location', '')),
cookies=self.nif_jar,
allow_redirects=False)
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, r3.cookies)
return True, r3
return False, None
[docs] def buypass(self):
"""Returns
login url
challenge
"""
status, mi = self.nif_realm()
if status is True:
r = requests.get(mi.headers.get('Location', ''),
allow_redirects=False)
if r.status_code == 200:
self.bp_jar = r.cookies
bp_html = BeautifulSoup(r.text, 'lxml')
login_url = bp_html.find('form', attrs={'class': 'nif-login-form'}).get_attribute_list('action')[0]
challenge = bp_html.find('input', attrs={'name': 'challenge'}).get_attribute_list('value')[0]
# Login
login = requests.post(url=login_url, data={'challenge': challenge,
'password': self.password,
'username': self.username,
'rememberMe': 'off',
'authMethod': ''},
cookies=self.bp_jar,
allow_redirects=False
)
if login.status_code == 200:
self.bp_jar = requests.cookies.merge_cookies(self.bp_jar, login.cookies)
return True, login
return False, None
[docs] def nif_id(self):
status, buypass = self.buypass()
if status is True:
bp_html = BeautifulSoup(buypass.text, 'lxml')
login_url = bp_html.find('form').get_attribute_list('action')[0]
code = bp_html.find('input', attrs={'name': 'code'}).get_attribute_list('value')[0]
state = bp_html.find('input', attrs={'name': 'state'}).get_attribute_list('value')[0]
session_state = bp_html.find('input', attrs={'name': 'session_state'}).get_attribute_list('value')[0]
resp = requests.post(url=login_url,
data={'code': code,
'state': state,
'session_state': session_state},
cookies=self.nif_jar,
allow_redirects=False)
if resp.status_code == 302:
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, resp.cookies)
callback = requests.get(url='https://id.nif.no{}'.format(resp.headers.get('Location', '')),
cookies=self.nif_jar,
allow_redirects=False
)
if callback.status_code == 302:
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, callback.cookies)
connect = requests.get(url='https://id.nif.no{}'.format(callback.headers.get('Location', '')),
cookies=self.nif_jar,
allow_redirects=False
)
if connect.status_code == 200:
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, connect.cookies)
mi_html = BeautifulSoup(connect.text, 'lxml')
id_url = mi_html.find('form').get_attribute_list('action')[0]
id_token = mi_html.find('input', attrs={'name': 'id_token'}).get_attribute_list('value')[0]
scope = mi_html.find('input', attrs={'name': 'scope'}).get_attribute_list('value')[0]
state = mi_html.find('input', attrs={'name': 'state'}).get_attribute_list('value')[0]
session_state = \
mi_html.find('input', attrs={'name': 'session_state'}).get_attribute_list('value')[0]
resp = requests.post(url=id_url,
data={'id_token': id_token,
'scope': scope,
'state': state,
'session_state': session_state},
cookies=self.nif_jar,
allow_redirects=False)
if resp.status_code == 302:
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, resp.cookies)
return True, resp
return False, None
[docs] def minidrett(self):
status, nif_id = self.nif_id()
if status is True:
resp = requests.get(url=nif_id.headers.get('Location', ''),
cookies=self.nif_jar,
allow_redirects=False
)
if resp.status_code == 302:
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, resp.cookies)
if self.realm in ['mi', 'minidrett']:
# Get profile and find person_id
profile = requests.get(url='https://minidrett.nif.no/MyProfile/Profiles',
cookies=self.nif_jar,
allow_redirects=False
)
if profile.status_code == 200:
# soup = BeautifulSoup(profile.text, 'lxml')
# profile_img_id = soup.find(alt='Profilbilde')['id']
# self.person_id = int(profile_img_id.split('_')[1])
try:
self.person_id = int(
profile.text.split('onclick="javaScript:DownloadCV(')[1].split(');"')[0])
except:
self.person_id = None
return True, self.person_id, self.nif_jar
return False, None, None
[docs] def ka_login(self):
if self.person_id is not None:
r = requests.get('https://ka.nif.no/Members', cookies=self.nif_jar, allow_redirects=False)
if r.status_code == 302:
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, r.cookies)
frm = requests.get(r.headers.get('Location', ''),
cookies=self.nif_jar,
allow_redirects=False)
if frm.status_code == 200:
self.nif_jar = requests.cookies.merge_cookies(self.nif_jar, frm.cookies)
ka_html = BeautifulSoup(frm.text, 'lxml')
id_url = ka_html.find('form').get_attribute_list('action')[0]
id_token = ka_html.find('input', attrs={'name': 'id_token'}).get_attribute_list('value')[0]
scope = ka_html.find('input', attrs={'name': 'scope'}).get_attribute_list('value')[0]
state = ka_html.find('input', attrs={'name': 'state'}).get_attribute_list('value')[0]
