TemplateTool.py 98.6 KB
Newer Older
1
# -*- coding: utf-8 -*-
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2 3 4
##############################################################################
#
# Copyright (c) 2002 Nexedi SARL and Contributors. All Rights Reserved.
Jean-Paul Smets's avatar
Jean-Paul Smets committed
5
#                    Jean-Paul Smets-Solanes <jp@nexedi.com>
Jean-Paul Smets's avatar
Jean-Paul Smets committed
6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# This program is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################

30
from webdav.client import Resource
Jean-Paul Smets's avatar
Jean-Paul Smets committed
31

Yoshinori Okuji's avatar
Yoshinori Okuji committed
32
from App.config import getConfiguration
33
import os
34
import shutil
35
import sys
36
import hashlib
37
import pprint
Yoshinori Okuji's avatar
Yoshinori Okuji committed
38

39
from Acquisition import Implicit, Explicit
Jean-Paul Smets's avatar
Jean-Paul Smets committed
40
from AccessControl import ClassSecurityInfo
41
from AccessControl.SecurityInfo import ModuleSecurityInfo
42
from Products.CMFActivity.ActiveResult import ActiveResult
43
from Products.PythonScripts.PythonScript import PythonScript
44
from Products.ERP5Type.Globals import InitializeClass, DTMLFile, PersistentMapping
45
from Products.ERP5Type.DiffUtils import DiffFile
Jean-Paul Smets's avatar
Jean-Paul Smets committed
46
from Products.ERP5Type.Tool.BaseTool import BaseTool
47
from Products.ERP5Type.Cache import transactional_cached
48
from Products.ERP5Type import Permissions, interfaces
49
from Products.ERP5.Document.BusinessTemplate import BusinessTemplateMissingDependency
50
from Products.ERP5Type.Accessor.Constant import PropertyGetter as ConstantGetter
51
from Products.ERP5.genbt5list import generateInformation
52
from Acquisition import aq_base
53
from tempfile import mkstemp, mkdtemp
Jean-Paul Smets's avatar
Jean-Paul Smets committed
54
from Products.ERP5 import _dtmldir
Aurel's avatar
Aurel committed
55
from cStringIO import StringIO
56
from urllib import pathname2url, urlopen, splittype, urlretrieve
57
import urllib2
58 59
import re
from xml.dom.minidom import parse
60
from xml.parsers.expat import ExpatError
61 62
import struct
import cPickle
63
from base64 import b64encode, b64decode
64
from Products.ERP5Type.Message import translateString
65
from zLOG import LOG, INFO, WARNING
66
from base64 import decodestring
67
from difflib import unified_diff
68
from operator import attrgetter
69
import subprocess
70
import time
Jean-Paul Smets's avatar
Jean-Paul Smets committed
71

72
WIN = os.name == 'nt'
73

74 75
CATALOG_UPDATABLE = object()
ModuleSecurityInfo(__name__).declarePublic('CATALOG_UPDATABLE')
76

77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92
class BusinessTemplateUnknownError(Exception):
  """ Exception raised when the business template
      is impossible to find in the repositories
  """
  pass

class UnsupportedComparingOperator(Exception):
  """ Exception when the comparing string is unsupported
  """
  pass

class BusinessTemplateIsMeta(Exception):
  """ Exception when the business template is provided by another one
  """
  pass

93 94
ModuleSecurityInfo(__name__).declarePublic('BusinessTemplateUnknownError')

Jean-Paul Smets's avatar
Jean-Paul Smets committed
95
class TemplateTool (BaseTool):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
96
    """
97
      TemplateTool manages Business Templates.
Jean-Paul Smets's avatar
Jean-Paul Smets committed
98

99 100 101 102 103 104
      TemplateTool provides some methods to deal with Business Templates:
        - download
        - publish
        - install
        - update
        - save
Jean-Paul Smets's avatar
Jean-Paul Smets committed
105 106
    """
    id = 'portal_templates'
107
    title = 'Template Tool'
Jean-Paul Smets's avatar
Jean-Paul Smets committed
108
    meta_type = 'ERP5 Template Tool'
Jean-Paul Smets's avatar
Jean-Paul Smets committed
109
    portal_type = 'Template Tool'
110 111 112 113 114
    allowed_types = (
      'ERP5 Business Template',
      'ERP5 Business Package',
      'ERP5 Business Manager',
      )
115

116 117
    # This stores information on repositories.
    repository_dict = {}
Jean-Paul Smets's avatar
Jean-Paul Smets committed
118 119 120 121

    # Declarative Security
    security = ClassSecurityInfo()

Rafael Monnerat's avatar
Rafael Monnerat committed
122 123
    security.declareProtected(Permissions.ManagePortal, 'manage_overview')
    manage_overview = DTMLFile('explainTemplateTool', _dtmldir)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
124

125 126
    security.declareProtected(Permissions.AccessContentsInformation,
                              'getInstalledBusinessTemplate')
127
    def getInstalledBusinessTemplate(self, title, strict=False, **kw):
128
      """Returns an installed version of business template of a given title.
129

130
        Returns None if business template is not installed or has been uninstalled.
131 132
        It not "installed" business template is found, look at replaced ones.
        This is mostly usefull if we are looking for the installed business
133 134
        template in a transaction replacing an existing business template.
        If strict is true, we do not take care of "replaced" business templates.
135 136
      """
      # This can be slow if, say, 10000 business templates are present.
Vincent Pelletier's avatar
Vincent Pelletier committed
137 138 139
      # However, that unlikely happens, and using a Z SQL Method has a
      # potential danger because business templates may exchange catalog
      # methods, so the database could be broken temporarily.
140
      last_bt = last_time = None
141 142 143 144 145 146 147
      for bt in self.objectValues(portal_type=['Business Template',
                                               'Business Package',
                                               'Business Manager']):
        if bt.getPortalType() == 'Business Manager':
          if bt.getStatus() == 'installed':
            return bt
          return None
148
        if bt.getTitle() == title or title in bt.getProvisionList():
149 150 151
          state = bt.getInstallationState()
          if state == 'installed':
            return bt
152 153 154 155 156 157 158 159 160
          if state == 'not_installed':
            last_transition = bt.workflow_history \
              ['business_template_installation_workflow'][-1]
            if last_transition['action'] == 'uninstall': # There is not uninstalled state !
              t = last_transition['time']
              if last_time < t:
                last_bt = None
                last_time = t
          elif state == 'replaced' and not strict:
161 162 163 164 165 166
            t = bt.workflow_history \
              ['business_template_installation_workflow'][-1]['time']
            if last_time < t:
              last_bt = bt
              last_time = t
      return last_bt
167

168 169
    security.declareProtected(Permissions.AccessContentsInformation,
                              'getInstalledBusinessTemplatesList')
170
    def getInstalledBusinessTemplatesList(self):
171 172 173 174 175
      """Deprecated.
      """
      DeprecationWarning('getInstalledBusinessTemplatesList is deprecated; Use getInstalledBusinessTemplateList instead.', DeprecationWarning)
      return self.getInstalledBusinessTemplateList()

176
    def _getInstalledBusinessTemplateList(self, only_title=0):
177
      """Get the list of installed business templates.
178 179
      """
      installed_bts = []
180 181 182
      for bt in self.contentValues(portal_type=['Business Template',
                                                'Business Package',
                                                'Business Manager']):
183
        if bt.getInstallationState() == 'installed':
184 185 186 187
          bt5 = bt
          if only_title:
            bt5 = bt.getTitle()
          installed_bts.append(bt5)
188
      return installed_bts
189

190 191
    security.declareProtected(Permissions.AccessContentsInformation,
                              'getInstalledBusinessTemplateList')
192 193 194 195 196
    def getInstalledBusinessTemplateList(self):
      """Get the list of installed business templates.
      """
      return self._getInstalledBusinessTemplateList(only_title=0)

197 198
    security.declareProtected(Permissions.AccessContentsInformation,
                              'getInstalledBusinessTemplateTitleList')
199 200 201 202 203
    def getInstalledBusinessTemplateTitleList(self):
      """Get the list of installed business templates.
      """
      return self._getInstalledBusinessTemplateList(only_title=1)

204 205
    security.declareProtected(Permissions.AccessContentsInformation,
                              'getInstalledBusinessTemplateRevision')
206 207 208 209 210 211
    def getInstalledBusinessTemplateRevision(self, title, **kw):
      """
        Return the revision of business template installed with the title
        given
      """
      bt = self.getInstalledBusinessTemplate(title)
212 213 214
      if bt is not None:
        return bt.getRevision()
      return None
215

216 217
    security.declareProtected(Permissions.AccessContentsInformation,
                              'getBuiltBusinessTemplateList')
218 219
    def getBuiltBusinessTemplateList(self):
      """Get the list of built and not installed business templates.
220
      """
221 222 223
      return [bt for bt in self.objectValues(portal_type='Business Template')
                 if bt.getInstallationState() == 'not_installed' and
                    bt.getBuildingState() == 'built']
224

225 226 227 228 229 230
    @property
    def asRepository(self):
      class asRepository(Explicit):
        """Export business template by their title

        Provides a view of template tool allowing a user to download the last
231
        edited business template with a URL like:
232 233 234 235 236 237 238 239 240 241 242
          http://.../erp5/portal_templates/asRepository/erp5_core
        """
        def __before_publishing_traverse__(self, self2, request):
          path = request['TraversalRequestNameStack']
          self.subpath = tuple(reversed(path))
          del path[:]
        def __call__(self, REQUEST, RESPONSE):
          title, = self.subpath
          last_bt = None, None
          for bt in self.aq_parent.searchFolder(title=title):
            bt = bt.getObject()
243 244 245
            modified = bt.getModificationDate()
            if last_bt[0] < modified and bt.getInstallationState() != 'deleted':
              last_bt = modified, bt
246 247 248 249 250 251 252 253 254 255 256
          if last_bt[1] is None:
            return RESPONSE.notFoundError(title)
          RESPONSE.setHeader('Content-type', 'application/data')
          RESPONSE.setHeader('Content-Disposition',
                             'inline;filename=%s-%s.zexp' % (title, last_bt[0]))
          if REQUEST['REQUEST_METHOD'] == 'GET':
            bt = last_bt[1]
            if bt.getBuildingState() != 'built':
              bt.build()
            return self.aq_parent.manage_exportObject(bt.getId(), download=1)
      return asRepository().__of__(self)
257

258
    security.declareProtected(Permissions.ManagePortal,
259 260
                              'getDefaultBusinessTemplateDownloadURL')
    def getDefaultBusinessTemplateDownloadURL(self):
261 262 263 264 265
      """Returns the default download URL for business templates.
      """
      return "file://%s/" % pathname2url(
                  os.path.join(getConfiguration().instancehome, 'bt5'))

Rafael Monnerat's avatar
Rafael Monnerat committed
266
    security.declareProtected('Import/Export objects', 'save')
267
    def save(self, business_template, REQUEST=None, RESPONSE=None):
Yoshinori Okuji's avatar
Yoshinori Okuji committed
268
      """
Vincent Pelletier's avatar
Vincent Pelletier committed
269
        Save the BusinessTemplate in the servers's filesystem.
Yoshinori Okuji's avatar
Yoshinori Okuji committed
270 271
      """
      cfg = getConfiguration()
Vincent Pelletier's avatar
Vincent Pelletier committed
272 273
      path = os.path.join(cfg.clienthome,
                          '%s' % (business_template.getTitle(),))
274
      path = pathname2url(path)
275
      business_template.export(path=path, local=True)
276
      if REQUEST is not None:
277
        psm = translateString('Saved in ${path} .',
278
                              mapping={'path':pathname2url(path)})
279
        ret_url = '%s/%s?portal_status_message=%s' % \
Vincent Pelletier's avatar
Vincent Pelletier committed
280
                  (business_template.absolute_url(),
281
                   REQUEST.get('form_id', 'view'), psm)
Vincent Pelletier's avatar
Vincent Pelletier committed
282 283 284
        if RESPONSE is None:
          RESPONSE = REQUEST.RESPONSE
        return REQUEST.RESPONSE.redirect( ret_url )
285 286

    security.declareProtected( 'Import/Export objects', 'export' )
287
    def export(self, business_template, REQUEST=None, RESPONSE=None, is_package=False):
288
      """
Vincent Pelletier's avatar
Vincent Pelletier committed
289 290
        Export the Business Template as a bt5 file and offer the user to
        download it.
291
      """
292
      export_string = business_template.export()
Aurel's avatar
Aurel committed
293
      try:
294 295
        if RESPONSE is not None:
          RESPONSE.setHeader('Content-type','tar/x-gzip')
296
          if not is_package:
297 298 299 300 301
            RESPONSE.setHeader('Content-Disposition', 'inline;filename=%s-%s.bt5'
              % (business_template.getTitle(), business_template.getVersion()))
          else:
            RESPONSE.setHeader('Content-Disposition', 'inline;filename=%s.bp5'
            % business_template.getTitle())
