Mantis Bug Tracker - Forums

Code: Select all

import argparse
import re
import logging
from selenium import webdriver
from selenium.webdriver.support.ui import WebDriverWait
from selenium.webdriver.support import expected_conditions as EC
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import Select
from datetime import datetime, timedelta


logging.basicConfig(level=logging.INFO)

def mantisLogin(baseURL, fsa_user, fsa_pwd):
    options = webdriver.ChromeOptions()
    options.add_argument('--headless')
    options.add_argument('--no-sandbox')
    options.add_argument('--disable-gpu')

    driver = webdriver.Chrome(options=options)
    driver.get(baseURL)
    WebDriverWait(driver, 5).until(EC.url_contains("login"))
    username = driver.find_element("id", "id_username")
    password = driver.find_element("id", "id_password")
    username.clear()
    password.clear()
    username.send_keys(fsa_user)
    password.send_keys(fsa_pwd)
    driver.find_element(By.CLASS_NAME, "submit").click()
    WebDriverWait(driver, 5).until(EC.url_contains("main_page.php"))

    return driver

def main(args):
    credentials_file = "credentials.txt"
    try:
        username = args.username
        password = args.password
    except Exception as e:
        print(f"Failed to read credentials: {e}")
        return

    mantisURL = "http://mantis-dev.*********.com/"

    try:
        driver = mantisLogin(mantisURL, username, password)
    except Exception as e:
        print(f"Failed to login: {e}")
        return
    
    print(str(driver))
    driver.quit()

if __name__ == '__main__':
    parser = argparse.ArgumentParser()
    parser.add_argument("username")
    parser.add_argument("password")
    main(parser.parse_args())