school_api.client.api.utils

Complexity

Total Complexity

5

Size/Duplication

Total Lines	31
Duplicated Lines	0 %

Importance

Changes

0

2 Functions

Rating	Name	Duplication	Size	Complexity
A	get_view_state_from_html()	0	12	3
A	get_alert_tip()	0	6	2

# -*- coding: utf-8 -*-
from __future__ import absolute_import, unicode_literals

import re

from bs4 import BeautifulSoup

from school_api.exceptions import OtherException


def get_alert_tip(html):
    ''' 获取页面提示信息 '''
    tips = re.findall(r">alert\(\'(.*?)\'", html)
    if tips:
        return tips[0]
    return None


def get_view_state_from_html(html):
    ''' 获取 __VIEWSTATE 值 '''
    pre_soup = BeautifulSoup(html, "html.parser")
    view_state_soup = pre_soup.find(attrs={"name": "__VIEWSTATE"})
    try:
        view_state = view_state_soup['value']
    except TypeError:
        if html.find("网站防火墙") > -1:
            raise OtherException('', "请求被防火墙所拦截, 请降低请求频率")
        raise OtherException('', '获取view_state失败')

    return view_state