Aurel's avatar
Aurel committed
302 303 304
        return export_string.getvalue()
      finally:
        export_string.close()
Yoshinori Okuji's avatar
Yoshinori Okuji committed
305

306
    security.declareProtected( 'Import/Export objects', 'publish' )
307 308
    def publish(self, business_template, url, username=None, password=None):
      """
Vincent Pelletier's avatar
Vincent Pelletier committed
309
        Publish the given business template at the given URL.
310 311
      """
      business_template.build()
Vincent Pelletier's avatar
Vincent Pelletier committed
312
      export_string = self.manage_exportObject(id=business_template.getId(),
313
                                               download=True)
314
      bt = Resource(url, username=username, password=password)
Vincent Pelletier's avatar
Vincent Pelletier committed
315 316
      bt.put(file=export_string,
             content_type='application/x-erp5-business-template')
317
      business_template.setPublicationUrl(url)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
318

319
    security.declareProtected(Permissions.ManagePortal, 'update')
320 321
    def update(self, business_template):
      """
Vincent Pelletier's avatar
Vincent Pelletier committed
322
        Update an existing template from its publication URL.
323 324 325 326 327 328
      """
      url = business_template.getPublicationUrl()
      id = business_template.getId()
      bt = Resource(url)
      export_string = bt.get().get_body()
      self.deleteContent(id)
Aurel's avatar
Aurel committed
329
      self._importObjectFromFile(StringIO(export_string), id=id)
330

331

332
    security.declareProtected( Permissions.ManagePortal, 'manage_download' )
333 334
    def manage_download(self, url, id=None, REQUEST=None):
      """The management interface for download.
335
      """
336 337
      if REQUEST is None:
        REQUEST = getattr(self, 'REQUEST', None)
338

339
      bt = self.download(url, id=id)
340

341
      if REQUEST is not None:
342
        ret_url = bt.absolute_url()
Yusei Tahara's avatar
Yusei Tahara committed
343
        psm = translateString("Business template downloaded successfully.")
344
        REQUEST.RESPONSE.redirect("%s?portal_status_message=%s"
345
                                    % (ret_url, psm))
346

347
    def _download_local(self, path, bt_id, format_version=1):
348 349
      """Download Business Template from local directory or file
      """
350
      if format_version == 2:
351 352 353
        bp = self.newContent(bt_id, 'Business Package')
        bp.importFile(path)
        return bp
354 355 356 357
      elif format_version == 3:
        bm = self.newContent(bt_id, 'Business Manager')
        bm.importFile(path)
        return bm
358

359 360
      bt = self.newContent(bt_id, 'Business Template')
      bt.importFile(path)
361
      return bt
362 363 364 365

    def _download_url(self, url, bt_id):
      tempid, temppath = mkstemp()
      try:
366
        os.close(tempid) # Close the opened fd as soon as possible.
367
        file_path, headers = urlretrieve(url, temppath)
368
        if re.search(r'<title>.*Revision \d+:', open(file_path, 'r').read()):
369 370 371
          # this looks like a subversion repository, try to check it out
          LOG('ERP5', INFO, 'TemplateTool doing a svn checkout of %s' % url)
          return self._download_svn(url, bt_id)
Rafael Monnerat's avatar
Rafael Monnerat committed
372

373 374 375 376 377 378 379 380
        return self._download_local(file_path, bt_id)
      finally:
        os.remove(temppath)

    def _download_svn(self, url, bt_id):
      svn_checkout_tmp_dir = mkdtemp()
      svn_checkout_dir = os.path.join(svn_checkout_tmp_dir, 'bt')
      try:
381 382
        from Products.ERP5VCS.WorkingCopy import getVcsTool
        getVcsTool('svn').__of__(self).export(url, svn_checkout_dir)
383 384 385 386 387
        return self._download_local(svn_checkout_dir, bt_id)
      finally:
        shutil.rmtree(svn_checkout_tmp_dir)

    security.declareProtected( 'Import/Export objects', 'download' )
388
    def download(self, url, id=None, REQUEST=None):
389 390 391 392 393 394 395 396 397 398 399
      """
      Download Business Template from url, can be file or local directory
      """
      # For backward compatibility: If REQUEST is passed, it is likely that we
      # come from the management interface.
      if REQUEST is not None:
        return self.manage_download(url, id=id, REQUEST=REQUEST)
      if id is None:
        id = self.generateNewId()

      urltype, name = splittype(url)
400 401 402
      # Create a zexp path which would be used for Business Manager files
      zexp_path = name + '/' + name.split('/')[-1] + '.zexp'

403 404 405
      if WIN and urltype and '\\' in name:
        urltype = None
        name = url
406
      if urltype and urltype != 'file':
407
        if '/portal_templates/asRepository/' in url:
408 409 410 411 412
          # In this case, the downloaded BT is already built.
          bt = self._p_jar.importFile(urlopen(url))
          bt.id = id
          del bt.uid
          return self[self._setObject(id, bt)]
413
        bt = self._download_url(url, id)
414 415 416 417
      elif os.path.exists(zexp_path):
        # If the path exists, we create a Business Manager object after
        # downloading it from zexp path
        bt = self._download_local(os.path.normpath(zexp_path), id, format_version=3)
418
      else:
419 420 421 422 423 424
        template_version_path_list = [
                                      name+'/bp/template_format_version',
                                     ]

        for path in template_version_path_list:
          try:
425
            file = open(os.path.normpath(path))
426
          except IOError:
427 428 429 430 431 432 433
            continue
        try:
          format_version = int(file.read())
          file.close()
        except UnboundLocalError:
          # In case none of the above paths do have template_format_version
          format_version = 1
434
        # XXX: Download only needed in case the file is in directory
435
        bt = self._download_local(os.path.normpath(name), id, format_version)
436

437
      bt.build(no_action=True)
438
      return bt
Jean-Paul Smets's avatar
Jean-Paul Smets committed
439

440
    security.declareProtected('Import/Export objects', 'importBase64EncodedText')
441
    def importBase64EncodedText(self, file_data=None, id=None, REQUEST=None,
442
                                batch_mode=False, **kw):
443
      """
444 445 446
        Import Business Template from passed base64 encoded text.
      """
      import_file = StringIO(decodestring(file_data))
447
      return self.importFile(import_file = import_file, id = id, REQUEST = REQUEST,
448 449
                             batch_mode = batch_mode, **kw)

450
    security.declareProtected('Import/Export objects', 'importFile')
451
    def importFile(self, import_file=None, id=None, REQUEST=None,
452
                   batch_mode=False, **kw):
453
      """
Vincent Pelletier's avatar
Vincent Pelletier committed
454
        Import Business Template from one file
455
      """
456 457
      if REQUEST is None:
        REQUEST = getattr(self, 'REQUEST', None)
458

459 460 461 462 463
      if id is None:
        id = self.generateNewId()

      if (import_file is None) or (len(import_file.read()) == 0):
        if REQUEST is not None:
Yusei Tahara's avatar
Yusei Tahara committed
464
          psm = translateString('No file or an empty file was specified.')
465 466
          REQUEST.RESPONSE.redirect("%s?portal_status_message=%s"
                                    % (self.absolute_url(), psm))
Alexandre Boeglin's avatar
Alexandre Boeglin committed
467 468
          return
        else :
469
          raise RuntimeError, 'No file or an empty file was specified'
Aurel's avatar
Aurel committed
470
      # copy to a temp location
Alexandre Boeglin's avatar
Alexandre Boeglin committed
471
      import_file.seek(0) #Rewind to the beginning of file
472
      tempid, temppath = mkstemp()
473 474
      try:
        os.close(tempid) # Close the opened fd as soon as possible
475
        with open(temppath, 'wb') as tempfile:
476
          tempfile.write(import_file.read())
477
        bt = self._download_local(temppath, id)
478 479
      finally:
        os.remove(temppath)
480
      bt.build(no_action=True)
Aurel's avatar
Aurel committed
481
      bt.reindexObject()
482

483
      if not batch_mode and \
484
         (REQUEST is not None):
485
        ret_url = bt.absolute_url()
Yusei Tahara's avatar
Yusei Tahara committed
486
        psm = translateString("Business templates imported successfully.")
487 488
        REQUEST.RESPONSE.redirect("%s?portal_status_message=%s"
                                  % (ret_url, psm))
489
      elif batch_mode:
490
        return bt
491

492
    security.declareProtected(Permissions.ManagePortal, 'getDiffFilterScriptList')
493 494 495 496
    def getDiffFilterScriptList(self):
      """
      Return list of scripts usable to filter diff
      """
497
      # XXX, the "or ()" should not be there, the preference tool is
498 499
      # inconsistent, the called method should not return None when
      # nothing is selected
500
      portal = self.getPortalObject()
501 502 503 504 505 506 507 508
      script_list = []
      for script_id in portal.portal_preferences\
         .getPreferredDiffFilterScriptIdList() or ():
        try:
          script_list.append(getattr(portal, script_id))
        except AttributeError:
          LOG("TemplateTool", WARNING, "Unable to find %r script" % script_id)
      return script_list
509

510
    security.declareProtected(Permissions.ManagePortal, 'getFilteredDiffAsHTML')
511 512 513 514 515 516
    def getFilteredDiffAsHTML(self, diff):
      """
      Return the diff filtered by python scripts into html format
      """
      return self.getFilteredDiff(diff).toHTML()

517
    def _cleanUpTemplateFolder(self, folder_path):
518 519
      file_object_list = [x for x in os.listdir(folder_path)]
      for file_object in file_object_list:
520 521 522 523 524 525
        file_object_path = os.path.join(folder_path, file_object)
        if os.path.isfile(file_object_path):
          os.unlink(file_object_path)
        else:
          shutil.rmtree(file_object_path)

526 527
    security.declareProtected( 'Import/Export objects', 'importAndReExportBusinessTemplateFromPath' )
    def importAndReExportBusinessTemplateFromPath(self, template_path):
528 529 530 531 532 533 534 535 536 537 538 539 540 541 542 543 544 545 546 547 548 549
      """
        Imports the template that is in the template_path and exports it to the
        same path.

        We want to clean this directory, i.e. remove all files before
        the export. Because this is called as activity though, it could cause
        the following problem:
        - Activity imports the template
        - Activity removes all files from template_path
        - Activity fails in export.
        Then the folder contents will be changed, so when retrying the
        activity may succeed without the user understanding that files were
        erased. For this reason export is done in 3 steps:
        - First to a temporary directory
        - If there was no error delete contents of template_path
        - Copy the contents of the temporary directory to the template_path
      """
      import_template = self.download(url=template_path)
      export_dir = mkdtemp()
      try:
        import_template.export(path=export_dir, local=True)
        self._cleanUpTemplateFolder(template_path)
550 551 552 553 554
        file_name_list = [x for x in os.listdir(export_dir)]
        for file_name in file_name_list:
          temp_file_path = os.path.join(export_dir, file_name)
          destination_file_path = os.path.join(template_path, file_name)
          shutil.move(temp_file_path, destination_file_path)
555 556 557 558 559
      except:
        raise
      finally:
        shutil.rmtree(export_dir)

560 561
    security.declareProtected( 'Import/Export objects', 'importAndReExportBusinessTemplateListFromPath' )
    def importAndReExportBusinessTemplateListFromPath(self, repository_list, REQUEST=None, **kw):
562 563 564 565 566 567 568 569 570 571 572 573 574 575 576 577 578 579 580 581 582 583 584 585 586
      """
        Migrate business templates to new format where files like .py or .html
        are exported seprately than the xml.
      """
      repository_list = filter(bool, repository_list)

      if REQUEST is None:
        REQUEST = getattr(self, 'REQUEST', None)
        
      if len(repository_list) == 0 and REQUEST:
        ret_url = self.absolute_url()
        REQUEST.RESPONSE.redirect("%s?portal_status_message=%s"
                                  % (ret_url, 'No repository was defined'))
                                    
      for repository in repository_list:
        repository = repository.rstrip('\n')
        repository = repository.rstrip('\r')
        for business_template_id in os.listdir(repository):
          template_path = os.path.join(repository, business_template_id)
          if os.path.isfile(template_path):
            LOG(business_template_id,0,'is file, so it is skipped')
          else:
            if not os.path.exists((os.path.join(template_path, 'bt'))):
              LOG(business_template_id,0,'has no bt sub-folder, so it is skipped')
            else:
587 588
              self.activate(activity='SQLQueue').\
                importAndReExportBusinessTemplateFromPath(template_path)
589

590 591
    security.declareProtected( 'Import/Export objects', 'migrateBTToBM')
    def migrateBTToBM(self, template_path, REQUEST=None, **kw):
