Permalink
Cannot retrieve contributors at this time
Name already in use
A tag already exists with the provided branch name. Many Git commands accept both tag and branch names, so creating this branch may cause unexpected behavior. Are you sure you want to create this branch?
webctrl/wc_getReport.py
Go to fileThis commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
103 lines (85 sloc)
3.57 KB
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
import suds | |
import sys | |
import numpy | |
import xml | |
import argparse | |
import getpass | |
if len(sys.argv) < 2: | |
print "You haven't specified any arguments. Use -h to get more details on how to use this command." | |
parser = argparse.ArgumentParser() | |
parser.add_argument('--username', '-u', type=str, default=None, help='Username for the login to the WebCTRL server') | |
parser.add_argument('--password', '-p', type=str, default=None, help='Password for the login to the WebCTRL server') | |
parser.add_argument('--node', '-n', type=str, default=None, | |
help='Path to the point or node for which you want to retrieve a report.') | |
parser.add_argument('-url', type=str, default='https://webctrl.rz-berlin.mpg.de', help="URL of the WebCTRL server as e.g. http://google.de") | |
args = parser.parse_args() | |
if args.username is None: | |
print 'No user name specified. Login to WebCTRL needs a user name and password. Check all options for this command via -h' | |
sys.exit(1) | |
else: | |
username = args.username | |
if args.password is None: | |
password = getpass.getpass('No password specified via -p. Please enter your WebCTRL login password: ') | |
else: | |
password = args.password | |
if args.node is None: | |
print 'No path to a node specified. Check all options for this command via -h' | |
sys.exit(1) | |
if args.url is None: | |
print 'No URL given. Specify the URL to the WebCTRL server analogous to http://google.de' | |
sys.exit(1) | |
else: | |
wsdlFile = args.url + '/_common/webservices/Report?wsdl' | |
try: | |
client = suds.client.Client(wsdlFile, username=username, password=password) | |
except AttributeError: | |
print 'Error: Incorrect username and/or password' | |
except xml.sax._exceptions.SAXParseException: | |
print 'Error: Incorrect/Misspelled WSDL file. It should be: http(s)://URL?wsdl' | |
sys.exit(1) | |
except: | |
print("Unexpected error:", sys.exc_info()[0]) | |
sys.exit(1) | |
# Get the report | |
try: | |
report = client.service.runReport(args.node, '~point-list-report', 'csv') | |
except suds.WebFault as fault: | |
print fault | |
sys.exit(1) | |
# Split the report into separate lines and calculate the number of columns | |
lines = report.split("\n") | |
nColumns = len(lines[0].split(",")) | |
# Declare arrays | |
maxColumnWidth = [0 for i in range(nColumns)] | |
emptyColumns = numpy.full(nColumns, True, dtype=bool) | |
# Initialize the maximum column widths by the width of the column headers | |
column = lines[0].replace('"', '').split(",") | |
for i, col in enumerate(column): | |
maxColumnWidth[i] = len(col) | |
# Calculate the maximum column width taking into account all entries of each column and check whether a column is completely empty. | |
# If only one entry of a specific column is not empty the whole column will be marked as not empty. | |
for l, line in enumerate(lines): | |
if l > 0 and len(line) > 0: | |
column = line.replace('"', '').split(",") | |
for i, col in enumerate(column): | |
if len(col) > maxColumnWidth[i]: | |
maxColumnWidth[i] = len(col) | |
if col.strip() != "": | |
emptyColumns[i] = False | |
# Print the report in column format skipping those columns that are marked as completely empty. | |
for l, line in enumerate(lines): | |
if len(line) > 0: | |
column = line.replace('"', '').split(",") | |
c = '|' | |
for i, col in enumerate(column): | |
if not emptyColumns[i]: | |
formatStr = ' {:<'+str(maxColumnWidth[i])+'}' | |
c = c + formatStr.format(col) + ' |' | |
print c | |
if l == 0: | |
c = '|' | |
for i, n in enumerate(maxColumnWidth): | |
if not emptyColumns[i]: | |
c = c + n * '-' + '---' | |
print c | |
sys.exit(0) |