Delete update_dcu_by_nodejs-temp.py
This commit is contained in:
@ -1,290 +0,0 @@
|
|||||||
import time
|
|
||||||
import signal
|
|
||||||
import socketio
|
|
||||||
print(dir(socketio))
|
|
||||||
import openpyxl
|
|
||||||
import eventlet
|
|
||||||
from threading import Lock
|
|
||||||
import os
|
|
||||||
import csv
|
|
||||||
import pexpect
|
|
||||||
import sys
|
|
||||||
import re
|
|
||||||
from PIL import Image
|
|
||||||
import pytesseract
|
|
||||||
from selenium import webdriver
|
|
||||||
from selenium.webdriver.common.by import By
|
|
||||||
from selenium.webdriver.common.keys import Keys
|
|
||||||
from selenium.webdriver.chrome.service import Service
|
|
||||||
from selenium.webdriver.chrome.options import Options
|
|
||||||
from selenium.webdriver.support.ui import WebDriverWait
|
|
||||||
from selenium.webdriver.support import expected_conditions as EC
|
|
||||||
from selenium.common.exceptions import TimeoutException, NoSuchElementException, ElementClickInterceptedException
|
|
||||||
from webdriver_manager.chrome import ChromeDriverManager
|
|
||||||
|
|
||||||
# Initialize Socket.IO server
|
|
||||||
sio = socketio.Server(cors_allowed_origins='*')
|
|
||||||
app = socketio.WSGIApp(sio)
|
|
||||||
|
|
||||||
# Global variables
|
|
||||||
ssh_tunnel = None
|
|
||||||
ssh_lock = Lock()
|
|
||||||
driver = None
|
|
||||||
stop_requested = False
|
|
||||||
processing_log = "/Users/macos/Documents/SINDIGILIVE/CLIENTS/ICON+/Project-HES/Update_DCU/processing_log.txt"
|
|
||||||
|
|
||||||
# Signal handler for terminating processes
|
|
||||||
def handle_stop_signal(signum, frame):
|
|
||||||
global stop_requested
|
|
||||||
stop_requested = True
|
|
||||||
log_processing("Stop signal received, stopping process...", processing_log)
|
|
||||||
stop_ssh_tunnel()
|
|
||||||
|
|
||||||
signal.signal(signal.SIGTERM, handle_stop_signal)
|
|
||||||
|
|
||||||
# Function to log processing details
|
|
||||||
def log_processing(message, log_file):
|
|
||||||
timestamp = time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())
|
|
||||||
with open(log_file, "a") as log:
|
|
||||||
log.write(f"{timestamp} - {message}\n")
|
|
||||||
|
|
||||||
# Function to create an SSH tunnel
|
|
||||||
def create_ssh_tunnel(dcu_ip, ssh_user="callysta_icon", ssh_host="10.232.4.113", ssh_port=21112, ssh_password="Callysta_icon2024!"):
|
|
||||||
global ssh_tunnel
|
|
||||||
ssh_command = f"ssh -L 8888:{dcu_ip}:80 {ssh_user}@{ssh_host} -p {ssh_port}"
|
|
||||||
log_processing(f"Starting SSH tunnel to {dcu_ip}", processing_log)
|
|
||||||
try:
|
|
||||||
child = pexpect.spawn(ssh_command, timeout=60)
|
|
||||||
child.expect(f"{ssh_user}@{ssh_host}'s password: ")
|
|
||||||
child.sendline(ssh_password)
|
|
||||||
ssh_tunnel = child
|
|
||||||
log_processing(f"SSH tunnel to {dcu_ip} established.", processing_log)
|
|
||||||
return True
|
|
||||||
except pexpect.exceptions.TIMEOUT as e:
|
|
||||||
log_processing(f"SSH connection timeout to {dcu_ip}: {str(e)}", processing_log)
|
|
||||||
except pexpect.exceptions.EOF as e:
|
|
||||||
log_processing(f"SSH connection error to {dcu_ip}: {str(e)}", processing_log)
|
|
||||||
return False
|
|
||||||
|
|
||||||
# Function to terminate the SSH tunnel and WebDriver
|
|
||||||
def stop_ssh_tunnel():
|
|
||||||
global ssh_tunnel, driver
|
|
||||||
if ssh_tunnel and ssh_tunnel.isalive():
|
|
||||||
ssh_tunnel.terminate()
|
|
||||||
log_processing("SSH tunnel terminated successfully.", processing_log)
|
|
||||||
ssh_tunnel = None
|
|
||||||
else:
|
|
||||||
log_processing("No active SSH tunnel to terminate.", processing_log)
|
|
||||||
if driver:
|
|
||||||
try:
|
|
||||||
driver.quit()
|
|
||||||
log_processing("Driver process quit successfully.", processing_log)
|
|
||||||
except Exception as e:
|
|
||||||
log_processing(f"Error quitting driver: {str(e)}", processing_log)
|
|
||||||
finally:
|
|
||||||
driver = None
|
|
||||||
|
|
||||||
# Socket.IO events for starting and stopping the tunnel
|
|
||||||
@sio.event
|
|
||||||
def connect(sid, environ):
|
|
||||||
log_processing(f"Client connected: {sid}", processing_log)
|
|
||||||
|
|
||||||
@sio.event
|
|
||||||
def disconnect(sid):
|
|
||||||
log_processing(f"Client disconnected: {sid}", processing_log)
|
|
||||||
stop_ssh_tunnel() # Stop SSH tunnel if the client disconnects
|
|
||||||
|
|
||||||
@sio.event
|
|
||||||
def stop_tunnel(sid):
|
|
||||||
with ssh_lock:
|
|
||||||
if stop_ssh_tunnel():
|
|
||||||
sio.emit("tunnel_status", {"status": "terminated"}, room=sid)
|
|
||||||
else:
|
|
||||||
sio.emit("tunnel_status", {"status": "not_active"}, room=sid)
|
|
||||||
|
|
||||||
# Placeholder functions for Selenium operations
|
|
||||||
def wait_for_page_to_load(driver, timeout=30, processing_log="processing_log.txt"):
|
|
||||||
"""Wait for the page to completely load."""
|
|
||||||
log_processing("Waiting for the page to load", processing_log)
|
|
||||||
try:
|
|
||||||
WebDriverWait(driver, timeout).until(
|
|
||||||
lambda d: d.execute_script("return document.readyState") == "complete"
|
|
||||||
)
|
|
||||||
log_processing("Page fully loaded.", processing_log)
|
|
||||||
except TimeoutException:
|
|
||||||
log_processing("Timeout waiting for page load.", processing_log)
|
|
||||||
raise Exception("Timeout waiting for page load.")
|
|
||||||
|
|
||||||
def click_running_state_tab(driver, processing_log="processing_log.txt"):
|
|
||||||
"""Ensure the 'Running State' tab is clickable and click it."""
|
|
||||||
log_processing("Attempting to click 'Running State' tab", processing_log)
|
|
||||||
try:
|
|
||||||
running_state_tab = WebDriverWait(driver, 10).until(
|
|
||||||
EC.presence_of_element_located((By.ID, "running-state"))
|
|
||||||
)
|
|
||||||
driver.execute_script("arguments[0].scrollIntoView(true);", running_state_tab)
|
|
||||||
WebDriverWait(driver, 10).until(EC.element_to_be_clickable((By.ID, "running-state")))
|
|
||||||
|
|
||||||
if "sun-navbar_item is-active" not in running_state_tab.get_attribute("class"):
|
|
||||||
running_state_tab.click()
|
|
||||||
log_processing("Clicked 'Running State' tab.", processing_log)
|
|
||||||
else:
|
|
||||||
log_processing("'Running State' tab already active.", processing_log)
|
|
||||||
except TimeoutException:
|
|
||||||
log_processing("Timeout waiting for 'Running State' tab.", processing_log)
|
|
||||||
raise Exception("Timeout waiting for 'Running State' tab.")
|
|
||||||
except Exception as e:
|
|
||||||
log_processing(f"Error clicking 'Running State' tab: {e}", processing_log)
|
|
||||||
raise
|
|
||||||
|
|
||||||
def extract_text_and_verify(dcu_ip, expected_ip, processing_log="processing_log.txt"):
|
|
||||||
screenshot_path = f"/Users/macos/Documents/SINDIGILIVE/CLIENTS/ICON+/Project-HES/Update_DCU/evidence/{dcu_ip}.png"
|
|
||||||
status = "Failed"
|
|
||||||
dcu_id = "Not found"
|
|
||||||
|
|
||||||
log_processing(f"Starting text extraction for {dcu_ip}", processing_log)
|
|
||||||
try:
|
|
||||||
image = Image.open(screenshot_path)
|
|
||||||
log_processing(f"Screenshot opened for {dcu_ip}", processing_log)
|
|
||||||
|
|
||||||
extracted_text = pytesseract.image_to_string(image)
|
|
||||||
log_processing(f"Extracted text: {extracted_text}", processing_log)
|
|
||||||
|
|
||||||
dcu_id_match = re.search(r'\b\d{14}\b', extracted_text)
|
|
||||||
if dcu_id_match:
|
|
||||||
dcu_id = dcu_id_match.group(0)
|
|
||||||
log_processing(f"DCU ID found: {dcu_id}", processing_log)
|
|
||||||
else:
|
|
||||||
log_processing("DCU ID not found in text.", processing_log)
|
|
||||||
|
|
||||||
if expected_ip in extracted_text and "Success" in extracted_text:
|
|
||||||
status = "Success"
|
|
||||||
log_processing(f"Expected IP and Success found for {dcu_ip}", processing_log)
|
|
||||||
else:
|
|
||||||
log_processing(f"Expected IP or 'Success' not found in text for {dcu_ip}", processing_log)
|
|
||||||
|
|
||||||
except FileNotFoundError:
|
|
||||||
log_processing(f"Screenshot not found for {dcu_ip} at {screenshot_path}", processing_log)
|
|
||||||
status = "Screenshot not found"
|
|
||||||
|
|
||||||
return dcu_id, status
|
|
||||||
|
|
||||||
def verify_dcu_update(driver, expected_ip, dcu_ip, retries=5, processing_log="processing_log.txt"):
|
|
||||||
save_directory = "/Users/macos/Documents/SINDIGILIVE/CLIENTS/ICON+/Project-HES/Update_DCU/evidence"
|
|
||||||
os.makedirs(save_directory, exist_ok=True)
|
|
||||||
screenshot_path = os.path.join(save_directory, f"{dcu_ip}.png")
|
|
||||||
|
|
||||||
log_processing(f"Starting verification for {dcu_ip}", processing_log)
|
|
||||||
|
|
||||||
for attempt in range(retries):
|
|
||||||
try:
|
|
||||||
log_processing(f"Attempt {attempt + 1} for {dcu_ip}", processing_log)
|
|
||||||
WebDriverWait(driver, 15).until(lambda d: d.execute_script("return document.readyState") == "complete")
|
|
||||||
|
|
||||||
iframe = driver.find_element(By.ID, 'home_content_iframe')
|
|
||||||
driver.switch_to.frame(iframe)
|
|
||||||
|
|
||||||
element = WebDriverWait(driver, 30).until(
|
|
||||||
EC.presence_of_element_located((By.XPATH, '/html/body/div[1]/div/div[2]/div[2]/div[1]/div/div[1]/div[2]'))
|
|
||||||
)
|
|
||||||
|
|
||||||
actual_text = element.text
|
|
||||||
log_processing(f"Text found for {dcu_ip}: {actual_text}", processing_log)
|
|
||||||
|
|
||||||
if actual_text == expected_ip:
|
|
||||||
log_processing(f"Verification successful for {dcu_ip}", processing_log)
|
|
||||||
driver.save_screenshot(screenshot_path)
|
|
||||||
return
|
|
||||||
else:
|
|
||||||
log_processing(f"Expected: {expected_ip}, found: {actual_text} for {dcu_ip}", processing_log)
|
|
||||||
raise Exception("Verification failed: Text does not match expected IP.")
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
log_processing(f"Error during attempt {attempt + 1} for {dcu_ip}: {e}", processing_log)
|
|
||||||
driver.switch_to.default_content()
|
|
||||||
driver.refresh()
|
|
||||||
time.sleep(5)
|
|
||||||
|
|
||||||
raise Exception(f"Verification failed for {dcu_ip} after {retries} attempts.")
|
|
||||||
|
|
||||||
# Function to update DCU IP
|
|
||||||
def update_dcu_ip(updated_log, ip_list, processing_log):
|
|
||||||
global stop_requested, driver
|
|
||||||
log_processing("Starting DCU IP update process", processing_log)
|
|
||||||
for ip in ip_list:
|
|
||||||
if stop_requested:
|
|
||||||
log_processing("Stop requested, terminating process.", processing_log)
|
|
||||||
break
|
|
||||||
|
|
||||||
log_processing(f"Processing DCU IP: {ip}", processing_log)
|
|
||||||
|
|
||||||
if not create_ssh_tunnel(ip):
|
|
||||||
log_processing(f"Failed to create SSH tunnel for {ip}", processing_log)
|
|
||||||
continue
|
|
||||||
|
|
||||||
with open(updated_log, "a", newline="") as log:
|
|
||||||
writer = csv.writer(log)
|
|
||||||
if log.tell() == 0:
|
|
||||||
writer.writerow(["DCU IP", "DCU ID", "Updated IP", "Update Status"])
|
|
||||||
|
|
||||||
chrome_options = webdriver.ChromeOptions()
|
|
||||||
chrome_options.binary_location = "/Applications/Google Chrome.app/Contents/MacOS/Google Chrome"
|
|
||||||
chrome_options.add_argument("--headless")
|
|
||||||
driver_path = "/Users/macos/Documents/SINDIGILIVE/CLIENTS/ICON+/Project-HES/Update_DCU/chromedriver/chromedriver"
|
|
||||||
driver = webdriver.Chrome(service=Service(driver_path), options=chrome_options)
|
|
||||||
|
|
||||||
try:
|
|
||||||
driver.get("http://localhost:8888")
|
|
||||||
WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "username")))
|
|
||||||
driver.find_element(By.ID, "username").send_keys("admin")
|
|
||||||
driver.find_element(By.ID, "password").send_keys("jh1296")
|
|
||||||
driver.find_element(By.ID, "submit_btn").click()
|
|
||||||
|
|
||||||
log_processing("Login to HES successful", processing_log)
|
|
||||||
WebDriverWait(driver, 20).until(EC.element_to_be_clickable((By.ID, "parameter-setting"))).click()
|
|
||||||
|
|
||||||
iframes = driver.find_elements(By.TAG_NAME, "iframe")
|
|
||||||
if iframes:
|
|
||||||
driver.switch_to.frame(iframes[0])
|
|
||||||
|
|
||||||
new_ip = "10.232.107.250:9032"
|
|
||||||
WebDriverWait(driver, 10).until(EC.presence_of_element_located((By.ID, "main_ip")))
|
|
||||||
ip_field = driver.find_element(By.ID, "main_ip")
|
|
||||||
dcu_id = "Not found"
|
|
||||||
|
|
||||||
for _ in range(3):
|
|
||||||
ip_field.clear()
|
|
||||||
current_value = ip_field.get_attribute("value").strip()
|
|
||||||
if current_value == "":
|
|
||||||
break
|
|
||||||
time.sleep(2)
|
|
||||||
else:
|
|
||||||
raise Exception("Failed to clear the IP field!")
|
|
||||||
|
|
||||||
ip_field.send_keys(new_ip)
|
|
||||||
log_processing("Updated IP in Parameter Setting page", processing_log)
|
|
||||||
|
|
||||||
driver.find_element(By.ID, "main_host_submit").click()
|
|
||||||
log_processing("Saved changes", processing_log)
|
|
||||||
|
|
||||||
driver.back()
|
|
||||||
wait_for_page_to_load(driver)
|
|
||||||
click_running_state_tab(driver)
|
|
||||||
verify_dcu_update(driver, expected_ip=new_ip, dcu_ip=ip)
|
|
||||||
|
|
||||||
dcu_id, status = extract_text_and_verify(ip, new_ip)
|
|
||||||
writer.writerow([ip, dcu_id, new_ip, status])
|
|
||||||
|
|
||||||
except Exception as e:
|
|
||||||
log_processing(f"Error processing {ip}: {e}", processing_log)
|
|
||||||
writer.writerow([ip, "Not found", "Not found", "Failed - Error"])
|
|
||||||
finally:
|
|
||||||
stop_ssh_tunnel()
|
|
||||||
|
|
||||||
# Main function to start the server
|
|
||||||
if __name__ == "__main__":
|
|
||||||
updated_log = "/Users/macos/Documents/SINDIGILIVE/CLIENTS/ICON+/Project-HES/Update_DCU/update_log.csv"
|
|
||||||
processing_log = "/Users/macos/Documents/SINDIGILIVE/CLIENTS/ICON+/Project-HES/Update_DCU/processing_log.txt"
|
|
||||||
ip_list = sys.argv[1:]
|
|
||||||
eventlet.wsgi.server(eventlet.listen(('localhost', 5000)), app)
|
|
||||||
update_dcu_ip(updated_log, ip_list, processing_log)
|
|
||||||
Reference in New Issue
Block a user