def _get_storage_manager(self, tenant_name):
app = flask_utils.setup_flask_app()
admin = flask_utils.set_admin_current_user(app)
storage_manager = get_storage_manager()
tenant = flask_utils.get_tenant_by_name(tenant_name)
self._set_tenant_in_app(tenant, app, storage_manager, admin)
return storage_manager
def restore_inter_deployment_dependencies(deployment_id,
update_service_composition,
tenant_name):
with get_storage_manager_instance() as sm:
set_tenant_in_app(get_tenant_by_name(tenant_name))
deployment = sm.get(models.Deployment, deployment_id)
blueprint = deployment.blueprint
app_dir = join(FILE_SERVER_BLUEPRINTS_FOLDER,
utils.current_tenant.name, blueprint.id)
app_blueprint = blueprint.main_file_name
logger.info('{0}: BP in {1}/{2}'.format(deployment_id, app_dir,
app_blueprint))
parsed_deployment = rest_utils.get_parsed_deployment(
blueprint, app_dir, app_blueprint)
deployment_plan = rest_utils.get_deployment_plan(
parsed_deployment, deployment.inputs)
logger.info('{0}: Parsed plan'.format(deployment.id))
rest_utils.update_deployment_dependencies_from_plan(
deployment.id, deployment_plan, sm, lambda *_: True)
logger.info('{0}: Updated dependencies from plan'.format(
deployment.id))
if update_service_composition:
create_service_composition_dependencies(deployment_plan,
deployment, sm)
def _import_secret(self, secret, colliding_secrets, override_collisions):
try:
tenant = get_tenant_by_name(secret['tenant_name'])
create_secret(key=secret['key'], secret=secret, tenant=tenant)
except manager_exceptions.ConflictError:
if override_collisions:
existing_secret = self._get_secret_object(secret)
update_imported_secret(existing_secret, secret)
add_to_dict_values(colliding_secrets, secret['tenant_name'],
secret['key'])
def main():
tenant_name = sys.argv[1]
update_service_composition = (sys.argv[2] == 'True')
with get_storage_manager_instance() as sm:
set_tenant_in_app(get_tenant_by_name(tenant_name))
blueprints = sm.list(models.Blueprint)
idd_capable_blueprints = []
for bp in blueprints:
if is_capable_for_idd(bp.plan):
idd_capable_blueprints.append(bp.id)
logger.info(
'IDD capable blueprints: {0}'.format(idd_capable_blueprints))
if not idd_capable_blueprints:
return
deployments = sm.list(models.Deployment,
filters={'blueprint_id': idd_capable_blueprints})
deployments_queue = queue.Queue()
failed_deployments_queue = queue.Queue()
for dep in deployments:
deployments_queue.put((dep.id, dep.tenant.name))
logger.info('IDD capable deployments: {0}'.format(
[dep.id for dep in deployments]))
for i in range(min(NUM_THREADS, deployments_queue.qsize())):
t = threading.Thread(target=create_inter_deployment_dependencies,
args=(deployments_queue, failed_deployments_queue,
update_service_composition, tenant_name))
t.start()
deployments_queue.join()
if not failed_deployments_queue.empty():
deployments = list(failed_deployments_queue.queue)
logger.critical('Failed create the inter deployment '
'dependencies from the following '
'deployments {0}. See exception '
'tracebacks logged above for more '
'details'.format(deployments))
exit(1)
def setup_flask_app(tenant_name):
app = flask_utils.setup_flask_app()
flask_utils.set_admin_current_user(app)
tenant = flask_utils.get_tenant_by_name(tenant_name)
flask_utils.set_tenant_in_app(tenant)
def setup_flask_app(tenant_name):
app = flask_utils.setup_flask_app()
config.instance.load_configuration()
flask_utils.set_admin_current_user(app)
tenant = flask_utils.get_tenant_by_name(tenant_name)
flask_utils.set_tenant_in_app(tenant)
def main(tenant_names, all_tenants, plugin_names, blueprint_ids, mapping_file,
correct):
def update_suggestions(new_suggestion: dict):
for plugin_name, plugin_version in new_suggestion.items():
if plugin_name not in install_suggestions:
install_suggestions[plugin_name] = []
if plugin_version not in install_suggestions[plugin_name]:
install_suggestions[plugin_name].append(plugin_version)
if all_tenants and tenant_names:
print('--all-tenants and --tenant-name options are mutually exclusive')
exit(1)
if not tenant_names:
tenant_names = (DEFAULT_TENANT, )
setup_environment()
set_tenant_in_app(get_tenant_by_name(DEFAULT_TENANT))
sm = get_storage_manager()
# Prepare the resolver
cloudify_section = sm.get(models.ProviderContext, PROVIDER_CONTEXT_ID).\
context.get(CLOUDIFY, {})
resolver_section = cloudify_section.get(IMPORT_RESOLVER_KEY, {})
resolver_section.setdefault(
'implementation', 'manager_rest.'
'resolver_with_catalog_support:ResolverWithCatalogSupport')
resolver = dsl_parser_utils.create_import_resolver(resolver_section)
if all_tenants:
tenants = sm.list(models.Tenant, get_all_results=True)
else:
tenants = [get_tenant_by_name(name) for name in tenant_names]
blueprint_filter = {'tenant': None}
if blueprint_ids:
blueprint_filter['id'] = blueprint_ids
mappings = load_mappings(mapping_file) if correct else {}
stats, install_suggestions, blueprints_processed = {}, {}, 0
for tenant in tenants:
set_tenant_in_app(get_tenant_by_name(tenant.name))
sm = get_storage_manager()
blueprint_filter['tenant'] = tenant
blueprints = sm.list(models.Blueprint,
filters=blueprint_filter,
get_all_results=True)
for blueprint in blueprints:
print('Processing blueprint of {0}: {1} '.format(
tenant.name, blueprint.id))
blueprint_identifier = '{0}::{1}'.format(tenant.name, blueprint.id)
if correct:
try:
result = correct_blueprint(
blueprint, plugin_names,
mappings.get(tenant.name, {}).get(blueprint.id))
except UpdateException as ex:
print(ex)
else:
blueprints_processed += 1
if result not in stats:
stats[result] = [blueprint_identifier]
else:
stats[result].append(blueprint_identifier)
else:
try:
a_mapping, a_statistic, a_suggestion = \
scan_blueprint(resolver, blueprint, plugin_names)
except UpdateException as ex:
print(ex)
else:
blueprints_processed += 1
if a_mapping:
if tenant.name not in mappings:
mappings[tenant.name] = {}
mappings[tenant.name][blueprint.id] = a_mapping
if a_statistic:
stats[blueprint_identifier] = a_statistic
if a_suggestion:
update_suggestions(a_suggestion)
# Wrap it up
if correct:
printout_correction_stats(stats)
else:
with open(mapping_file, 'w') as output_file:
yaml.dump(mappings, output_file, default_flow_style=False)
chmod(mapping_file, 0o644)
print('\nSaved mapping file to the {0}'.format(mapping_file))
printout_scanning_stats(blueprints_processed, mappings, stats)
printout_install_suggestions(install_suggestions)
def setup_flask_app(tenant_name):
app = flask_utils.setup_flask_app()
flask_utils.set_admin_current_user(app)
tenant = flask_utils.get_tenant_by_name(tenant_name)
flask_utils.set_tenant_in_app(tenant)