diff --git a/interface.py b/interface.py index 20a96d1..ece1853 100644 --- a/interface.py +++ b/interface.py @@ -106,7 +106,7 @@ class Kuser: return False else: if result == False: - self._api.login(self._username, self._password) + self._api.login(self._username, self._password, 2) result = self._api.check_online() if result == False: self.login_failed(chat_id, message_id) diff --git a/kcoj.py b/kcoj.py index 687e119..ae2ff1d 100644 --- a/kcoj.py +++ b/kcoj.py @@ -7,11 +7,11 @@ class KCOJ: self._url = url self._session = requests.Session() # login KCOJ - def login(self, username, password): + def login(self, username, password, course): try: payload = {'name': username, 'passwd': password, - 'rdoCourse': 1} + 'rdoCourse': course} return self._session.post(self._url + '/Login', data=payload, timeout=0.5) except requests.exceptions.Timeout: return None