/usr/share/pyshared/psphere/scripting.py is in python-psphere 0.5.2-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 | """
Parse command line options, allow users to append their own options and
read predefined configuration from the users .visdkrc file.
"""
# Copyright 2010 Jonathan Kinred
#
# 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 optparse
class BaseScript(object):
def __init__(self, client):
self.client = client
self.required_opts = []
usage = ('usage: %prog --url https://<host>/sdk --username <username> '
'--password <password>')
self.parser = optparse.OptionParser(usage)
self.parser.add_option('--url', dest='url',
help='the url of the vSphere server')
self.parser.add_option('--username', dest='username',
help='the username to connnect with')
self.parser.add_option('--password', dest='password',
help='the password to connect with')
def add_option(self, opt, dest, help, required):
self.parser.add_option(opt, dest=dest, help=help)
# TODO: Append to usage
# Add to the list of required options which we'll use later
if required:
self.required_opts.append(dest)
def get_options(self):
"""Get the options that have been set.
Called after the user has added all their own options
and is ready to use the variables.
"""
(options, args) = self.parser.parse_args()
# Set values from .visdkrc, but only if they haven't already been set
visdkrc_opts = self.read_visdkrc()
for opt in self.config_vars:
if not getattr(options, opt):
# Try and use value from visdkrc
if visdkrc_opts:
if opt in visdkrc_opts:
setattr(options, opt, visdkrc_opts[opt])
# Ensure all the required options are set
for opt in self.required_opts:
if opt not in dir(options) or getattr(options, opt) == None:
self.parser.error('%s must be set!' % opt)
return options
def read_visdkrc(self):
try:
config = open(self.visdkrc)
except IOError, e:
if e.errno == 2:
# Doesn't exist, ignore it
return None
elif e.errno == 13:
print('ERROR: Permission denied opening %s' % self.visdkrc)
return None
else:
print('ERROR: Could not open %s: %s' % (self.visdkrc, e.strerror))
return None
lines = config.readlines()
config.close()
parsed_opts = {}
for line in lines:
(key, value) = line.split('=')
parsed_opts[key] = value.rstrip('\n')
visdkrc_opts = {}
if('VI_PROTOCOL' in parsed_opts and 'VI_SERVER' in parsed_opts and
'VI_SERVICEPATH' in parsed_opts):
visdkrc_opts['url'] = '%s://%s%s' % (parsed_opts['VI_PROTOCOL'],
parsed_opts['VI_SERVER'],
parsed_opts['VI_SERVICEPATH'])
if 'VI_USERNAME' in parsed_opts:
visdkrc_opts['username'] = parsed_opts['VI_USERNAME']
if 'VI_PASSWORD' in parsed_opts:
visdkrc_opts['password'] = parsed_opts['VI_PASSWORD']
return visdkrc_opts
|