Generate ADMX template for Chrome OS policies

components/policy/tools/template_writers/template_formatter.py

 #!/usr/bin/env python
 # Copyright 2017 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.
 
 '''Takes translated policy_template.json files as input, applies template
 writers and emits various template and doc files (admx, html, json etc.).
 '''
 
 import codecs
 import collections
 import optparse
 import os
 import sys
 
 import writer_configuration
 import policy_template_generator
 
 from writers import adm_writer, adml_writer, admx_writer, \
+                    chromeos_admx_writer, chromeos_adml_writer, \
                     google_admx_writer, google_adml_writer, \
                     android_policy_writer, reg_writer, doc_writer, \
                     json_writer, plist_writer, plist_strings_writer
 
 def MacLanguageMap(lang):
   '''Handles slightly different path naming convention for Macs:
   - 'en-US' -> 'en'
   - '-' -> '_'
 
   Args:
@@ skipping 13 matching lines @@
 WriterDesc = collections.namedtuple(
     'WriterDesc', ['type', 'is_per_language', 'encoding', 'language_map'])
 
 
 _WRITER_DESCS = [
   WriterDesc('adm', True, 'utf-16', None),
   WriterDesc('adml', True, 'utf-16', None),
   WriterDesc('admx', False, 'utf-16', None),
   WriterDesc('google_adml', True, 'utf-8', None),
   WriterDesc('google_admx', False, 'utf-8', None),
+  WriterDesc('chromeos_adml', True, 'utf-8', None),
+  WriterDesc('chromeos_admx', False, 'utf-8', None),
   WriterDesc('android_policy', False, 'utf-8', None),
   WriterDesc('reg', False, 'utf-16', None),
   WriterDesc('doc', True, 'utf-8', None),
   WriterDesc('json', False, 'utf-8', None),
   WriterDesc('plist', False, 'utf-8', None),
   WriterDesc('plist_strings', True, 'utf-8', MacLanguageMap)
 ]
 
 
 # Template writers that are not per-language use policy_templates.json from
@@ skipping 47 matching lines @@
     admx, google_admx, android_policy, reg, json, plist: Absolute path of the
         corresponding file types. Must NOT contain a ${lang} placeholder. There
         is only one output file, not one per language.
     D: List of grit defines, used to assemble writer configuration.
     E, t: Grit environment variables and target platform. Unused, but
         grit_rule.gni adds them, so OptionParser must handle them.
   '''
   parser = optparse.OptionParser()
   parser.add_option('--translations', dest='translations')
   parser.add_option('--languages', dest='languages')
-  parser.add_option('--adm', dest='adm')
-  parser.add_option('--adml', dest='adml')
-  parser.add_option('--admx', dest='admx')
-  parser.add_option('--google_adml', dest='google_adml')
-  parser.add_option('--google_admx', dest='google_admx')
-  parser.add_option('--reg', dest='reg')
-  parser.add_option('--doc', dest='doc')
-  parser.add_option('--json', dest='json')
-  parser.add_option('--plist', dest='plist')
-  parser.add_option('--plist_strings', dest='plist_strings')
-  parser.add_option('--android_policy', dest='android_policy')
-  parser.add_option('-D', action='append', dest='grit_defines', default=[])
-  parser.add_option('-E', action='append', dest='grit_build_env', default=[])
-  parser.add_option('-t', action='append', dest='grit_target', default=[])
+  parser.add_option('--adm', action='append', dest='adm')
+  parser.add_option('--adml', action='append', dest='adml')
+  parser.add_option('--admx', action='append', dest='admx')
+  parser.add_option('--chromeos_adml', action='append', dest='chromeos_adml')
+  parser.add_option('--chromeos_admx', action='append', dest='chromeos_admx')
+  parser.add_option('--google_adml', action='append', dest='google_adml')
+  parser.add_option('--google_admx', action='append', dest='google_admx')
+  parser.add_option('--reg', action='append', dest='reg')
+  parser.add_option('--doc', action='append', dest='doc')
+  parser.add_option('--json', action='append', dest='json')
+  parser.add_option('--plist', action='append', dest='plist')
+  parser.add_option('--plist_strings', action='append', dest='plist_strings')
+  parser.add_option('--android_policy', action='append', dest='android_policy')
+  parser.add_option('-D', action='append', dest='grit_defines')
+  parser.add_option('-E', action='append', dest='grit_build_env')
+  parser.add_option('-t', action='append', dest='grit_target')
   options, args = parser.parse_args(argv[1:])
 
   _LANG_PLACEHOLDER = "${lang}";
   assert _LANG_PLACEHOLDER in options.translations
 
   languages = filter(bool, options.languages.split(','))
   assert _DEFAULT_LANGUAGE in languages
 
   config = _GetWriterConfiguration(options.grit_defines);
 
   # For each language, load policy data once and run all writers on it.
   for lang in languages:
     # Load the policy data.
     policy_templates_json_path = \
       options.translations.replace(_LANG_PLACEHOLDER, lang)
     with codecs.open(policy_templates_json_path, 'r', 'utf-16') as policy_file:
       policy_data = eval(policy_file.read())
 
     # Preprocess the policy data.
     policy_generator = \
         policy_template_generator.PolicyTemplateGenerator(config, policy_data)
 
     for writer_desc in _WRITER_DESCS:
       # For writer types that are not per language (e.g. admx), only do it once.
       if (not writer_desc.is_per_language and lang != _DEFAULT_LANGUAGE):
         continue
 
       # Was the current writer type passed as argument, e.g. --admx <path>?
-      output_path = getattr(options, writer_desc.type, '')
-      if (not output_path):
+      # Note that all paths are arrays and we loop over all of them.
+      output_paths = getattr(options, writer_desc.type, '')
+      if (not output_paths):
         continue
+      for output_path in output_paths:
+        # Substitute language placeholder in output file.
+        if (writer_desc.is_per_language):
+          assert _LANG_PLACEHOLDER in output_path
+          mapped_lang = writer_desc.language_map(lang) \
+              if writer_desc.language_map else lang
+          output_path = output_path.replace(_LANG_PLACEHOLDER, mapped_lang)
+        else:
+          assert _LANG_PLACEHOLDER not in output_path
 
-      # Substitute language placeholder in output file.
-      if (writer_desc.is_per_language):
-        assert _LANG_PLACEHOLDER in output_path
-        mapped_lang = \
-            writer_desc.language_map(lang) if writer_desc.language_map else lang
-        output_path = output_path.replace(_LANG_PLACEHOLDER, mapped_lang)
-      else:
-        assert _LANG_PLACEHOLDER not in output_path
+        # Run the template writer on th policy data.
+        writer = GetWriter(writer_desc.type, config)
+        output_data = policy_generator.GetTemplateText(writer)
 
-      # Run the template writer on th policy data.
-      writer = GetWriter(writer_desc.type, config)
-      output_data = policy_generator.GetTemplateText(writer)
+        # Make output directory if it doesn't exist yet.
+        output_dir = os.path.split(output_path)[0]
+        if not os.path.exists(output_dir):
+          os.makedirs(output_dir)
 
-      # Make output directory if it doesn't exist yet.
-      output_dir = os.path.split(output_path)[0]
-      if not os.path.exists(output_dir):
-        os.makedirs(output_dir)
-
-      # Write output file.
-      with codecs.open(output_path, 'w', writer_desc.encoding) as output_file:
-        output_file.write(output_data)
+        # Write output file.
+        with codecs.open(output_path, 'w', writer_desc.encoding) as output_file:
+          output_file.write(output_data)
 
 
 if '__main__' == __name__:
   sys.exit(main(sys.argv))