Files
Backup/_NDGOV_CS/ITD.Cohesity/updateAzPolicy.py
T
Zack Meier 1d304511b8 update
2026-04-15 15:45:50 -05:00

108 lines
4.0 KiB
Python

#!/usr/bin/python
import sys,argparse,json,time
import time
sys.path.insert(0, './classes/')
import cohesityAPI as cohesity
import sharePointAPI as sharePoint
import serviceNowAPI as snow
import automationsAPI as dashboard
# Global variables that will be used across all functions
# Begin Functions
def GetArgs():
parser = argparse.ArgumentParser(add_help=False)
parser.add_argument('--cluster', '-c', type=str, action='store')
parser.add_argument('--verify', '-v', action='store_true')
parser.add_argument('--help', '-h', action='store_true')
return (parser.parse_args())
def PrintHelp():
print("\nBasic Usage:")
print("\npython3 sqlServerRegistration.py -c cluster.fqdn.tld")
print("\t -c FQDN of Cohesity cluster address")
print("\t -v force confirmation on change")
print("\t -h Prints this help message")
args = GetArgs()
# Validate arguments & assign variables if necessary
if args.help:
PrintHelp()
exit(1)
if not args.cluster:
PrintHelp()
exit(1)
# ServiceNow object
ticketSystem = snow.SnowAPI("northdakota.service-now.com")
# Sharepoint Object
cmdb = sharePoint.API()
# Connect to the Cohesity cluster
cluster = cohesity.API(args.cluster)
authToken = cluster.GetAuthToken()
cluster.UpdateHeaders(authToken['accessToken'])
vmWareJobs=cluster.GetFilteredRequest("/public/protectionJobs","?environments=kAzureNative&isActive=True")
for job in vmWareJobs:
preChangeNotes = json.dumps(job)
if 'name' in job:
jobName = job['name']
if '@cares01' in jobName:
continue
print("\nUpdating policy for {0}".format(jobName))
policyData = {"policyId": "7780085755317378:1694019083558:285"
if '@infra01' in jobName:
continue
print("\nUpdating policy for {0}".format(jobName))
policyData = {"policyId": "7780085755317378:1694019083558:286"}
job.update(policyData)
elif '@npd01' in jobName:
continue
print("\nUpdating policy for {0}".format(jobName))
policyData = {"policyId": "7780085755317378:1694019083558:287"}
job.update(policyData)
elif '@prd01' in jobName:
print("\nUpdating policy for {0}".format(jobName))
policyData = {"policyId": "7780085755317378:1694019083558:288"}
job.update(policyData)
else:
continue
if args.verify:
confirm = input("About to update protection job {0}. Are you sure? [yes|no] ".format(jobName))
else:
confirm = "yes"
if confirm.lower() == "yes":
changeID = ticketSystem.CreateStandardChange("NDIT-SPS-Cohesity Data Protection", "NDIT-Cloud Platforms", "svccohesityadm", "ndheck", "mlaverdure", "Computer Systems", "Backup/Restore", "Automated Change - Cohesity job update for {}".format(jobName), "Update protection policy to enable replication.")
changeSysID = changeID['result']['sys_id']['value']
ticketSystem.scheduleStandardChange(changeSysID)
ticketSystem.implementStandardChange(changeSysID)
ticketSystem.addStandardChangeNotes(changeSysID, "Original job details\n\n{}".format(preChangeNotes))
ticketSystem.reviewStandardChange(changeSysID)
updateStats = cluster.UpdateVMProtectionJob(job)
if updateStats.status_code != 200 and updateStats.status_code != 201:
postChangeNotes = "Job update failed\n\n" + updateStats.text
ticketSystem.addStandardChangeNotes(changeSysID, postChangeNotes)
ticketSystem.closeStandardChange(changeSysID, "Unsuccessful", "Change Failed")
else:
postChangeNotes = "Updated the policy\n\n" + updateStats.text
ticketSystem.addStandardChangeNotes(changeSysID, postChangeNotes)
ticketSystem.closeStandardChange(changeSysID, "Successful", "Change Complete")
print("Change {} complete".format(changeID['result']['number']['value']))