简单的自动化脚本

代码

""" 注意点：
1.文件读取，特别注意【文件的路径】
2.产生驱动，特别注意【浏览器的类型】
"""


import os,yaml
from selenium import webdriver
from time import sleep


class BaiDuTest(object):
    def __init__(self,randomVar=None):
        if randomVar == None:
            self.data = self.readYaml()
            self.browser = self.webstart()

    # 文件读取
    @staticmethod
    def readYaml():
        filePath = (os.path.dirname(os.path.dirname(__file__))) + "/pythonProject/element.yaml"
        with open(filePath, mode="r", encoding="utf-8") as f1:
            res = yaml.load(f1, Loader=yaml.FullLoader)
        return res

    # 产生驱动
    @staticmethod
    def webstart():
        browser = webdriver.Firefox()
        return browser

    # 具体业务
    def baiDu(self):
        self.browser.get(self.data["URL"])
        sleep(2)
        self.browser.find_element_by_id(self.data["searchID"]).send_keys("Fractures")
        sleep(2)
        self.browser.find_element_by_id(self.data["buttonID"]).click()
        sleep(2)
        self.browser.close()

if __name__ == '__main__':
    test1= BaiDuTest()
    test1.baiDu()

yaml文件

URL: https://www.baidu.com/
searchID: kw
buttonID: su