# 導(dǎo)入包
import numpy as np
import pandas as pd
import requests
import json
from fake_useragent import UserAgent
import time

# 獲取一頁(yè)
def get_one_page(page_num):
    # 獲取URL
    url = 'https://api.eol.cn/gkcx/api/'

    # 構(gòu)造headers
    headers = {
        'User-Agent': UserAgent().random,
        'Origin': 'https://gkcx.eol.cn',
        'Referer': 'https://gkcx.eol.cn/school/search?province=&schoolflag=&recomschprop=',
    }

    # 構(gòu)造data
    data = {
        'access_token': "",
        'admissions': "",
        'central': "",
        'department': "",
        'dual_class': "",
        'f211': "",
        'f985': "",
        'is_dual_class': "",
        'keyword': "",
        'page': page_num,
        'province_id': "",
        'request_type': 1,
        'school_type': "",
        'size': 20,
        'sort': "view_total",
        'type': "",
        'uri': "apigkcx/api/school/hotlists",
    }

    # 發(fā)起請(qǐng)求
    try:
        response = requests.post(url=url, data=data, headers=headers)
    except Exception as e:
        print(e) 
        time.sleep(3) 
        response = requests.post(url=url, data=data, headers=headers)

# 解析獲取數(shù)據(jù)
    school_data = json.loads(response.text)['data']['item']

    # 學(xué)校名
    school_name = [i.get('name') for i in school_data]
    # 隸屬部門(mén)
    belong = [i.get('belong') for i in school_data]
    # 高校層次
    dual_class_name = [i.get('dual_class_name') for i in school_data]
    # 是否985
    f985 = [i.get('f985') for i in school_data]
    # 是否211
    f211 = [i.get('f211') for i in school_data]
    # 辦學(xué)類(lèi)型
    level_name = [i.get('level_name') for i in school_data]
    # 院校類(lèi)型
    type_name = [i.get('type_name') for i in school_data]
    # 是否公辦
    nature_name = [i.get('nature_name') for i in school_data]
    # 人氣值
    view_total = [i.get('view_total') for i in school_data]
    # 省份
    province_name = [i.get('province_name') for i in school_data]
    # 城市
    city_name = [i.get('city_name') for i in school_data]
    # 區(qū)域
    county_name = [i.get('county_name') for i in school_data]
    
    # 保存數(shù)據(jù)
    df_one = pd.DataFrame({
        'school_name': school_name,
        'belong': belong,
        'dual_class_name': dual_class_name,
        'f985': f985,
        'f211': f211,
        'level_name': level_name,
        'type_name': type_name,
        'nature_name': nature_name,
        'view_total': view_total,
        'province_name': province_name,
        'city_name': city_name,
        'county_name': county_name,
    })

    return df_one

# 獲取多頁(yè)
def get_all_page(all_page_num):
    # 存儲(chǔ)表
    df_all = pd.DataFrame()

    # 循環(huán)頁(yè)數(shù)
    for i in range(all_page_num):
        # 打印進(jìn)度
        print(f'正在獲取第{i + 1}頁(yè)的高校信息')
        # 調(diào)用函數(shù)
        df_one = get_one_page(page_num=i+1)
        # 追加
        df_all = df_all.append(df_one, ignore_index=True)
        # 休眠
        time.sleep(np.random.uniform(2))

    return df_all


# 運(yùn)行函數(shù)
df_school = get_all_page(all_page_num=143)

# 讀出數(shù)據(jù)
df_school.to_excel('./data/全國(guó)高校數(shù)據(jù).xlsx', index=False)