希望新增支持下360渠道服

[Lmy-696]

根据之前信息结合AI写了下，希望有所帮助

src/channelHandler/qihooLogin/qihooChannel.py

import requests
import re
from bs4 import BeautifulSoup

class QihooLogin:
    def __init__(self, channel_id, oauthData, account_type):
        self.channel_id = channel_id
        self.oauthData = oauthData
        self.account_type = account_type
        self.session = requests.Session()

    def webLogin(self, username, password):
        """
        模拟 360 网页登录
        :param username: 用户名
        :param password: 密码
        """
        login_url = "https://i.360.cn/login/wap"
        headers = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36"
        }
        response = self.session.get(login_url, headers=headers)
        soup = BeautifulSoup(response.text, 'html.parser')
        # 假设页面中有隐藏的表单字段需要提取，例如 csrf_token
        csrf_token = soup.find('input', attrs={'name': 'csrf_token'})['value'] if soup.find('input', attrs={'name': 'csrf_token'}) else ""

        data = {
            "username": username,
            "password": password,
            "csrf_token": csrf_token
        }
        post_response = self.session.post(login_url, data=data, headers=headers)
        # 检查登录是否成功，这里简单根据页面内容判断，实际可能需要更复杂的逻辑
        if "登录成功" in post_response.text:
            # 提取登录后的相关信息，例如 token、session 等
            self.oauthData["token"] = self._extract_token(post_response.text)
            self.oauthData["session_id"] = self.session.cookies.get('session_id')
        else:
            raise Exception("360 登录失败")

    def _extract_token(self, response_text):
        """
        从登录后的页面中提取 token
        :param response_text: 登录后的页面文本
        :return: token 值
        """
        # 假设 token 是通过正则表达式提取的，实际根据页面结构调整
        token_match = re.search(r'"token":"([^"]+)"', response_text)
        return token_match.group(1) if token_match else ""

    def initAccountData(self):
        """
        初始化账户数据
        """
        # 这里假设需要向 360 的某个接口发送请求获取账户数据
        account_url = "https://i.360.cn/api/account/data"
        headers = {
            "User-Agent": "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/91.0.4472.124 Safari/537.36",
            "Authorization": f"Bearer {self.oauthData['token']}"
        }
        response = self.session.get(account_url, headers=headers)
        if response.status_code == 200:
            data = response.json()
            return {
                "appAccountId": data.get("account_id", ""),
                "session": self.oauthData["session_id"]
            }
        else:
            raise Exception("获取账户数据失败")

src/channelHandler/qihooChannelHandler.py

import json
import time
import base64
import channelmgr
from envmgr import genv
from logutil import setup_logger
from channelHandler.qihooLogin.qihooChannel import QihooLogin
from channelHandler.channelUtils import getShortGameId

