pandorafms/tests/console/include/bulk_operations.py

# -*- coding: utf-8 -*-
from selenium import selenium
from selenium.webdriver.common.by import By
from selenium.webdriver.support.ui import WebDriverWait, Select
from selenium.webdriver.support import expected_conditions as EC
from common_functions_60 import *

import random, time
import string

def delete_agents_in_bulk(driver,agent_names_list):

        click_menu_element(driver,"Agent operations")
        driver.find_element_by_id("option").click()
        Select(driver.find_element_by_id("option")).select_by_visible_text("Delete agents in bulk")

        for agent_name in agent_names_list:
                Select(driver.find_element_by_id("id_agents")).select_by_visible_text(agent_name)

        driver.find_element_by_id("submit-go").click()


def edit_agents_in_bulk(driver,agent_names_list,new_group=None,new_description=None):

	click_menu_element(driver,"Agent operations")
	driver.find_element_by_id("option").click()
	Select(driver.find_element_by_id("option")).select_by_visible_text("Edit agents in bulk")

	for agent_name in agent_names_list:
		Select(driver.find_element_by_id("id_agents")).select_by_visible_text(agent_name)

	time.sleep(3)

	if new_group != None:
		
		Select(driver.find_element_by_id("group")).select_by_visible_text(new_group)

	if new_description != None:
		
		driver.find_element_by_id("text-description").clear()
		driver.find_element_by_id("text-description").send_keys(new_description)

	driver.find_element_by_id("submit-updbutton").click()


def delete_modules_in_bulk(driver,agent_name_list,module_name_list,select_agent_first=None):

	#If select_agent_first is None, select the modules first
	
	#To erase all modules with this names, add "Any" in agent_name_list
	#To erase all modules of the agent put "Any" in module_name_list

	click_menu_element(driver,"Module operations")
	
	driver.find_element_by_id("option").click()
	Select(driver.find_element_by_id("option")).select_by_visible_text("Delete modules in bulk")
	
	if select_agent_first != None:
		
		driver.find_element_by_id("radiobtn0002").click()
		
		for agent_name in agent_name_list:
			Select(driver.find_element_by_id("id_agents")).select_by_visible_text(agent_name)

		time.sleep(3)

		
		for module_name in module_name_list:
			Select(driver.find_element_by_id("module")).select_by_visible_text(module_name)

	else:

		#driver.find_element_by_id("module_type").click()
		Select(driver.find_element_by_id("module_type")).select_by_visible_text("All")

		time.sleep(3)

		for module_name in module_name_list:
			Select(driver.find_element_by_id("module_name")).select_by_visible_text(module_name)
		
		time.sleep(3)
		
		for agent_name in agent_name_list:
			Select(driver.find_element_by_id("agents")).select_by_visible_text(agent_name)


	driver.find_element_by_id("submit-go").click()


def copy_modules_in_bulk(driver,owner_agent_name,module_name_list,destiny_agents_list):
	
	# module_name_list is a list of modules to copy	

	click_menu_element(driver,"Module operations")

	driver.find_element_by_id("option").click()
	Select(driver.find_element_by_id("option")).select_by_visible_text("Copy modules in bulk")
	time.sleep(2)
	
	Select(driver.find_element_by_id("source_id_agent")).select_by_visible_text(owner_agent_name)

	time.sleep(3)

	for module_name in module_name_list:
		Select(driver.find_element_by_id("target_modules")).select_by_visible_text(module_name)

	for agent_name in destiny_agents_list:
		Select(driver.find_element_by_id("destiny_id_agent")).select_by_visible_text(agent_name)

	driver.find_element_by_id("submit-go").click()
	alert = driver.switch_to_alert()
	alert.accept()

def edit_modules_in_bulk(driver,module_name_list,agent_name_list,new_module_group=None,new_min=None,new_max=None,ff_threshold_list=None):

	#ff_threshold_list example: ff_threshold_list = [normal value,warning value,critical value] -> Mode Change each status

	click_menu_element(driver,"Module operations")

	driver.find_element_by_id("option").click()
	Select(driver.find_element_by_id("option")).select_by_visible_text("Edit modules in bulk")

	Select(driver.find_element_by_id("module_type")).select_by_visible_text("All")

	for module_name in module_name_list:
		Select(driver.find_element_by_id("module_name")).select_by_visible_text(module_name)

	time.sleep(3)

	for agent_name in agent_name_list:
		Select(driver.find_element_by_id("agents")).select_by_visible_text(agent_name)

	time.sleep(3)

	if new_module_group != None:
		Select(driver.find_element_by_id("id_module_group")).select_by_visible_text(new_module_group)

	if new_min != None:
		driver.find_element_by_id("text-min").clear()
		driver.find_element_by_id("text-min").send_keys(new_min)

	if new_max != None:
		driver.find_element_by_id("text-max").clear()
		driver.find_element_by_id("text-max").send_keys(new_max)

	if ff_threshold_list != None:

		Select(driver.find_element_by_id("each_ff")).select_by_visible_text("Change each status")

		time.sleep(3)

		driver.find_element_by_id("text-min_ff_event_normal").clear()
		driver.find_element_by_id("text-min_ff_event_normal").send_keys(ff_threshold_list[0])

		driver.find_element_by_id("text-min_ff_event_warning").clear()
		driver.find_element_by_id("text-min_ff_event_warning").send_keys(ff_threshold_list[1])

		driver.find_element_by_id("text-min_ff_event_critical").clear()
		driver.find_element_by_id("text-min_ff_event_critical").send_keys(ff_threshold_list[2])

	driver.find_element_by_id("submit-go").click()