爬取页面的封装函数

from urllib import request,parse
from urllib.error import HTTPError,URLError

def get(url, headers=None):
    return urlrequests(url, headers=headers)

def post(url, from, headers=None):
    return urlrequests(url, from, headers=headers)

def urlrequests(url, form=None, headers=None):
    user_agent = 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36'

    if headers == None:
        headers = {
            'User-Agent':user_agent
        }
    html_bytes = b''
    
    try:
        if from:
            form_str = parse.urlencode(form)
            form_bytes = form_str.encode('utf-8')
            req = request.Request(url,data=form_bytes,headers=headers)
        else:
            req = request.Request(url,headers=headers)
        
        response = request.urlopen(req)
        html_bytes = response.read()
    
    except HTTPError as e:
        print(e)
    
    except URLError as e:
        print(e)
    return html_bytes

if __name__ == '__main__':
    url = 'http://www.baidu.com'
    html_byte = get(url)
    print(html_byte)
    
    

猜你喜欢

转载自blog.csdn.net/zhan9le/article/details/81638404