import requests
from lxml import etree
import re

data ={
"ssdm":"","dwmc":"","mldm":"zyxw","mlmc":"","yjxkdm":"","zymc":"","xxfs":"","pageno":""

}

class UniversityInfo:
    def __init__(self):
        # 招生单位、考试方式、院系所、专业、学习方式、研究方向、指导老师、拟招人数、备注、政治、外语、业务课一、业务课二
        self.EnrollmentUnit = ""
        self.ExaminationMethod = ""
        self.CollegesDepartments = ""
        self.Major = ""
        self.learningStyle = ""
        self.ResearchDirection = ""
        self.Instructor = ""
        self.Number = ""
        self.Remarks = ""
        self.Politics = ""
        self.English = ""
        self.BusinessClass1 = ""
        self.BusinessClass2 = ""


class YanZhaoWang:

    def __init__(self,code):
        self.url = "//yz.chsi.com.cn/zsml/queryAction.do?ssdm&dwmc&mldm=zyxw&mlmc&yjxkdm={}&zymc&xxfs&pageno={}".format(code)
        self.data = data
        self.data["yjxkdm"]=code
        self.page = 1
        self.schoolUrl = []
        self.facultyUrl = []
        self.schoolInfo = []
    def GetFacultyUrl(self):
        for i in self.schoolUrl:
            response = requests.post("//yz.chsi.com.cn"+i).text
            response = etree.HTML(response)
            url = response.xpath('//table/tbody//td[8]/a/@href')
            for i in url:
                print(i)
                self.facultyUrl.append(i)

    def GetSchoolInfo(self):
        for i in self.facultyUrl:
            response = requests.post("//yz.chsi.com.cn/"+i).text
            response = etree.HTML(response)
            schoolinfo = UniversityInfo()
            schoolinfo.EnrollmentUnit = response.xpath('//table[@class="zsml-condition"]/tbody/tr[1]/td[2]/text()')[0]
            schoolinfo.ExaminationMethod = response.xpath('//table[@class="zsml-condition"]/tbody/tr[1]/td[4]/text()')[0]
            schoolinfo.CollegesDepartments = response.xpath('//table[@class="zsml-condition"]/tbody/tr[2]/td[2]/text()')[0]
            schoolinfo.Major = response.xpath('//table[@class="zsml-condition"]/tbody/tr[2]/td[4]/text()')[0]
            schoolinfo.learningStyle = response.xpath('//table[@class="zsml-condition"]/tbody/tr[3]/td[2]/text()')[0]
            schoolinfo.ResearchDirection = response.xpath('//table[@class="zsml-condition"]/tbody/tr[3]/td[4]/text()')[0]
            schoolinfo.Instructor = response.xpath('//table[@class="zsml-condition"]/tbody/tr[4]/td[2]/text()')
            schoolinfo.Number = response.xpath('//table[@class="zsml-condition"]/tbody/tr[4]/td[4]/text()')[0]
            schoolinfo.Remarks = response.xpath('//table[@class="zsml-condition"]/tbody/tr[5]/text()')[0]
            table = response.xpath('//tbody[@class="zsml-res-items"]')
            for i in table:
                schoolinfo.Politics = i.xpath('tr/td[1]/text()')[0]
                schoolinfo.English = i.xpath('tr/td[2]/text()')[0]
                schoolinfo.BusinessClass1 = i.xpath('tr/td[3]/text()')[0]
                schoolinfo.BusinessClass1 = i.xpath('tr/td[4]/text()')[0]
                self.schoolInfo.append(schoolinfo)

    def GetSchoolUrl(self,url):
        response = requests.post(url).text
        response = etree.HTML(response)
        page = response.xpath('//li[@class="lip lip-last"]/a/@onclick')
        url = response.xpath('//*[@id="form3"]/a/@href')
        for i in url:
            self.schoolUrl.append(i)
        print(page)
        if page!=[]:
            self.page += 1
            self.GetSchoolUrl(self.url.format(self.page))

    def WriteSchool(self):
        with open("./text.txt",'a',encoding="utf-8")as f:
            for i in self.schoolInfo:
                print(i.EnrollmentUnit)
                f.write(self.Tostring(i.EnrollmentUnit)+"\t"+self.Tostring(i.ExaminationMethod)+"\t"+self.Tostring(i.CollegesDepartments)+"\t"+self.Tostring(i.Major)+"\t"+self.Tostring(i.learningStyle)+"\t"+self.Tostring(i.ResearchDirection)+"\t"+self.Tostring(i.Instructor)+"\t"+self.Tostring(i.Number)+"\t"+self.Tostring(i.Remarks)+"\t"+self.Tostring(i.Politics)+"\t"+self.Tostring(i.English)+"\t"+self.Tostring(i.BusinessClass1)+'\t'+self.Tostring(i.BusinessClass2)+'\n')
    def Tostring(self,str):
        str ="".join(str)
        return re.sub("[\t\r\n]+","",str)
    def run(self):
        self.GetSchoolUrl(self.url.format(self.page))
        self.GetFacultyUrl()
        self.GetSchoolInfo()
        self.WriteSchool()


if __name__ == '__main__':

    # 括号内填专业代码
    yzw = YanZhaoWang('1055')
    yzw.run()