amhaats1
commented
7 months ago Oh I actually got it figured out, if you want to adjust or add it to your code as an option, just a thought
!/usr/bin/env python
"""refresh protection source"""
import pyhesity wrapper module
from pyhesity import * from time import sleep
command line arguments
import argparse parser = argparse.ArgumentParser() parser.add_argument('-v', '--vip', type=str, required=True) # cluster to connect to parser.add_argument('-u', '--username', type=str, default='helios') # username parser.add_argument('-d', '--domain', type=str, default='local') # domain - defaults to local parser.add_argument('-i', '--useApiKey', action='store_true') # use API key authentication parser.add_argument('-pwd', '--password', type=str, default=None) # optional password parser.add_argument('-n', '--sourcename', type=str, action='append') # optional name of vcenter parser.add_argument('-l', '--sourcefile', type=str, help='Path to a text file containing source names')
args = parser.parse_args()
vip = args.vip username = args.username domain = args.domain password = args.password useApiKey = args.useApiKey sourceNames = args.sourcename
Read source names from file if provided
if args.sourcefile: with open(args.sourcefile, 'r') as file: sourceNames_from_file = file.read().splitlines() if sourceNames: sourceNames.extend(sourceNames_from_file) else: sourceNames = sourceNames_from_file
authenticate
apiauth(vip=vip, username=username, domain=domain, password=password, useApiKey=useApiKey)
if sourceNames is None: print('No sources specified') exit()
sources = api('get', 'protectionSources/registrationInfo?allUnderHierarchy=false') if 'rootNodes' not in sources: print('No sources found') exit()
def getObjectId(sourcename): for source in sources['rootNodes']: if source['rootNode']['name'].lower() == sourcename.lower(): return source['rootNode']['id'] return None
def waitForRefresh(sourcename): authStatus = '' while authStatus != 'Finished': rootFinished = False appsFinished = False sleep(5) rootNodes = api('get', 'protectionSources/registrationInfo?includeApplicationsTreeInfo=false') rootNode = [r for r in rootNodes['rootNodes'] if r['rootNode']['name'].lower() == sourcename.lower()] if rootNode[0]['registrationInfo']['authenticationStatus'] == 'kFinished': rootFinished = True if 'registeredAppsInfo' in rootNode[0]['registrationInfo']: for app in rootNode[0]['registrationInfo']['registeredAppsInfo']: if app['authenticationStatus'] == 'kFinished': appsFinished = True
return rootNode[0]['rootNode']['id']
else:
appsFinished = False
else:
appsFinished = True
if rootFinished is True and appsFinished is True:
authStatus = 'Finished'
return rootNode[0]['rootNode']['id']for sourcename in sourceNames: objectId = getObjectId(sourcename) if objectId is not None: print('refreshing %s...' % sourcename) result = api('post', 'protectionSources/refresh/%s' % objectId) result = waitForRefresh(sourcename) else: print('%s not found' % sourcename)
bseltz-cohesity
python/refreshSource/refreshSource.py
Would you be able to add some code to be able to refresh sources from a list text file?
such as
-l, --sourceList: text file containing source names to be refreshed (one job per line)