NJUlogin 3.3__py3-none-any.whl

NJUlogin/QRlogin.py

@@ -0,0 +1,133 @@
+import requests
+import numpy as np
+import time
+from PIL import Image
+from io import BytesIO
+import os
+from lxml import etree
+from user_agents import parse
+
+from .utils import config, urls, get_post
+from .utils.clear import clear
+from .base import baseLogin
+
+
+class QR(object):
+    def __init__(self, session: requests.Session, getTimeout: int):
+        self.session = session
+        self.getTimeout = getTimeout
+
+    def getQR(self) -> np.ndarray:
+        """获取二维码图片，返回numpy数组"""
+        self.ts = int(time.time() * 1000)
+        url = urls.QRid % self.ts
+        QRid = get_post.get(self.session, url, timeout=self.getTimeout).text
+        self.QRid = QRid
+        url = urls.QRimg % QRid
+        QRdata = get_post.get(self.session, url, timeout=self.getTimeout).content
+        qr = Image.open(BytesIO(QRdata)).convert("L")
+        return np.array(qr)[6:-6, 6:-6]
+
+    def printQR(self):
+        """打印二维码至终端，同时也会保存一份QR.png到当前目录"""
+        QRimg = self.getQR()
+        Image.fromarray(QRimg).save("QR.png")
+        char_full = "\u2588"
+        char_up = "\u2580"
+        char_down = "\u2584"
+        for i in range(0, QRimg.shape[0] - 3, 6):
+            for j in range(0, QRimg.shape[1], 3):
+                if QRimg[i, j] < 128 and QRimg[i + 3, j] < 128:
+                    print(char_full, end="")
+                elif QRimg[i, j] < 128 and QRimg[i + 3, j] >= 128:
+                    print(char_up, end="")
+                elif QRimg[i, j] >= 128 and QRimg[i + 3, j] < 128:
+                    print(char_down, end="")
+                else:
+                    print(" ", end="")
+            print("")
+        for j in range(0, QRimg.shape[1], 3):
+            if QRimg[-1, j] < 128:
+                print(char_up, end="")
+            else:
+                print(" ", end="")
+        print("")
+
+
+class QRlogin(baseLogin):
+    def __init__(self, loginTimeout: int = config.loginTimeout, *args, **kwargs):
+        """
+        QRlogin(loginTimeout: int = config.loginTimeout, headers: dict = config.headers, getTimeout: int = config.getTimeout)
+        @description:
+        二维码登录
+        -------
+        @param:
+        loginTimeout: int, 登录超时时间，即二维码有效时间
+        headers: dict, 请求头
+        getTimeout: int, 请求超时时间，即在getTimeout秒内未获取到响应则抛出TimeoutError
+        -------
+        """
+        super().__init__(*args, **kwargs)
+        assert parse(self.session.headers["User-Agent"]).is_pc, "扫码登录只支持PC端"
+        self.loginTimeout = loginTimeout
+
+    def getStatus(self, qr: QR) -> str:
+        """等候扫码，返回扫码状态"""
+        url = urls.status % (qr.ts, qr.QRid)
+        status = self.get(url, timeout=self.getTimeout).text
+        return status
+
+    def waitingLogin(self, qr: QR) -> bool:
+        """等候登录，返回登录状态"""
+        # 0: 未扫码, 1: 登录成功, 2: 已扫码未确认登录
+        first0, first2 = False, False
+        for _ in range(self.loginTimeout):
+            status = self.getStatus(qr)
+            try:
+                status = int(status)
+                if status not in [0, 1, 2]:
+                    raise ValueError
+            except ValueError:
+                raise ValueError("未知状态，代码可能需要维护")
+            if status == 0 and not first0:
+                print("微信或南京大学APP扫码登录")
+                first0 = True
+            elif status == 2 and not first2:
+                print("扫描成功，请在手机上『确认登录』")
+                first2 = True
+            elif status == 1:
+                clear()
+                return True
+            time.sleep(1)
+        clear()
+        print("登录超时")
+        return False
+
+    def login(self, dest: str = None) -> requests.Session:
+        if dest is not None:
+            url = urls.login % dest
+        else:
+            url = urls.login.split("?")[0]
+        html = self.get(url, timeout=self.getTimeout).text
+        qr = QR(self.session, self.getTimeout)
+        clear()
+        qr.printQR()
+        if not self.waitingLogin(qr):
+            return None
+
+        selector = etree.HTML(html)
+        data = {
+            "lt": selector.xpath('//input[@name="lt"]/@value')[1],
+            "uuid": qr.QRid,
+            "dllt": selector.xpath('//input[@name="dllt"]/@value')[1],
+            "execution": selector.xpath('//input[@name="execution"]/@value')[1],
+            "_eventId": selector.xpath('//input[@name="_eventId"]/@value')[1],
+            "rmShown": selector.xpath('//input[@name="rmShown"]/@value')[1],
+        }
+        res = self.post(url, data=data, timeout=self.getTimeout)
+        if self.judge_not_login(res, url):
+            print("登录失败")
+            return None
+        os.remove("QR.png")
+        self.response = res
+        return self.session