592
      """
593 594
        Migrate business template repository to Business Manager repo.
        Business Manager completely rely only on BusinessItem and to show
595 596 597 598
        the difference between both of them

        So, the steps should be:
        1. Install the business template which is going to be migrated
599
        2. Create a new Business Manager with random id and title
600 601
        3. Add the path, build and export the template
        4. Remove the business template from the directory and add the business
602 603 604
        manager there instead
        5. Change the ID and title of the business manager
        6. Export the business manager to the directory, leaving anything in
605
        the installed erp5 unchanged
606 607
      """
      import_template = self.download(url=template_path)
608
      if import_template.getPortalType() == 'Business Manager':
609 610 611
        LOG(import_template.getTitle(),0,'Already migrated')
        return

612 613
      export_dir = mkdtemp()

614
      installed_bt_list = self.getInstalledBusinessTemplatesList()
615 616
      installed_bt_title_list = [bt.title for bt in installed_bt_list]

617
      is_installed = False
618 619 620
      if import_template.getTitle() not in installed_bt_title_list:
        # Install the business template
        import_template.install(**kw)
621
        is_installed = True
622

623
      # Make list of object paths which needs to be added in the bm5
624 625 626 627 628 629 630 631 632
      # This can be decided by looping over every type of items we do have in
      # bt5 and checking if there have been any changes being made to it via this
      # bt5 installation or not.
      # For ex:
      # CatalogTempalteMethodItem, CatalogResultsKeyItem, etc. do make some
      # changes in erp5_mysql_innodb(by adding properties, by adding sub-objects),
      # so we need to add portal_catalog/erp5_mysql_innodb everytime we find
      # a bt5 making changes in any of these items.

633 634
      portal_path = self.getPortalObject()
      template_path_list = []
635
      property_path_list = []
636 637

      # For modules, we don't need to create path for the module
638
      module_list = import_template.getTemplateModuleIdList()
639 640
      for path in module_list:
        template_path_list.append(path + ' | 1 | 1')
641

642
      # For portal_types, we have to add path and subobjects
643 644
      portal_type_id_list = import_template.getTemplatePortalTypeIdList()
      portal_type_path_list = []
645
      for id in portal_type_id_list:
646 647
        portal_type_path_list.append('portal_types/'+id + ' | 1 | 1')
        #portal_type_path_list.append('portal_types/'+id+'/**')
648 649
      template_path_list.extend(portal_type_path_list)

650
      # For categories, we create path for category objects as well as the subcategories
651
      category_list = import_template.getTemplateBaseCategoryList()
652 653
      category_path_list = []
      for base_category in category_list:
654 655
        category_path_list.append('portal_categories/'+base_category + ' | 1 | 1')
        #category_path_list.append('portal_categories/'+base_category+'/**')
656 657
      template_path_list.extend(category_path_list)

658 659 660 661 662 663 664
      # Adding business template skin selection property on the portal_tempaltes
      template_skin_selection_list = import_template.getTemplateRegisteredSkinSelectionList()
      selection_list = []
      for selection in template_skin_selection_list:
        skin, selection = selection.split(' | ')
        selection_list.append('portal_skins/%s#business_template_registered_skin_selections'%skin)

665
      # For portal_skins, we export the folder
666
      portal_skin_list = import_template.getTemplateSkinIdList()
667 668
      portal_skin_path_list = []
      for skin in portal_skin_list:
669 670
        portal_skin_path_list.append('portal_skins/'+skin + ' | 1 | 1')
        #portal_skin_path_list.append('portal_skins/'+skin+'/**')
671 672 673 674 675 676 677 678 679 680 681 682
      template_path_list.extend(portal_skin_path_list)

      # For workflow chains,
      # We have 2 objects in the Business Template design where we deal with
      # workflow objects, we deal with the installation separately:
      # 1. Workflow_id : We export the whole workflow objects in this case
      # 2. Portal Workflow chain: It is already being exported via portal_types
      # XXX: CHECK For 2, keeping in mind the migration of workflow would be merged
      # before this part where we make workflow_list as property of portal_type
      workflow_id_list = import_template.getTemplateWorkflowIdList()
      workflow_path_list = []
      for workflow in workflow_id_list:
683 684
        workflow_path_list.append('portal_workflow/' + workflow + ' | 1 | 1')
        #workflow_path_list.append('portal_workflow/' + workflow + '/**')
685 686 687
      template_path_list.extend(workflow_path_list)

      # For paths, we add them directly to the path list
688 689 690
      path_list = import_template.getTemplatePathList()
      for path in path_list:
        template_path_list.append(path  + ' | 1 | 1')
691

692
      # Catalog methods would be added as sub objects
693 694 695
      catalog_method_item_list = import_template.getTemplateCatalogMethodIdList()
      catalog_method_path_list = []
      for method in catalog_method_item_list:
696
        catalog_method_path_list.append('portal_catalog/' + method + ' | 1 | 1')
697
      template_path_list.extend(catalog_method_path_list)
698 699

      # For catalog objects, we check if there is any catalog object, and then
700
      # add catalog object also in the path if there is
701 702 703 704
      template_catalog_datetime_key   = import_template.getTemplateCatalogDatetimeKeyList()
      template_catalog_full_text_key  = import_template.getTemplateCatalogFullTextKeyList()
      template_catalog_keyword_key    = import_template.getTemplateCatalogKeywordKeyList()
      template_catalog_local_role_key = import_template.getTemplateCatalogLocalRoleKeyList()
705
      template_catalog_method_id      = import_template.getTemplateCatalogMethodIdList()
706 707 708 709 710 711 712 713 714 715 716
      template_catalog_multivalue_key = import_template.getTemplateCatalogMultivalueKeyList()
      template_catalog_related_key    = import_template.getTemplateCatalogRelatedKeyList()
      template_catalog_request_key    = import_template.getTemplateCatalogRequestKeyList()
      template_catalog_result_key     = import_template.getTemplateCatalogResultKeyList()
      template_catalog_result_table   = import_template.getTemplateCatalogResultTableList()
      template_catalog_role_key       = import_template.getTemplateCatalogRoleKeyList()
      template_catalog_scriptable_key = import_template.getTemplateCatalogScriptableKeyList()
      template_catalog_search_key     = import_template.getTemplateCatalogSearchKeyList()
      template_catalog_security_uid_column = import_template.getTemplateCatalogSecurityUidColumnList()
      template_catalog_topic_key      = import_template.getTemplateCatalogTopicKeyList()

717 718 719 720 721
      catalog_property_list = [
        template_catalog_datetime_key,
        template_catalog_full_text_key,
        template_catalog_keyword_key,
        template_catalog_local_role_key,
722
        template_catalog_method_id,
723 724 725 726 727 728 729 730 731 732 733 734 735 736 737 738 739 740
        template_catalog_multivalue_key,
        template_catalog_related_key,
        template_catalog_request_key,
        template_catalog_result_key,
        template_catalog_result_table,
        template_catalog_role_key,
        template_catalog_scriptable_key,
        template_catalog_search_key,
        template_catalog_security_uid_column,
        template_catalog_topic_key,
        ]
      is_property_added = any(catalog_property_list)

      properties_removed = [
        'sql_catalog_datetime_search_keys_list',
        'sql_catalog_full_text_search_keys_list',
        'sql_catalog_keyword_search_keys_list',
        'sql_catalog_local_role_keys_list',
741
        'sql_catalog_method_id_list',
742 743 744 745 746 747 748 749 750
        'sql_catalog_multivalue_keys_list',
        'sql_catalog_related_keys_list',
        'sql_catalog_request_keys_list',
        'sql_search_result_keys_list',
        'sql_search_tables_list',
        'sql_catalog_role_keys_list',
        'sql_catalog_scriptable_keys_list',
        'sql_catalog_search_keys_list',
        'sql_catalog_security_uid_columns_list',
751
        'sql_catalog_topic_search_keys_list',
752 753 754
        ]

      removable_property = {}
755
      removable_sub_object_path = []
756 757 758 759 760 761

      if is_property_added:
        if catalog_method_path_list:
          catalog_path = catalog_method_path_list[0].rsplit('/', 1)[0]
        else:
          catalog_path = 'portal_catalog/erp5_mysql_innodb'
762 763
          removable_sub_object_path.append(catalog_path)
        template_path_list.append(catalog_path + ' | 1 | 1')
764 765
        removable_property[catalog_path] = properties_removed
        for prop in properties_removed:
766
            property_path_list.append('%s#%s | 1 | 1' % (catalog_path, prop))
767

768 769 770
      # Add these catalog items in the object_property instead of adding
      # dummy path item for them
      if import_template.getTitle() == 'erp5_mysql_innodb_catalog':
771
        template_path_list.extend('portal_catalog/erp5_mysql_innodb | 1 | 1')
772

773 774 775 776
      # Add portal_property_sheets
      property_sheet_id_list = import_template.getTemplatePropertySheetIdList()
      property_sheet_path_list = []
      for property_sheet in property_sheet_id_list:
777 778
        property_sheet_path_list.append('portal_property_sheets/' + property_sheet + ' | 1 | 1')
        #property_sheet_path_list.append('portal_property_sheets/' + property_sheet + '/**')
779
      template_path_list.extend(property_sheet_path_list)
780

781
      # Create new objects for business package
782 783
      migrated_bm = self.newContent(
                                    portal_type='Business Manager',
784
                                    title=import_template.getTitle()
785
                                    )
786

787
      template_path_list.extend(property_path_list)
788
      template_path_list.extend(selection_list)
789 790
      template_path_list = self.cleanTemplatePathList(template_path_list)
      migrated_bm.setProperty('template_path_list', template_path_list)
791 792

      kw['removable_property'] = removable_property
793 794 795 796
      kw['removable_sub_object_path'] = removable_sub_object_path

      migrated_bm.build(**kw)

797
      # Export the newly built business package to the export directory
798 799
      migrated_bm.export(path=export_dir, local=True)

800
      if is_installed:
801
        import_template.uninstall()
802

803 804 805 806 807 808 809 810 811 812 813 814 815 816 817 818 819 820 821 822 823 824 825 826 827 828 829 830 831 832 833
    def cleanTemplatePathList(self, path_list):
      """
      Remove redundant paths and sub-objects' path if the object path already
      exist.
      """
      # Remove layer and sign
      a1 = [l.split(' | ')[0] for l in path_list]
      # Split path into list
      a2 = [l.split('/') for l in a1]
      # Create new list for paths with **
      a3 = [l for l in a2 if l[-1] == '**']
      # Create new list for paths without **
      a4 = [l for l in a2 if l[-1] != '**']
      # Remove ** from paths in a3
      reserved_id = ('portal_transforms', 'portal_ids')
      a3 = [l[:-1] for l in a3 if l[0] not in reserved_id] + [l for l in a3 if l[0] in reserved_id]
      # Create new final path list
      a2 = a3+a4
      # Join the path list
      a2 = [('/').join(l) for l in a2]
      # Remove the redundant paths
      a2 = list(set(a2))
      # Sort the path list
      a2.sort()
      # Add the layer and signs, for now all 1
      a2 = [l+' | 1 | 1' for l in a2]

      return a2

    security.declareProtected( 'Import/Export objects', 'migrateBTListToBM')
    def migrateBTListToBM(self, repository_list, REQUEST=None, **kw):
834 835 836 837 838 839 840 841 842 843 844 845 846 847 848 849 850 851 852 853 854 855 856 857 858
      """
      Run migration for BT5 one by one in a given repository. This will be done
      via activities.
      """
      repository_list = filter(bool, repository_list)

      if REQUEST is None:
        REQUEST = getattr(self, 'REQUEST', None)

      if len(repository_list) == 0 and REQUEST:
        ret_url = self.absolute_url()
        REQUEST.RESPONSE.redirect("%s?portal_status_message=%s"
                                  % (ret_url, 'No repository was defined'))

      for repository in repository_list:
        repository = repository.rstrip('\n')
        repository = repository.rstrip('\r')
        for business_template_id in os.listdir(repository):
          template_path = os.path.join(repository, business_template_id)
          if os.path.isfile(template_path):
            LOG(business_template_id,0,'is file, so it is skipped')
          else:
            if not os.path.exists((os.path.join(template_path, 'bt'))):
              LOG(business_template_id,0,'has no bt sub-folder, so it is skipped')
            else:
859
              self.migrateBTToBM(template_path)
860 861 862
              #self.activate(activity='SQLQueue').\
              #  migrateBTToBP(template_path)

863
    security.declareProtected(Permissions.ManagePortal, 'getFilteredDiff')
864 865 866 867 868 869
    def getFilteredDiff(self, diff):
      """
      Filter the diff using python scripts
      """
      diff_file_object = DiffFile(diff)
      diff_block_list = diff_file_object.getModifiedBlockList()
870 871 872 873
      if diff_block_list:
        script_list = self.getDiffFilterScriptList()
        for block, line_tuple in diff_block_list:
          for script in script_list:
874 875
            if script(line_tuple[0], line_tuple[1]):
              diff_file_object.children.remove(block)
876
              break
