93 lines
3.0 KiB
Python
Executable File
93 lines
3.0 KiB
Python
Executable File
#!/usr/bin/env python
|
|
|
|
from optparse import OptionParser
|
|
|
|
import getpass
|
|
import os
|
|
import sys
|
|
import yaml
|
|
|
|
possible_topdir = os.path.normpath(os.path.join(os.path.abspath(sys.argv[0]),
|
|
os.pardir,
|
|
os.pardir))
|
|
|
|
if os.path.exists(os.path.join(possible_topdir,
|
|
'anvil',
|
|
'__init__.py')):
|
|
sys.path.insert(0, possible_topdir)
|
|
|
|
|
|
from anvil import log as logging
|
|
from anvil import importer
|
|
from anvil import passwords
|
|
from anvil.components.helpers import keystone
|
|
from anvil import utils
|
|
|
|
|
|
def get_token():
|
|
pw_storage = passwords.KeyringProxy(path='/etc/anvil/passwords.cfg')
|
|
lookup_name = "service_token"
|
|
prompt = "Please enter the password for %s: " % ('/etc/anvil/passwords.cfg')
|
|
(exists, token) = pw_storage.read(lookup_name, prompt)
|
|
if not exists:
|
|
pw_storage.save(lookup_name, token)
|
|
return token
|
|
|
|
|
|
def replace_services_endpoints(token, options):
|
|
client = importer.construct_entry_point("keystoneclient.v2_0.client:Client",
|
|
token=token, endpoint=options.keystone_uri)
|
|
current_endpoints = client.endpoints.list()
|
|
current_services = client.services.list()
|
|
|
|
def filter_resource(r):
|
|
raw = dict(r.__dict__) # Can't access the raw attrs, arg...
|
|
raw_cleaned = {}
|
|
for k, v in raw.items():
|
|
if k == 'manager' or k.startswith('_'):
|
|
continue
|
|
raw_cleaned[k] = v
|
|
return raw_cleaned
|
|
|
|
for e in current_endpoints:
|
|
print("Deleting endpoint: ")
|
|
print(utils.prettify_yaml(filter_resource(e)))
|
|
client.endpoints.delete(e.id)
|
|
|
|
for s in current_services:
|
|
print("Deleting service: ")
|
|
print(utils.prettify_yaml(filter_resource(s)))
|
|
client.services.delete(s.id)
|
|
|
|
if options.file:
|
|
with(open(options.file, 'r')) as fh:
|
|
contents = yaml.load(fh)
|
|
set_contents = {
|
|
'services': contents.get('services', []),
|
|
'endpoints': contents.get('endpoints', []),
|
|
}
|
|
print("Regenerating with:")
|
|
print(utils.prettify_yaml(set_contents))
|
|
set_contents['users'] = []
|
|
set_contents['roles'] = []
|
|
set_contents['tenants'] = []
|
|
initer = keystone.Initializer(token, options.keystone_uri)
|
|
initer.initialize(**set_contents)
|
|
|
|
|
|
def main():
|
|
parser = OptionParser()
|
|
parser.add_option("-k", '--keystone', dest='keystone_uri',
|
|
help='keystone endpoint uri to authenticate with', metavar='KEYSTONE')
|
|
parser.add_option("-f", '--file', dest='file',
|
|
help='service and endpoint creation file', metavar='FILE')
|
|
(options, args) = parser.parse_args()
|
|
if not options.keystone_uri or not options.file:
|
|
parser.error("options are missing, please try -h")
|
|
logging.setupLogging(logging.DEBUG)
|
|
replace_services_endpoints(get_token(), options)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|