bt5_prototype: Add InstallationTree class

product/ERP5/Document/BusinessPackage.py

@@ -28,6 +28,7 @@
 ##############################################################################
 
 import fnmatch, re, gc
+import hashlib
 import transaction
 from copy import deepcopy
 from collections import defaultdict
@@ -292,46 +293,53 @@ class PathTemplatePackageItem(Implicit, Persistent):
         container._setObject(object_id, obj)
         obj = container._getOb(object_id)
 
-    # Regenerate local roles for all paths in this business template
-    p = context.getPortalObject()
-    portal_type_role_list_len_dict = {}
-    update_dict = defaultdict(list)
-    for path in self._objects:
-      obj = p.unrestrictedTraverse(path, None)
-      # Ignore any object without PortalType (non-ERP5 objects)
-      try:
-        portal_type = aq_base(obj).getPortalType()
-      except Exception, e:
-        pass
+
+def createInstallationData(package_list):
+  """
+  Create installation object as well as adds new node on the installation tree
+  from the installed state
+  """
+  data = {}
+  path_list = []
+
+  # Create path_list to be installed by the installation
+  for package in package_list:
+    path_list.extend(package.getTemplatePathList())
+    path_list = list(set(path_list))
+
+  for package in package_list:
+    obj_dict = package._path_item._objects
+    for path in path_list:
+      if not data.has_key(path):
+        data[path] = [obj_dict[path]]
+      else:
+        data[path].append(obj_dict[path])
+
+  # Compare the objects which are present in multiple numbers in data_list
+  for path, obj_list in data_list.iteritems():
+    # We use iteritems so that it'd be okay while we change the values
+    if len(obj_list) == 1:
+      data[path] = obj_list[0]
+    else:
+      hash_func = hashlib.sha1
+      hash_list = [hash_fuc(obj.asXML()).hexdigest() for obj in obj_list]
+      hash_list = set(hash_list)
+      if not len(hash_list) == 1:
+        raise ValueError('There is a conflict')
       else:
-        if portal_type not in p.portal_types:
-          LOG("BusinessTemplate", WARNING,
-              "Could not update Local Roles as Portal Type '%s' could not "
-              "be found" % portal_type)
-
-          continue
-
-        if portal_type not in portal_type_role_list_len_dict:
-          portal_type_role_list_len_dict[portal_type] = \
-              len(p.portal_types[portal_type].getRoleInformationList())
-
-        if portal_type_role_list_len_dict[portal_type]:
-          update_dict[portal_type].append(obj)
-
-    if update_dict:
-      def updateLocalRolesOnDocument():
-        for portal_type, obj_list in update_dict.iteritems():
-          update = p.portal_types[portal_type].updateLocalRolesOnDocument
-          for obj in obj_list:
-            update(obj)
-            LOG("BusinessTemplate", INFO,
-                "Updated Local Roles for '%s' (%s)"
-                % (portal_type, obj.getRelativeUrl()))
-      transaction.get().addBeforeCommitHook(updateLocalRolesOnDocument)
-
-class InstallationState(object):
+        data[path] = obj_list[0]
+
+  return data
+
+class InstallationTree(object):
   """
   Tree implemetation to manage install/update/remove between states.
+  This is a very rough code to explain what can be achieved. In real case,
+  this class should be a well defined ERP5 object and most possibly act as a
+  portal tool, cause there should be one installation tree per site(agree ??)
+
+  Data at every node:
+  ('_path_item': PathTemplateItem, }
 
   State Number:
   1)  ERP5Site
@@ -342,6 +350,9 @@ class InstallationState(object):
   Leaf node: OFS State(with some default BP installed)
   Trying to install a new BT5 should be like adding new node to the tree
 
+  Will show if faced by any conflict between states, but mostly will try to
+  solve by itself
+
   How to pickle:
   http://stackoverflow.com/questions/2134706/hitting-maximum-recursion-depth-using-pythons-pickle-cpickle
 
@@ -350,27 +361,25 @@ class InstallationState(object):
 
   """
 
-  def __init__(self, state):
-    """
-    """
-    self.root_status = False   # Installtion status of the combined packages
-    self.data_list = []        # To be installed/update/deleted list of packages
-    self.current_level = 1
+  def __init__(self, data):
+    self.data = data          # To be installed/update/deleted list of packages
+    self.children = []        # List of child nodes
 
-  def setNewState(self, state):
+  def addNewState(self, state):
     """
-    In tree language, should act as setNext node to the tree
+    In tree language, should act as set next node to the tree
 
-    This should  add package list after comparing the states of
+    This should add package list after comparing the states of
     packages with the installed state. So even if we try to install multiple
     packages at a time, it should be counted as one state being implented on
     another installed state, i.e, the state of ERP5Site
     """
-    pass
+    self.children.append(state)
 
-  def install_package_list(self, package_list):
+  def mapToERP5Site(self):
     """
     Create a new state by comparing all BP combined built and the ERP5Site,
     then calls setNewState to update the state
     """
+    # No need to save sha here, save it in business package itself
     pass