def getDriver():
"""
返回selenium的driver
:return:
"""
config = ReadConfig()
path = config.find_path("config.ini")
config.__read__(path)
browser_path = config.get_browser("path")
driver_path = config.get_driver("path")
# self.driver = webdriver.Firefox()
# self.driver.set_window_position(0, 0)
# self.driver.set_window_size(1400, 900)
# self.driver.maximize_window() # 让窗口最大化
# 使用以下三行代码可以不弹出界面，实现无界面爬取
options = Options()
options.add_argument('--headless')
options.add_argument('--disable-gpu')
options.binary_location = browser_path
# chrome浏览器配置，添加options参数， executable_path 可选，配置了环境变量后可省略，不然传该驱动的绝对路径
driver = webdriver.Chrome(executable_path=driver_path, options=options)
# 火狐浏览器配置
# driver = webdriver.Firefox(executable_path='geckodriver', options=options)
return driver
def get_stock_data_path():
"""
:return: 保存股票数据的根路径
"""
config = ReadConfig()
path = config.find_path("config.ini")
config.__read__(path)
stock_path = config.get_stock("path")
if not os.path.exists(stock_path):
os.makedirs(stock_path)
return stock_path
