python接口自動化8-參數化

前言

前面一篇實現了參數的關聯，那種只是記流水賬的完成功能，不便於維護，也沒什么可讀性，接下來這篇可以把每一個動作寫成一個函數，這樣更方便了。

參數化的思維只需記住一點：不要寫死

一、登錄函數

1.s參數是session的一個實例類，先放這里，方便寫后面代碼

2.登錄函數傳三個參數，s是需要調用前面的session類，所以必傳，可以傳個登錄的url,然后payload是賬號和密碼

二、保存草稿

1.編輯內容的標題title和正文body_data參數化了，這樣后面可以方便傳不同值

2.這里返回了獲取到新的url地址，因為后面的postid參數需要在這里提取

 

三、提取postid

1.這里用正則表達式提取url里面的postid

 

四、刪除草稿

1.傳個url和postid就可以了

 

五、參考代碼：

# coding:utf-8
import requests

def login(s, url, payload):
    '''登錄'''
    headers = {"User-Agent": "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:44.0) Gecko/20100101 Firefox/44.0",
               "Accept": "application/json, text/javascript, */*; q=0.01",
               "Accept-Language": "zh-CN,zh;q=0.8,en-US;q=0.5,en;q=0.3",
               "Accept-Encoding": "gzip, deflate, br",
               "Content-Type": "application/json; charset=utf-8",
               "X-Requested-With": "XMLHttpRequest",
               "Content-Length": "385",
               "Cookie": "xxx已省略",
               "Connection": "keep-alive"
               }
    r = s.post(url, json=payload, headers=headers, verify=False)
    result = r.json()
    print result
    return result['success']    # 返回True或False

def save_box(s, url2, title, body_data):
    '''# 獲取報存之后url地址'''
    body = {"__VIEWSTATE": "",
        "__VIEWSTATEGENERATOR": "FE27D343",
        "Editor$Edit$txbTitle": title,
        "Editor$Edit$EditorBody": "<p>"+body_data+"</p>",
        "Editor$Edit$Advanced$ckbPublished": "on",
        "Editor$Edit$Advanced$chkDisplayHomePage": "on",
        "Editor$Edit$Advanced$chkComments": "on",
        "Editor$Edit$Advanced$chkMainSyndication": "on",
        "Editor$Edit$lkbDraft": "存為草稿",
        }
    r2 = s.post(url2, data=body, verify=False)
    print r2.url
    return r2.url

def get_postid(u):
    '''正則提取postid'''
    import re
    postid = re.findall(r"postid=(.+?)&", u)
    print postid  # 這里是list
    if len(postid) < 1:
        return ''
    else:
        return postid[0]

def delete_box(s,url3, postid):
    '''刪除草稿箱'''
    json3 = {"postId": postid}
    r3 = s.post(url3, json=json3, verify=False)
    print r3.json()

if __name__ == "__main__":
    url = "https://passport.cnblogs.com/user/signin"
    payload = {
                "input1": "xxx",
                "input2": "xxx",
                "remember": True
               }
    s = requests.session()
    login(s, url, payload,)
    url2 = "https://i.cnblogs.com/EditPosts.aspx?opt=1"
    u = save_box(s, url2, "標題", "正文內容")
    postid = get_postid(u)
    url3 = "https://i.cnblogs.com/post/delete"
    delete_box(s, url3, postid)

轉自：http://www.cnblogs.com/yoyoketang/p/6891710.html