从零搭建教务抢课系统（二）

目录

（一）核心功能：模拟登陆
（二）使用Cookie进行模拟登录
（三）获取教务网选课列表
（四）循环选课
（五）断线重连

Github链接： https://github.com/njuwuyuxin/CourseGrabber

使用cookie模拟登录

在成功实现了基本登陆后，为了方便使用，这里尝试使用cookie进行登录

首先我们在创建session时，初始cookie为空，可以打印进行查看

s = requests.session()
print(s.cookies.get_dict())

在构造登陆请求体，成功登陆之后，session中的cookies被自动更新，可以打印查看，南大教务平台的Cookie形如

{
  'user_id':'"1612xxxxx 1577448172273"',
  'ARRAffinity':'80372ade9da56061dc1cfb0f216b6917726c2c01e3d804e60cad7fce0e0af662',
  'JSESSIONID':'8D6204D5EDBE04DC6088DB9BE43A5924'
}

可以看到共有三个表项，这里在成功登录之后，手动将其保存到本地文件中。这里没有使用相关库函数，而是手动实现了简单的cookie存储

def SaveCookie(session):
    with open(".cookie",'w') as f:
        for key,val in session.cookies.get_dict().items():
            f.write(key+":"+val+'\n')

同时实现了从文件读取cookie的方法

def GetCookie(session):
    cookie = {}
    if ".cookie" not in os.listdir():
        return cookie
    with open(".cookie",'r') as f:
        for line in f:
            line = line.replace('\n','').replace('\r','')
            item = line.split(':')
            cookie[item[0]] = item[1]
    return cookie

cookie的保存与读取实现之后，登陆部分的逻辑可以改为：

1. 首先检查本地是否存在cookie
2. 如果存在cookie，尝试使用cookie登录；如果不存在，直接使用账号密码登录
3. 如果cookie登录成功，直接进入系统；如果cookie已过期，则重新使用账号密码登录，并更新本地cookie

#获取cookie，如果本地有cookie，尝试使用cookie登录
    c = GetCookie(session)
    session.cookies.update(c)
    if c:
        rs = session.get(host+"student/index.do")    #教务平台首页，如果能够进入，说明已成功登录
        if rs.content.__len__() > 5000:
            print("登陆成功!")
            return True
        else:
            print("登录已过期，请重新登录")

这样我们基本实现了使用cookie进行登录，避免了重复输入账号密码及验证码的验证