抢课Python

 

import requests

from PIL import Image

from bs4 import BeautifulSoup

import copy

import time

import re

import os

import json

import threading

 

 

class Spider:

    class Lesson:

 

        def __init__(self, name, code, teacher_name, Time, number):

            self.name = name

            self.code = code

            self.teacher_name = teacher_name

            self.time = Time

            self.number = number

 

        def show(self):

            print(' name:' + self.name + ' code:' + self.code + ' teacher_name:' + self.teacher_name + ' time:' + self.time)

 

    def __init__(self, url):

        self.__uid = ''

        self.__real_base_url = ''

        self.__base_url = url

        self.__name = ''

        self.__base_data = {

            '__EVENTTARGET': '',

            '__EVENTARGUMENT': '',

            '__VIEWSTATE': '',

            'ddl_kcxz': '',

            'ddl_ywyl': '',

            'ddl_kcgs': '',

            'ddl_xqbs': '',

            'ddl_sksj': '',

            'TextBox1': '',

            'dpkcmcGrid:txtChoosePage': '1',

            'dpkcmcGrid:txtPageSize': '200',

        }

        self.__headers = {

            'User-Agent': 'Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.62 Safari/537.36',

        }

        self.session = requests.Session()

        self.__now_lessons_number = 0

 

 

    def __set_real_url(self):

        '''

        得到真实的登录地址（无Cookie）

        获取Cookie（有Cookie)

        :return: 该请求

        '''

        request = self.session.get(self.__base_url, headers=self.__headers)

        real_url = request.url

        if real_url != 'http://218.75.197.123:83/' and real_url != 'http://218.75.197.123:83/index.apsx': # 湖南工业大学

            self.__real_base_url = real_url[:len(real_url) - len('default2.aspx')]

        else:

            if real_url.find('index') > 0:

                self.__real_base_url = real_url[:len(real_url) - len('index.aspx')]

            else:

                self.__real_base_url = real_url

        return request

 

    def __get_code(self):

        '''

        获取验证码

        :return: 验证码

        '''

        if self.__real_base_url != 'http://218.75.197.123:83/':

            request = self.session.get(self.__real_base_url + 'CheckCode.aspx', headers=self.__headers)

        else:

            request = self.session.get(self.__real_base_url + 'CheckCode.aspx?', headers=self.__headers)

        with open('code.jpg', 'wb')as f:

            f.write(request.content)

        im = Image.open('code.jpg')

        im.show()

        print('Please input the code:')

        code = input()

        return code

 

    def __get_login_data(self, uid, password):

        '''

        得到登录包

        :param uid: 学号

        :param password: 密码

        :return: 含登录包的data字典

        '''

        self.__uid = uid

        request = self.__set_real_url()

        soup = BeautifulSoup(request.text, 'lxml')

        form_tag = soup.find('input')

        __VIEWSTATE = form_tag['value']

        code = self.__get_code()

        data = {

            '__VIEWSTATE': __VIEWSTATE,

            'txtUserName': self.__uid,

            'TextBox2': password,

            'txtSecretCode': code,

            'RadioButtonList1': '学生'.encode('gb2312'),

            'Button1': '',

            'lbLanguage': '',

            'hidPdrs': '',

            'hidsc': '',

        }

        return data

 

    def login(self, uid, password):

        '''

        外露的登录接口

        :param uid: 学号

        :param password: 密码

        :return: 抛出异常或返回是否登录成功的布尔值

        '''

        while True:

            data = self.__get_login_data(uid, password)

            if self.__real_base_url != 'http://218.75.197.123:83/':

                request = self.session.post(self.__real_base_url + 'default2.aspx', headers=self.__headers, data=data)

            else:

                request = self.session.post(self.__real_base_url + 'index.aspx', headers=self.__headers, data=data)

            soup = BeautifulSoup(request.text, 'lxml')

            if request.status_code != requests.codes.ok:

                print('4XX or 5XX Error,try to login again')

                time.sleep(0.5)

                continue

            if request.text.find('验证码不正确') > -1:

                print('验证码错误')

                continue

            if request.text.find('密码错误') > -1:

                print('密码错误')

                return False

            if request.text.find('用户名不存在') > -1:

                print('用户名错误')

                return False

            try:

                name_tag = soup.find(id='xhxm')

                self.__name = name_tag.string[:len(name_tag.string) - 2]

                print('欢迎' + self.__name)

                self.__enter_lessons_first()

                return True

            except:

                print('未知错误，尝试再次登录')

                time.sleep(0.5)

                continue

 

    def __enter_lessons_first(self):

        '''

        首次进入选课界面

        :return: none

        '''

        data = {

            'xh': self.__uid,