selenium简单代码入门

 #!/usr/bin/env python
 #-*- coding:utf-8 -*-

 import os,sys,string
 import time
 import unittest
 from selenium import webdriver
 from selenium.webdriver.common.by import By

 #unittest是python下的单元测试模块
 class WeiboDate(unittest.TestCase):

     #初始化,并使用Firefox浏览器驱动
     def setUp(self):
         self.driver = webdriver.Firefox()

     def test_WeiboSearch(self):
         driver = self.driver
         driver.get("http://data.weibo.com/index/")
         #wait
         driver.implicitly_wait(10)
         driver.find_element(By.XPATH,"//div[@id='pl_index_searchMain']//input").send_keys("apple watch")
         driver.find_element_by_css_selector("a.index-search").click()
         time.sleep(3)
         #截图
         driver.get_screenshot_as_file("/opt/selenium/script/jietu.png")
         #最大化浏览器
         driver.maximize_window()
         driver.forward()
         #cookie
         cookie = driver.get_cookies()
         file=open('cookie','w+').writelines("cookie")

     def tearDown(self):
         self.driver.close()

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