class QihooChannel(channelmgr.channel):
    def __init__(
        self,
        login_info: dict,
        user_info: dict = {},
        ext_info: dict = {},
        device_info: dict = {},
        create_time: int = int(time.time()),
        last_login_time: int = 0,
        name: str = "",
        oAuthData: dict = {},
        game_id: str = "",
        account_type: int = 4,
    ) -> None:
        super().__init__(
            login_info,
            user_info,
            ext_info,
            device_info,
            create_time,
            last_login_time,
            name,
        )
        self.oAuthData = oAuthData
        self.logger = setup_logger()
        self.crossGames = False
        self.game_id = game_id
        real_game_id = getShortGameId(game_id)
        cloudRes = genv.get("CLOUD_RES")
        res = cloudRes.get_channelData(self.channel_name, real_game_id) if cloudRes else None
        if res is None:
            self.logger.error(f"Failed to get channel config for {self.name}")
            raise Exception(
                f"游戏{real_game_id}-渠道{self.channel_name}暂不支持，请参照教程联系开发者发起添加请求。"
            )
        self.qihooLogin = QihooLogin(
            res.get(self.channel_name).replace("qihoo_", ""), self.oAuthData, account_type
        )
        self.realGameId = real_game_id
        self.uniBody = None
        self.uniData = None
        self.account_type = account_type

    def request_user_login(self, username, password):
        """
        请求用户登录
        :param username: 用户名
        :param password: 密码
        """
        genv.set("GLOB_LOGIN_UUID", self.uuid)
        self.qihooLogin.webLogin(username, password)
        self.oAuthData = self.qihooLogin.oauthData
        self.account_type = self.qihooLogin.account_type
        self.logger.info(f"360 登录类型：{self.account_type}")
        self.logger.debug(self.oAuthData)
        return self.oAuthData is not None

    def _get_session(self):
        try:
            data = self.qihooLogin.initAccountData()
        except Exception as e:
            self.logger.error(f"Failed to get session data {e}")
            self.oAuthData = None
            return None, None
        return data["appAccountId"], data["session"]

    def is_token_valid(self):
        if self.oAuthData is None:
            self.logger.info(f"Token is invalid for {self.name}")
            return False
        return True

    @classmethod
    def from_dict(cls, data: dict):
        return cls(
            login_info=data.get("login_info", {}),
            user_info=data.get("user_info", {}),
            ext_info=data.get("ext_info", {}),
            device_info=data.get("device_info", {}),
            create_time=data.get("create_time", int(time.time())),
            last_login_time=data.get("last_login_time", 0),
            name=data.get("name", ""),
            oAuthData=data.get("oAuthData", None),
            game_id=data.get("game_id", ""),
            account_type=data.get("account_type", 4),
        )

    def get_sdk_udid(self):
        return self.oAuthData["uuid"] if "uuid" in self.oAuthData else ""

    def _build_extra_unisdk_data(self) -> str:
        res = {
            "SAUTH_STR": "",
            "SAUTH_JSON": "",
            "extra_data": "",
            "realname": "",
            "get_access_token": "1",
        }
        extra = json.dumps({"adv_channel": "0", "adid": "0"})
        realname = json.dumps({"realname_type": 0, "age": 18})
        json_data = {
            "extra_data": extra,
            "get_access_token": "1",
            "sdk_udid": self.get_sdk_udid(),
            "realname": realname,
        }
        json_data.update(self.uniBody)

        str_data = json_data.copy()
        str_data.update({"username": self.uniSDKJSON["username"]})
        str_data = "&".join([f"{k}={v}" for k, v in str_data.items()])

        res["SAUTH_STR"] = base64.b64encode(str_data.encode()).decode()
        res["SAUTH_JSON"] = base64.b64encode(json.dumps(json_data).encode()).decode()
        res["extra_data"] = extra
        res["realname"] = realname
        return json.dumps(res)

    def get_uniSdk_data(self, game_id: str = ""):
        genv.set("GLOB_LOGIN_UUID", self.uuid)
        if game_id == "":
            game_id = self.game_id
        self.logger.info(f"Get unisdk data for {self.name}")
        import channelHandler.channelUtils as channelUtils

        if not self.is_token_valid():
            self.request_user_login(self.login_info["username"], self.login_info["password"])
        appAccountId, session = self._get_session()
        self.uniBody = channelUtils.buildSAUTH(
            self.channel_name,
            self.channel_name,
            str(appAccountId),
            session,
            getShortGameId(game_id),
            "3.3.0.7",
        )
        fd = genv.get("FAKE_DEVICE")
        self.uniData = channelUtils.postSignedData(
            self.uniBody, getShortGameId(game_id)
        )
        self.logger.info(f"Get unisdk data for {self.uniData}")
        self.uniSDKJSON = json.loads(
            base64.b64decode(self.uniData["unisdk_login_json"]).decode()
        )
        res = {
            "user_id": self.get_sdk_udid(),
            "token": base64.b64encode(session.encode()).decode(),
            "login_channel": self.channel_name,
            "udid": fd["udid"],
            "app_channel": self.channel_name,
            "sdk_version": "3.0.5.002",
            "jf_game_id": getShortGameId(game_id),
            "pay_channel": self.channel_name,
            "extra_data": "",
            "extra_unisdk_data": self._build_extra_unisdk_data(),
            "gv": "157",
            "gvn": "1.5.80",
            "cv": "a1.5.0",
        }
        return res

[KKeygen]

360渠道应该是用客户端扫码登录导入账号记录的，希望对你有帮助

[Lmy-696]

好的，试了可以，之前没见有描述就没试