Add optional mandatory policy setting for template generation

grit/format/policy_templates/writers/template_writer.py

 #!/usr/bin/env python
 # Copyright (c) 2012 The Chromium Authors. All rights reserved.
 # Use of this source code is governed by a BSD-style license that can be
 # found in the LICENSE file.
 
 
 class TemplateWriter(object):
   '''Abstract base class for writing policy templates in various formats.
   The methods of this class will be called by PolicyTemplateGenerator.
   '''
@@ skipping 69 matching lines @@
     for supported_on in policy['supported_on']:
       for supported_on_platform in supported_on['platforms']:
         if supported_on_platform in self.platforms:
           return True
     return False
 
   def CanBeRecommended(self, policy):
     '''Checks if the given policy can be recommended.'''
     return policy.get('features', {}).get('can_be_recommended', False)
 
+  def CanBeMandatory(self, policy):
+    '''Checks if the given policy can be mandatory.'''
+    return policy.get('features', {}).get('can_be_mandatory', True)

[Joao da Silva, 2014/09/05 14:55:04]

The default dict for 'features' must contain 'can_be_mandatory: True' for the
default case to work for policies that don't set any features

+
   def IsPolicySupportedOnPlatform(self, policy, platform):
     '''Checks if |policy| is supported on |platform|.
 
     Args:
       policy: The dictionary of the policy.
       platform: The platform to check; one of 'win', 'mac', 'linux' or
         'chrome_os'.
     '''
     is_supported = lambda x: platform in x['platforms']
     return any(filter(is_supported, policy['supported_on']))
@@ skipping 48 matching lines @@
           for child_policy in child_policies:
             # Nesting of groups is currently not supported.
             self.WritePolicy(child_policy)
           self.EndPolicyGroup()
         if child_recommended_policies:
           self.BeginRecommendedPolicyGroup(policy)
           for child_policy in child_recommended_policies:
             self.WriteRecommendedPolicy(child_policy)
           self.EndRecommendedPolicyGroup()
       elif self.IsPolicySupported(policy):
-        self.WritePolicy(policy)
+        if self.CanBeMandatory(policy):
+          self.WritePolicy(policy)
         if self.CanBeRecommended(policy):
           self.WriteRecommendedPolicy(policy)
     self.EndTemplate()
 
     return self.GetTemplateText()
 
   def PreprocessPolicies(self, policy_list):
     '''Preprocesses a list of policies according to a given writer's needs.
     Preprocessing steps include sorting policies and stripping unneeded
     information such as groups (for writers that ignore them).
@@ skipping 120 matching lines @@
     '''
     is_group = policy['type'] == 'group'
     if is_group:
       # Groups are sorted by caption.
       str_key = policy['caption']
     else:
       # Regular policies are sorted by name.
       str_key = policy['name']
     # Groups come before regular policies.
     return (not is_group, str_key)