post get 两种方法

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


# a. 定义一个get函数 get(url,headers=None)
def get(url,headers=None):
    return urlrequests(url,headers=headers)   #内部可以调用urlrequests   #如果不写headers=headers会被默认为None
# b. 定义一个post函数 post(url,headers=None)
def post(url,form,headers=None):
    return urlrequests(url,form,headers=headers) # 这个写不写都可以   但是建议所有None都写上headers=headers
#1. 传入url
#2. user_agent
#3. headers
#4. 定义Request
#5. urlopen
#6. 返回byte数组
def urlrequests(url,form=None,headers=None): #如果传了form就认为是post   不传就不是post
            # 如果传了form就能把form_bytrs获得到

    user_agent = 'Mozilla/5.0 (Windows NT 6.1; WOW64) AppleWebKit/537.36 (KHTML, like Gecko) Chrome/67.0.3396.99 Safari/537.36'
    # 如果用户需要自行传入headers, 则覆盖之前的headers
    if headers == None:
        headers = {
            'User-Agent': user_agent
        }
    html_bytes = b''
    try:
        if form:                      # 判断form是否存在  也有可能是None
            # 有form的情况就说明是 POST
            # 转换成str
            form_str = parse.urlencode(form)          # 想让用户传进来的是字典类型
            # 转换成bytes
            form_bytes = form_str.encode('utf-8')
            req = request.Request(url, data=form_bytes, headers=headers)  # form传到Request里
        else:
            # 没有form就是GET
            req = request.Request(url,headers=headers)   #form传到Request里

        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__':

 # 用post方法
    # url = 'http://fanyi.baidu.com/sug'
    # form = {
    #     'kw':'嘿嘿'
    # }
    # html_bytes = post(url,form=form)
    # print(html_bytes)

 # 用get方法
    url = 'http://www.baidu.com'
    html_byte = get(url)
    print(html_byte)

# 当封装过之后的都是三个阶段
# 第一个阶段  准备数据
# 第二个阶段  调用函数
# 第三个阶段  存储\打印