class Tag(AzureBaseAction): """Adds tags to Azure resources :example: This policy will tag all existing resource groups with a value such as Environment .. code-block:: yaml policies: - name: azure-tag-resourcegroups resource: azure.resourcegroup description: | Tag all existing resource groups with a value such as Environment actions: - type: tag tag: Environment value: Test """ schema = utils.type_schema( 'tag', **{ 'value': Lookup.lookup_type({'type': 'string'}), 'tag': Lookup.lookup_type({'type': 'string'}), 'tags': { 'type': 'object' } }) schema_alias = True def __init__(self, data=None, manager=None, log_dir=None): super(Tag, self).__init__(data, manager, log_dir) def validate(self): if not self.data.get('tags') and not (self.data.get('tag') and self.data.get('value')): raise FilterValidationError( "Must specify either tags or a tag and value") if self.data.get('tags') and self.data.get('tag'): raise FilterValidationError( "Can't specify both tags and tag, choose one") return self def _process_resource(self, resource): new_tags = self._get_tags(resource) TagHelper.add_tags(self, resource, new_tags) def _get_tags(self, resource): return self.data.get('tags') or { Lookup.extract(self.data.get('tag'), resource): Lookup.extract(self.data.get('value'), resource) }
def test_lookup_type(self): number_schema = {'type': 'number'} lookup_default_number = Lookup.lookup_type(number_schema) string_schema = {'type': 'string'} lookup_default_string = Lookup.lookup_type(string_schema) self.assertEqual(number_schema, lookup_default_number['oneOf'][1]) self.assertEqual( number_schema, lookup_default_number['oneOf'][0]['oneOf'][0] ['properties']['default-value']) self.assertEqual(string_schema, lookup_default_string['oneOf'][1]) self.assertEqual( string_schema, lookup_default_string['oneOf'][0]['oneOf'][0] ['properties']['default-value'])
class ResizePlan(AzureBaseAction): """Resize App Service Plans :example: Resize App Service Plan to B1 plan with 2 instance. .. code-block:: yaml policies: - name: azure-resize-plan resource: azure.appserviceplan actions: - type: resize-plan size: B1 count: 2 :example: Resize app service plans with on/off hours and resource tagging .. code-block:: yaml policies: - name: on-hours resource: azure.appserviceplan filters: - type: onhour default_tz: pt onhour: 8 tag: onoffhour_schedule actions: - type: resize-plan size: type: resource key: tags.on_hour_sku default-value: P1 - name: off-hours resource: azure.appserviceplan filters: - type: offhour default_tz: pt offhour: 19 tag: onoffhour_schedule actions: - type: tag tag: on_hour_sku value: type: resource key: sku.name - type: resize-plan size: size: S1 """ schema = { 'type': 'object', 'anyOf': [{ 'required': ['size'] }, { 'required': ['count'] }], 'properties': { 'type': { 'enum': ['resize-plan'] }, 'size': Lookup.lookup_type({ 'type': 'string', 'enum': [ 'F1', 'B1', 'B2', 'B3', 'D1', 'S1', 'S2', 'S3', 'P1', 'P2', 'P3', 'P1V2', 'P2V2', 'P3v2', 'PC2', 'PC3', 'PC4' ] }), 'count': Lookup.lookup_type({'type': 'integer'}) }, 'additionalProperties': False } schema_alias = True def _prepare_processing(self): self.client = self.manager.get_client( ) # type azure.mgmt.web.WebSiteManagementClient def _process_resource(self, resource): model = models.AppServicePlan(location=resource['location']) if resource['kind'] == 'functionapp': self.log.info("Skipping %s, because this App Service Plan " "is for Consumption Azure Functions." % resource['name']) return if resource['kind'] == 'linux': model.reserved = True size = Lookup.extract(self.data.get('size'), resource) # get existing tier model.sku = models.SkuDescription() model.sku.tier = resource['sku']['tier'] model.sku.name = resource['sku']['name'] if 'size' in self.data: model.sku.tier = ResizePlan.get_sku_name(size) model.sku.name = size if 'count' in self.data: model.sku.capacity = Lookup.extract(self.data.get('count'), resource) try: self.client.app_service_plans.update(resource['resourceGroup'], resource['name'], model) except models.DefaultErrorResponseException as e: self.log.error("Failed to resize %s. Inner exception: %s" % (resource['name'], e.inner_exception)) @staticmethod def get_sku_name(tier): tier = tier.upper() if tier == 'F1': return 'FREE' elif tier == 'D1': return 'SHARED' elif tier in ['B1', 'B2', 'B3']: return 'BASIC' elif tier in ['S1', 'S2', 'S3']: return 'STANDARD' elif tier in ['P1', 'P2', 'P3']: return 'PREMIUM' elif tier in ['P1V2', 'P2V2', 'P3V2']: return 'PREMIUMV2' elif tier in ['PC2', 'PC3', 'PC4']: return 'PremiumContainer' return None