You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
60 lines
2.1 KiB
60 lines
2.1 KiB
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# importfoldertree.py March 2018
|
|
#
|
|
# Pass in a directory and this tool will import all the json files in the directory. It depends on the admin CLI
|
|
#
|
|
# Change History
|
|
#
|
|
#
|
|
# 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
|
|
|
|
# CHANGE THIS VARIABLE IF YOUR CLI IS IN A DIFFERENT LOCATION
|
|
clidir='/opt/sas/viya/home/bin/'
|
|
|
|
# get input parameters
|
|
parser = argparse.ArgumentParser(description="Delete a folder and its sub-folders")
|
|
parser.add_argument("-d","--directory", help="Directory that contains JSON files to import",required='True')
|
|
args= parser.parse_args()
|
|
basedir=args.directory
|
|
|
|
# check that directory exists
|
|
if os.path.isdir(basedir):
|
|
|
|
# loop files in the directory
|
|
for filename in os.listdir( basedir ):
|
|
|
|
if filename.lower().endswith('.json'):
|
|
|
|
# upload the json package
|
|
command=clidir+'sas-admin transfer upload --file '+os.path.join(basedir,filename)+'> /tmp/packageid.json'
|
|
print(command)
|
|
subprocess.call(command, shell=True)
|
|
|
|
#print the json from the upload
|
|
with open('/tmp/packageid.json') as json_file:
|
|
package_data = json.load(json_file)
|
|
|
|
print(json.dumps(package_data,indent=2))
|
|
|
|
# get the packageid and import the package
|
|
packageid=package_data["id"]
|
|
command=clidir+'sas-admin -q transfer import --id '+packageid
|
|
|
|
print(command)
|
|
|
|
subprocess.call(command, shell=True)
|
|
print("NOTE: Viya root folders imported from json files in "+basedir)
|
|
else: print("ERROR: Directory does not exist")
|
|
|
|
|