Модуль входа для Selenium Webdriver

Question

Я пытаюсь зайти на сайт и запустить несколько тестов. Есть так много ссылок, на которые мне нужно нажать, и на веб-страницах должны выполняться другие задачи. Я написал скрипт на python, который принимает аргументы командной строки (браузер, имя пользователя и пароль) и входит на сайт. Это отлично работает. Модуль входа необходимо импортировать в другие сценарии, где я могу выполнять тесты с помощью этого модуля входа.

testlogin.py

import sys
import argparse
from selenium import webdriver

parser = argparse.ArgumentParser()
parser.add_argument('browser', default='chrome', help='Types of browser:chrome, firefox, ie')
parser.add_argument('username', help='This is the username')
parser.add_argument('password', help='This is the password')
args = parser.parse_args()

setup_parameters = sys.argv[1:]


class Testsite(object):


      test_url = 'someurl'

# Initialization Method
    def __init__(self):
        self.username = setup_parameters[1]
        self.password = setup_parameters[2]
        if setup_parameters[0] == 'chrome':
            self.browser = webdriver.Chrome()
            print("Running tests on Chrome browser")
            self.browser.implicitly_wait(15)
            self.premier_login()
            self.premier_close()
        elif setup_parameters[0] == 'ie':
            self.browser = webdriver.Ie()
            print("Running tests on Internet Explorer browser")
            self.browser.implicitly_wait(15)
            self.premier_login()
            self.premier_close()
        elif setup_parameters[0] == 'firefox':
            self.browser = webdriver.Firefox()
            print("Running tests on Firefox browser")
            self.browser.implicitly_wait(15)
            self.premier_login()
            self.premier_close()
        elif setup_parameters[0] == 'None':
            print('No browser type specified.... continuing with the default browser')
            self.browser = webdriver.Chrome()

#Method to login to the site

    def site_login(self):
        self.browser.get(self.test_url)
        self.browser.find_element_by_id("Username").send_keys(self.username)
        self.browser.find_element_by_id("Password").send_keys(self.password)
        self.browser.find_element_by_id("btnLogin").click()
        self.browser.maximize_window()
        self.browser.implicitly_wait(10)

# Closing the browser window and terminating the test
    def site_close(self):
        self.browser.quit()


if __name__ == '__main__':

    Testsite().site_login()

Как мне установить драйвер из вышеуказанного модуля, если я хочу импортировать его в другие скрипты.

Я попытался импортировать вышеуказанный модуль в другие скрипты, но не смог установить драйвер.