路飞学城-Python爬虫实战密训-第1章

 1 import requests
 2 from bs4 import BeautifulSoup
 3 
 4 
 5 # 登录github并返回帐户名称
 6 def get_info(user_name, user_password):
 7     headers = {
 8         ‘User-Agent‘: ‘Mozilla/5.0 (Windows NT 10.0; WOW64) AppleWebKit/537.36 ‘
 9                       ‘(KHTML, like Gecko) Chrome/66.0.3359.139 Safari/537.36‘}
10     try:
11         # 获取cookie及token信息
12         ret = requests.get(
13             url=‘https://github.com/login‘,
14             headers=headers
15         )
16         cookie_dict1 = ret.cookies.get_dict()
17         res = BeautifulSoup(ret.text, ‘html.parser‘)
18         token = res.find(name=‘input‘, attrs={‘type‘: ‘hidden‘, ‘name‘: ‘authenticity_token‘}).get(‘value‘)
19         # 请求头，完成github登录
20         ret2 = requests.post(
21             url=‘https://github.com/session‘,
22             data={
23                 ‘commit‘: ‘Sign in‘,
24                 ‘utf8‘: ‘?‘,
25                 ‘authenticity_token‘: token,
26                 ‘login‘: user_name,
27                 ‘password‘: user_password
28             },
29             headers=headers,
30             cookies=cookie_dict1
31         )
32         cookie_dict2 = ret2.cookies.get_dict()
33         # 获取用户信息数据
34         ret3 = requests.get(
35             url=‘https://github.com/‘+user_name,
36             headers=headers,
37             cookies=cookie_dict2
38         )
39         res3 = BeautifulSoup(ret3.text, ‘html.parser‘)
40         div = res3.find(name=‘div‘, attrs={‘id‘: ‘js-pjax-container‘})
41         h1 = div.find(name=‘h1‘, attrs={‘class‘: ‘vcard-names‘})
42         name = h1.find(name=‘span‘, attrs={‘class‘: ‘p-name vcard-fullname d-block overflow-hidden‘}).get_text()
43         print(‘The user name is:%s‘ % name)
44         a = div.find(name=‘a‘, attrs={‘class‘: ‘u-photo d-block tooltipped tooltipped-s‘})
45         img = a.find(name=‘img‘, attrs={‘class‘: ‘avatar width-full rounded-2‘}).get(‘src‘)
46         print(‘The user photo address is:%s‘ % img)
47     except:
48         print(‘sorry,cannot get the information‘)
49 
50 
51 if __name__ == ‘__main__‘:
52     user_name = input(‘Please input your login name:‘).strip()
53     user_password = input(‘Please input your password:‘).strip()
54     get_info(user_name, user_password)