Make json_schema_compiler remove 'nocompile' nodes from JSON at the JSON level

tools/json_schema_compiler/json_schema.py

 # 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.
 
 import copy
 import json
 
 def StripJSONComments(stream):
   """Strips //-style comments from a stream of JSON. Allows un-escaped //
   inside string values.
@@ skipping 30 matching lines @@
       else:
         if last_char == '/' and not inside_string:
           result += '/'
         if char == '"':
           inside_string = not inside_string
     last_char = char
     result += char
 
   return result
 
+def DeleteNocompileNodes(item):
+  def HasNocompile(thing):
+    return type(thing) == dict and thing.get('nocompile', False)
+
+  if type(item) == dict:
+    toDelete = []
+    for key, value in item.items():
+      if HasNocompile(value):
+        toDelete.append(key)
+      else:
+        DeleteNocompileNodes(value)
+    for delete in toDelete:

[Yoyo Zhou, 2012/04/16 23:21:46]

I'd call 'delete' 'key' instead.

[not at google - send to devlin, 2012/04/17 04:05:45]

Done.

+      del item[delete]
+  elif type(item) == list:

[Yoyo Zhou, 2012/04/16 23:21:46]

You could make this a list comprehension.
item[:] = [DeleteNoCompileNodes(thing) for thing in item if not
HasNocompile(thing)]

[not at google - send to devlin, 2012/04/17 04:05:45]

Done.

+    i = 0
+    while i < len(item):
+      if HasNocompile(item[i]):
+        del item[i]
+      else:
+        DeleteNocompileNodes(item[i])
+        i += 1
+
+  return item
+
 def Load(filename):
   with open(filename, 'r') as handle:
-    return json.loads(StripJSONComments(handle.read()))
-
+    return DeleteNocompileNodes(json.loads(StripJSONComments(handle.read())))
 
 # A dictionary mapping |filename| to the object resulting from loading the JSON
 # at |filename|.
 _cache = {}
 
 def CachedLoad(filename):
   """Equivalent to Load(filename), but caches results for subsequent calls"""
   if filename not in _cache:
     _cache[filename] = Load(filename)
   # Return a copy of the object so that any changes a caller makes won't affect
   # the next caller.
   return copy.deepcopy(_cache[filename])