877 878 879 880
      # XXX-Aurel : this method should return a text diff but
      # DiffFile does not provide yet such feature
      return diff_file_object

881
    security.declareProtected(Permissions.ManagePortal, 'diffObjectAsHTML')
882 883 884
    def diffObjectAsHTML(self, REQUEST, **kw):
      """
        Convert diff into a HTML format before reply
885
        This is compatible with ERP5VCS look and feel but
886 887 888 889
        it is preferred in future we use more difflib python library.
      """
      return DiffFile(self.diffObject(REQUEST, **kw)).toHTML()

890
    security.declareProtected(Permissions.ManagePortal, 'diffObject')
891
    def diffObject(self, REQUEST, **kw):
Aurel's avatar
Aurel committed
892
      """
Vincent Pelletier's avatar
Vincent Pelletier committed
893 894
        Make diff between two objects, whose paths are stored in values bt1
        and bt2 in the REQUEST object.
Aurel's avatar
Aurel committed
895
      """
896 897
      bt1_id = getattr(REQUEST, 'bt1', None)
      bt2_id = getattr(REQUEST, 'bt2', None)
898 899 900 901 902 903 904
      if bt1_id is not None and bt2_id is not None:
        bt1 = self._getOb(bt1_id)
        bt2 = self._getOb(bt2_id)
        if self.compareVersions(bt1.getVersion(), bt2.getVersion()) < 0:
          return bt2.diffObject(REQUEST, compare_with=bt1_id)
        else:
          return bt1.diffObject(REQUEST, compare_with=bt2_id)
Aurel's avatar
Aurel committed
905
      else:
906 907 908 909 910
        object_id = getattr(REQUEST, 'object_id', None)
        bt1_id = object_id.split('|')[0]
        bt1 = self._getOb(bt1_id)
        REQUEST.set('object_id', object_id.split('|')[1])
        return bt1.diffObject(REQUEST)
911

Vincent Pelletier's avatar
Vincent Pelletier committed
912 913 914 915
    security.declareProtected( 'Import/Export objects',
                               'updateRepositoryBusinessTemplateList' )

    def updateRepositoryBusinessTemplateList(self, repository_list,
916
        REQUEST=None, RESPONSE=None, genbt5list=0, **kw):
Vincent Pelletier's avatar
Vincent Pelletier committed
917 918
      """
        Update the information on Business Templates from repositories.
919 920 921

      For local repositories, if bt5list is missing or if genbt5list > 1,
      bt5list is automatically generated (but not saved on disk).
922 923
      """
      self.repository_dict = PersistentMapping()
924
      property_list = ('title', 'version', 'revision', 'description', 'license',
925 926
                       'dependency', 'test_dependency', 'provision', 'copyright',
                       'force_install')
Vincent Pelletier's avatar
Vincent Pelletier committed
927 928
      #LOG('updateRepositoryBusiessTemplateList', 0,
      #    'repository_list = %r' % (repository_list,))
929
      for repository in repository_list:
930 931 932 933 934 935 936 937 938 939 940 941 942
        urltype, url = splittype(repository)
        if WIN and urltype and '\\' in url:
          urltype = None
          url = repository
        if urltype and urltype != 'file':
          f = urlopen(repository + '/bt5list')
        else:
          bt5list = os.path.join(url, 'bt5list')
          if genbt5list > os.path.exists(bt5list):
            f = generateInformation(url)
            f.seek(0)
          else:
            f = open(bt5list, 'rb')
943
        try:
944 945 946 947 948 949 950 951 952 953 954
          try:
            doc = parse(f)
          except ExpatError:
            if REQUEST is not None:
              psm = translateString('Invalid repository: ${repo}',
                                    mapping={'repo':repository})
              REQUEST.RESPONSE.redirect("%s?portal_status_message=%s"
                                       % (self.absolute_url(), psm))
              return
            else:
              raise RuntimeError, 'Invalid repository: %s' % repository
955
          try:
956
            property_dict_list = []
957 958 959 960 961 962 963 964 965 966 967 968 969 970 971 972 973 974 975 976
            root = doc.documentElement
            for template in root.getElementsByTagName("template"):
              id = template.getAttribute('id')
              if type(id) == type(u''):
                id = id.encode('utf-8')
              temp_property_dict = {}
              for node in template.childNodes:
                if node.nodeName in property_list:
                  value = ''
                  for text in node.childNodes:
                    if text.nodeType == text.TEXT_NODE:
                      value = text.data
                      if type(value) == type(u''):
                        value = value.encode('utf-8')
                      break
                  temp_property_dict.setdefault(node.nodeName, []).append(value)

              property_dict = {}
              property_dict['id'] = id
              property_dict['title'] = temp_property_dict.get('title', [''])[0]
Vincent Pelletier's avatar
Vincent Pelletier committed
977 978
              property_dict['version'] = \
                  temp_property_dict.get('version', [''])[0]
Jérome Perrin's avatar
Jérome Perrin committed
979 980
              property_dict['revision'] = \
                  temp_property_dict.get('revision', [''])[0]
Vincent Pelletier's avatar
Vincent Pelletier committed
981 982 983 984 985 986
              property_dict['description'] = \
                  temp_property_dict.get('description', [''])[0]
              property_dict['license'] = \
                  temp_property_dict.get('license', [''])[0]
              property_dict['dependency_list'] = \
                  temp_property_dict.get('dependency', ())
987 988
              property_dict['test_dependency_list'] = \
                  temp_property_dict.get('test_dependency', ())
989 990
              property_dict['provision_list'] = \
                  temp_property_dict.get('provision', ())
Vincent Pelletier's avatar
Vincent Pelletier committed
991 992
              property_dict['copyright_list'] = \
                  temp_property_dict.get('copyright', ())
993 994
              property_dict['force_install'] = \
                  int(temp_property_dict.get('force_install', [0])[0])
995

996 997 998 999 1000
              property_dict_list.append(property_dict)
          finally:
            doc.unlink()
        finally:
          f.close()
1001

1002
        #XXX: Hardcoding 'erp5_core_proxy_field_legacy' BP in the list
1003 1004 1005 1006 1007
        bp_dict ={
          'copyright_list': ['Copyright (c) 2001-2017 Nexedi SA'],
          'dependency_list': [],
          'description': '',
          'force_install': 0,
1008
          'id': 'erp5_core_proxy_field_legacy',
1009 1010 1011
          'license': 'GPL',
          'revision': '',
          'test_dependency_list': [],
1012 1013
          'provision_list': [],
          'title': 'erp5_core_proxy_field_legacy',
1014 1015 1016 1017
          'version': '1.0'}
        if repository.endswith('/bt5'):
          property_dict_list.append(bp_dict)

1018
        bm_dict_1 ={
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029
          'copyright_list': ['Copyright (c) 2001-2017 Nexedi SA'],
          'dependency_list': [],
          'description': '',
          'force_install': 0,
          'id': 'erp5_mysql_innodb_catalog',
          'license': 'GPL',
          'revision': '',
          'test_dependency_list': [],
          'provision_list': ['erp5_catalog'],
          'title': 'erp5_mysql_innodb_catalog',
          'version': '1.0'}
1030 1031 1032 1033 1034 1035 1036 1037 1038 1039 1040 1041 1042 1043

        bm_dict_2 ={
          'copyright_list': ['Copyright (c) 2001-2017 Nexedi SA'],
          'dependency_list': ['erp5_core'],
          'description': '',
          'force_install': 0,
          'id': 'erp5_xhtml_style',
          'license': 'GPL',
          'revision': '',
          'test_dependency_list': [],
          'provision_list': ['erp5_view_style'],
          'title': 'erp5_xhtml_style',
          'version': '1.0'}

1044 1045 1046 1047 1048 1049 1050 1051 1052 1053 1054 1055 1056
        bm_dict_3 ={
          'copyright_list': ['Copyright (c) 2001-2017 Nexedi SA'],
          'dependency_list': [],
          'description': '',
          'force_install': 0,
          'id': 'erp5_mysql_ndb_catalog',
          'license': 'GPL',
          'revision': '',
          'test_dependency_list': [],
          'provision_list': ['erp5_catalog'],
          'title': 'erp5_mysql_ndb_catalog',
          'version': '1.0'}

1057
        if repository.endswith('/bootstrap'):
1058
          property_dict_list.append(bm_dict_1)
1059 1060
          property_dict_list.append(bm_dict_2)
          property_dict_list.append(bm_dict_3)
1061

1062
        self.repository_dict[repository] = tuple(property_dict_list)
1063

1064
      if REQUEST is not None:
1065
        ret_url = self.absolute_url() + '/' + REQUEST.get('dialog_id', 'view')
Yusei Tahara's avatar
Yusei Tahara committed
1066
        psm = translateString("Business templates updated successfully.")
1067 1068
        REQUEST.RESPONSE.redirect("%s?cancel_url=%s&portal_status_message=%s&dialog_category=object_exchange&selection_name=business_template_selection"
                                  % (ret_url, REQUEST.form.get('cancel_url', ''), psm))
1069

Vincent Pelletier's avatar
Vincent Pelletier committed
1070 1071
    security.declareProtected( Permissions.AccessContentsInformation,
                               'getRepositoryList' )
1072
    def getRepositoryList(self):
Vincent Pelletier's avatar
Vincent Pelletier committed
1073 1074
      """
        Get the list of repositories.
1075 1076
      """
      return self.repository_dict.keys()
1077

1078 1079
    security.declarePublic( 'decodeRepositoryBusinessTemplateUid' )
    def decodeRepositoryBusinessTemplateUid(self, uid):
Vincent Pelletier's avatar
Vincent Pelletier committed
1080 1081 1082
      """
        Decode the uid of a business template from a repository.
        Return a repository and an id.
1083
      """
1084
      return cPickle.loads(b64decode(uid))
1085

1086 1087 1088 1089 1090 1091 1092 1093
    security.declarePublic( 'encodeRepositoryBusinessTemplateUid' )
    def encodeRepositoryBusinessTemplateUid(self, repository, id):
      """
        encode the repository and the id of a business template.
        Return an uid.
      """
      return b64encode(cPickle.dumps((repository, id)))

1094
    security.declarePublic('compareVersionStrings')
1095 1096 1097 1098 1099 1100 1101 1102 1103 1104 1105 1106 1107 1108 1109 1110 1111 1112 1113 1114 1115 1116 1117 1118 1119 1120 1121 1122
    def compareVersionStrings(self, version, comparing_string):
      """
       comparing_string is like "<= 0.2" | "operator version"
       operators supported: '<=', '<' or '<<', '>' or '>>', '>=', '=' or '=='
      """
      operator, comp_version = comparing_string.split(' ')
      diff_version = self.compareVersions(version, comp_version)
      if operator == '<' or operator == '<<':
        if diff_version < 0:
          return True;
        return False;
      if operator == '<=':
        if diff_version <= 0:
          return True;
        return False;
      if operator == '>' or operator == '>>':
        if diff_version > 0:
          return True;
        return False;
      if operator == '>=':
        if diff_version >= 0:
          return True;
        return False;
      if operator == '=' or operator == '==':
        if diff_version == 0:
          return True;
        return False;
      raise UnsupportedComparingOperator, 'Unsupported comparing operator: %s'%(operator,)
1123

1124 1125 1126 1127 1128 1129 1130 1131 1132 1133 1134 1135 1136 1137
    security.declareProtected(Permissions.AccessContentsInformation,
                              'IsOneProviderInstalled')
    def IsOneProviderInstalled(self, title):
      """
        return true if a business template that
        provides the bt with the given title is
        installed
      """
      installed_bt_list = self.getInstalledBusinessTemplatesList()
      for bt in installed_bt_list:
        provision_list = bt.getProvisionList()
        if title in provision_list:
          return True
      return False
1138

1139 1140 1141 1142 1143
    security.declareProtected(Permissions.AccessContentsInformation,
                               'getLastestBTOnRepos')
    def getLastestBTOnRepos(self, title, version_restriction=None):
      """
       It's possible we have different versions of the same BT
1144
       available on various repositories or on the same repository.
1145 1146 1147 1148 1149 1150
       This function returns the latest one that meet the version_restriction
       (i.e "<= 0.2") in the following form :
       tuple (repository, id)
      """
      result = None
      for repository, property_dict_list in self.repository_dict.items():
Jérome Perrin's avatar
Jérome Perrin committed
1151
        for property_dict in property_dict_list:
1152 1153 1154
          provision_list = property_dict.get('provision_list', [])
          if title in provision_list:
            raise BusinessTemplateIsMeta, 'Business Template %s is provided by another one'%(title,)
Jérome Perrin's avatar
Jérome Perrin committed
1155
          if title == property_dict['title']:
1156 1157
            if (version_restriction is None) or (self.compareVersionStrings(property_dict['version'], version_restriction)):
              if (result is None) or (self.compareVersions(property_dict['version'], result[2]) > 0):
