(05)selenium封装模块的调用

若有不对的地方，请大神们指出~

python的内部模块调用

1.需要用到的文件们都在同一个目录下

目录格式是这个样子的

pub.py的代码：

#!/usr/bin/env python
# -*- coding: utf-8 -*-
def add(a, b):
    return a+b

count.py的代码：

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

from pub import add #从模块pub 导入 add这个方法

print(add(4,5))

2.需要用到的文件们不在同一个目录下：

目录格式是这个样子的

model文件夹下pub.py的代码：（嗯，跟之前的是一模一样）

#!/usr/bin/env python
# -*- coding: utf-8 -*-
def add(a, b):
    return a+b

count.py的代码：

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

from  model.pub import add  #这里就有变化啦，model.pub
print(add(4,5))

3.稍微洋气一点点封装~~~

目录格式是这个样子的

show u the code：

mymodule作为package包需要有__init__.py的文件，可以为空文件；module.py里面可以存放需要的模块方法。

module.py

from time import sleep

class DiaoYong(object):
    """这是某些方法的调用"""

    def __init__(self, driver):
        self.driver = driver #这里不能再文件顶部导入webdriver，不然之后会有冲突
        # self.driver = None

    def swipeDown(self,site):
         """模拟H5端滑倒的代码"""
        # sleep(2)
        # button = self.driver.find_element_by_xpath(site)
        # sleep(3)
        # """从button元素像下滑动200元素"""
        # Action = TouchActions(self.driver)
        # Action.scroll_from_element(button, 0, -200).perform()
        # sleep(3)
        """通过找到百度来调试是否成功调用了这个方法"""
        sleep(2)
        self.driver.find_element_by_xpath(site).click()
        print("swipeDown调用成功的啦~~~")
        

02unittest.py

from selenium import webdriver
import unittest
from time import sleep

from mymodule.module import DiaoYong

class unitTest01(unittest.TestCase):  #unitTest01继承unittest.TestCase

    def setUp(self):
        print ("--------start---------")
        self.driver = webdriver.Chrome()
        # sd = diaoyong(driver)
        self.driver.implicitly_wait(30)
        self.base_url = "http://www.baidu.com"
        # 脚本运行时，错误的信息将被打印到这个列表中
        self.verificationErrors = []
        # 是否继续接受下一个警告
        self.accept_next_alert = True
        # diaoyong=module.diaoyong(self)

    def tearDown(self):
        self.driver.quit()
        print ("-------end--------")
        # pass

    def test_Button(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_id("kw").send_keys("selenium webdriver")
        driver.find_element_by_id("su").click()
        sd = DiaoYong(driver)  #调用前将对象实例化；这个地方传入的是driver，跟前面modle里面diaoyong对应
        sd.swipeDown('//*[@id="1"]/h3/a/em[2]')


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

如果你觉得实例化比较麻烦的话，可以直接在代码里面def 方法

from selenium import webdriver
import unittest
from time import sleep
from selenium.webdriver.common.touch_actions import TouchActions


class unitTest01(unittest.TestCase):

    def setUp(self):
        print ("--------start---------")
        self.driver = webdriver.Chrome()
        self.driver.implicitly_wait(30)
        self.base_url = "http://www.baidu.com"
        # 脚本运行时，错误的信息将被打印到这个列表中
        self.verificationErrors = []
        # 是否继续接受下一个警告
        self.accept_next_alert = True
        

    def tearDown(self):
        self.driver.quit()
        print ("-------end--------")
        # pass

   def swipeDown(self, site):
       driver = self.driver
       sleep(2)
       button = driver.find_element_by_xpath(site)
       sleep(3)
       """从button元素像下滑动200元素"""
       Action = TouchActions(driver)
       Action.scroll_from_element(button, 0, -200).perform()
       sleep(3)
       print("success")
 

    def test_Button(self):
        driver = self.driver
        driver.get(self.base_url + "/")
        driver.find_element_by_id("kw").send_keys("selenium webdriver")
        driver.find_element_by_id("su").click()
        self.swipeDown('//*[@id="kw"]')     #class内部调用
        

Tips：

想要Pycharm正确识别模块的话，需要右键点击该文件夹->选择"Sources Root"，这样就可以正确的导入模块啦~

image.png