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.
76 lines
2.2 KiB
76 lines
2.2 KiB
#!/usr/bin/python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
# importfoldertree.py
|
|
# March 20187
|
|
#
|
|
# Pass in a directory and this tool will import the complete viya folder
|
|
#
|
|
# 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, uuid, string, time, os, json
|
|
|
|
from sharedfunctions import callrestapi
|
|
|
|
# get python version
|
|
version=int(str(sys.version_info[0]))
|
|
|
|
# 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 for Export",required='True')
|
|
args= parser.parse_args()
|
|
basedir=args.directory
|
|
|
|
|
|
# retrieve root folders
|
|
#reqtype='get'
|
|
#reqval='/folders/rootFolders'
|
|
#resultdata=callrestapi(reqval,reqtype)
|
|
|
|
|
|
# loop files in the directory
|
|
|
|
for filename in os.listdir( basedir ):
|
|
|
|
# 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)
|
|
|
|
|