Rafael Monnerat's avatar
Rafael Monnerat committed
1158
                result = (repository, property_dict['id'], property_dict['version'])
1159 1160 1161 1162
      if result is not None:
        return (result[0], result[1])
      else:
        raise BusinessTemplateUnknownError, 'Business Template %s (%s) could not be found in the repositories'%(title, version_restriction or '')
1163

1164 1165 1166 1167 1168 1169 1170 1171 1172 1173 1174 1175 1176 1177
    security.declareProtected(Permissions.AccessContentsInformation,
                              'getProviderList')
    def getProviderList(self, title):
      """
       return a list of business templates that provides
       the given business template
      """
      result_list = []
      for repository, property_dict_list in self.repository_dict.items():
        for property_dict in property_dict_list:
          provision_list = property_dict['provision_list']
          if (title in provision_list) and (property_dict['title'] not in result_list):
            result_list.append(property_dict['title'])
      return result_list
1178

1179 1180
    security.declareProtected(Permissions.AccessContentsInformation,
                               'getDependencyList')
1181 1182 1183
    @transactional_cached(lambda self, bt, with_test_dependency_list=False:
                          (bt, with_test_dependency_list))
    def getDependencyList(self, bt, with_test_dependency_list=False):
1184 1185 1186 1187
      """
       Return the list of missing dependencies for a business
       template, given a tuple : (repository, id)
      """
1188 1189 1190
      # We do not take into consideration the dependencies
      # for meta business templates
      if bt[0] != 'meta':
1191 1192 1193 1194 1195
        result_list = []
        for repository, property_dict_list in self.repository_dict.items():
          if repository == bt[0]:
            for property_dict in property_dict_list:
              if property_dict['id'] == bt[1]:
1196 1197 1198 1199 1200
                dependency_list = [q.strip() for q in
                                   property_dict['dependency_list'] if q]
                if with_test_dependency_list:
                  dependency_list.extend([q.strip() for q in
                                          property_dict['test_dependency_list'] if q])
1201 1202 1203 1204 1205 1206 1207 1208 1209 1210 1211 1212 1213 1214 1215 1216 1217 1218 1219 1220 1221 1222 1223
                for dependency_couple in dependency_list:
                  # dependency_couple is like "erp5_xhtml_style (>= 0.2)"
                  dependency_couple_list = dependency_couple.split(' ', 1)
                  dependency = dependency_couple_list[0]
                  version_restriction = None
                  if len(dependency_couple_list) > 1:
                    version_restriction = dependency_couple_list[1]
                    if version_restriction.startswith('('):
                      # Something like "(>= 1.0rc6)".
                      version_restriction = version_restriction[1:-1]
                  require_update = False
                  if dependency not in result_list:
                    # Get the lastest version of the dependency on the
                    # repository that meet the version restriction
                    provider_installed = False
                    bt_dep = None
                    try:
                      bt_dep = self.getLastestBTOnRepos(dependency, version_restriction)
                    except BusinessTemplateUnknownError:
                      raise BusinessTemplateMissingDependency, 'While analysing %s the following dependency could not be satisfied: %s (%s)\nReason: Business Template could not be found in the repositories'%(bt[1], dependency, version_restriction or '')
                    except BusinessTemplateIsMeta:
                      provider_list = self.getProviderList(dependency)
                      for provider in provider_list:
1224
                        if self.getInstalledBusinessTemplate(provider) is not None:
1225 1226 1227 1228 1229 1230 1231 1232 1233 1234 1235
                          bt_dep = self.getLastestBTOnRepos(provider)
                          break
                      if bt_dep is None:
                        bt_dep = ('meta', dependency)
                    sub_dep_list = self.getDependencyList(bt_dep)
                    for sub_dep in sub_dep_list:
                      if sub_dep not in result_list:
                        result_list.append(sub_dep)
                    result_list.append(bt_dep)
                return result_list
        raise BusinessTemplateUnknownError, 'The Business Template %s could not be found on repository %s'%(bt[1], bt[0])
1236
      return []
1237

1238 1239
    security.declareProtected(Permissions.ManagePortal,
                              'findProviderInBTList')
1240 1241 1242 1243 1244 1245 1246 1247 1248 1249 1250
    def findProviderInBTList(self, provider_list, bt_list):
      """
       Find one provider in provider_list which is present in
       bt_list and returns the found tuple (repository, id)
       in bt_list.
      """
      for provider in provider_list:
        for repository, id in bt_list:
          if id.startswith(provider):
            return (repository, id)
      raise BusinessTemplateUnknownError, 'Provider not found in bt_list'
1251

1252 1253 1254 1255
    security.declareProtected(Permissions.AccessContentsInformation,
                              'sortBusinessTemplateList')
    def sortBusinessTemplateList(self, bt_list):
      """
1256 1257 1258 1259 1260 1261
      Sort a list of business template in repositories according to
      dependencies

      bt_list : list of (repository, id) tuple.
      """
      sorted_bt_list = []
1262
      title_id_mapping = {}
1263 1264 1265 1266 1267 1268 1269

      # Calculate the dependency graph
      dependency_dict = {}
      provition_dict = {}
      repository_dict = {}
      undependent_list = []

1270 1271 1272
      for repository, bt_id in bt_list:
        bt = [x for x in self.repository_dict[repository] \
              if x['id'] == bt_id][0]
1273 1274 1275 1276 1277 1278
        bt_title = bt['title']
        repository_dict[bt_title] = repository
        dependency_dict[bt_title] = [x.split(' ')[0] for x in bt['dependency_list']]
        title_id_mapping[bt_title] = bt_id
        if not dependency_dict[bt_title]:
          del dependency_dict[bt_title]
1279
        for provision in list(bt['provision_list']):
1280 1281
          provition_dict[provision] = bt_title
        undependent_list.append(bt_title)
1282 1283 1284 1285 1286 1287 1288 1289 1290 1291 1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309 1310

      # Calculate the reverse dependency graph
      reverse_dependency_dict = {}
      for bt_id, dependency_id_list in dependency_dict.items():
        update_dependency_id_list = []
        for dependency_id in dependency_id_list:

          # Get ride of provision id
          if dependency_id in provition_dict:
            dependency_id = provition_dict[dependency_id]
          update_dependency_id_list.append(dependency_id)

          # Fill incoming edge dict
          if dependency_id in reverse_dependency_dict:
            reverse_dependency_dict[dependency_id].append(bt_id)
          else:
            reverse_dependency_dict[dependency_id] = [bt_id]

          # Remove from free node list
          try:
            undependent_list.remove(dependency_id)
          except ValueError:
            pass

        dependency_dict[bt_id] = update_dependency_id_list

      # Let's sort the bt5!
      while undependent_list:
        bt_id = undependent_list.pop(0)
1311 1312 1313
        if bt_id not in repository_dict:
          continue
        sorted_bt_list.insert(0, (repository_dict[bt_id], title_id_mapping[bt_id]))
1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328
        for dependency_id in dependency_dict.get(bt_id, []):

          local_dependency_list = reverse_dependency_dict[dependency_id]
          local_dependency_list.remove(bt_id)
          if local_dependency_list:
            reverse_dependency_dict[dependency_id] = local_dependency_list
          else:
            del reverse_dependency_dict[dependency_id]
            undependent_list.append(dependency_id)

      if len(sorted_bt_list) != len(bt_list):
        raise NotImplementedError, \
          "Circular dependencies on %s" % reverse_dependency_dict.keys()
      else:
        return sorted_bt_list
1329

1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341 1342 1343 1344 1345 1346 1347 1348 1349 1350 1351 1352 1353 1354 1355 1356 1357 1358 1359
    security.declareProtected(Permissions.AccessContentsInformation,
                              'sortDownloadedBusinessTemplateList')
    def sortDownloadedBusinessTemplateList(self, id_list):
      """
      Sort a list of already downloaded business templates according to
      dependencies

      id_list : list of business template's id in portal_templates.
      """
      def isDepend(a, b):
        # return True if a depends on b.
        dependency_list = [x.split(' ')[0] for x in a.getDependencyList()]
        provision_list = list(b.getProvisionList()) + [b.getTitle()]
        for i in provision_list:
          if i in dependency_list:
            return True
          return False

      sorted_bt_list = []
      for bt_id in id_list:
        bt = self._getOb(bt_id)
        for j in range(len(sorted_bt_list)):
          if isDepend(sorted_bt_list[j], bt):
            sorted_bt_list.insert(j, bt)
            break
        else:
           sorted_bt_list.append(bt)
      sorted_bt_list = [bt.getId() for bt in sorted_bt_list]
      return sorted_bt_list

Vincent Pelletier's avatar
Vincent Pelletier committed
1360 1361
    security.declareProtected( Permissions.AccessContentsInformation,
                               'getRepositoryBusinessTemplateList' )
1362
    def getRepositoryBusinessTemplateList(self, update_only=False,
1363
             template_list=None, **kw):
1364
      """Get the list of Business Templates in repositories.
1365 1366 1367

         update_only: return only bt that needs to be updated
         template_list: only returns bt within the given list
1368 1369
      """
      from Products.ERP5Type.Document import newTempBusinessTemplate
1370 1371 1372 1373
      result_list = []
      template_set = None
      if template_list is not None:
        template_set = set(template_list)
1374 1375

      template_item_list = []
1376 1377 1378 1379 1380 1381 1382 1383 1384 1385
      # First of all, filter Business Templates in repositories.
      template_item_dict = {}
      for repository, property_dict_list in self.repository_dict.items():
        for property_dict in property_dict_list:
          title = property_dict['title']
          if template_set and not(title in template_set):
            continue
          if not update_only:
            template_item_list.append((repository, property_dict))
          else:
1386
            if title not in template_item_dict:
Vincent Pelletier's avatar
Vincent Pelletier committed
1387 1388
              # If this is the first time to see this business template,
              # insert it.
1389 1390
              template_item_dict[title] = (repository, property_dict)
            else:
Vincent Pelletier's avatar
Vincent Pelletier committed
1391 1392 1393 1394
              # If this business template has been seen before, insert it only
              # if this business template is newer.
              previous_repository, previous_property_dict = \
                  template_item_dict[title]
1395 1396
              if self.compareVersions(previous_property_dict['version'],
                                      property_dict['version']) < 0:
1397
                template_item_dict[title] = (repository, property_dict)
1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409
      # Next, select only updated business templates.
      if update_only:
        for repository, property_dict in template_item_dict.values():
          installed_bt = \
              self.getInstalledBusinessTemplate(property_dict['title'], strict=True)
          if installed_bt is not None:
            diff_version = self.compareVersions(installed_bt.getVersion(),
                                                property_dict['version'])
            if diff_version < 0:
              template_item_list.append((repository, property_dict))
            elif diff_version == 0 \
                  and property_dict['revision'] \
1410
                  and installed_bt.getRevision() != property_dict['revision']:
1411 1412
                    template_item_list.append((repository, property_dict))
          elif template_list is not None:
1413 1414 1415 1416 1417
            template_item_list.append((repository, property_dict))

      # Create temporary Business Template objects for displaying.
      for repository, property_dict in template_item_list:
        property_dict = property_dict.copy()
1418
        id = filename = property_dict.pop('id')
1419 1420 1421 1422
        installed_bt = \
            self.getInstalledBusinessTemplate(property_dict['title'])
        if installed_bt is not None:
          installed_version = installed_bt.getVersion()
1423 1424
          installed_revision = installed_bt.getShortRevision()
          if installed_bt.getRevision() == property_dict['revision']:
1425
            version_state = 'present'
1426 1427
          else:
            version_state = 'different'
1428 1429 1430
        else:
          installed_version = ''
          installed_revision = ''
1431
          version_state = 'new'
1432
        uid = self.encodeRepositoryBusinessTemplateUid(repository, id)
1433 1434
        obj = newTempBusinessTemplate(self, 'temp_' + uid,
                                      version_state = version_state,
1435
                                      version_state_title=version_state.title(),
1436
                                      filename = filename,
1437 1438
                                      installed_version = installed_version,
                                      installed_revision = installed_revision,
1439 1440
                                      repository = repository, **property_dict)
        obj.setUid(uid)
1441 1442 1443
        result_list.append(obj)
      result_list.sort(key=lambda x: x.getTitle())
      return result_list
1444

Vincent Pelletier's avatar
Vincent Pelletier committed
1445 1446
    security.declareProtected( Permissions.AccessContentsInformation,
                               'getUpdatedRepositoryBusinessTemplateList' )
1447 1448 1449 1450
    def getUpdatedRepositoryBusinessTemplateList(self, **kw):
      """Get the list of updated Business Templates in repositories.
      """
      #LOG('getUpdatedRepositoryBusinessTemplateList', 0, 'kw = %r' % (kw,))
1451
      return self.getRepositoryBusinessTemplateList(update_only=True, **kw)
1452

1453
    security.declarePublic('compareVersions')
1454
    def compareVersions(self, version1, version2):
