cookies的使用

保存cookies

#!/usr/bin/env/python
#encoding: UTF-8

import re
import requests
import http.cookiejar
from PIL import Image
import time
import json

headers = {‘User-Agent’: 'Mozilla/5.0 (Windows NT 6.1; Win64; x64) AppleWebKit/537.36 ’
‘(KHTML, like Gecko) Chrome/57.0.2987.98 Safari/537.36’,
“Host”: “www.zhihu.com”,
“Referer”: “https://www.zhihu.com/”,
}

建立一个会话，可以把同一用户的不同请求联系起来；直到会话结束都会自动处理cookies

session = requests.Session()

建立LWPCookieJar实例，可以存Set-Cookie3类型的文件。

而MozillaCookieJar类是存为’/.txt’格式的文件

session.cookies = http.cookiejar.LWPCookieJar(“cookie”)

若本地有cookie则不用再post数据了

try:
session.cookies.load(ignore_discard=True)
except IOError:
print(‘Cookie未加载！’)

def get_xsrf():
“”"
获取参数_xsrf
“”"
response = session.get(‘https://www.zhihu.com’, headers=headers)
html = response.text
get_xsrf_pattern = re.compile(r’<input type=“hidden” name="_xsrf" value="(.*?)"’)
_xsrf = re.findall(get_xsrf_pattern, html)[0]
return _xsrf

def get_captcha():
“”"
获取验证码本地显示
返回你输入的验证码
“”"
t = str(int(time.time() * 1000))
captcha_url = ‘http://www.zhihu.com/captcha.gif?r=’ + t + “&type=login”
response = session.get(captcha_url, headers=headers)
with open(‘cptcha.gif’, ‘wb’) as f:
f.write(response.content)
# Pillow显示验证码
im = Image.open(‘cptcha.gif’)
im.show()
captcha = input('本次登录需要输入验证码： ')
return captcha

def login(username, password):
“”"
输入自己的账号密码，模拟登录知乎
“”"
# 检测到11位数字则是手机登录
if re.match(r’\d{11}$’, username):
url = ‘http://www.zhihu.com/login/phone_num’
data = {’_xsrf’: get_xsrf(),
‘password’: password,
‘remember_me’: ‘true’,
‘phone_num’: username
}
else:
url = ‘https://www.zhihu.com/login/email’
data = {’_xsrf’: get_xsrf(),
‘password’: password,
‘remember_me’: ‘true’,
‘email’: username
}
# 若不用验证码，直接登录
result = session.post(url, data=data, headers=headers)
# 打印返回的响应，r = 1代表响应失败，msg里是失败的原因
# loads可以反序列化内置数据类型，而load可以从文件读取
if (json.loads(result.text))[“r”] == 1:
# 要用验证码，post后登录
data[‘captcha’] = get_captcha()
result = session.post(url, data=data, headers=headers)
print((json.loads(result.text))[‘msg’])
# 保存cookie到本地
session.cookies.save(ignore_discard=True, ignore_expires=True)

def isLogin():
# 通过查看用户个人信息来判断是否已经登录
url = “https://www.zhihu.com/settings/profile”
# 禁止重定向，否则登录失败重定向到首页也是响应200
login_code = session.get(url, headers=headers, allow_redirects=False).status_code
if login_code == 200:
return True
else:
return False

if name == ‘main’:
if isLogin():
print(‘您已经登录’)
else:
account = input(‘输入账号：’)
secret = input(‘输入密码：’)
login(account, secret)

---

作者：haliofwu
来源：CSDN
原文：https://blog.csdn.net/xixi880928/article/details/78339157
版权声明：本文为博主原创文章，转载请附上博文链接！