creator.py

from selenium import webdriver
from selenium.webdriver.chrome.options import Options as ChromeOptions
from selenium.webdriver.firefox.options import Options as FirefoxOptions
from base import *

import unittest
import configparser


class IssueCreator(unittest.TestCase):

    def __init__(self):
        self.config = configparser.ConfigParser()
        self.config.read("config.ini")

        headless_mode = self.config.getboolean("BROWSER", "HEADLESS_MODE")

        if self.config.get("BROWSER", "USE_BROWSER") == "CHROME":
            chrome_options = ChromeOptions()

            if headless_mode:
                chrome_options.add_argument("--headless")

            self.driver = webdriver.Chrome(options=chrome_options)
        elif self.config.get("BROWSER", "USE_BROWSER") == "FIREFOX":
            firefox_options = FirefoxOptions()

            if headless_mode:
                firefox_options.add_argument("--headless")

            self.driver = webdriver.Firefox(options=firefox_options)
        else:
            raise ValueError("USE_BROWSER should be one of CHROME, FIREFOX")

        self.start_page_url = self.config.get('URL', 'START_PAGE_URL')

    def create_issue(self):
        self.driver.get(self.get_url_from_config())
        login = LoginPage(self.driver)
        login.fill_sign_in()
        ci = IssuesPage(self.driver)
        ci.create_issue()
        # Close browser
        ci.close()

    def get_url_from_config(self):
        self.config.read('config.ini')
        return self.config['URL']['DEFAULT_URL']


if __name__ == "__main__":
    ic = IssueCreator()
    ic.create_issue()