Vincent Pelletier's avatar
Vincent Pelletier committed
1455 1456 1457
      """
        Return negative if version1 < version2, 0 if version1 == version2,
        positive if version1 > version2.
1458 1459

      Here is the algorithm:
Vincent Pelletier's avatar
Vincent Pelletier committed
1460 1461
        - Non-alphanumeric characters are not significant, besides the function
          of delimiters.
1462 1463 1464 1465
        - If a level of a version number is missing, it is assumed to be zero.
        - An alphabetical character is less than any numerical value.
        - Numerical values are compared as integers.

Vincent Pelletier's avatar
Vincent Pelletier committed
1466
      This implements the following predicates:
1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490
        - 1.0 < 1.0.1
        - 1.0rc1 < 1.0
        - 1.0a < 1.0.1
        - 1.1 < 2.0
        - 1.0.0 = 1.0
      """
      r = re.compile('(\d+|[a-zA-Z])')
      v1 = r.findall(version1)
      v2 = r.findall(version2)

      def convert(v, i):
        """Convert the ith element of v to an interger for a comparison.
        """
        #LOG('convert', 0, 'v = %r, i = %r' % (v, i))
        try:
          e = v[i]
          try:
            e = int(e)
          except ValueError:
            # ASCII code is one byte, so this produces negative.
            e = struct.unpack('b', e)[0] - 0x200
        except IndexError:
          e = 0
        return e
1491

1492 1493 1494 1495 1496 1497 1498 1499
      for i in xrange(max(len(v1), len(v2))):
        e1 = convert(v1, i)
        e2 = convert(v2, i)
        result = cmp(e1, e2)
        if result != 0:
          return result

      return 0
1500

1501
    def _getBusinessTemplateUrlDict(self):
1502
      business_template_url_dict = {}
1503
      for bt in self.getRepositoryBusinessTemplateList():
1504
        url, name = self.decodeRepositoryBusinessTemplateUid(bt.getUid())
1505 1506 1507
        if name.endswith('.bt5'):
          name = name[:-4]
        business_template_url_dict[name] = {
Rafael Monnerat's avatar
Rafael Monnerat committed
1508
          'url': '%s/%s' % (url, bt.filename),
1509 1510 1511 1512 1513
          'revision': bt.getRevision()
          }
      return business_template_url_dict

    security.declareProtected(Permissions.ManagePortal,
Rafael Monnerat's avatar
Rafael Monnerat committed
1514
        'installBusinessTemplatesFromRepositories')
1515
    def installBusinessTemplatesFromRepositories(self, *args, **kw):
1516 1517
      """Deprecated.
      """
1518
      DeprecationWarning('installBusinessTemplatesFromRepositories is deprecated; Use self.installBusinessTemplateListFromRepository instead.', DeprecationWarning)
1519
      return self.installBusinessTemplateListFromRepository(*args, **kw)
1520

1521 1522
    security.declareProtected(Permissions.ManagePortal,
         'resolveBusinessTemplateListDependency')
1523 1524 1525
    def resolveBusinessTemplateListDependency(self,
                                              template_title_list,
                                              with_test_dependency_list=False):
1526
      available_bt5_list = self.getRepositoryBusinessTemplateList()
1527
      template_title_list = set(template_title_list)
1528
      installed_bt5_title_list = self.getInstalledBusinessTemplateTitleList()
1529
      bt5_set = set()
1530 1531
      for available_bt5 in available_bt5_list:
        if available_bt5.title in template_title_list:
1532
          template_title_list.remove(available_bt5.title)
1533 1534
          bt5 = self.decodeRepositoryBusinessTemplateUid(available_bt5.uid)
          bt5_set.add(bt5)
1535
          meta_dependency_set = set()
1536 1537 1538
          for dep_repository, dep_id in self.getDependencyList(
              bt5,
              with_test_dependency_list):
1539 1540 1541
            if dep_repository != 'meta':
              bt5_set.add((dep_repository, dep_id))
            else:
1542 1543 1544 1545 1546 1547 1548 1549 1550 1551
              meta_dependency_set.add((dep_repository, dep_id))
          for dep_repository, dep_id in meta_dependency_set:
            provider_list = self.getProviderList(dep_id)
            provider_installed = False
            provider_title = None
            for provider in provider_list:
              if provider in [i[1].replace(".bt5", "") for i in bt5_set] or \
                    provider in installed_bt5_title_list or \
                    provider in template_title_list:
                provider_title = provider
1552
                for candidate in available_bt5_list:
1553
                  if candidate.title == provider:
1554 1555 1556
                    bt5_set.add(\
                      self.decodeRepositoryBusinessTemplateUid(
                          candidate.uid))
1557
                    break
1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570 1571 1572
                break
            if provider_title is None and len(provider_list) == 1:
              provider_title = provider_list[0]
            LOG('resolveBT, provider_title', 0, provider_title)
            if provider_title:
              for candidate in available_bt5_list:
                if candidate.title == provider_title:
                  bt5_set.add(\
                    self.decodeRepositoryBusinessTemplateUid(
                        candidate.uid))
                  break
            else:
              raise BusinessTemplateMissingDependency,\
                "Unable to resolve dependencies for %s, options are %s" \
                    % (dep_id, provider_list)
1573 1574 1575 1576

      if len(template_title_list) > 0:
         raise BusinessTemplateUnknownError, 'The Business Template %s could not be found on repositories %s' % \
             (list(template_title_list), self.getRepositoryList())
1577 1578
      return self.sortBusinessTemplateList(list(bt5_set))

1579 1580 1581
    security.declareProtected(Permissions.ManagePortal,
        'installBusinessTemplateListFromRepository')
    def installBusinessTemplateListFromRepository(self, template_list,
1582
        only_different=True, update_catalog=False, activate=False,
1583
        install_dependency=False):
1584 1585 1586 1587
      """Installs template_list from configured repositories by default only newest"""
      # XXX-Luke: This method could replace
      # TemplateTool_installRepositoryBusinessTemplateList while still being
      # possible to reuse by external callers
1588

1589 1590
      operation_log = []
      resolved_template_list = self.resolveBusinessTemplateListDependency(
1591
                   template_list)
1592 1593
      installed_bt5_dict = {x.getTitle(): x.getRevision()
        for x in self.getInstalledBusinessTemplateList()}
1594 1595
      if only_different:
        template_url_dict = self._getBusinessTemplateUrlDict()
1596 1597

      def checkAvailability(bt_title):
1598
        return bt_title in template_list or bt_title in installed_bt5_dict
1599 1600 1601 1602 1603 1604 1605
      missing_dependency_list = [i for i in resolved_template_list
                                 if not checkAvailability(i[1].replace(".bt5", ""))]

      if not install_dependency and len(missing_dependency_list) > 0:
        raise BusinessTemplateMissingDependency,\
            "Impossible to install, please install the following dependencies before: %s" \
            % [x[1] for x in missing_dependency_list]
1606 1607

      activate_kw =  dict(activity="SQLQueue", tag="start_%s" % (time.time()))
1608
      for repository, bt_id in resolved_template_list:
1609 1610 1611
        if only_different:
          bt = template_url_dict.get(bt_id)
          if bt is not None and bt['revision'] == installed_bt5_dict.get(bt_id):
1612
            continue
1613
        bt_url = '%s/%s' % (repository, bt_id)
1614
        param_dict = dict(download_url=bt_url, only_different=only_different)
1615
        param_dict["update_catalog"] = update_catalog
1616 1617 1618 1619 1620 1621 1622

        if activate:
          self.activate(**activate_kw).\
                updateBusinessTemplateFromUrl(**param_dict)
          activate_kw["after_tag"] = activate_kw["tag"]
          activate_kw["tag"] = bt_id
          operation_log.append('Installed %s using activities' % (bt_id))
1623
        else:
1624 1625
          document = self.updateBusinessTemplateFromUrl(**param_dict)
          operation_log.append('Installed %s with revision %s' % (
1626
              document.getTitle(), document.getShortRevision()))
1627 1628

      return operation_log
1629

1630 1631 1632
    security.declareProtected(Permissions.ManagePortal,
            'updateBusinessTemplateFromUrl')
    def updateBusinessTemplateFromUrl(self, download_url, id=None,
1633 1634 1635
                                         keep_original_list=None,
                                         before_triggered_bt5_id_list=None,
                                         after_triggered_bt5_id_list=None,
1636
                                         update_catalog=False,
1637
                                         reinstall=False,
1638
                                         active_process=None,
Rafael Monnerat's avatar
Rafael Monnerat committed
1639
                                         force_keep_list=None,
1640
                                         only_different=True):
Rafael Monnerat's avatar
Rafael Monnerat committed
1641
      """
1642
        This method download and install a bt5, from a URL.
1643 1644 1645 1646 1647

        keep_original_list can be used to make paths not touched at all

        force_keep_list can be used to force path to be modified or removed
        even if template system proposes not touching it
1648
      """
1649 1650 1651 1652 1653 1654 1655 1656
      if keep_original_list is None:
        keep_original_list = []
      if before_triggered_bt5_id_list is None:
        before_triggered_bt5_id_list = []
      if after_triggered_bt5_id_list is None:
        after_triggered_bt5_id_list = []
      if force_keep_list is None:
        force_keep_list = []
1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669
      if active_process is None:
        installed_dict = {}
        def log(msg):
          LOG('TemplateTool.updateBusinessTemplateFromUrl', INFO, msg)
      else:
        active_process = self.unrestrictedTraverse(active_process)
        if getattr(aq_base(active_process), 'installed_dict', None) is None:
          active_process.installed_dict = PersistentMapping()
        installed_dict = active_process.installed_dict
        message_list = []
        log = message_list.append

      log("Installing %s ..." % download_url)
1670
      imported_bt5 = self.download(url = download_url, id = id)
1671 1672
      bt_title = imported_bt5.getTitle()

1673 1674 1675
      if reinstall:
        install_kw = None
      else:
1676 1677 1678 1679 1680 1681 1682
        if only_different:
          previous_bt5 = self.getInstalledBusinessTemplate(bt_title)
          if previous_bt5 and \
             imported_bt5.getRevision() == previous_bt5.getRevision():
            log("%s is already installed with revision %s"
                % (bt_title, imported_bt5.getShortRevision()))
            return imported_bt5
1683 1684 1685

        install_kw = {}
        for listbox_line in imported_bt5.BusinessTemplate_getModifiedObject():
1686 1687
          item = listbox_line.object_id
          state = listbox_line.object_state
1688
          if state.startswith('Removed'):
1689 1690 1691 1692 1693 1694 1695 1696
            # The following condition could not be used to automatically decide
            # if an item must be kept or not. For example, this would not work
            # for items installed by PortalTypeWorkflowChainTemplateItem.
            maybe_moved = installed_dict.get(listbox_line.object_id, '')
            log('%s: %s%s' % (state, item,
              maybe_moved and ' (moved to %s ?)' % maybe_moved))
          else:
            installed_dict[item] = bt_title
1697 1698 1699

          # For actions which suggest that item shall be kept and item is not
          # explicitely forced, keep the default -- do nothing
1700 1701
          # XXX: 'force_keep_list' variable is misnamed.
          should_keep = item not in force_keep_list and state in (
1702 1703
            'Modified but should be kept', 'Removed but should be kept')
          # If item is forced to be untouched, do not touch it
1704 1705
          if item in keep_original_list or should_keep:
            if not should_keep:
1706 1707 1708
              log('Item %r is in force_keep_list and keep_original_list,'
                  ' as keep_original_list has precedence item is NOT MODIFIED'
                  % item)
1709 1710 1711
            install_kw[item] = 'nothing'
          else:
            install_kw[item] = listbox_line.choice_item_list[0][1]
1712

1713 1714
      # Run before script list
      for before_triggered_bt5_id in before_triggered_bt5_id_list:
1715 1716 1717
        log('Execute %r' % before_triggered_bt5_id)
        imported_bt5.unrestrictedTraverse(before_triggered_bt5_id)()

1718 1719 1720 1721 1722 1723 1724
      # Note: CATALOG_UPDATABLE should only be used in eceptional cases
      #       where the caller installs several bts and does not know
      #       which ones need to update catalog. Handling catalog should be
      #       usually done at upgrader level.
      if update_catalog is CATALOG_UPDATABLE and install_kw != {}:
        update_catalog = imported_bt5.isCatalogUpdatable()

1725 1726
      imported_bt5.install(object_to_update=install_kw,
                           update_catalog=update_catalog)
1727

1728 1729
      # Run After script list
      for after_triggered_bt5_id in after_triggered_bt5_id_list:
1730 1731 1732 1733 1734 1735 1736 1737
        log('Execute %r' % after_triggered_bt5_id)
        imported_bt5.unrestrictedTraverse(after_triggered_bt5_id)()
      if active_process is not None:
        active_process.postResult(ActiveResult(
          '%03u. %s' % (len(active_process.getResultList()) + 1, bt_title),
          detail='\n'.join(message_list)))
      else:
        log("Updated %s from %s" % (bt_title, download_url))
