#!/usr/bin/env python

"""Create a Cohesity NFS View Using python"""

# import pyhesity wrapper module

from pyhesity import *

# command line arguments

import argparse

parser = argparse.ArgumentParser()

parser.add_argument('-v', '--vip', type=str, default='helios.cohesity.com')

parser.add_argument('-u', '--username', type=str, default='helios')

parser.add_argument('-d', '--domain', type=str, default='local')

parser.add_argument('-t', '--tenant', type=str, default=None)

parser.add_argument('-c', '--clustername', type=str, default=None)

parser.add_argument('-mcm', '--mcm', action='store_true')

parser.add_argument('-i', '--useApiKey', action='store_true')

parser.add_argument('-pwd', '--password', type=str, default=None)

parser.add_argument('-np', '--noprompt', action='store_true')

parser.add_argument('-m', '--mfacode', type=str, default=None)

parser.add_argument('-e', '--emailmfacode', action='store_true')

parser.add_argument('-n', '--viewname', type=str, required=True) # name view to create

parser.add_argument('-a', '--aliasname', type=str, required=True) # name of alias to create

parser.add_argument('-p', '--folderpath', type=str, default='/') # folder path of alias

args = parser.parse_args()

vip = args.vip

username = args.username

domain = args.domain

tenant = args.tenant

clustername = args.clustername

mcm = args.mcm

useApiKey = args.useApiKey

password = args.password

noprompt = args.noprompt

mfacode = args.mfacode

emailmfacode = args.emailmfacode

viewName = args.viewname

aliasName = args.aliasname

folderPath = args.folderpath

# authentication =========================================================

# demand clustername if connecting to helios or mcm

if (mcm or vip.lower() == 'helios.cohesity.com') and clustername is None:

print('-c, --clustername is required when connecting to Helios or MCM')

exit(1)

# authenticate

apiauth(vip=vip, username=username, domain=domain, password=password, useApiKey=useApiKey, helios=mcm, prompt=(not noprompt), mfaCode=mfacode, emailMfaCode=emailmfacode, tenantId=tenant)

# exit if not authenticated

if apiconnected() is False:

print('authentication failed')

exit(1)

# if connected to helios or mcm, select access cluster

if mcm or vip.lower() == 'helios.cohesity.com':

heliosCluster(clustername)

if LAST_API_ERROR() != 'OK':

exit(1)

# end authentication =====================================================

view = api('get', 'views/%s' % viewName)

if view is None:

print('view %s not found')

exit()

newAlias = {

"viewName": view['name'],

"viewPath": folderPath,

"aliasName": aliasName,

"sharePermissions": view.get('sharePermissions', []),

"subnetWhitelist": view.get('subnetWhitelist', [])

}

print('Creating view alias %s -> %s%s' % (aliasName, viewName, folderPath))

result = api('post', 'viewAliases', newAlias)