importcaslibs.py

#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# importcaslibs.py
# April 2021
#
# Pass in a directory and this tool will import all the json files in the directory. It depends on the admin CLI
# The json files should be standard caslib definitions
#
# File format
#{
#"attributes": { "active": false, "personal": false, "subDirs": false},
#"description": "",
#"name": "Sales2",
# "path": "/tmp/sales",
# "scope": "global",
# "server": "cas-shared-default",
# "type": "PATH"
#}
#
#
# Change History
#
# 14APR2023 - Added 'superuser' argument
# 01JUN2034 - Include 'superuser' with authorization apply
#
# Copyright © 2019, SAS Institute Inc., Cary, NC, USA.  All Rights Reserved.
#
#  Licensed under the Apache License, Version 2.0 (the License); you may not use this file except in compliance with the License.
# You may obtain a copy of the License at http://www.apache.org/licenses/LICENSE-2.0
#
#  Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS
#  OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License.
#
#
# Import Python modules
import argparse, sys, subprocess, os, json
from sharedfunctions import callrestapi, getapplicationproperties, file_accessible, getclicommand


# get input parameters
parser = argparse.ArgumentParser(description="Import JSON files that define path-based CASLIBS from directory. All json files in directory will be imported.")
parser.add_argument("-d","--directory", help="Directory that contains JSON caslib definition files to import",required='True')
parser.add_argument("-q","--quiet", help="Suppress the are you sure prompt.", action='store_true')
parser.add_argument("-su","--superuser", help="Runs the CASLIB create process with superuser permissions.", action='store_true')
args= parser.parse_args()
basedir=args.directory
quietmode=args.quiet
su=args.superuser


# get cli location from properties, check that cli is there if not ERROR and stop
clicommand=getclicommand()

# get python version
version=int(str(sys.version_info[0]))

# if the quiet mode flag is not passed then prompt to continue
if not quietmode:

      if version  > 2:
            areyousure=input("WARNING: Are you sure? (Y)")
      else:
            areyousure=raw_input("WARNING:   Are you sure? (Y)")
else:
      areyousure="Y"

tryimport=0

if areyousure.upper() =='Y':

      # check that directory exists
      if os.path.isdir(basedir):

            # loop files in the directory
            for filename in os.listdir( basedir ):

                  fullfile=os.path.join(basedir,filename)
                  
                  # only process json files
                  if filename.lower().endswith('.json'):

                        #create the caslib
                        if '_authorization_' not in filename:

                              # get some caslib attributes for the authorization import
                              with open(fullfile) as json_file:
                                    data = json.load(json_file)
                              
                              caslibname=data['name']
                              casserver=data['server']
                              # creates then runs the caslib creation command, with superuser perms where selected
                              if su:
                                    command=clicommand+' cas caslibs create path --source-file "'+fullfile+'" --su'
                              else:
                                    command=clicommand+' cas caslibs create path --source-file "'+fullfile+'"'
                              print("NOTE: Viya Caslib import attempted from json file "+filename+" in  directory "+basedir  )
                              print(command)
                              subprocess.call(command, shell=True)
                              tryimport=tryimport+1

                              # apply the authorization if authorization file exists
                              authfile=os.path.join(basedir,caslibname+'_authorization_.json')
                              access_file=file_accessible(authfile,'r')

                              if access_file==True:
                                    
                                    if su:
                                          command=clicommand+' cas caslibs replace-controls --server '+casserver+' --name "'+ caslibname+'" --force --su --source-file "'+authfile+'"'
                                    else:
                                          command=clicommand+' cas caslibs replace-controls --server '+casserver+' --name "'+ caslibname+'" --force --source-file "'+authfile+'"'

                                    print("NOTE: Viya Caslib authorization import attempted from json file "+filename+" in  directory "+basedir  )
                                    print(command)
                                    subprocess.call(command, shell=True)           

            if not tryimport: print("NOTE: No caslib files available for import.")

      else: print("ERROR: Directory does not exist.")
else:
       print("NOTE: Operation cancelled.")