Changes to support roll 39 IDLs

tools/dom/scripts/database.py

 #!/usr/bin/python
 # Copyright (c) 2011, the Dart project authors.  Please see the AUTHORS file
 # for details. All rights reserved. Use of this source code is governed by a
 # BSD-style license that can be found in the LICENSE file.
 
 """Module to manage IDL files."""
 
 import copy
 import pickle
 import logging
@@ skipping 25 matching lines @@
       root_dir -- a directory. If directory does not exist, it will
       be created.
     """
     self._root_dir = root_dir
     if not os.path.exists(root_dir):
       _logger.debug('creating root directory %s' % root_dir)
       os.makedirs(root_dir)
     self._all_interfaces = {}
     self._interfaces_to_delete = []
     self._enums = {}
+    self._all_dictionaries = {}
 
   def Clone(self):
     new_database = Database(self._root_dir)
     new_database._all_interfaces = copy.deepcopy(self._all_interfaces)
     new_database._interfaces_to_delete = copy.deepcopy(
         self._interfaces_to_delete)
     new_database._enums = copy.deepcopy(self._enums)
+    new_database._all_dictionaries = copy.deepcopy(self._all_dictionaries)
+
     return new_database
 
   def Delete(self):
     """Deletes the database by deleting its directory"""
     if os.path.exists(self._root_dir):
       shutil.rmtree(self._root_dir)
     # reset in-memory constructs
     self._all_interfaces = {}
 
   def _ScanForInterfaces(self):
@@ skipping 175 matching lines @@
 
   def HasEnum(self, enum_name):
     return enum_name in self._enums
 
   def GetEnum(self, enum_name):
     return self._enums[enum_name]
 
   def AddEnum(self, enum):
     self._enums[enum.id] = enum
 
+  def HasDictionary(self, dictionary_name):
+    """Returns True if the dictionary is in memory"""
+    return dictionary_name in self._all_dictionaries
+
+  def GetDictionary(self, dictionary_name):
+    """Returns an IDLDictionary corresponding to the dictionary_name
+    from memory.
+
+    Args:
+      dictionary_name -- the name of the dictionary.
+    """
+    if dictionary_name not in self._all_dictionaries:
+      raise RuntimeError('Dictionary %s is not loaded' % dictionary_name)
+    return self._all_dictionaries[dictionary_name]
+
+  def AddDictionary(self, dictionary):
+    """Returns an IDLDictionary corresponding to the dictionary_name
+    from memory.
+
+    Args:
+      dictionary -- the name of the dictionary.
+    """
+    dictionary_name = dictionary.id
+    if dictionary_name in self._all_dictionaries:
+      raise RuntimeError('Dictionary %s already exists' % dictionary_name)
+    self._all_dictionaries[dictionary_name] = dictionary
+
   def TransitiveSecondaryParents(self, interface, propagate_event_target):
     """Returns a list of all non-primary parents.
 
     The list contains the interface objects for interfaces defined in the
     database, and the name for undefined interfaces.
     """
     def walk(parents):
       for parent in parents:
         parent_name = parent.type.id
         if IsDartCollectionType(parent_name):
           result.append(parent_name)
           continue
         if self.HasInterface(parent_name):
           parent_interface = self.GetInterface(parent_name)
           result.append(parent_interface)
           walk(parent_interface.parents)
 
     result = []
     if interface.parents:
       parent = interface.parents[0]
       if (IsPureInterface(parent.type.id) or
           (propagate_event_target and parent.type.id == 'EventTarget')):
         walk(interface.parents)
       else:
         walk(interface.parents[1:])
     return result