fix: update auth logic

bjut_tech/auth/jwglxt.py

@@ -43,6 +43,10 @@ def check(self) -> bool:
         return response.status_code == 200
 
     def authenticate(self):
+        if self.check():
+            # also for setting initial session cookie
+            return
+
         self._get_key()
         pub_key = rsa.PublicKey(int(self.key[0], 16), int(self.key[1], 16))
         password_encrypted = rsa.encrypt(self.password.encode('utf-8'), pub_key)

bjut_tech/auth/libziyuan.py

@@ -39,15 +39,15 @@ def authenticate(self, session: Optional[Client] = None) -> Client:
         if session is None:
             session = self._new_session()
 
-        saved_cookies = self.persistence.load(self.persistence_key)
-        if saved_cookies is not None:
-            session.cookies.update(saved_cookies)
-
         if self.check(session):
-            print('[libziyuan] authenticated session reused')
             return session
 
+        saved_cookies = self.persistence.load(self.persistence_key)
         if saved_cookies is not None:
+            session.cookies.update(saved_cookies)
+            if self.check(session):
+                print('[libziyuan] authenticated session reused')
+                return session
             self.persistence.delete(self.persistence_key)
 
         response = session.get('https://libziyuan.bjut.edu.cn/por/login_auth.csp', params={

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "bjut-tech"
-version = "0.2.1"
+version = "0.2.2"
 description = "The bjut.tech Python package for some common stuff."
 readme = "README.md"
 dependencies = [