1738

1739 1740
      return imported_bt5

1741 1742 1743 1744 1745 1746 1747 1748 1749 1750 1751 1752 1753 1754 1755 1756 1757 1758 1759 1760
    security.declareProtected(Permissions.ManagePortal,
            'installMultipleBusinessPackage')
    def installMultipleBusinessPackage(self, bp5_list):
      """
      Install multiple Business Package at the same time
      """
      # XXX: Compare before calling install on path object property items
      from Products.ERP5.Document.BusinessPackage import \
                    ObjectPropertyTemplatePackageItem, PathTemplatePackageItem

      final_path_item = bp5_list[0]._path_item
      final_prop_item = bp5_list[0]._object_property_item

      for bp5 in bp5_list:
        final_path_item += bp5._path_item
        final_prop_item += bp5._object_property_item

      final_path_item.install()
      final_prop_item.install()

1761
    security.declareProtected(Permissions.ManagePortal,
1762 1763
            'installBusinessManager')
    def installBusinessManager(self, bm):
1764
      """
1765
      Run installation on flattened Business Manager
1766
      """
1767 1768 1769 1770 1771 1772 1773 1774 1775 1776 1777 1778 1779 1780 1781 1782 1783 1784 1785 1786 1787 1788 1789 1790
      # Run install on separate Business Item one by one
      for path_item in bm._path_item_list:
        path_item.install(self)

      bm.setStatus('installed')

    security.declareProtected(Permissions.ManagePortal,
            'updateInstallationState')
    def updateInstallationState(self, bm_list):
      """
      Run installation after comparing combined Business Manager status

      Steps:
      1. Create combinedBM for the bm_list
      2. Get the old combinedBM by checking the 'installed' status for it or
      by checking timestamp (?? which is better)
      CombinedBM: Collection of all Business item(s) whether installed or
      uninstalled
      3. Build BM from the filesystem
      4. Compare the combinedBM state to the last combinedBM state
      5. Compare the installation state to the OFS state
      6. If conflict while comaprison at 3, raise the error
      7. In all other case, install the BM List
      """
1791
      # Create old installation state from Installed Business Manager
1792 1793 1794
      installed_bm_list = self.getInstalledBusinessManagerList()
      combined_installed_path_item = [item for bm
                                      in installed_bm_list
1795
                                      for item in bm.objectValues()]
1796 1797 1798 1799 1800 1801

      # Create BM for old installation state and update its path item list
      old_installation_state = self.newContent(
                                  portal_type='Business Manager',
                                  title='Old Installation State',
                                  )
1802
      for item in combined_installed_path_item:
1803 1804 1805
        item.isIndexable = ConstantGetter('isIndexable', value=False)
        old_installation_state._setObject(item.getId(), aq_base(item),
                                          suppress_events=True)
1806 1807 1808 1809 1810 1811 1812 1813 1814 1815 1816 1817

      forbidden_bm_title_list = ['Old Installation State',]
      for bm in bm_list:
        forbidden_bm_title_list.append(bm.title)

      new_installed_bm_list = [l for l
                               in self.getInstalledBusinessManagerList()
                               if l.title not in forbidden_bm_title_list]
      new_installed_bm_list.extend(bm_list)

      combined_new_path_item = [item for bm
                                in new_installed_bm_list
1818
                                for item in bm.objectValues()]
1819 1820 1821 1822 1823 1824

      # Create BM for new installation state and update its path item list
      new_installation_state = self.newContent(
                                  portal_type='Business Manager',
                                  title='New Installation State',
                                  )
1825

1826
      for item in combined_new_path_item:
1827
        item.isIndexable = ConstantGetter('isIndexable', value=False)
1828 1829
        new_id = new_installation_state.generateNewId()
        new_installation_state._setObject(new_id, aq_base(item),
1830
                                          suppress_events=True)
1831 1832 1833 1834 1835 1836 1837 1838 1839 1840 1841 1842 1843 1844 1845 1846 1847 1848 1849 1850 1851 1852

      # Create installation process, which have the changes to be made in the
      # OFS during installation. Importantly, it should also be a Business Manager
      installation_process = self.newContent(
                                  portal_type='Business Manager',
                                  title='Installation Process',
                                  )

      # Get path list for old and new states
      old_state_path_list = old_installation_state.getPathList()
      new_state_path_list = new_installation_state.getPathList()

      to_install_path_item_list = []

      # Get the path which has been removed in new installation_state
      removed_path_list = [path for path
                           in old_state_path_list
                           if path not in new_state_path_list]

      # Add the removed path with negative sign in the to_install_path_item_list
      for path in removed_path_list:
        old_item = old_installation_state.getBusinessItemByPath(path)
1853
        old_item.setProperty('item_sign', '-1')
1854 1855
        to_install_path_item_list.append(old_item)

1856 1857
      # XXX: At this point, we expect all the Business Manager objects as 'reduced',
      # thus all the BusinessItem sub-objects should have single value
1858
      # Update hashes of item in old state before installation
1859
      for item in old_installation_state.objectValues():
1860
        if item.isProperty:
1861
          value = item.getProperty('item_property_value')
1862 1863
        else:
          value_list = item.objectValues()
1864 1865
          if value_list:
            value = value_list[0]
1866 1867
          else:
            value = ''
1868
        if value:
1869
          item.setProperty('item_sha', self.calculateComparableHash(
1870
                                                              value,
1871 1872
                                                              item.isProperty,
                                                              ))
1873

1874 1875
      # Path Item List for installation_process should be the difference between
      # old and new installation state
1876
      for item in new_installation_state.objectValues():
1877
        # If the path has been removed, then add it with sign = -1
1878 1879
        old_item = old_installation_state.getBusinessItemByPath(item.getProperty('item_path'))
        # Calculate sha for the items in new_insatallation_state
1880
        if item.isProperty:
1881
          value = item.getProperty('item_property_value')
1882
        else:
1883 1884 1885 1886 1887 1888 1889
          value_list = item.objectValues()
          if value_list:
            value = value_list[0]
          else:
            value = ''
        if value:
          item.setProperty('item_sha', self.calculateComparableHash(
1890 1891 1892
                                                                value,
                                                                item.isProperty,
                                                                ))
1893 1894 1895
        if old_item:
          # If the old_item exists, we match the hashes and if it differs, then
          # add the new item
1896
          if old_item.getProperty('item_sha') != item.getProperty('item_sha'):
1897 1898 1899 1900
            to_install_path_item_list.append(item)
        else:
          to_install_path_item_list.append(item)

1901
      for item in to_install_path_item_list:
1902 1903 1904
        item.isIndexable = ConstantGetter('isIndexable', value=False)
        installation_process._setObject(item.getId(), aq_base(item),
                                        suppress_events=True)
1905

1906
      error_list = self.compareOldStateToOFS(installation_process, old_installation_state)
1907 1908 1909 1910 1911

      # Change status of all BM installed
      for bm in bm_list:
        bm.setStatus('installed')

1912 1913 1914
      if error_list:
        raise ValueError(' '.join(error_list))

1915 1916
    installMultipleBusinessManager = updateInstallationState

1917
    def calculateComparableHash(self, object, isProperty=False):
1918 1919
      """
      Remove some attributes before comparing hashses
Ayush Tiwari's avatar
Ayush Tiwari committed
1920 1921
      and return hash of the comparable object dict, in case the object is
      an erp5 object.
1922 1923 1924 1925 1926

      Use shallow copy of the dict of the object at ZODB after removing
      attributes which changes at small updation, like workflow_history,
      uid, volatile attributes(which starts with _v)
      """
1927
      if isProperty:
Ayush Tiwari's avatar
Ayush Tiwari committed
1928 1929
        obj_dict = object
      else:
1930 1931
        klass = object.__class__
        classname = klass.__name__
Ayush Tiwari's avatar
Ayush Tiwari committed
1932
        obj_dict = object.__dict__.copy()
1933 1934 1935 1936 1937 1938 1939 1940 1941 1942 1943 1944 1945 1946 1947 1948 1949 1950 1951 1952 1953 1954
        attr_set = {'_dav_writelocks', '_filepath', '_owner', '_related_index',
                    'last_id', 'uid',
                    '__ac_local_roles__', '__ac_local_roles_group_id_dict__'}

        attr_set.update(('isIndexable',))

        if classname in ('File', 'Image'):
          attr_set.update(('_EtagSupport__etag', 'size'))
        elif classname == 'Types Tool' and klass.__module__ == 'erp5.portal_type':
          attr_set.add('type_provider_list')

        for attr in object.__dict__.keys():
          if attr in attr_set or attr.startswith('_cache_cookie_') or attr.startswith('_v'):
            try:
              del obj_dict[attr]
            except AttributeError:
              # XXX: Continue in cases where we want to delete some properties which
              # are not in attribute list
              # Raise an error
              continue

        if 'data' in obj_dict:
1955
          try:
1956 1957 1958
            obj_dict['data'] = obj_dict.get('data').__dict__
          except AttributeError:
            pass
1959 1960 1961 1962

      obj_sha = hash(pprint.pformat(obj_dict))
      return obj_sha

1963 1964 1965 1966 1967 1968 1969 1970 1971 1972
    def compareOldStateToOFS(self, installation_process, old_state):

      # Get the paths about which we are concerned about
      to_update_path_list = installation_process.getPathList()
      portal = self.getPortalObject()

      error_list = []

      for path in to_update_path_list:
        try:
Ayush Tiwari's avatar
Ayush Tiwari committed
1973
          if '#' in str(path):
1974
            isProperty = True
Ayush Tiwari's avatar
Ayush Tiwari committed
1975 1976 1977 1978 1979 1980 1981 1982 1983 1984 1985 1986
            relative_url, property_id = path.split('#')
            obj = portal.restrictedTraverse(relative_url)
            property_value = obj.getProperty(property_id)

            # If the value at ZODB for the property is none, raise KeyError
            # This is important to have compatibility between the way we check
            # path as well as property. Otherwise, if we install a new property,
            # we are always be getting an Error that there is change made at
            # ZODB for this property
            if not property_value:
              raise KeyError
            property_type = obj.getPropertyType(property_id)
1987
            obj = property_value
Ayush Tiwari's avatar
Ayush Tiwari committed
1988
          else:
1989
            isProperty = False
Ayush Tiwari's avatar
Ayush Tiwari committed
1990 1991
            obj = portal.restrictedTraverse(path)

1992
          obj_sha = self.calculateComparableHash(obj, isProperty)
1993

1994
          # Get item at old state
1995
          old_item = old_state.getBusinessItemByPath(path)
1996 1997 1998 1999 2000
          # Check if there is an object at old state at this path

          if old_item:
            # Compare hash with ZODB

2001
            if old_item.getProperty('item_sha') == obj_sha:
2002 2003 2004 2005
              # No change at ZODB on old item, so get the new item
              new_item = installation_process.getBusinessItemByPath(path)
              # Compare new item hash with ZODB

2006 2007
              if new_item.getProperty('item_sha') == obj_sha:
                if int(new_item.getProperty('item_sign')) == -1:
2008 2009 2010 2011 2012
                  # If the sign is negative, remove the value from the path
                  new_item.install(installation_process)
                else:
                  # If same hash, and +1 sign, do nothing
                  continue
2013 2014 2015 2016 2017 2018 2019 2020 2021 2022

              else:
                # Install the new_item
                new_item.install(installation_process)

            else:
              # Change at ZODB, so get the new item
              new_item = installation_process.getBusinessItemByPath(path)
              # Compare new item hash with ZODB

2023
              if new_item.getProperty('item_sha') == obj_sha:
2024 2025 2026 2027 2028 2029 2030 2031 2032 2033 2034 2035
                # If same hash, do nothing
                continue

              else:
                # Raise error
                error_list.append('Trying to remove changes at ZODB at %s' % path)

          else:
            # Object created at ZODB by the user
            # Compare with the new_item

            new_item = installation_process.getBusinessItemByPath(path)
2036
            if new_item.getProperty('item_sha') == obj_sha:
2037 2038 2039 2040 2041 2042 2043
              # If same hash, do nothing
              continue

            else:
              # Raise error
              error_list.append('Trying to remove changes at ZODB at %s' % path)

2044
        except KeyError:
2045 2046 2047 2048 2049 2050 2051 2052 2053 2054
          # Get item at old state
          old_item = old_state.getBusinessItemByPath(path)
          # Check if there is an object at old state at this path

          if old_item:
            # This means that the user had removed the object at this path
            # Check what the sign is for the new_item
            new_item = installation_process.getBusinessItemByPath(path)
            # Check sign of new_item

2055
            if int(new_item.getProperty('item_sign')) == 1:
2056 2057 2058 2059 2060
              error_list.append('Object at %s removed by user' % path)

          else:
            # If there is  no item at old state, install the new_item
            new_item = installation_process.getBusinessItemByPath(path)
