addproject

base_framework/public_tools/get_token.py

@@ -0,0 +1,911 @@
+import json
+from urllib import parse
+
+import requests
+from retrying import retry
+from base_framework.public_tools.read_config import InitConfig
+from base_framework.public_tools.read_config import ReadConfig, get_current_config, get_current_env
+from base_framework.base_config.current_pth import *
+from requests.packages.urllib3.connectionpool import InsecureRequestWarning
+from base_framework.public_tools import log
+from Crypto.Cipher import PKCS1_v1_5
+from Crypto.PublicKey import RSA
+import base64
+
+
+requests.packages.urllib3.disable_warnings(InsecureRequestWarning)
+import time
+import re
+# get_cfg = ReadConfig(env_choose_path)
+token_dict = dict()
+student_token_cache = {}
+obj_log = log.get_logger()
+
+
+class LazyProperty:
+    def __init__(self, fun):
+        self.fun = fun
+
+    def __get__(self, instance, owner):
+        if instance is None:
+            return self
+        value = self.fun(instance)
+        setattr(instance, self.fun.__name__, value)
+        return value
+
+
+class LoginSys(InitConfig):
+    def __init__(self, host=None, curt_user=None, current_evn=None):
+        try:
+            InitConfig.__init__(self, run_user_name=curt_user, current_evn=current_evn)
+        except Exception as e:
+            print(e)
+        self.host = host
+        self.curt_user = self.current_user
+        self.env = get_current_env()
+
+    @LazyProperty
+    def get_session(self):
+        """
+        获取SSO session
+        :return:
+        """
+        sys_type = self.host.split(".")
+        if 'visparklearning' in sys_type and ('cc-manage' or 'cti-manage' in sys_type):
+            api_url = self.curt_sso_url
+        elif 'hulk-content-audit-server' in sys_type or 'hulk-content-manage-gateway' in sys_type or 'hulk-class-help-manager' in sys_type or 'hulk-ark-gateway' in sys_type or 'manage-api' in sys_type :
+            api_url = self.all_school_sso_url
+        else:
+            api_url = self.curt_sso_url
+        post_data = dict()
+        post_data['showUsername'] = self.show_username
+        post_data['username'] = self.username
+        post_data['password'] = self.password
+        req_session = requests.session()
+        resp = req_session.post(
+            url=api_url,
+            data=post_data,
+            allow_redirects=False,
+            verify=False)
+        return req_session
+
+    @LazyProperty
+    def get_code(self):
+        """
+        获取code
+        :return:
+        """
+        post_data = dict()
+        sys_type = self.host.split(".")
+        if 'visparklearning' in sys_type and ('cc-manage' or 'cti-manage' in sys_type):
+            post_data['client_id'] = "vispark-crm"
+            post_data['response_type'] = 'code'
+            post_data['redirect_uri'] = self.crm_vispark_url
+            authorization = self.cc_auth
+        # elif sys_type == "smm":
+        #     post_data['client_id'] = "sms-manage"
+        #     post_data['response_type'] = 'code'
+        #     post_data['redirect_uri'] = self.sms_url
+        if 'visparklearning' in sys_type and ('cc-manage' or 'cti-manage' in sys_type):
+            api_url = self.get_code_url
+        elif 'hulk-content-audit-server' in sys_type or 'hulk-content-manage-gateway' in sys_type or 'hulk-class-help-manager' in sys_type or 'hulk-ark-gateway' in sys_type or 'manage-api' in sys_type:
+            api_url = self.get_all_school_code_url
+        else:
+            api_url = self.get_code_url
+        req_session = self.get_session
+        if 'mq-console' in sys_type:
+            allow_redirects = True
+        else:
+            allow_redirects = False
+        obj_log.info(api_url)
+        obj_log.info(post_data)
+        resp = req_session.get(
+            url=api_url,
+            params=post_data,
+            allow_redirects=allow_redirects,
+            verify=False)
+        if 'mq-console' in sys_type:
+            return [req_session, 'MQ']
+        code_temp = resp.headers
+        obj_log.info(code_temp)
+        code = code_temp.get('location')
+        obj_log.info(code)
+        if 'client_id=gmp' in code:
+            return code
+        try:
+            code = code.split("=")[-1]
+        except Exception:
+            raise IndexError("获取token失败，请检查环境或者登录信息是否正确！")
+        resp_list = [code, post_data['redirect_uri'], authorization]
+        return resp_list
+
+    # @LazyProperty
+    def get_token(self, type_name=None):
+        """
+        获取token
+        :return:
+        """
+        try:
+            token = token_dict[self.host].get(self.curt_user, None)
+        except KeyError as e:
+            token = False
+        if token:
+            return token
+        else:
+            token_dict_temp = dict()
+            sys_type = self.host.split(".")
+
+            post_data = dict()
+            temp_code = self.get_code
+            if isinstance(temp_code, str):
+                session = self.get_session
+                resp = session.get(url=temp_code)
+                for key, value in session.cookies.items():
+                    if key == 'peppa_sso_token':
+                        token = value
+                        break
+                token_header = {"accesstoken": token}
+                session.headers.update(token_header)
+                if self.jira_id:
+                    podenv = {"huohua-podenv": self.jira_id}
+                    session.headers.update(podenv)
+                token_dict_temp[self.curt_user] = session
+                token_dict[self.host] = token_dict_temp
+                return session
+
+            if temp_code[1] == 'MQ':  # MQ特殊处理
+                return temp_code[0]
+            post_data['code'] = temp_code[0]
+            post_data['redirect_uri'] = temp_code[1]
+            post_data['grant_type'] = 'authorization_code'
+            req_session = self.get_session
+            authorization = temp_code[2]
+            if authorization:
+                header = {'authorization': authorization}
+                if 'visparklearning' in sys_type and ('cc-manage' or 'cti-manage' in sys_type):
+                    resp = req_session.get(
+                        url=api_url+'&code='+temp_code[0], data=post_data, headers=header, allow_redirects=False, verify=False)
+                elif 'visparklearning' in sys_type and 'manage-gw' in sys_type:
+                    resp = req_session.get(url=api_url + '&code=' + temp_code[0], data=post_data, headers=header,
+                                           allow_redirects=False, verify=False)
+                else:
+                    resp = req_session.post(
+                        url=api_url, data=post_data, headers=header, allow_redirects=False, verify=False)
+                if 'visparklearning' in sys_type and ('cc-manage' or 'cti-manage' in sys_type):
+                    token_temp = parse.parse_qs(parse.urlparse(resp.headers['location']).query)
+                elif 'visparklearning' in sys_type and 'manage-gw' in sys_type:
+                    token_temp = parse.parse_qs(parse.urlparse(resp.headers['location']).query)
+                else:
+                    token_temp = resp.json()
+                if type_name == 'BFF':  # bff架构使用
+                    token = {"token": token_temp['access_token']}
+                elif 'visparklearning' in sys_type and ('cc-manage' or 'cti-manage' in sys_type):
+                    token = {"accesstoken": token_temp['peppa_sso_token'][0]}
+                elif 'visparklearning' in sys_type and 'manage-gw' in sys_type:
+                    token = {"accesstoken": token_temp['peppa_sso_token'][0]}
+                else:
+                    if "access_token" not in token_temp:
+                        obj_log.warning("---------获取token失败！-------")
+                        obj_log.warning("type_name:{}\ntoken_temp:{}".format(type_name, token_temp))
+                        obj_log.warning("-----------------------------")
+                    token = {"accesstoken": token_temp['access_token']}
+                req_session.headers.update(token)
+                if self.jira_id:
+                    podenv = {"huohua-podenv": self.jira_id}
+                    req_session.headers.update(podenv)
+                token_dict_temp[self.curt_user] = req_session
+                token_dict[self.host] = token_dict_temp
+                return req_session
+            else:
+                if 'kunpeng' in post_data['redirect_uri']:  # 鲲鹏/mp系统登录
+                    req_session.get(url=post_data['redirect_uri'] + '&code=' + temp_code[0], verify=False)
+                    return req_session
+                elif 'allschool-mp' in post_data['redirect_uri']:
+                    if self.jira_id:
+                        podenv = {"huohua-podenv": self.jira_id}
+                        req_session.headers.update(podenv)
+                    req_session.get(url=post_data['redirect_uri'] + '&code=' + temp_code[0], verify=False)
+                    access_token = req_session.cookies.get("peppa_sso_token")
+                    req_session.headers.update({"accesstoken": access_token})
+                    return req_session
+                run_session = self.sso_oauth_login(session=req_session, code=temp_code[0])
+                if self.jira_id:
+                    podenv = {"huohua-podenv": self.jira_id}
+                    run_session.headers.update(podenv)
+                token_dict_temp[self.curt_user] = run_session
+                token_dict[self.host] = token_dict_temp
+            return run_session
+
+    def sso_oauth_login(self, type='manage', session=None, code=None):
+        """
+            | 功能说明： |  |
+            | 输入参数： |  |  |
+            | 返回参数： | XXX |
+            | 作者信息： | 作者     | 修改时间      |
+            举例说明：
+            | sso_oauth_login   | |
+        """
+        if type == 'manage':
+            url = self.manage_url + '?code=' + code
+        elif type == 'opengalaxy':  # 独立环境发布系统
+            url = self.opengalaxy_url + '?code=' + code
+        session.get(url=url, allow_redirects=True, verify=False)
+        return session
+
+    def ug_hhi_sso_login_by_session_id(self):
+        """
+        ug的manage后台登录获取sessionId
+        :return:
+        """
+        req_session = requests.session()
+        headers = {"Content-Type": "application/json"}
+
+        # sso登录
+        url_form = self.curt_sso_url
+        obj_log.info(url_form)
+        post_data = dict()
+        post_data['showUsername'] = self.show_username
+        post_data['username'] = self.username
+        post_data['password'] = self.password
+        res_form = req_session.post(url=url_form, data=post_data, allow_redirects=False, verify=False)
+        obj_log.info("ug_hhi_sso_login_by_session_id_url_form:{}".format(url_form))
+        obj_log.info("ug_hhi_sso_login_by_session_id_res_form:{}".format(res_form.headers))
+
+
+        # 进行sso.qa.sparkedu.com/oauth/authorize
+        url_authorize = "{}?client_id=manage&response_type=code&redirect_uri={}".format(
+            self.get_code_url, self.manage_url)
+        res_authorize = req_session.get(url_authorize, allow_redirects=False)
+        obj_log.info("ug_hhi_sso_login_by_session_id_url_authorize:{}".format(url_authorize))
+        obj_log.info("ug_hhi_sso_login_by_session_id_res_authorize:{}".format(res_authorize.headers))
+
+        # 获取到res_authorize的location地址(主要是用到code)
+        url_sso_oauth_login = res_authorize.headers.get('location')
+        obj_log.info("ug_hhi_sso_login_by_session_id_url_sso_oauth_login:{}".format(url_sso_oauth_login))
+        res_url_sso_oauth_login = req_session.get(url_sso_oauth_login, allow_redirects=False)
+        obj_log.info("ug_hhi_sso_login_by_session_id_res_url_sso_oauth_login:{}".format(res_url_sso_oauth_login.headers))
+
+        # 因为环境问题所以多请求一次域名
+        res_manage_host = req_session.get(self.manage_host)
+        obj_log.info("ug_hhi_sso_login_by_session_id_url_manage:{}".format(self.manage_host))
+        obj_log.info(
+            "ug_hhi_sso_login_by_session_id_res_manage_host:{}".format(res_manage_host.headers))
+
+        token_dict_temp = dict()
+        token_dict_temp[self.curt_user] = req_session
+        token_dict[self.host] = token_dict_temp
+
+        return req_session
+
+    def ug_sso_login_by_starlight(self):
+        """
+        ug的starlight.qa.huohua.cn登录
+        :return:
+        """
+        req_session = requests.session()
+        headers = {"Content-Type": "application/json"}
+
+        # 访问starlight.qa.huohua.cn/生成JSESSION_STARLIGHT
+        starlight_url = self.starlight_url
+        obj_log.info(starlight_url)
+        req_session.get(starlight_url, allow_redirects=False)
+
+        # sso登录
+        url_form = self.curt_sso_url
+        obj_log.info(url_form)
+        post_data = dict()
+        post_data['showUsername'] = self.show_username
+        post_data['username'] = self.username
+        post_data['password'] = self.password
+        res_form = req_session.post(url=url_form, data=post_data, allow_redirects=False, verify=False)
+        obj_log.info(f'res_form的header：{res_form.headers}')
+
+        # 进行sso.qa.sparkedu.com/oauth/authorize
+        url_authorize = "{}?client_id=starlight&response_type=code&redirect_uri={}".format(
+            self.get_code_url, self.suyang_manage_sso_login_url)
+        res_authorize = req_session.get(url_authorize, allow_redirects=False)
+        obj_log.info(f"res_authorize的header：{res_authorize.headers.get('location')}")
+
+        # 获取到res_authorize的location地址(主要是用到code)
+        req_session.get(res_authorize.headers.get('location'), allow_redirects=False)
+
+        # 访问https://starlight.qa.huohua.cn/
+        req_session.get(self.starlight_url, allow_redirects=False)
+
+        token_dict_temp = dict()
+        token_dict_temp[self.curt_user] = req_session
+        token_dict[self.host] = token_dict_temp
+        return req_session
+
+class SparkleLogin(InitConfig):
+    def __init__(self, host=None, curt_user=None, current_evn=None):
+        # super().__init__(run_user_name=None, current_evn=current_evn)
+        InitConfig.__init__(self, run_user_name=None, current_evn=current_evn)
+        self.pc_token = None
+        self.session = requests.session()
+        # self.jira_id = get_cfg.get_value(sections='run_jira_id', options='huohua-podenv')
+        self.jira_id = ReadConfig(env_choose_path).get_value(sections='run_jira_id', options='huohua-podenv')
+
+    # def sparkle_pc_login(self, phone, passwd=123456):
+    def sparkle_pc_login(self, phone, passwd='Mima@123'):
+        if token_dict.get(str(phone)):
+            return token_dict.get(str(phone))
+        url = self.sparkle_pc_token_url
+        data = {"username": phone, "userType": 2, "password": str(passwd)}
+        self.session.headers.update({"Authorization": self.spark_pc_auth})
+        resp = json.loads(self.session.post(url, params=data).content)
+        header = {"account-token": resp.get("data").get("accessToken")}
+        if self.jira_id:
+            podenv = {"huohua-podenv": self.jira_id}
+            self.session.headers.update(podenv)
+        self.session.headers.update(header)
+        token_dict[str(phone)] = self.session
+        return self.session
+
+
+class AgentApiLogin(InitConfig):
+    def __init__(self, host=None, curt_user=None, current_evn=None):
+        # super().__init__(run_user_name=None, current_evn=current_evn)
+        InitConfig.__init__(self, run_user_name=None, current_evn=current_evn)
+        self.agent_api_token = None
+        self.session = requests.session()
+        # self.jira_id = get_cfg.get_value(sections='run_jira_id', options='huohua-podenv')
+        self.jira_id = ReadConfig(env_choose_path).get_value(sections='run_jira_id', options='huohua-podenv')
+
+    def agent_api_login(self, phone, authCode, countryCode=86):
+        if token_dict.get(str(phone)):
+            return token_dict.get(str(phone))
+        url = '{}/login/loginByAuthCode'.format(self.hhr_api_host)
+        data = {"phone": str(phone), "authCode": str(authCode), "countryCode": str(countryCode)}
+        resp = json.loads(self.session.post(url, json=data).content)
+        header = {"user-token": resp.get("data").get("token")}
+        if self.jira_id:
+            podenv = {"huohua-podenv": self.jira_id}
+            self.session.headers.update(podenv)
+        self.session.headers.update(header)
+        token_dict[str(phone)] = self.session
+        return self.session
+
+class MarketApiLogin(InitConfig):
+    def __init__(self, host=None, curt_user=None, current_evn=None):
+        InitConfig.__init__(self, run_user_name=None, current_evn=current_evn)
+        self.session = requests.session()
+        self.jira_id = ReadConfig(env_choose_path).get_value(sections='run_jira_id', options='huohua-podenv')
+
+    def market_api_login(self, user=None):
+        login_url = self.market_url
+        result = parse.urlparse(url=login_url ,allow_fragments=True)
+        host = result.hostname
+        choose_user = ReadConfig(env_choose_path).get_options('run_user_name')
+        if host in choose_user:
+            default_user = ReadConfig(env_choose_path).get_value(
+                sections='run_user_name' ,options=host)
+            self.current_user = default_user
+        if user:
+            self.current_user = user
+        try:
+            login_user = ReadConfig().get_value(sections=self.current_user ,options='username')
+            login_password = ReadConfig().get_value(sections=self.current_user ,options='password')
+        except Exception as e:
+            login_user = ReadConfig(astwb_config).get_value(sections=self.current_user ,options='username')
+            login_password = ReadConfig(astwb_config).get_value(sections=self.current_user ,options='password')
+
+        post_data = {'phone': login_user ,'password': login_password}
+        resp = self.session.post(url=login_url ,json=post_data , verify=False,
+                                 headers={'huohua-podenv': self.jira_id})
+
+        if resp.status_code != 200:
+            raise KeyError('获取 市场管理投放系统 token失败！')
+        set_cookie = resp.headers['set-cookie'].split(";")[-1].split(",")[1]
+        headers = {"Cookie":"gray_id=3077c7810744b47fc06ec513cf07801e; gray_tag=stable; "+set_cookie}
+        self.session.headers.update(headers)
+        return self.session
+
+
+
+class AllSchool(InitConfig):
+    def __init__(self, curt_user=None, current_evn=None):
+        # super().__init__(run_user_name=curt_user, current_evn=current_evn)
+        InitConfig.__init__(self, run_user_name=curt_user, current_evn=current_evn)
+        self.jira_id = ReadConfig(env_choose_path).get_value(sections='run_jira_id', options='huohua-podenv')
+
+    def all_school_token(self, user=None, login_type=1):
+        """
+        :param user: 登录用户
+        :param login_type: 登录系统 1：教师工作台
+                                  2：机构工作台
+                                  3：选课平台
+        :return: token
+        """
+        if login_type == 1:
+            login_url = self.all_school_url
+        elif login_type == 2:
+            login_url = self.all_school_org_url
+        elif login_type == 3:
+            login_url = self.find_classes_url
+        result = parse.urlparse(url=login_url, allow_fragments=True)
+        host = result.hostname
+        choose_user = ReadConfig(env_choose_path).get_options('run_user_name')
+        if host in choose_user:
+            if login_type == 2:
+                host = 'org-' + host
+            elif login_type == 3:
+                host = 'find-' + host
+            default_user = ReadConfig(env_choose_path).get_value(
+                sections='run_user_name', options=host)
+            self.current_user = default_user
+        if user:
+            self.current_user = user
+        try:
+            login_user = ReadConfig().get_value(sections=self.current_user, options='username')
+            login_password = ReadConfig().get_value(sections=self.current_user, options='password')
+        except Exception as e:
+            login_user = ReadConfig(astwb_config).get_value(sections=self.current_user, options='username')
+            login_password = ReadConfig(astwb_config).get_value(sections=self.current_user, options='password')
+        if int(login_type) == 3:
+            post_data = {"data": {"email": login_user, "password": login_password, "countryCode": "86",
+                                  "loginType": "EMAIL_PSW"}}
+        else:
+            post_data = {'email': login_user, 'password': login_password}
+        as_token = requests.post(url=login_url, json=post_data, verify=False,
+                                 headers={'huohua-podenv': self.jira_id})
+        rtn_temp = as_token.json()
+        if as_token.status_code != 200:
+            raise KeyError('获取 allSchool token失败！')
+        user_token = rtn_temp['data']['token']
+        return user_token
+
+
+class ParentLogin(InitConfig):
+    def __init__(self,host=None, curt_user=None, current_evn=None):
+        # super().__init__(run_user_name=curt_user, current_evn=current_evn)
+        InitConfig.__init__(self, run_user_name=curt_user, current_evn=current_evn)
+        self.jira_id = ReadConfig(env_choose_path).get_value(sections='run_jira_id', options='huohua-podenv')
+
+    @retry(stop_max_attempt_number=5, wait_fixed=3000)
+    def parent_send_msg(self, phone, authType=2, countryCode=None,
+                   sessionId=None, **kwargs):
+        """
+            | 功能说明： | M站（学生端）发送短信 |
+            | 输入参数： | phone | 手机号 |
+            | 返回参数： | XXX |
+            | 作者信息： | 作者     | 修改时间      |
+            举例说明：
+            | m_send_msg   | |
+        """
+        url = self.m_host + '/passport/auth_code/send'
+        post_data = {
+            "phone": phone,
+            "authType": authType,
+            "countryCode": countryCode,
+            "isLogin": False,
+            "sessionId": sessionId}
+        resp = requests.post(
+            url=url,
+            json=post_data,
+            verify=False)
+        time.sleep(2)
+        return resp
+
+    def get_parent_sms_code(self, phone):
+        """
+        M站获取手机验证码
+        :param phone:电话号码
+        :return: token
+        """
+        auth_url = self.curt_sso_url
+        get_code_url = self.get_code_url
+        token_url = self.get_token_url
+        sms_url = "https://smm.qa.huohua.cn/sms/log/page/all?WHERE.%5BEQ%5Dphone={}&WHERE.%5BEQ%5DtypeName=&WHERE.%5BEQ%5Dchannel=&WHERE.%5BEQ%5DgatewayType=&WHERE.%5BGT%5DcreateTime=&WHERE.%5BLT%5DcreateTime=&WHERE.%5BGT%5DsubmitTime=&WHERE.%5BLT%5DsubmitTime=&pageNum=1&pageSize=20&orderBy=id%20desc".format(
+            phone)
+        session = requests.session()
+        session.post(url=auth_url, verify=False,
+                     data={'showUsername': 'liuruiquan', 'username': 'liuruiquan@huohua.cn', 'password': 'lrq5823LRQ'})
+        get_code = session.get(url=get_code_url, verify=False,
+                               params={'client_id': 'crmnew', 'response_type': 'code',
+                                       'redirect_uri': 'https://crmv2.qa.huohua.cn'}, allow_redirects=False)
+        code_temp = get_code.headers
+        code = code_temp['location']
+        code = code.split("=")[-1]
+        token_get = session.post(url=token_url, verify=False, data={'code': code, 'redirect_uri': 'https://crmv2.qa.huohua.cn',
+                                                        'grant_type': 'authorization_code'},
+                     headers={'authorization': 'Basic Y3JtbmV3OmNybW5ldw=='})
+        token_temp = token_get.json()
+        token = {"accesstoken": token_temp['access_token']}
+        sms_temp = session.post(url=sms_url, verify=False, headers=token).json()
+        smm_code = sms_temp['list'][0]['msg']
+        reg = re.compile(r"(?<=手机短信验证码：)\d+")
+        match = reg.search(smm_code)
+        smm_code = match.group(0)
+        return smm_code
+
+    def get_parent_token(self, phone=None):
+        """
+        M站token获取
+        :param phone:电话号码
+        :return: token
+        """
+        login_url = "http://m.qa.huohua.cn/passport/login"
+        result = parse.urlparse(url=login_url, allow_fragments=True)
+        host = 'parent-' + result.hostname
+        choose_user = ReadConfig(env_choose_path).get_options('run_user_name')
+        if host in choose_user:
+            default_user = ReadConfig(env_choose_path).get_value(
+                sections='run_user_name', options=host)
+            if phone:
+                user = ReadConfig().get_value(sections=phone, options='username')
+                self.current_user = user
+            else:
+                phone = ReadConfig().get_value(
+                    sections=default_user, options='username')
+                self.current_user = phone
+        if student_token_cache.get(self.current_user):
+            return student_token_cache[self.current_user]
+        send_code = self.parent_send_msg(phone=self.current_user).json()
+        # if not send_code['success']:
+        #     raise ValueError(send_code)
+        verify_code = self.get_parent_sms_code(phone=self.current_user)
+        post_data = {"phone": self.current_user, "authCode": verify_code, "countryCode": None, "loginType": 1,
+                     "userType": 1,
+                     "subjectType": 0}
+        m_token = requests.post(url=login_url, json=post_data, verify=False, headers={'huohua-podenv': self.jira_id})
+        if m_token.status_code == 200:
+            token = m_token.json()
+            token = token['data']['token']
+            student_token_cache[self.current_user] = token
+            return token
+        else:
+            raise KeyError('获取 parent token失败！')
+
+    def get_parent_token_by_pwd(self, phone=None):
+        env_name = get_current_config(section='run_evn_name', key='current_business')
+        obj_log.info("env_name:{}".format(env_name))
+        if env_name and env_name=='hhi':
+            login_url = "https://hhi-user-auth-api.qa.visparklearning.com/login"
+        elif 'hhi' in phone.split('-'):
+            login_url = "https://pst-gw.qa.huohua.cn/passport/login"
+        else:
+            login_url = "http://m.qa.huohua.cn/passport/login"
+        result = parse.urlparse(url=login_url, allow_fragments=True)
+        host = 'parent-' + result.hostname
+        choose_user = ReadConfig(env_choose_path).get_options('run_user_name')
+        obj_log.info("host:{},choose_user:{}".format(host, choose_user))
+        if host in choose_user:
+            default_user = ReadConfig(env_choose_path).get_value(
+                sections='run_user_name', options=host)
+            if phone.split('-')[1]:
+                self.current_user = phone.split('-')[1]
+            else:
+                phone = ReadConfig(config_file_path).get_value(
+                    sections=default_user, options='username')
+                self.current_user = phone
+        else:
+            phone = ReadConfig(config_file_path).get_value(
+                sections='lzp', options='username')
+            self.current_user = phone
+        if student_token_cache.get(self.current_user):
+            return student_token_cache[self.current_user]
+        #  支持传入海外区号用户
+        if '+' in self.current_user :
+            phone_list =self.current_user.split("+")
+            countryCode = phone_list[1]
+            self.current_user = phone_list[0]
+        else:
+            countryCode = 86
+        post_data = {"phone": self.current_user, "password": "A123456", "countryCode": countryCode, "loginType": 2,
+                     "userType": 1,
+                     "subjectType": 0}
+
+        try:
+            m_token = requests.post(url=login_url, json=post_data, verify=False,
+                                    headers={'huohua-podenv': self.jira_id})
+            if m_token.status_code != 200:
+                raise KeyError('获取 parent token失败！')
+            token = m_token.json()
+            token = token['data']['token']
+            student_token_cache[self.current_user] = token
+            return token
+        except :
+            raise TypeError('获取 parent token失败！')
+
+    def get_xdu_parent_token_by_pwd(self, phone=None, host=None):
+        env_name = get_current_config(section='run_evn_name', key='current_business')
+        obj_log.info("env_name:{}".format(env_name))
+        login_url = 'http://'+host+"/app/sign/phone"
+        result = parse.urlparse(url=login_url, allow_fragments=True)
+        host = result.hostname
+        choose_user = ReadConfig(env_choose_path).get_options('run_user_name')
+        obj_log.info("host:{},choose_user:{}".format(host, choose_user))
+        if host in choose_user:
+            default_user = ReadConfig(env_choose_path).get_value(
+                sections='run_user_name', options=host)
+        post_data = {"phone": phone, "verifyCode": 1111, "countryCode": 86, "type": 1}
+        try:
+            m_token = requests.post(url=login_url, json=post_data, verify=False,
+                                    headers={'huohua-podenv': self.jira_id})
+            if m_token.status_code != 200:
+                raise KeyError('获取 parent token失败！')
+            token = m_token.json()
+            token = token['data']['token']
+            student_token_cache[self.current_user] = token
+            return token
+        except :
+            raise TypeError('获取 parent token失败！')
+
+
+class StudentLogin(InitConfig):
+    def __init__(self, host=None, curt_user=None, current_evn=None):
+        # super().__init__(run_user_name=curt_user, current_evn=current_evn)
+        InitConfig.__init__(self, run_user_name=curt_user, current_evn=current_evn)
+        self.jira_id = ReadConfig(env_choose_path).get_value(sections='run_jira_id', options='huohua-podenv')
+
+    def get_student_token_by_pwd(self, phone=None):
+        env_name = get_current_config(section='run_evn_name', key='current_business')
+        if env_name and env_name=='hhi':
+            login_url = "https://hhi-core-api.qa.visparklearning.com/token"
+        else:
+            login_url = "https://core-api.qa.huohua.cn/token"
+        result = parse.urlparse(url=login_url, allow_fragments=True)
+        host = result.hostname
+        choose_user = ReadConfig(env_choose_path).get_options('run_user_name')
+        if host in choose_user:
+            default_user = ReadConfig(env_choose_path).get_value(
+                sections='run_user_name', options=host)
+            if phone:
+                self.current_user = phone
+            else:
+                phone = ReadConfig(config_file_path).get_value(
+                    sections=default_user, options='username')
+                self.current_user = phone
+        else:
+            phone = ReadConfig(config_file_path).get_value(
+                sections='lzp', options='username')
+            self.current_user = phone
+        if student_token_cache.get(self.current_user):
+            return student_token_cache[self.current_user]
+        #  支持传入海外区号用户
+        if '+' in self.current_user :
+            phone_list =self.current_user.split("+")
+            countryCode = phone_list[1]
+            self.current_user = phone_list[0]
+        else:
+            countryCode = 86
+        post_data = {"phone": self.current_user, "password": "A123456", "countryCode": countryCode, "authCodeType":2,"loginType": 2}
+        try:
+            m_token = requests.post(url=login_url, json=post_data, verify=False, headers={'huohua-podenv': self.jira_id})
+
+            if m_token.status_code != 200:
+                raise KeyError('获取 student token失败！')
+            token = m_token.json()
+            token = token['data']['token']
+            student_token_cache[self.current_user] = token
+            return token
+        except :
+            raise TypeError('获取 student token失败！')
+
+
+class SparkEduLogin(InitConfig):
+    def __init__(self, host=None, curt_user=None, current_evn=None):
+        # super().__init__(run_user_name=curt_user, current_evn=current_evn)
+        InitConfig.__init__(self, run_user_name=curt_user, current_evn=current_evn)
+        self.jira_id = ReadConfig(env_choose_path).get_value(sections='run_jira_id', options='huohua-podenv')
+        team_name = ReadConfig(env_choose_path).get_value(sections='run_evn_name', options="current_team")
+
+    def get_spark_edu_graphic_code(self):
+        """
+        功能：获取图形验证码
+        """
+        code_url = "https://api.qa.sparkedu.com/third/passport/captcha"
+        post_data = {"width":240,"height":90}
+        resp = requests.post(url=code_url, json=post_data, verify=False, headers={'huohua-podenv': self.jira_id})
+        resp = resp.json()
+        return resp['data']['captchaId']
+
+    def check_spark_edu_graphic_code(self, captcha_id, captcha='1234',):
+        """
+        功能：校验图形验证码
+        captcha_id：图形验证码id，由get_spark_edu_graphic_code返回
+        captcha：图形验证码，默认1234，需事先配置好
+        """
+        check_url = "https://api.qa.sparkedu.com/third/passport/captcha/check"
+        post_data = {"captchaId":captcha_id,"captcha":"1234"}
+        resp = requests.post(url=check_url, json=post_data, verify=False, headers={'huohua-podenv': self.jira_id})
+        resp = resp.json()
+        print(resp)
+
+    def spark_edu_login_by_code(self, phone=None, auth_code=None,
+                                auth_code_type=17, country_code=86, language="zh-HK", platform="1",
+                                login_type=1, user_type=1):
+        """
+        功能：验证码登录
+        phone: 登录用户手机号
+        auth_code：验证码，这里默认验证码为1234，需提前配置
+        其余参数：默认
+        """
+        login_url = "https://api.qa.sparkedu.com/third/passport/login"
+        if not phone:
+            # 如果没有传入phone，则从小组配置文件中读取
+            team_config = os.path.abspath(os.path.join(ROOT_PATH, 'UBJ/library/Config/team_config.ini'))
+            phone = ReadConfig(team_config).get_value(sections='default_spark_edu_user', options='phone')
+        if not auth_code:
+            # 如果没有传入验证码，则从小组配置文件中读取
+            team_config = os.path.abspath(os.path.join(ROOT_PATH, 'UBJ/library/Config/team_config.ini'))
+            auth_code = ReadConfig(team_config).get_value(sections='default_spark_edu_user', options='auth_code')
+        post_data = {"phone":phone,
+                     "authCode":auth_code,
+                     "countryCode":country_code,
+                     "authCodeType":auth_code_type,
+                     "language":language,
+                     "platform":platform,
+                     "loginType":login_type,
+                     "userType":user_type}
+        m_token = requests.post(url=login_url, json=post_data, verify=False, headers={'huohua-podenv': self.jira_id})
+        if m_token.status_code == 200:
+            token = m_token.json()
+            return token['data']['token']
+        else:
+            print(m_token.content)
+            raise KeyError('获取 student token失败！')
+
+
+class SparkSaasLogin(InitConfig):
+    def __init__(self, host=None, curt_user=None, current_evn=None):
+        # super().__init__(run_user_name=curt_user, current_evn=current_evn)
+        InitConfig.__init__(self, run_user_name=curt_user, current_evn=current_evn)
+        self.team_config = os.path.abspath(os.path.join(ROOT_PATH, 'BIZ/library/Config/team_config.ini'))
+        self.jira_id = ReadConfig(env_choose_path).get_value(sections='run_jira_id', options='huohua-podenv')
+
+    def login_biz_saas(self, phone="", email="", password="", country_code="", auth_code="9999"):
+        """
+        Args:
+            phone:
+            email:
+            password: 密码，
+            country_code: 86, config.country_code - 目前有问题，暂时写到86
+            auth_code: 验证码
+        Returns:
+        """
+        url = "https://sc-saas-api.qa.huohua.cn/api/session/integration/login"
+        team_config = os.path.abspath(os.path.join(ROOT_PATH, 'BIZ/library/Config/team_config.ini'))
+        a = ReadConfig(team_config).get_value(sections='saas_org_info', options='SAAS_BELONG_SCHOOL')
+        country_code = ReadConfig(team_config).get_value(sections='saas_org_info', options='COUNTRY_CODE')
+        data = {
+        }
+        if phone and password == "":
+            data["authCode"] = auth_code
+            data["loginType"] = 2  # 登录类型
+            data["phone"] = str(phone)
+            data["countryCode"] = country_code
+        elif phone and password:
+            data["password"] = password
+            data["loginType"] = 1  # 登录类型
+            data["phone"] = str(phone)
+            data["countryCode"] = country_code
+        elif email and password:
+            data["password"] = password
+            data["loginType"] = 3  # 登录类型
+            data["email"] = email
+        elif phone == "" and password == "":
+            phone = ReadConfig(team_config).get_value(sections='saas_org_info', options='SAAS_NAME')
+            data["authCode"] = auth_code
+            data["loginType"] = 2  # 登录类型
+            data["phone"] = str(phone)
+            data["countryCode"] = country_code
+            print(str(country_code),str(phone),str(auth_code))
+        else:
+            print("登录方式错误...")
+            return False
+
+        print("开始登录 saas 系统..., 打印登录信息: {}".format(phone))
+        response = requests.post(url=url, json=data, verify=False, headers={'huohua-podenv': self.jira_id, "Content-Type": "application/json;charset=UTF-8"})
+        if response:
+            pass
+        else:
+            print("登录 SaaS 系统失败...{}".format(response))
+            return False
+        print(response.text)
+
+        token = response.json()['data']['token']
+        status = response.status_code
+        if status:
+            if status == 200 and token:
+                print("登录 SaaS 系统成功..., Token: {}".format(token[0]))
+                print("写入到config配置文件...")
+                return token
+            else:
+                print("登录 SaaS 系统失败...{}".format(response))
+                return False
+        else:
+            print("登录 SaaS 系统失败...{}".format(response))
+            return False
+
+class SparkSaasTeacherLogin(InitConfig):
+    def __init__(self, host=None, curt_user=None, current_evn=None):
+        # super().__init__(run_user_name=curt_user, current_evn=current_evn)
+        InitConfig.__init__(self, run_user_name=curt_user, current_evn=current_evn)
+        self.team_config = os.path.abspath(os.path.join(ROOT_PATH, 'BIZ/library/Config/team_config.ini'))
+        self.jira_id = ReadConfig(env_choose_path).get_value(sections='run_jira_id', options='huohua-podenv')
+
+    def rsa_encrypt(self, text):
+        """
+        Args:
+            text:
+        Returns:
+        """
+        public_key = "MIGfMA0GCSqGSIb3DQEBAQUAA4GNADCBiQKBgQDGsNh8Pu1zY9SrblsJDuITHQ+ObAyjGbQARgh3TQtaz5vYQ+avTZJJsJLxiZ5hpf1/5OCb3QZRzbWebrk9cgU892GGc8MDBtH7l/I1GrZq2mY9LO4wWjA0gC7qK0o3yqAwwNyh3uXkSsN6KlpXnac1G1o+Sjlr1P1IXoB4I5MlAQIDAQAB"
+        public_key_str = "-----BEGIN PUBLIC KEY-----\n{}\n-----END PUBLIC KEY-----".format(public_key)
+
+        # 字符串指定编码（转为bytes）
+        text = text.encode('utf-8')
+        # 构建公钥对象
+        cipher_public = PKCS1_v1_5.new(RSA.importKey(public_key_str))
+        # 加密（bytes）
+        text_encrypted = cipher_public.encrypt(text)
+        # base64编码，并转为字符串
+        text_encrypted_base64 = base64.b64encode(text_encrypted).decode()
+        return text_encrypted_base64
+
+    def spark_login_teacher(self, phone, auth_code='9999', auth_code_type=2, country_code="", pass_word="abc123", login_type="验证码"):
+        """
+        Args:
+            phone:
+            auth_code:
+            auth_code_type:
+            country_code:
+            login_type: 验证码，密码
+            pass_word: 密码
+        Returns: token
+        """
+        team_config = os.path.abspath(os.path.join(ROOT_PATH, 'BIZ/library/Config/team_config.ini'))
+        country_code = ReadConfig(team_config).get_value(sections='saas_org_info', options='TEACHER_COUNTRY_CODE')
+        data = {
+            "authCodeType": auth_code_type,
+            "countryCode": country_code
+        }
+
+        phone_encrypt = SparkSaasTeacherLogin().rsa_encrypt(str(phone))
+        data["phone"] = phone_encrypt
+
+        if login_type == "验证码":
+            print("教师端验证码登录")
+            url = "https://sc-mce-teacher-api.qa.huohua.cn/api/session"
+            data["authCode"] = auth_code
+            print(data)
+
+        else:
+            print("教师端密码登录")
+            url = "https://sc-mce-teacher-api.qa.huohua.cn/api/session/phone_pass"
+            pass_word_encrypt = SparkSaasTeacherLogin().rsa_encrypt(pass_word)
+            data["password"] = pass_word
+
+        print("开始登录 教师端 系统..., 打印登录手机号码, {}".format(phone))
+        # response = self.request.http_request(url, method='post', json=data)
+        response = requests.post(url=url, json=data, verify=False, headers={'huohua-podenv': self.jira_id, "Content-Type": "application/json;charset=UTF-8"})
+        if response:
+            pass
+        else:
+            print("登录 教师端 系统失败...{}".format(response))
+            return False
+        token = response.json()['data']['token']
+        status = response.status_code
+        if status:
+            if status and token:
+                print("登录 教师端 系统成功..., Token: {}".format(token))
+                print("写入到config配置文件...")
+                return token
+            else:
+                print("登录 教师端 系统失败...{}".format(response))
+                return False
+        else:
+            print("登录 教师端 系统失败...{}".format(response))
+            return False
+
+if __name__ == '__main__':
+    test = LoginSys()
+    id = test.ug_sso_login_by_starlight()
+    print(id)
+    headers = {"Content-Type": "application/json;charset=UTF-8"}
+    kwargs = {"lessonQueryModel":{},"pageModel":{"pageNum":1,"pageSize":10}}
+    url = "https://starlight.qa.huohua.cn/api/lesson/queryList"
+    resp = id.post(url=url, json=kwargs, headers=headers, allow_redirects=False, verify=False)
+    print(resp)