未定义的变量“驱动程序” /不创建新的Chrome实例

时间:2019-03-19 19:04:11

标签: python selenium driver

我正试图在我的/main.py文件中调用一个click函数。

/main.py

"""Start Point"""

from data.find_pending_records import FindPendingRecords
from vital.vital_entry import VitalEntry
import sys
from selenium import webdriver
from selenium.webdriver.common.by import By
from selenium.webdriver.common.keys import Keys
from selenium.webdriver.common.action_chains import ActionChains
import pandas as pd

if __name__ == "__main__":
    try: 
        # for PENDING_RECORDS in FindPendingRecords().get_excel_data(): begin to loop through entire directory

            PENDING_RECORDS = FindPendingRecords().get_excel_data()
            # Do operations on PENDING_RECORDS

            # Reads excel to map data from excel to vital
            MAP_DATA = FindPendingRecords().get_mapping_data()

            # Configures Driver
            VITAL_ENTRY = VitalEntry()

            # Start chrome and navigate to vital website
            VITAL_ENTRY.instantiate_chrome()

            # Begin processing Records
            VITAL_ENTRY.process_records(PENDING_RECORDS, MAP_DATA)

            # Save Record
            VITAL_ENTRY.save_contact(driver)

            print (PENDING_RECORDS)
            print("All done")

    except Exception as exc:
        # print(exc)
        raise

/vital_entry.py

class VitalEntry:
    """Vital Entry"""

    def save_contact (self, driver):
        driver.implicitly_wait(15)
        driver.find_element_by_css_selector("#publishButton").click() 

我在提示中不断收到此错误:

Traceback (most recent call last):
  File "main.py", line 32, in <module>
    VITAL_ENTRY.save_contact(driver)
NameError: name 'driver' is not defined

我不想创建新的chrome会话或窗口...我也尝试将其链接到上面的VITAL_ENTRY.process_records(PENDING_RECORDS, MAP_DATA)上。如您所见,我已经在导入驱动程序了。并且我在上面的调用中使用了它-我不想创建一个新的浏览器实例。

以下是.instantiate_chrome()

def instantiate_chrome(self):
    """Create Chrome webdriver instance."""
    self.options.headless = config.HEADLESS

    if not self.options.headless:
        self.options.add_argument("--start-maximized")

    self.options.add_argument('--disable-infobars')
    self.options.add_argument('--disable-gpu')

    self.driver = webdriver.Chrome(options=self.options)
    self.driver.set_page_load_timeout(30)
    self.driver.implicitly_wait(15)
    self.driver.get(config.VITAL_URL)

1 个答案:

答案 0 :(得分:1)

因此,您创建浏览器会话,然后再将其传递出该函数。如果您想在其他地方使用它,则instantiate_chrome()代码将需要return driver,那么您将需要按照我之前的评论中所述进行分配

driver= VITAL_ENTRY.instantiate_chrome()