NJUlogin/__init__.py

@@ -0,0 +1,3 @@
+from .base import baseLogin
+from .QRlogin import QRlogin
+from .pwdLogin import pwdLogin

NJUlogin/base.py

@@ -0,0 +1,101 @@
+import requests
+import json
+from lxml import etree
+from cryptography.hazmat.primitives.kdf.pbkdf2 import PBKDF2HMAC
+from cryptography.hazmat.primitives import hashes
+from cryptography.hazmat.backends import default_backend
+from cryptography.fernet import Fernet
+import base64
+import os
+
+from .utils import config, get_post, urls
+
+
+class baseLogin(object):
+    def __init__(
+        self, headers: dict = config.headers, getTimeout: int = config.getTimeout
+    ):
+        self.session = requests.Session()
+        self.session.headers.update(headers)
+        self.getTimeout = getTimeout
+
+    def get(self, url: str, **kwargs) -> requests.Response:
+        return get_post.get(self.session, url, **kwargs)
+
+    def post(self, url: str, data: dict, **kwargs) -> requests.Response:
+        return get_post.post(self.session, url, data, **kwargs)
+
+    def judge_not_login(self, html: requests.Response, loginurl: str) -> bool:
+        """判断是否登录成功"""
+        return (
+            html is None or html.url == loginurl or html.url.endswith("security_check")
+        )
+
+    def logout(self) -> None:
+        """退出登录"""
+        self.get(urls.logout, timeout=self.getTimeout)
+
+    def logout_all(self) -> None:
+        """退出所有设备的登录"""
+        html = self.get(urls.onlineList, timeout=self.getTimeout)
+        selector = etree.HTML(html.text)
+        OnlineList = selector.xpath('//input[@value="踢出"]/@onclick')
+        for item in OnlineList:
+            tokenId = item.split("'")[1]
+            self.post(urls.logoutOthers, data={"tokenId": tokenId})
+
+    @property
+    def available(self) -> bool:
+        """判断是否登录成功"""
+        html = self.get(urls.index, timeout=self.getTimeout)
+        return html.url == urls.index
+
+    def export(self, filename: str, password: str = None) -> None:
+        """导出登录信息"""
+        if not self.available:
+            raise ValueError("未登录，无法导出登录信息")
+        if password is None:
+            with open(filename, "w") as f:
+                json.dump(self.session.cookies.get_dict(), f, indent=2)
+        else:
+            cookies = json.dumps(self.session.cookies.get_dict(), indent=2).encode()
+            salt = os.urandom(16)
+            kdf = PBKDF2HMAC(
+                algorithm=hashes.SHA256(),
+                length=32,
+                salt=salt,
+                iterations=100000,
+                backend=default_backend(),
+            )
+            key = base64.urlsafe_b64encode(kdf.derive(password.encode()))
+            fernet = Fernet(key)
+            enc_cookie = fernet.encrypt(cookies)
+            with open(filename, "wb") as f:
+                f.write(enc_cookie + salt)
+
+    def load(self, filename: str, password: str = None) -> None:
+        """加载登录信息"""
+        if password is None:
+            try:
+                with open(filename, "r") as f:
+                    cookies = json.load(f)
+            except UnicodeDecodeError:
+                raise RuntimeError("请提供密码")
+        else:
+            with open(filename, "rb") as f:
+                enc_cookie = f.read()
+            salt = enc_cookie[-16:]
+            enc_cookie = enc_cookie[:-16]
+            kdf = PBKDF2HMAC(
+                algorithm=hashes.SHA256(),
+                length=32,
+                salt=salt,
+                iterations=100000,
+                backend=default_backend(),
+            )
+            key = base64.urlsafe_b64encode(kdf.derive(password.encode()))
+            fernet = Fernet(key)
+            cookies = json.loads(fernet.decrypt(enc_cookie))
+        self.session.cookies.update(cookies)
+        if not self.available:
+            raise ValueError("登录信息已失效，请重新登录")

NJUlogin/captchaOCR/__init__.py

@@ -0,0 +1 @@
+from .ocr import CaptchaOCR