Python-Automation/Marathon.py at main · mozbakanlgi/Python-Automation · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
from selenium import webdriver
from selenium.webdriver.support.ui import Select
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.common.keys import Keys
from selenium.common.exceptions import NoAlertPresentException
from selenium.common.exceptions import NoSuchElementException
from Confidential import odhusername,odhpassword
import selenium.webdriver.chrome.service
import win32com.client
import sys
import time
class Bot():
    def __init__(self):
        self.driver=webdriver.Chrome()

    def get_input(self, prompt):
        while True:
            try:
                value = input(prompt)
                if not value:
                    raise ValueError('Jobname is not defined')
            except ValueError as e:
                print("Sorry, I didn't understand that.", e)
                continue

            if value.isdigit():
                print("Sorry, job name must be a string.")
                continue
            else:
                break
        return value

    def login(self):
        # Switch the control to the Alert window
        print(f"{'#' * 50} Python Script To Access Marathon {'#' * 50}")
        # odhusername = input("Enter the user name")
        # odhpassword = input("Enter the password")
        self.driver.maximize_window()
        url = "http://connectivity.odh.lgi.io/marathon/ui/#/apps"
        self.driver.get(url)
        try:
            shell = win32com.client.Dispatch("WScript.Shell")
            shell.Sendkeys("odhuser")
            time.sleep(2)
            shell.Sendkeys("{TAB}")
            time.sleep(2)
            shell.Sendkeys("password")
            time.sleep(2)
            shell.Sendkeys("{ENTER}")
            time.sleep(2)
        except NoAlertPresentException as e:
            print("no alert",e)

        try:
            jobName = self.get_input("Please enter job name: ")

            self.driver.find_element(By.XPATH,'//*[@id="marathon"]/div/nav/div/div[2]/div[1]/input').send_keys(jobName)
            WebDriverWait(self.driver, 10).until(
                EC.visibility_of_element_located((By.XPATH, '//*[@id="marathon"]/div/nav/div/div[2]/div[1]/input'))
            ).send_keys(Keys.ENTER)

            try:
                self.driver.find_element(By.XPATH,'//*[@id="marathon"]/div/div/div/div/main/div[2]/table/tbody/tr[4]').click()
                time.sleep(2.0)
                jobStatus = self.driver.find_element(By.XPATH,'//*[@id="marathon"]/div/div/div/div[1]/span/span[1]').text
                if (jobStatus == 'Suspended'):
                    print("The job "+jobName+" is already in suspended state")
                    pass
                else:
                    # suspend the marathon
                    print(f"{'#' * 50} Suspending The Marathon Job {jobName}  {'#' * 50}")
                    self.driver.find_element(By.XPATH,'//*[@id="marathon"]/div/div/div/div[1]/div[2]/div/button').click()
                    time.sleep(1.0)
                    self.driver.find_element(By.XPATH,'//*[@id="marathon"]/div/div/div/div[1]/div[2]/div/ul/li[1]/a').click()
                    print("The job "+ jobName +" is Suspended")
                # Switch the control to the Alert window
                try:
                    time.sleep(2.0)
                    shell.Sendkeys("{ENTER}")
                    # WebDriverWait(self.driver, 3).until(EC.alert_is_present()),
                    #                                     'Timed out waiting for PA creation ' +
                    #                                     'confirmation popup to appear.')
                    #
                    # alert = self.driver.switch_to.alert
                    # alert.accept()
                    # print("alert accepted")
                except:
                    print("no alert")
                #get the hdfs path
                print(f"{'#' * 50} Fetching the HDFS Path for {jobName}...  {'#' * 50}")
                time.sleep(10.0)
                self.driver.find_element(By.XPATH,'//*[@id="marathon"]/div/div/div/div[2]/ul/li[2]/a').click()
                time.sleep(1.0)
                command = self.driver.find_element(By.XPATH,'//*[@id="marathon"]/div/div/div/div[2]/div/div/div[1]/dl/dd[2]/span[1]').text
                hdfsPath=command.rsplit(' ', 1)[1]
                print('HDFS PATH:',hdfsPath)
            except NoSuchElementException as e:
                print("Element not found:", e)
        except ValueError as e:
            print(e)
        except:
            print("Unexpected error:", sys.exc_info()[0])
            raise

        #eosstb-eosstb-streamparser-ch
bot = Bot()
bot.login()
sys.exit()