98552376f3
The cfg API is now available via the oslo-config library, so switch to it and remove the copied-and-pasted version. Add the 2013.1b3 tarball to tools/pip-requires - this will be changed to 'oslo-config>=2013.1' when oslo-config is published to pypi. This will happen in time for grizzly final. Remove the 'deps = pep8==1.3.3' from tox.ini as it means all the other deps get installed with easy_install which can't install oslo-config from the URL. Retain dummy cfg.py file until keystoneclient middleware has been updated (I18c450174277c8e2d15ed93879da6cd92074c27a). Change-Id: I4815aeb8a9341a31a250e920157f15ee15cfc5bc
98 lines
3.1 KiB
Python
98 lines
3.1 KiB
Python
#!/usr/bin/python
|
|
|
|
import sys
|
|
|
|
import keystoneclient.v2_0.client
|
|
from oslo.config import cfg
|
|
|
|
import glance.context
|
|
import glance.openstack.common.log as logging
|
|
import glance.registry.context
|
|
import glance.db.sqlalchemy.api as db_api
|
|
|
|
LOG = logging.getLogger(__name__)
|
|
LOG.addHandler(logging.StreamHandler())
|
|
LOG.setLevel(logging.DEBUG)
|
|
|
|
|
|
def get_owner_map(ksclient, owner_is_tenant=True):
|
|
if owner_is_tenant:
|
|
entities = ksclient.tenants.list()
|
|
else:
|
|
entities = ksclient.users.list()
|
|
# build mapping of (user or tenant) name to id
|
|
return dict([(entity.name, entity.id) for entity in entities])
|
|
|
|
|
|
def build_image_owner_map(owner_map, db, context):
|
|
image_owner_map = {}
|
|
for image in db.image_get_all(context):
|
|
image_id = image['id']
|
|
owner_name = image['owner']
|
|
|
|
if not owner_name:
|
|
LOG.info('Image %s has no owner. Skipping.' % image_id)
|
|
continue
|
|
|
|
try:
|
|
owner_id = owner_map[owner_name]
|
|
except KeyError:
|
|
msg = 'Image %s owner %s was not found. Skipping.'
|
|
LOG.error(msg % (image_id, owner_name))
|
|
continue
|
|
|
|
image_owner_map[image_id] = owner_id
|
|
|
|
msg = 'Image %s owner %s -> %s' % (image_id, owner_name, owner_id)
|
|
LOG.info(msg)
|
|
|
|
return image_owner_map
|
|
|
|
|
|
def update_image_owners(image_owner_map, db, context):
|
|
for (image_id, image_owner) in image_owner_map.items():
|
|
db.image_update(context, image_id, {'owner': image_owner})
|
|
LOG.info('Image %s successfully updated.' % image_id)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
config = cfg.CONF
|
|
extra_cli_opts = [
|
|
cfg.BoolOpt('dry-run',
|
|
help='Print output but do not make db changes.'),
|
|
cfg.StrOpt('keystone-auth-uri',
|
|
help='Authentication endpoint'),
|
|
cfg.StrOpt('keystone-admin-tenant-name',
|
|
help='Administrative user\'s tenant name'),
|
|
cfg.StrOpt('keystone-admin-user',
|
|
help='Administrative user\'s id'),
|
|
cfg.StrOpt('keystone-admin-password',
|
|
help='Administrative user\'s password',
|
|
secret=True),
|
|
]
|
|
config.register_cli_opts(extra_cli_opts)
|
|
config(project='glance', prog='glance-registry')
|
|
|
|
db_api.configure_db()
|
|
|
|
context = glance.common.context.RequestContext(is_admin=True)
|
|
|
|
auth_uri = config.keystone_auth_uri
|
|
admin_tenant_name = config.keystone_admin_tenant_name
|
|
admin_user = config.keystone_admin_user
|
|
admin_password = config.keystone_admin_password
|
|
|
|
if not (auth_uri and admin_tenant_name and admin_user and admin_password):
|
|
LOG.critical('Missing authentication arguments')
|
|
sys.exit(1)
|
|
|
|
ks = keystoneclient.v2_0.client.Client(username=admin_user,
|
|
password=admin_password,
|
|
tenant_name=admin_tenant_name,
|
|
auth_url=auth_uri)
|
|
|
|
owner_map = get_owner_map(ks, config.owner_is_tenant)
|
|
image_updates = build_image_owner_map(owner_map, db_api, context)
|
|
if not config.dry_run:
|
|
update_image_owners(image_updates, db_api, context)
|