接口登录CSDN发布博客---封装方法，使用unittest框架

一个简单的跑接口流程：登录后发表带图片的博客。这里涉及到的知识点：

1、登录时通过cookies去保持登录状态，把cookies添加到一个session中，这样可以保持长时间登录状态；

2、我们通过爬虫获取页面的title标签的文本判断登录成功；

3、通过抓包找到上传图片的地址，通过post方式来上传图片；

4、抓包找到发布博客的地址，最后发布博客；

5、使用了unittest框架来完成接口流程测试，作为一个测试用例。

ApiMethod.py的代码如下：

 # coding:utf-8
 import requests
 from bs4 import BeautifulSoup
 class ApiMethod():
     def __init__(self,userAgent):
         self.h = {"User-Agent": userAgent}
     #通过session来登录CSDN，保持一直登录状态
     def loginCsdn(self):
         try:
             s= requests.session()
             c = requests.cookies.RequestsCookieJar()
             c.set("UserInfo","ltxlmnyec9obNIiCirM0KBmlYeijHfUcQ%2BZE7J%2B%2Ffk1djT8xd36OsnAqrYJ46KvVN8f%2BdG6KiVfFSHuxAmSJTT8VFZF%2FpmyBVj3McsPss%2FAH6gR3SuyN3ms9A9pd7xaq8T%2FZ%2Fcr5ZOVFqSGjNvpRqQ%3D%3D")
             c.set("UserNick","%E9%A3%8E%E6%A5%9A%E9%A2%9C")
             c.set("AU","FCF")
             c.set("Hm_lpvt_6bcd52f51e9b3dce32bec4a3997715ac","")
             c.set("dc_tos","p65d8s")
             c.set("UserName","maybe_frank")
             s.cookies.update(c)
             re = s.get("https://mp.csdn.net/postedit",headers = self.h)
             soup = BeautifulSoup(re.content,"html.parser")
             title = soup.title.string
             if title == "写文章-CSDN博客":
                 print("登录成功！")
             else:
                 print("登录失败")
             return s
         except Exception as e:
             print(e)

     def upload(self,session,urlPath,fileName,file):
         try:
             f = {"file":(fileName,open(file,"rb"),"image/png")}
             re = session.post(url=urlPath,headers = self.h,files = f)
             return re.json()["content"]
         except Exception as e:
             print(e)

     def write(self,session,urlPath,data):
         try:
             body = data
             re = session.post(url=urlPath,headers = self.h,data=body)
             #返回上传图片后的图片url
             return  re.json()["content"]
         except Exception as e:
             print(e)

 if __name__ == '__main__':
     sf = ApiMethod("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:58.0) Gecko/20100101 Firefox/58.0")
     s = sf.loginCsdn()
     sf.upload(s,"https://mp.csdn.net/UploadImage","Q1.png","C:\\Users\\ztf.YFK\\Desktop\\1.png")
     data = {"titl":"wers",
              "typ":1,
              "cont":'<p>上传文件：</p><p><img src="https://img-blog.csdn.net/20180326110705916" alt="" /><br /></p>',
              "categories":"接口",
              "chnl":16,
              "level":0,
              "tag2":"",
              "artid":0,
              "private":"false",
              "stat":"publish"}
     content = sf.write(s,"https://mp.csdn.net/postedit/saveArticle?isPub=1",data)
     print(content)

write.py的代码如下：

 from csdn.apiMethod import ApiMethod
 import unittest
 import time
 import random
 class WriteBlog(unittest.TestCase):
     @classmethod
     def setUpClass(cls):
         cls.sf = ApiMethod("Mozilla/5.0 (Windows NT 10.0; Win64; x64; rv:58.0) Gecko/20100101 Firefox/58.0")
         cls.s = cls.sf.loginCsdn()

     @classmethod
     def tearDownClass(cls):
         pass

     def test_writeBlog(self):
         file_url = self.upload(self.s,"https://mp.csdn.net/UploadImage","Q1.png","C:\\Users\\ztf.YFK\\Desktop\\1.png")
         print("上传文件后的url:",end="")
         print(file_url)
         #发布博客时的传的body的数据
         num= random.randint(1,1000)
         self.data = {"titl":"wers%s"%num,
          "typ":1,
          "cont":'<p>上传文件：</p><p><img src="https://img-blog.csdn.net/20180326110705916" alt="" /><br /></p>',
          "categories":"接口",
          "chnl":16,
          "level":0,
          "tag2":"",
          "artid":0,
          "private":"false",
          "stat":"publish"}

         #调用发布博客接口
         content = self.sf.write(self.s,"https://mp.csdn.net/postedit/saveArticle?isPub=1",self.data)
         print(content)
         #self.assertTrue(self,content,"发布成功！")
         self.assertEqual(self,content,"发布成功","调用发布博客接口请求失败！")

     if __name__ == '__main__':
         unittest.main()

运行后，可以去CSDN博客查看是否发布成功。代码经实际调试可以跑成功。