Index: tools/metrics/histograms/update_policies.py |
diff --git a/tools/metrics/histograms/update_policies.py b/tools/metrics/histograms/update_policies.py |
index 5cad317512b3916541dc3d4d34ea3e55ea00b37d..79f524f690b97a526fb139a90d0fd5233679b797 100644 |
--- a/tools/metrics/histograms/update_policies.py |
+++ b/tools/metrics/histograms/update_policies.py |
@@ -8,6 +8,7 @@ definitions read from policy_templates.json. |
If the file was pretty-printed, the updated version is pretty-printed too. |
""" |
+import os |
import re |
import sys |
@@ -15,12 +16,15 @@ from ast import literal_eval |
from optparse import OptionParser |
from xml.dom import minidom |
-from diffutil import PromptUserToAcceptDiff |
-from pretty_print import PrettyPrintNode |
+import print_style |
+ |
+# Import the metrics/common module. |
+sys.path.append(os.path.join(os.path.dirname(__file__), '..', 'common')) |
+from diff_util import PromptUserToAcceptDiff |
HISTOGRAMS_PATH = 'histograms.xml' |
-POLICY_TEMPLATES_PATH = |
- '../../../components/policy/resources/policy_templates.json' |
+POLICY_TEMPLATES_PATH = ( |
+ '../../../components/policy/resources/policy_templates.json') |
ENUM_NAME = 'EnterprisePolicies' |
class UserError(Exception): |
@@ -118,8 +122,7 @@ def main(): |
xml = f.read() |
UpdateHistogramDefinitions(policy_templates, histograms_doc) |
- |
- new_xml = PrettyPrintNode(histograms_doc) |
+ new_xml = print_style.GetPrintStyle().PrettyPrintNode(histograms_doc) |
if PromptUserToAcceptDiff(xml, new_xml, 'Is the updated version acceptable?'): |
with open(HISTOGRAMS_PATH, 'wb') as f: |
f.write(new_xml) |