improvements (78909c89) · Commits · Zeyu DONG / CAS_Auto_Login

.gitignore

+3 −1

Original line number	Diff line number	Diff line
		config.json
		*.xml
		.idea/CAS_Auto_Login.iml
		No newline at end of file

CAS_Auto_Login.py

+45 −62

Original line number	Diff line number	Diff line
		import requests
		from bs4 import BeautifulSoup
		from time import sleep
		import time
		import json
		import sys
		import os
		import datetime
		import logging

		config = None


		def log(s):
		ms = int(datetime.datetime.now().microsecond / 1000)
		ms = '%03d' % ms
		print('[' + time.strftime('%Y-%m-%d %H:%M:%S', time.localtime(time.time())) + '.' + ms + '] ' + s)
		logging.basicConfig(format="%(asctime)s - %(levelname)s - %(message)s", level=logging.INFO)


		def load_config():
		@@ -23,60 +18,48 @@ def load_config():


		def main():
		log("Program started.")
		logging.info('Program started.')
		os.chdir(os.path.dirname(sys.argv[0])) # To read config in the same directory
		log('Reading configurations...')
		logging.info('Reading configurations...')
		config = load_config()
		times_retry_login = config['max_times_retry_login']
		log('Configurations successfully imported.')
		test_url = config['captive_portal_server'] + '/generate_204'
		logging.info('Configurations successfully imported.')
		while True:
		log('Checking network status...')
		logging.info('Checking network status...')
		try:
		login = requests.Session()
		test = login.get(config['testUrl'])
		test = login.get(test_url)
		except:
		log('Connection FAILED. Try again in ' + str(config['interval_retry_connection']) + ' sec.')
		logging.info('Connection FAILED. Try again in ' + str(config['interval_retry_connection']) + ' sec.')
		sleep(config['interval_retry_connection'])
		continue
		while test.status_code != 204:

		log('You are offline. Starting login...')
		log('Start to get login information')

		start = test.text.find(r'action') + 8
		end = test.text.find(r'" method="post"')
		action = test.text[start:end]
		log('action= ' + action + ';')

		start = test.text.find(r'name="lt"') + 17
		lt = test.text[start:]
		end = lt.find(r'" />')
		lt = lt[0:end]
		# Get parameter "lt"
		log('lt= ' + lt + ';')

		start = test.text.find(r'name="execution"') + 24
		execution = test.text[start:]
		end = execution.find(r'" />')
		execution = execution[0:end]
		# Get parameter "execution"
		log('execution= ' + execution + ';')

		start = test.text.find(r'name="_eventId"') + 23
		end = start + 10
		_eventId = test.text[start:end]
		end = _eventId.find(r'" />')
		_eventId = _eventId[0:end]
		# Get parameter "_eventld"
		log('_eventId= ' + _eventId + ';')

		log('Login information acquired.')

		url = 'http://weblogin.sustc.edu.cn' + action
		username = config['username']
		password = config['password']
		# SUSTC CAS username (StudentID) and password
		data = 'username=' + username + '&password=' + password + '&lt=' + lt + '&execution=' + execution + '&_eventId=submit&submit=LOGIN'
		soup_login = BeautifulSoup(test.content, 'html5lib')
		if 'CAS' not in soup_login.title.string:
		logging.warning('Not connected to a SUSTC network')
		sleep(config['interval_retry_connection'])
		break

		logging.info('You are offline. Starting login...')
		logging.info('Start to get login information')

		action = soup_login.find('form', id='fm1')['action'] # get login url
		logging.debug('action= ' + action)

		lt = soup_login.find('input', attrs={'name': 'lt'})['value'] # Get parameter "lt"
		logging.debug('lt= ' + lt)

		execution = soup_login.find('input', attrs={'name': 'execution'})['value'] # Get parameter "execution"
		logging.debug('execution= ' + execution)

		logging.info('Login information acquired.')

		url = 'http://weblogin.sustc.edu.cn{}'.format(action)

		data = 'username={}&password={}&lt={}&execution={}&_eventId=submit&submit=LOGIN'. \
		format(config['username'], config['password'], lt, execution)

		h = {'Host': 'weblogin.sustc.edu.cn', 'Connection': 'keep-alive', 'Cache-Control': 'max-age=0',
		'Accept': 'text/html,application/xhtml+xml,application/xml;q=0.9,image/webp,/;q=0.8',
		@@ -85,30 +68,30 @@ def main():
		'Content-Type': 'application/x-www-form-urlencoded', 'DNT': '1', 'Referer': url,
		'Accept-Encoding': 'gzip, deflate', 'Accept-Language': 'en-US,en;q=0.8,zh-CN;q=0.6,zh;q=0.4'}

		log('Login as ' + username)

		login.post(url, data=data, headers=h)
		log('Login information posted to the CAS server.')
		logging.info('Login as ' + config['username'])

		test = requests.get(config['testUrl'])
		r = login.post(url, data=data, headers=h)
		logging.info('Login information posted to the CAS server.')
		soup_response = BeautifulSoup(r.content, 'html5lib')

		if test.status_code != 204:
		if soup_response.find('div', id='msg')['class'][0] != 'success':
		times_retry_login -= 1
		if times_retry_login > 0: # If keep trying to login too many times, it may trigger security alarm on the CAS server
		log(
		logging.info(
		'Login FAILED. Try again in ' + str(config['interval_retry_login']) + ' sec. ' + str(
		times_retry_login) + r' attempt(s) remaining.')
		else:
		log('Login FAILED.' + r'Attempts used up. The program will quit.')
		logging.info('Login FAILED.' + r'Attempts used up. The program will quit.')
		sys.exit('Login FAILED')

		sleep(config['interval_retry_login'])
		else:
		log('Login successful. Current user: ' + username)
		logging.info('Login successful')
		times_retry_login = config['max_times_retry_login']
		log('Login attempts reset to ' + str(times_retry_login) + ' .')
		logging.info('Login attempts reset to ' + str(times_retry_login) + ' .')
		break

		log('Online. Re-check status in ' + str(config['interval_check_status']) + ' sec.')
		logging.info('Online. Re-check status in ' + str(config['interval_check_status']) + ' sec.')

		sleep(config['interval_check_status'])

config_sample.json

+1 −1

Original line number	Diff line number	Diff line
		{"testUrl": "http://www.v2ex.com/generate_204", "interval_retry_connection": "30", "interval_retry_login": "30", "password": "", "interval_check_status": "60", "username": "", "max_times_retry_login": "5"}
		No newline at end of file
		{"captive_portal_server": "http://captive.v2ex.co", "interval_retry_connection": 30, "max_times_retry_login": 5, "interval_retry_login": 30, "interval_check_status": 60, "password": "YOUR_PASSWD", "username": "YOUR_ID"}
		No newline at end of file

Admin message