Add a helper tool that can be used to reset the endpoints & services

tools/endpoint-service-replace.py

@@ -0,0 +1,92 @@
+#!/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())