def get_stack(stack_id: str, region: Optional[str]=None) -> dict: """ GET /stacks/{id} """ sentry_client.capture_breadcrumb(data={ 'stack_id': stack_id, 'region': region, }) stack_name, stack_version = stack_id.rsplit('-', 1) stack_dict = Stack.get(stack_name, stack_version, region=region or config.region) metrics.count('get_stack') return stack_dict, 200, _make_headers()
def all_stacks(references: str = None, region: str = None) -> dict: """ GET /stacks/ """ sentry_client.capture_breadcrumb(data={ 'references': references, 'region': region }) if not references: references = [] stacks = Stack.list(*references, region=region or config.region) stacks.sort(key=lambda stack: stack.creation_time) metrics.count('get_all_stacks') return stacks, 200, _make_headers()
def all_stacks(references: str=None, region: str=None) -> dict: """ GET /stacks/ """ sentry_client.capture_breadcrumb(data={ 'references': references, 'region': region }) if not references: references = [] stacks = Stack.list(*references, region=region or config.region) stacks.sort(key=lambda stack: stack.creation_time) metrics.count('get_all_stacks') return stacks, 200, _make_headers()
def exception_to_connexion_problem(func, *args, **kwargs): try: return func(*args, **kwargs) except ObjectNotFound as exception: problem = connexion.problem(404, 'Not Found', "Stack not found: {}".format(exception.uid), headers=_make_headers()) metrics.count('generic.{}.stack_not_found'.format(func.__name__)) return problem except ExecutionError as error: sentry_client.captureException() metrics.count('generic.{}.senza_cmd_error'.format(func.__name__)) return connexion.problem(500, title='Execution Error', detail=error.output, headers=_make_headers()) except Exception: sentry_client.captureException() raise
def create_stack(new_stack: dict) -> dict: """ POST /stacks/ :param new_stack: New stack """ keep_stacks = new_stack['keep_stacks'] # type: int new_traffic = new_stack['new_traffic'] # type: int stack_version = new_stack['stack_version'] # type: str senza_yaml = new_stack['senza_yaml'] # type: str parameters = new_stack.get('parameters', []) disable_rollback = new_stack.get('disable_rollback', False) region = new_stack.get('region', config.region) # type: Optional[str] dry_run = new_stack.get('dry_run', False) tags = new_stack.get('tags', []) running_time = MeasureRunningTime('create_stack.success') sentry_client.capture_breadcrumb( data={ 'keep_stacks': keep_stacks, 'new_traffic': new_traffic, 'stack_version': stack_version, 'disable_rollback': disable_rollback, }) try: senza_definition = yaml.load(senza_yaml) except yaml.YAMLError as exception: metrics.count('create_stack.invalid_senza_file') return connexion.problem(400, 'Invalid senza yaml', exception.message, headers=_make_headers()) try: stack_name = senza_definition['SenzaInfo']['StackName'] except KeyError as exception: logger.error("Couldn't get stack name from definition.", extra={'senza_yaml': repr(senza_yaml)}) missing_property = str(exception) problem = connexion.problem( 400, 'Invalid senza yaml', "Missing property in senza yaml: {}".format(missing_property), headers=_make_headers()) metrics.count('create_stack.missing_stack_name_prop') return problem # Create the Stack logger.info("Creating stack %s...", stack_name) senza = Senza(region) tags = [ 'LizzyKeepStacks={}'.format(keep_stacks), 'LizzyTargetTraffic={}'.format(new_traffic), *tags ] output = senza.create(senza_yaml, stack_version, parameters, disable_rollback, dry_run, tags) logger.info("Stack created.", extra={ 'stack_name': stack_name, 'stack_version': stack_version, 'parameters': parameters }) stack_dict = (Stack.get(stack_name, stack_version, region=region) if not dry_run else { 'stack_name': stack_name, 'creation_time': '', 'description': '', 'status': 'DRY-RUN', 'version': stack_version }) running_time.finish() metrics.count('create_stack.region.{}'.format(region)) return stack_dict, 201, _make_headers(output=output)
def create_stack(new_stack: dict) -> dict: """ POST /stacks/ :param new_stack: New stack """ keep_stacks = new_stack['keep_stacks'] # type: int new_traffic = new_stack['new_traffic'] # type: int stack_version = new_stack['stack_version'] # type: str senza_yaml = new_stack['senza_yaml'] # type: str parameters = new_stack.get('parameters', []) disable_rollback = new_stack.get('disable_rollback', False) region = new_stack.get('region', config.region) # type: Optional[str] dry_run = new_stack.get('dry_run', False) tags = new_stack.get('tags', []) running_time = MeasureRunningTime('create_stack.success') sentry_client.capture_breadcrumb(data={ 'keep_stacks': keep_stacks, 'new_traffic': new_traffic, 'stack_version': stack_version, 'disable_rollback': disable_rollback, }) try: senza_definition = yaml.load(senza_yaml) except yaml.YAMLError as exception: metrics.count('create_stack.invalid_senza_file') return connexion.problem(400, 'Invalid senza yaml', exception.message, headers=_make_headers()) try: stack_name = senza_definition['SenzaInfo']['StackName'] except KeyError as exception: logger.error("Couldn't get stack name from definition.", extra={'senza_yaml': repr(senza_yaml)}) missing_property = str(exception) problem = connexion.problem(400, 'Invalid senza yaml', "Missing property in senza yaml: {}".format( missing_property), headers=_make_headers()) metrics.count('create_stack.missing_stack_name_prop') return problem # Create the Stack logger.info("Creating stack %s...", stack_name) senza = Senza(region) tags = ['LizzyKeepStacks={}'.format(keep_stacks), 'LizzyTargetTraffic={}'.format(new_traffic), *tags] output = senza.create(senza_yaml, stack_version, parameters, disable_rollback, dry_run, tags) logger.info("Stack created.", extra={'stack_name': stack_name, 'stack_version': stack_version, 'parameters': parameters}) stack_dict = (Stack.get(stack_name, stack_version, region=region) if not dry_run else {'stack_name': stack_name, 'creation_time': '', 'description': '', 'status': 'DRY-RUN', 'version': stack_version}) running_time.finish() metrics.count('create_stack.region.{}'.format(region)) return stack_dict, 201, _make_headers(output=output)