2061
            # XXX: Hack for not trying to install the sub-objects from zexp,
2062
            # This should rather be implemented while exporting the object,
2063
            # where we shouldn't export sub-objects in the zexp
2064 2065 2066 2067 2068
            if not isProperty:
              try:
                value =  new_item.objectValues()[0]
              except IndexError:
                continue
2069 2070
            new_item.install(installation_process)

2071 2072
      return error_list

2073 2074 2075 2076 2077 2078 2079 2080 2081
    security.declareProtected(Permissions.ManagePortal,
            'createNewInstallationState')
    def createNewInstallationState(self, bm_list, old_installation_state):
      """
      Combines multiple BM to form single BM which would be the new
      installtion state
      """
      new_bm_list = bm_list[:]

2082 2083 2084 2085 2086 2087 2088
      # Create a list of path and hashes to be used for comparison
      path_list = []
      sha_list = []
      forbidden_bm_title_list = ['Installation State',]

      for bm in new_bm_list:
        forbidden_bm_title_list.append(bm.title)
2089 2090 2091
        for item in bm.path_item_list:
          path_list.append(item.path)
          sha_list.append(item.sha)
2092 2093 2094 2095 2096 2097

      installed_bm_list = [l for l
                           in self.getInstalledBusinessManagerList()
                           if l.title not in forbidden_bm_title_list]
      new_bm_list.extend(installed_bm_list)

2098 2099 2100 2101 2102 2103 2104 2105
      # Summation should also consider arithmetic on the Business Item(s)
      # having same path and layer and combine them.
      combinedBM = self.newContent(portal_type='Business Manager',
                                   title='Combined Business Manager')
      combinedBM.build()
      new_bm_list.insert(0, combinedBM)
      combinedBM = reduce(lambda x, y: x+y, new_bm_list)

2106
      final_path_list = combinedBM.getTemplatePathList()
2107 2108 2109 2110
      final_path_item_list = []

      final_path_item_list.extend(combinedBM._path_item_list)

2111 2112 2113
      removable_sha_list = []
      removable_path_list = []

2114
      for item in old_installation_state._path_item_list:
2115
        if item.path in path_list and item.sha in sha_list:
2116 2117 2118 2119 2120 2121 2122
          # If there is Business Item which have no change on updation, then
          # no need to reinstall it, cause in that case we prefer the changes
          # at ZODB

          # XXX: BAD DESIGN: Should compare both path as well as hash and keep
          # them together in a dictionary,using them separately can lead to
          # conflict in case two paths have same hash.
2123 2124
          removable_sha_list.append(item.sha)
          removable_path_list.append(item.path)
2125
        else:
2126 2127
          # If there is update of path item, change the sign of the last
          # version of that Business Item and add it to final_path_item_list
2128
          item.sign = -1
2129 2130 2131 2132 2133 2134 2135 2136 2137 2138 2139 2140
          final_path_item_list.append(item)

      final_path_list.extend(old_installation_state.getTemplatePathList())

      # Remove the Business Item objects from final_path_item_list which have
      # same sha and path
      final_path_list = [path for path
                         in final_path_list
                         if path not in removable_path_list]

      final_path_item_list = [item for item
                              in final_path_item_list
2141
                              if item.sha not in removable_sha_list]
2142

2143
      final_path_item_list.sort(key=lambda x: x.sign)
2144

2145 2146 2147
      # Remove the old installation state
      self._delObject(old_installation_state.getId())

2148 2149 2150 2151
      combinedBM._setTemplatePathList(final_path_list)
      combinedBM._path_item_list = final_path_item_list

      # Change the title of the combined BM
2152
      combinedBM.setTitle('Installation State')
2153 2154 2155

      # XXX: We are missing the part of creating installed_BM for all the BM
      # we have in bm_list, because this would be needed in case we build
2156
      # Business Manager again.
2157 2158

      # Reduce the final Business Manager
2159
      #combinedBM.reduceBusinessManager()
2160 2161

      return combinedBM
2162

2163
    security.declareProtected(Permissions.ManagePortal,
2164
            'compareInstallationStateWithOFS')
2165 2166
    def compareInstallationStateWithOFS(self, buildBM, new_installation_state,
            old_installation_state):
2167
      """
2168 2169 2170 2171 2172 2173 2174 2175 2176 2177 2178 2179
      Compare the buildBM and to be installed BM and show the changes in a way
      it can be notified or installed.

      1. Compare the hash of the objects and create a diff file in case of
      conflict
      2. If forced installation, delete the old object and install the ones
      from update Business Manager

      Returns:

      - has_confict: True , if there is a conflict between the 2 BMs
      - DiffFile: In case there is conflict between the two states
2180
      """
2181 2182 2183
      # XXX: BAD DESIGN: Should compare both path as well as hash, just path
      # can lead to conflict in case two paths have same sha.
      built_item_dict = {
2184
                        item.path: item.sha for item
2185
                        in buildBM._path_item_list
2186 2187 2188
                        }

      old_item_dict = {
2189
                      item.path: item.sha for item
2190
                      in old_installation_state._path_item_list
2191 2192 2193 2194 2195 2196
                      }

      # For creating new_item_dict, we have to take use of template_path_list
      # property as there can be case where we have path but not path_item as
      # the new state already gets filtered while creation
      new_item_dict = {
2197
                      item.path: item.sha for item
2198
                      in new_installation_state._path_item_list
2199
                      if item.sign == 1
2200 2201 2202
                      }

      build_sha_list = built_item_dict.values()
2203 2204
      final_item_list = []

2205
      for item in new_installation_state._path_item_list:
2206
        if item.sign == 1:
2207
          if path in old_item_dict.keys():
2208
            if old_item_dict[path] == item.sha:
2209 2210
              pass

2211
      for item in new_installation_state._path_item_list:
2212
        if item.sha in build_sha_list and item.sign == 1:
2213 2214
          # No need to install value which we already have
          continue
2215
        else:
2216 2217 2218
          final_item_list.append(item)

      new_installation_state._path_item_list = final_item_list
2219

2220
    security.declareProtected(Permissions.ManagePortal,
2221 2222
            'compareMultipleBusinessManager')
    def compareBusinessManager(self, new_bm, old_bm):
2223
      """
2224 2225 2226
      Compare two business manager and return a new Business manager based on
      the difference. This is specially required to compare two versions of
      Business Manager(s).
2227
      """
2228 2229 2230
      compared_bm = new_bm - old_bm
      # Return the subtraction of the Business Manager
      return compared_bm
2231

2232 2233
    security.declareProtected(Permissions.ManagePortal,
            'cleanInstallationState')
2234 2235 2236 2237
    def cleanInstallationState(self, installation_state):
      """
      Installation State is the Business Manager which has been installed
      Cleaning means removal of all the Business Item with negative sign
2238

2239 2240 2241
      **WARNING**:This should only be done after installing the Business Manager
      """
      if installation_state.getStatus() != 'installed':
2242
        LOG('WARNING', 0, 'Trying to clean installation state before installing')
2243
        raise ValueError, "Can't clean before installing"
2244

2245
      final_path_item_list = []
2246

2247
      for item in installation_state._path_item_list:
2248
        if item.sign == 1:
2249
          final_path_item_list.append(item)
2250

2251
      installation_state._path_item_list = final_path_item_list
2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264

    def getInstalledBusinessManagerList(self):
      bm_list = self.objectValues(portal_type='Business Manager')
      installed_bm_list = [bm for bm in bm_list if bm.getStatus() == 'installed']
      return installed_bm_list

    def getInstalledBusinessManagerTitleList(self):
      installed_bm_list = self.getInstalledBusinessManagerList()
      if not len(installed_bm_list):
        return []
      installed_bm_title_list = [bm.title for bm in installed_bm_list]
      return installed_bm_title_list

2265 2266 2267 2268 2269 2270 2271
    security.declareProtected(Permissions.ManagePortal,
            'getBusinessTemplateUrl')
    def getBusinessTemplateUrl(self, base_url_list, bt5_title):
      """
        This method verify if the business template are available
        into one url (repository).
      """
2272 2273
      if base_url_list is None:
        base_url_list = self.getRepositoryList()
2274 2275 2276 2277 2278
      # This list could be preconfigured at some properties or
      # at preferences.
      for base_url in base_url_list:
        url = "%s/%s" % (base_url, bt5_title)
        if base_url == "INSTANCE_HOME_REPOSITORY":
Rafael Monnerat's avatar
Rafael Monnerat committed
2279
          url = "file://%s/bt5/%s" % (getConfiguration().instancehome,
2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294
                                      bt5_title)
          LOG('ERP5', INFO, "TemplateTool: INSTANCE_HOME_REPOSITORY is %s." \
              % url)
        try:
          urllib2.urlopen(url)
          return url
        except (urllib2.HTTPError, OSError):
          # XXX Try again with ".bt5" in case the folder format be used
          # Instead tgz one.
          url = "%s.bt5" % url
          try:
            urllib2.urlopen(url)
            return url
          except (urllib2.HTTPError, OSError):
            pass
Rafael Monnerat's avatar
Rafael Monnerat committed
2295
      LOG('ERP5', INFO, 'TemplateTool: %s was not found into the url list: '
2296 2297 2298
                        '%s.' % (bt5_title, base_url_list))
      return None

2299 2300 2301
    security.declareProtected(Permissions.ManagePortal,
        'upgradeSite')
    def upgradeSite(self, bt5_list, deprecated_after_script_dict=None,
2302 2303
                    deprecated_reinstall_set=None, dry_run=False,
                    delete_orphaned=False,
2304 2305
                    keep_bt5_id_set=[],
                    update_catalog=False):
2306 2307 2308 2309 2310 2311 2312
      """
      Upgrade many business templates at a time. bt5_list should
      contains only final business templates, then all dependencies
      are calculated, and missing business templates will be added,
      old business templates will be updated, and orphelin business
      templates will be deleted

2313 2314 2315
      keep_bt5_id_set: business template that should not be deleted.
                       This is useful if we want to keep an old business
                       template without updating it and without removing it
2316

2317 2318
      deprecated_reinstall_set: this parameter is obsolete, please set
                                force_install property at business template level
2319 2320
                                It list all business templates who needs
                                reinstall
2321 2322 2323 2324 2325 2326

      update_catalog: handling catalog should be handled outside upgradeSite.
                      This option only exists for the case where it is not
                      known which bts need catalog update. In this case one
                      can pass CATALOG_UPDATABLE which will be propagated to
                      updateBusinessTemplateFromUrl.
2327
      """
2328
      # make sure that we updated information on repository
2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341
      self.updateRepositoryBusinessTemplateList(self.getRepositoryList())
      # do upgrade
      message_list = []
      deprecated_reinstall_set = deprecated_reinstall_set or set()
      def append(message):
        message_list.append(message)
        LOG('upgradeSite', 0, message)
      dependency_list = [x[1] for x in \
        self.resolveBusinessTemplateListDependency(bt5_list)]
      update_bt5_list = self.getRepositoryBusinessTemplateList(
        template_list=dependency_list)
      update_bt5_list.sort(key=lambda x: dependency_list.index(x.title))
      for bt5 in update_bt5_list:
2342
        reinstall = bt5.title in deprecated_reinstall_set or bt5.force_install
2343 2344
        if (not(reinstall) and bt5.version_state == 'present') or \
            bt5.title in keep_bt5_id_set:
2345 2346 2347 2348 2349
          continue
        append("Update %s business template in state %s%s" % \
          (bt5.title, bt5.version_state, (reinstall and ' (reinstall)') or ''))
        if not(dry_run):
          bt5_url = "%s/%s" % (bt5.repository, bt5.title)
2350 2351
          self.updateBusinessTemplateFromUrl(bt5_url, reinstall=reinstall,
                                             update_catalog=update_catalog)
2352 2353 2354 2355 2356 2357 2358 2359 2360 2361 2362 2363 2364 2365
      if delete_orphaned:
        if keep_bt5_id_set is None:
          keep_bt5_id_set = set()
        to_remove_bt5_list = [x for x in self.getInstalledBusinessTemplateList()
                              if x.title not in dependency_list]
        sorted_to_remove_bt5_id_list = self.sortDownloadedBusinessTemplateList(
                                  [x.id for x in to_remove_bt5_list])
        sorted_to_remove_bt5_id_list.reverse()
        to_remove_bt5_list.sort(
          key=lambda x: sorted_to_remove_bt5_id_list.index(x.id))
        for bt in to_remove_bt5_list:
          if bt.title in keep_bt5_id_set:
            continue
          append("Uninstall business template %s" % bt.title)
2366
          if not(dry_run):
2367 2368 2369
            # XXX Here is missing parameters to really remove stuff
            bt.uninstall()

2370 2371
      return message_list

Jean-Paul Smets's avatar
Jean-Paul Smets committed
2372
InitializeClass(TemplateTool)