1. 程式人生 > >簡單的混合驅動

簡單的混合驅動

cti ive selenium med 數據文件 簡單的 encoding exec edr

數據文件:

testdata.txt

visit||http://www.bing.com
visit||${e:\urls.txt}

urls.txt
http://www.sohu.com
http://www.sogou.com
http://www.baidu.com

#encoding=utf-8
from selenium import webdriver
import re

driver = webdriver.Chrome(executable_path="d:\\chromedriver")

def visit(url):
    global driver
    driver.get(url)

def main(filepath):
    with open(filepath) as fp:
        for line in fp:
            if line.strip():
                if re.search(r"\${(.*?)}",line):
                    action = line.split("||")[0]
                    data_file = re.search(r"\${(.*?)}",line).group(1)
                    with open(data_file)  as file_obj:
                        for url in  file_obj:
                            command = "%s(‘%s‘)" %(action,url.strip())
                            eval(command)

                else:
                    action = line.split("||")[0]
                    url = line.split("||")[1].strip()
                    command = "%s(‘%s‘)" %(action,url)
                    eval(command)

if __name__ == "__main__":
    main("E:\\python\\自動化\\testdata.txt")
    driver.quit()

簡單的混合驅動