用python+selenium将腾讯首页今日话题的内容自动发表到自己cnblog里

目的：使用pyhton下的unittest单元测试框架并结合selenium的webdriver来实现将腾讯首页的今日话题下的内容自动发表达到自己的cnblog里。

思路：创建QQDailyTopic类继承unittest的TestCase类，setUp()方法用于测试执行前的初始化工作，而最后的tearDown()与setUp()方法相呼应，用于测试执行之后的善后工作。

然后用方法get_qq_daily_topic_url获得qq首页今日话题的url；

方法get_title_and_content_from_qq_daily_topic从今日话题获得后面我们在cnblog下新blog需要的标题title和富文本content，提取今日话题title文本和内容的innerHTML；

方法login用于登录，定位元素并输入username和password然后点击登录按钮；

方法set_content用于向富文本框填入内容，这里借助js向添加新随笔的富文本框插入指定内容，通过scecute_scrit()执行js代码；

方法test_transpond_qq_daily_topic用来测试向cnblog里转发qq首页今日话题，包括自动用有效用户名和密码登录我的博客，在添加新随笔的标题和富文本框里自动填入今日话题的标题和内容，最后点击发布按钮。unittest下必须以“test”开头的方法。
实现代码如下：

#coding=utf-8
from selenium import webdriver
import unittest
from time import sleep

class QQDailyTopic(unittest.TestCase):

    def setUp(self): 
        self.dr = webdriver.Firefox()
        self.title, self.content = self.get_title_and_content_from_qq_daily_topic()

    def get_qq_daily_topic_url(self):
        return self.dr.find_element_by_css_selector('#todaytop a').get_attribute('href')

    def get_title_and_content_from_qq_daily_topic(self):
        self.dr.get('http://www.qq.com/')
        url = self.get_qq_daily_topic_url()
        self.dr.get(url)
        title = self.dr.find_element_by_id('sharetitle').text
        content = self.dr.find_element_by_id('articleContent').get_attribute('innerHTML')
        return (title, content)

    def login(self, username, password):
        self.dr.find_element_by_id('input1').send_keys(username)
        self.dr.find_element_by_id('input2').send_keys(password)
        self.dr.find_element_by_id('signin').click()

    #借助js向添加新随笔的富文本框插入指定内容
    def set_content(self, text):
        text = text.strip()
        js = 'document.getElementById("Editor_Edit_EditorBody_ifr").contentWindow.document.body.innerHTML=\'%s\'' %(text)
        print(js)
        self.dr.execute_script(js)

    def test_transpond_qq_daily_topic(self):
        self.dr.get('https://passport.cnblogs.com/user/signin')
        self.login('kemi_xxxx', 'kemi_xxxx')#自己博客园用户名和密码
        sleep(3)

        self.dr.get('https://i.cnblogs.com/EditPosts.aspx?opt=1')
        self.dr.find_element_by_id('Editor_Edit_txbTitle').send_keys(self.title)
        self.set_content(self.content)
        self.dr.find_element_by_id('Editor_Edit_lkbPost').click()

    def tearDown(self):
        sleep(5)
        self.dr.quit()

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

实现效果如下：

QQ首页的今日话题

cnblog转发今日话题

部分内容截图