selenium自动化（1）

因为要截图，所以封装了一个截图的类

#!/usr/bin/env python
# -- codin#!/usr/bin/env python
# -- coding: utf-8 --
# @Time : DATE{TIME}
# @Author :cinty
# @File : screen_png.py

import os
import time
import datetime
from datetime import datetime
#G该文件主要用于生成文件夹保存截图文件

#获取当前时间，datetime获取到的时间变量，类型为：datetime，通过strftime转换成时间字符串

class Screen_png():

    def get_screen(self):

        # print(browser)
        time = datetime.now().strftime("%Y-%m-%d %H.%M.%S")
        screen_name = time + '.png'

        path=os.getcwd() #获取当前目录
        #print(path)
        #把截图文件夹放在该路径下
        dirs=os.path.join(path,'screen_png') #把文件夹screen_png加入到当前目录下
        #print(dirs)
        if not os.path.exists(dirs):#如果文件不存在
            os.makedirs(dirs)
        else:
            pass

        # screen_dir=os.path.join(dirs,str(datetime.now().strftime("Y%m%d%")))
        # print(screen_dir)
        screen_dir=dirs+'\\'+screen_name
        print("截图保存成功")
        print(screen_dir)
        return screen_dir

代码实现，基本调用web端，并截图

#import configparser
#import logging
#import urllib
import screen_png
import time
#from PIL import ImageGrab
#import datetime
#from datetime import datetime
#from appium.webdriver.common.touch_action import TouchAction
from selenium import webdriver
import js

screen_dir=screen_png.Screen_png() #实例化截图类


browser=webdriver.Chrome()
#打开web端网页
a=browser.get("http://xxxx.com")
time.sleep(3)
#as_file的参数必须是文件名
# js.window.scroll(100,100)
screen = browser.get_screenshot_as_file(screen_dir.get_screen())

browser.find_element_by_class_name("imgBd").click()

browser.close()

# browser.find_element_by_id()