diff --git a/buptelecmon/electricitymonitor.py b/buptelecmon/electricitymonitor.py index 0d16f40..2fbc9cd 100644 --- a/buptelecmon/electricitymonitor.py +++ b/buptelecmon/electricitymonitor.py @@ -9,6 +9,7 @@ import re import buptelecmon.logger import buptelecmon.exceptions +from bupt_session_py.session import Session class ElectricityMonitor(object): _logger = buptelecmon.logger.register(__name__) @@ -16,37 +17,18 @@ class ElectricityMonitor(object): def __init__(self): self._username = '' self._password = '' - self._session = requests.Session() - self._session.headers.update({ - "user-agent": "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_15_7) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/101.0.4951.64 Safari/537.36", - }) + self._session = Session() self._looping = False # Login to the service def login(self, username, password): - login_url = 'https://auth.bupt.edu.cn/authserver/login' chong_url = 'https://app.bupt.edu.cn/buptdf/wap/default/chong' self._logger.debug('Logging in.') - res = self._session.get(login_url) - execution = re.findall(r'input name="execution" value="(.*)"/>