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.
 
 

142 lines
4.6 KiB

#!/usr/bin/python
# -*- coding: utf-8 -*-
#
# listcontent.py
# february 2018
#
# Pass in a folder path and list content
# Change History
#
# Copyright © 2018, 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
from sharedfunctions import getfolderid, callrestapi, printresult, getfolderid, getidsanduris, getpath, json
from datetime import datetime as dt, timedelta as td
# get python version
version=int(str(sys.version_info[0]))
# get input parameters
parser = argparse.ArgumentParser(description="List folder and its sub-folders and content.")
parser.add_argument("-f","--folderpath", help="Enter the path to the viya folder to start the listing.",required='True')
parser.add_argument("-v","--verbosecsv", help="Verbose CSV(only used with -o=csv) ", action='store_true' )
parser.add_argument("-o","--output", help="Output Style", choices=['csv','json','simple','simplejson'],default='json')
parser.add_argument("--debug", action='store_true', help="Debug")
args = parser.parse_args()
debug=args.debug
path_to_folder=args.folderpath
verbosecsv=args.verbosecsv
output_style=args.output
delimiter = ','
if verbosecsv: cols=cols=["id","pathtoitem","name","type","contentType","description","typeDefName","documentType","contentDisposition","fileStatus","searchable","size","createdBy","creationTimeStamp","modifiedBy","modifiedTimeStamp","expirationTimeStamp","encoding","parentUri"]
else: cols=["id","pathtoitem","name","contentType","createdBy","creationTimeStamp","modifiedBy","modifiedTimeStamp","uri"]
def getfoldercontent(path_to_folder):
# call getfolderid to get the folder id
targets=getfolderid(path_to_folder)
#if debug: print(targets)
# if the folder is found
if targets[0] is not None:
uri=targets[1]
#get folder content, recursive call returns all children
reqval=uri+"/members?recursive=true&limit=100000"
reqtype='get'
if debug: print(reqval)
folders_result_json=callrestapi(reqval,reqtype)
# add a loop of folder result that adds the parent folder path
# add the path back into the json for printing
total_items=folders_result_json['count']
itemlist=folders_result_json['items']
returned_items=len(itemlist)
for i in range(0,returned_items):
contenttype=itemlist[i]["contentType"]
itemuri=itemlist[i]["uri"]
name=itemlist[i]["name"]
parentFolderUri=itemlist[i]["parentFolderUri"]
path_to_item=getpath(itemuri)
itemlist[i]["pathtoitem"]=path_to_item
itemlist[i]["pathanditemname"]=path_to_item+name
#print(path_to_item,name,contenttype)
newitems = [ ]
# remove folders
for i in range(0,returned_items):
if folders_result_json['items'][i]["contentType"]!="folder":
newitems.append(folders_result_json['items'][i])
folders_result_json["count"]=len(newitems)
folders_result_json["items"]=newitems
return folders_result_json
# root folder, loop sub-folders and print
if path_to_folder=="/":
# get a list of root folders
# get the json result for each one
# append them together for printing
reqval='/folders/rootFolders'
reqtype='get'
rootfolderresult=callrestapi(reqval,reqtype)
rootfolders=rootfolderresult["items"]
z=0
for folder in rootfolders:
foldername="/"+folder["name"]
result_json=getfoldercontent(foldername)
content_count=result_json['count']
# print if there is content in the folder
if content_count:
# printer header only for the first group
if z==0: printresult(result_json,output_style,cols)
else: printresult(result_json,output_style,cols,header=0)
z=z+1
else:
# print folder content
result_json=getfoldercontent(path_to_folder)
printresult(result_json,output_style,cols)