SimulationTool.py 141 KB
Newer Older
Jean-Paul Smets's avatar
Jean-Paul Smets committed
1 2
##############################################################################
#
3
# Copyright (c) 2002, 2005 Nexedi SARL and Contributors. All Rights Reserved.
Jean-Paul Smets's avatar
Jean-Paul Smets committed
4
#                    Jean-Paul Smets-Solanes <jp@nexedi.com>
5
#                    Romain Courteaud <romain@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 Products.CMFCore.utils import getToolByName
Jean-Paul Smets's avatar
Jean-Paul Smets committed
31 32

from AccessControl import ClassSecurityInfo
33
from Products.ERP5Type.Globals import InitializeClass, DTMLFile
Jean-Paul Smets's avatar
Jean-Paul Smets committed
34
from Products.ERP5Type import Permissions
35
from Products.ERP5Type.Tool.BaseTool import BaseTool
Jean-Paul Smets's avatar
Jean-Paul Smets committed
36 37 38

from Products.ERP5 import _dtmldir

39
from zLOG import LOG, PROBLEM, WARNING, INFO
Jean-Paul Smets's avatar
Jean-Paul Smets committed
40 41

from Products.ERP5.Capacity.GLPK import solve
42
from numpy import zeros, resize
Alexandre Boeglin's avatar
Alexandre Boeglin committed
43
from DateTime import DateTime
Jean-Paul Smets's avatar
Jean-Paul Smets committed
44

45 46
from Products.ERP5 import DeliverySolver
from Products.ERP5 import TargetSolver
47
from Products.PythonScripts.Utility import allow_class
Jean-Paul Smets's avatar
Jean-Paul Smets committed
48

49
from Products.ZSQLCatalog.SQLCatalog import Query, ComplexQuery
50

51
from Shared.DC.ZRDB.Results import Results
52
from Products.ERP5Type.Utils import mergeZRDBResults
53 54
from App.Extensions import getBrain
from MySQLdb import ProgrammingError
55

56
from hashlib import md5
57
from warnings import warn
58 59 60 61 62 63 64 65
from cPickle import loads, dumps
from copy import deepcopy
from sys import exc_info

MYSQL_MIN_DATETIME_RESOLUTION = 1/86400.

class StockOptimisationError(Exception):
    pass
66

67
class SimulationTool(BaseTool):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86
    """
    The SimulationTool implements the ERP5
    simulation algorithmics.


    Examples of applications:

    -

    -
    ERP5 main purpose:

    -

    -

    """
    id = 'portal_simulation'
    meta_type = 'ERP5 Simulation Tool'
87
    portal_type = 'Simulation Tool'
Jean-Paul Smets's avatar
Jean-Paul Smets committed
88 89 90 91 92 93 94 95 96 97 98 99
    allowed_types = ( 'ERP5 Applied Rule', )

    # Declarative Security
    security = ClassSecurityInfo()

    #
    #   ZMI methods
    #
    security.declareProtected( Permissions.ManagePortal, 'manage_overview' )
    manage_overview = DTMLFile( 'explainSimulationTool', _dtmldir )

    def filtered_meta_types(self, user=None):
100 101 102 103 104 105 106
      # Filters the list of available meta types.
      all = SimulationTool.inheritedAttribute('filtered_meta_types')(self)
      meta_types = []
      for meta_type in self.all_meta_types():
        if meta_type['name'] in self.allowed_types:
          meta_types.append(meta_type)
      return meta_types
Jean-Paul Smets's avatar
Jean-Paul Smets committed
107

108 109 110
    def tpValues(self) :
      """ show the content in the left pane of the ZMI """
      return self.objectValues()
111

112 113 114
    security.declarePrivate('manage_afterAdd')
    def manage_afterAdd(self, item, container) :
      """Init permissions right after creation.
115

116 117 118 119 120 121 122 123 124 125 126 127
      Permissions in simulation tool are simple:
       o Each member can access and create some content.
       o Only manager can view, because simulation can be seen as
         sensitive information.
      """
      item.manage_permission(Permissions.AddPortalContent,
            ['Member', 'Author', 'Manager'])
      item.manage_permission(Permissions.AccessContentsInformation,
            ['Member', 'Auditor', 'Manager'])
      item.manage_permission(Permissions.View,
            ['Manager',])
      BaseTool.inheritedAttribute('manage_afterAdd')(self, item, container)
128

129 130
    def solveDelivery(self, delivery, delivery_solver_name, target_solver_name,
                      additional_parameters=None, **kw):
131
      """
132
        Solves a delivery by calling first DeliverySolver, then TargetSolver
133
      """
134 135 136
      return self._solveMovementOrDelivery(delivery, delivery_solver_name,
          target_solver_name, delivery=1,
          additional_parameters=additional_parameters, **kw)
137

138 139
    def solveMovement(self, movement, delivery_solver_name, target_solver_name,
                      additional_parameters=None, **kw):
140
      """
141
        Solves a movement by calling first DeliverySolver, then TargetSolver
142
      """
143 144 145
      return self._solveMovementOrDelivery(movement, delivery_solver_name,
          target_solver_name, movement=1,
          additional_parameters=additional_parameters, **kw)
146

147 148 149
    def _solveMovementOrDelivery(self, document, delivery_solver_name,
                                 target_solver_name, movement=0, delivery=0,
                                 additional_parameters=None,**kw):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
150
      """
151
        Solves a document by calling first DeliverySolver, then TargetSolver
Jean-Paul Smets's avatar
Jean-Paul Smets committed
152
      """
153 154 155 156 157 158 159 160
      if movement == delivery:
        raise ValueError('Parameters movement and delivery have to be'
                         ' different')

      solve_result = []
      for solver_name, solver_module in ((delivery_solver_name, DeliverySolver),
                                         (target_solver_name, TargetSolver)):
        result = None
161 162 163 164 165 166
        if solver_name is not None:
          solver_file_path = "%s.%s" % (solver_module.__name__,
                                        solver_name)
          __import__(solver_file_path)
          solver_file = getattr(solver_module, solver_name)
          solver_class = getattr(solver_file, solver_name)
167 168
          solver = solver_class(additional_parameters=additional_parameters,
              **kw)
169

170
          if movement:
171
            result = solver.solveMovement(document)
172
          if delivery:
173 174 175
            result = solver.solveDelivery(document)
        solve_result.append(result)
      return solve_result
176

Jean-Paul Smets's avatar
Jean-Paul Smets committed
177 178
    #######################################################
    # Stock Management
179

180
    def _generatePropertyUidList(self, prop, as_text=0):
181 182 183 184 185 186 187
      """
      converts relative_url or text (single element or list or dict)
        to an object usable by buildSQLQuery

      as_text == 0: tries to lookup an uid from the relative_url
      as_text == 1: directly passes the argument as text
      """
188
      if prop is None :
189 190
        return []
      category_tool = getToolByName(self, 'portal_categories')
191
      property_uid_list = []
192
      if isinstance(prop, str):
193
        if not as_text:
194
          prop_value = category_tool.getCategoryValue(prop)
195
          if prop_value is None:
196
            raise ValueError, 'Category %s does not exists' % prop
197
          property_uid_list.append(prop_value.getUid())
198
        else:
199 200 201
          property_uid_list.append(prop)
      elif isinstance(prop, (list, tuple)):
        for property_item in prop :
202
          if not as_text:
203 204 205 206
            prop_value = category_tool.getCategoryValue(property_item)
            if prop_value is None:
              raise ValueError, 'Category %s does not exists' % property_item
            property_uid_list.append(prop_value.getUid())
207 208
          else:
            property_uid_list.append(property_item)
209
      elif isinstance(prop, dict):
210
        tmp_uid_list = []
211 212 213
        if isinstance(prop['query'], str):
          prop['query'] = [prop['query']]
        for property_item in prop['query'] :
214
          if not as_text:
215 216 217 218
            prop_value = category_tool.getCategoryValue(property_item)
            if prop_value is None:
              raise ValueError, 'Category %s does not exists' % property_item
            tmp_uid_list.append(prop_value.getUid())
219 220
          else:
            tmp_uid_list.append(property_item)
221
        if tmp_uid_list:
222
          property_uid_list = {}
223
          property_uid_list['operator'] = prop['operator']
224 225 226
          property_uid_list['query'] = tmp_uid_list
      return property_uid_list

227 228
    def _getSimulationStateQuery(self, **kw):
      simulation_state_dict = self._getSimulationStateDict(**kw)
229
      return self._buildSimulationStateQuery(simulation_state_dict=simulation_state_dict)
Vincent Pelletier's avatar
Vincent Pelletier committed
230

231
    def _buildSimulationStateQuery(self, simulation_state_dict, table='stock'):
232 233 234 235 236 237 238
      input_simulation_state = simulation_state_dict.get(
                                 'input_simulation_state')
      output_simulation_state = simulation_state_dict.get(
                                 'output_simulation_state')
      simulation_state = simulation_state_dict.get('simulation_state')
      if simulation_state is not None:
        simulation_query = Query(operator='IN',
239
                                 **{'%s.simulation_state' % (table, ):
240 241
                                    simulation_state})
      elif input_simulation_state is not None:
242 243
        input_quantity_query = ComplexQuery(
                        ComplexQuery(
244
                            Query(**{'%s.quantity' % table: '>=0'}),
245 246 247 248 249 250 251
                            Query(**{'%s.is_cancellation' % table: 0}),
                            operator='AND'),
                        ComplexQuery(
                            Query(**{'%s.quantity' % table: '<0'}),
                            Query(**{'%s.is_cancellation' % table: 1}),
                            operator='AND'),
                        operator='OR')
252
        input_simulation_query = Query(operator='IN',
253
                                       **{'%s.simulation_state' % (table, ):
254 255 256 257 258
                                          input_simulation_state})
        simulation_query = ComplexQuery(input_quantity_query,
                                        input_simulation_query,
                                        operator='AND')
        if output_simulation_state is not None:
259 260 261 262 263 264
          output_quantity_query = ComplexQuery(
                        ComplexQuery(
                            Query(**{'%s.quantity' % table: '<0'}),
                            Query(**{'%s.is_cancellation' % table: 0}),
                            operator='AND'),
                        ComplexQuery(
265
                            Query(**{'%s.quantity' % table: '>=0'}),
266 267 268
                            Query(**{'%s.is_cancellation' % table: 1}),
                            operator='AND'),
                        operator='OR')
269
          output_simulation_query = Query(operator='IN',
270
                                          **{'%s.simulation_state' % (table, ):
271 272 273 274 275 276 277
                                             output_simulation_state})
          output_query = ComplexQuery(output_quantity_query,
                                      output_simulation_query,
                                      operator='AND')
          simulation_query = ComplexQuery(simulation_query, output_query,
                                          operator='OR')
      else:
278
        simulation_query = None
279 280 281
      return simulation_query

    def _getSimulationStateDict(self, simulation_state=None, omit_transit=0,
282 283 284 285 286 287 288 289 290 291 292 293
                                input_simulation_state=None,
                                output_simulation_state=None,
                                transit_simulation_state=None,
                                strict_simulation_state=None):
      """
      This method is used in order to give what should be
      the input_simulation_state or output_simulation_state
      depending on many parameters
      """
      string_or_list = (str, list, tuple)
      # Simulation States
      # If strict_simulation_state is set, we directly put it into the dictionary
294
      simulation_dict = {}
295 296 297
      if strict_simulation_state:
        if isinstance(simulation_state, string_or_list)\
                and simulation_state:
298 299
           simulation_query = Query(
                   **{'stock.simulation_state': simulation_state})
300 301 302 303 304 305 306 307 308 309 310 311 312 313 314 315 316 317 318 319 320 321 322 323 324 325 326 327 328 329 330 331 332 333 334 335 336 337 338 339 340 341 342 343 344 345 346 347 348 349 350 351 352 353 354
      else:
        # first, we evaluate simulation_state
        sql_kw = {}
        if simulation_state and isinstance(simulation_state, string_or_list):
          if isinstance(simulation_state, str):
            sql_kw['input_simulation_state'] = [simulation_state]
            sql_kw['output_simulation_state'] = [simulation_state]
          else:
            sql_kw['input_simulation_state'] = simulation_state
            sql_kw['output_simulation_state'] = simulation_state
        # then, if omit_transit == 1, we evaluate (simulation_state -
        # transit_simulation_state) for input_simulation_state
        if omit_transit:
          if isinstance(simulation_state, string_or_list)\
                and simulation_state:
            if isinstance(transit_simulation_state, string_or_list)\
                  and transit_simulation_state:
              # when we know both are usable, we try to calculate
              # (simulation_state - transit_simulation_state)
              if isinstance(simulation_state, str):
                simulation_state = [simulation_state]
              if isinstance(transit_simulation_state, str) :
                transit_simulation_state = [transit_simulation_state]
              delivered_simulation_state_list = []
              for state in simulation_state :
                if state not in transit_simulation_state :
                  delivered_simulation_state_list.append(state)
              sql_kw['input_simulation_state'] = delivered_simulation_state_list

        # alternatively, the user can directly define input_simulation_state
        # and output_simulation_state
        if input_simulation_state and isinstance(input_simulation_state,
                                                  string_or_list):
          if isinstance(input_simulation_state, str):
            input_simulation_state = [input_simulation_state]
          sql_kw['input_simulation_state'] = input_simulation_state
        if output_simulation_state and isinstance(output_simulation_state,
                                                  string_or_list):
          if isinstance(output_simulation_state, str):
            output_simulation_state = [output_simulation_state]
          sql_kw['output_simulation_state'] = output_simulation_state
        # XXX In this case, we must not set sql_kw[input_simumlation_state] before
        input_simulation_state = None
        output_simulation_state = None
        if sql_kw.has_key('input_simulation_state'):
          input_simulation_state = sql_kw.get('input_simulation_state')
        if sql_kw.has_key('output_simulation_state'):
          output_simulation_state = sql_kw.get('output_simulation_state')
        if input_simulation_state is not None \
           or output_simulation_state is not None:
          sql_kw.pop('input_simulation_state',None)
          sql_kw.pop('output_simulation_state',None)
        if input_simulation_state is not None:
          if output_simulation_state is not None:
            if input_simulation_state == output_simulation_state:
355
              simulation_dict['simulation_state'] = input_simulation_state
356
            else:
357 358
              simulation_dict['input_simulation_state'] = input_simulation_state
              simulation_dict['output_simulation_state'] = output_simulation_state
359
          else:
360
            simulation_dict['input_simulation_state'] = input_simulation_state
361
        elif output_simulation_state is not None:
362 363
          simulation_dict['simulation_state'] = output_simulation_state
      return simulation_dict
364

365 366 367 368 369 370
    def _getOmitQuery(self, query_table=None, omit_input=0, omit_output=0,
                      omit_asset_increase=0, omit_asset_decrease=0, **kw):
      omit_dict = self._getOmitDict(omit_input=omit_input,
                                    omit_output=omit_output,
                                    omit_asset_increase=omit_asset_increase,
                                    omit_asset_decrease=omit_asset_decrease)
371
      return self._buildOmitQuery(query_table=query_table, omit_dict=omit_dict)
Vincent Pelletier's avatar
Vincent Pelletier committed
372

373
    def _buildOmitQuery(self, query_table, omit_dict):
374 375 376 377
      """
      Build a specific query in order to take:
      - negatives quantity values if omit_input
      - postives quantity values if omit_output
378 379
      - negatives asset price values if omit_asset_increase
      - positives asset price values if omit_asset_decrease
380 381
      """
      omit_query = None
382 383
      omit_input = omit_dict.get('input', False)
      omit_output = omit_dict.get('output', False)
384 385 386 387
      omit_asset_increase = omit_dict.get('asset_increase', False)
      omit_asset_decrease = omit_dict.get('asset_decrease', False)
      if omit_input or omit_output\
          or omit_asset_increase or omit_asset_decrease:
388
        # Make sure to check some conditions
389 390 391 392 393 394 395
        condition_expression = \
          "%(query_table)s.node_uid <> %(query_table)s.mirror_node_uid \
         OR %(query_table)s.section_uid <> %(query_table)s.mirror_section_uid \
         OR %(query_table)s.mirror_node_uid IS NULL \
         OR %(query_table)s.mirror_section_uid IS NULL \
         OR %(query_table)s.payment_uid IS NOT NULL \
           " % {'query_table': query_table}
396
        if omit_input:
397 398 399 400 401 402
          quantity_query = ComplexQuery(
                        ComplexQuery(
                            Query(**{'%s.quantity' % query_table: '<0'}),
                            Query(**{'%s.is_cancellation' % query_table: 0}),
                            operator='AND'),
                        ComplexQuery(
403
                            Query(**{'%s.quantity' % query_table: '>=0'}),
404 405 406
                            Query(**{'%s.is_cancellation' % query_table: 1}),
                            operator='AND'),
                        operator='OR')
407 408 409
          omit_query = ComplexQuery(quantity_query, condition_expression,
                                    operator='AND')
        if omit_output:
410 411
          quantity_query = ComplexQuery(
                        ComplexQuery(
412
                            Query(**{'%s.quantity' % query_table: '>=0'}),
413 414 415 416 417 418 419
                            Query(**{'%s.is_cancellation' % query_table: 0}),
                            operator='AND'),
                        ComplexQuery(
                            Query(**{'%s.quantity' % query_table: '<0'}),
                            Query(**{'%s.is_cancellation' % query_table: 1}),
                            operator='AND'),
                        operator='OR')
420
          output_query = ComplexQuery(quantity_query, condition_expression,
421
                                      operator='AND')
422
          if omit_query is not None:
423
            omit_query = ComplexQuery(omit_query, output_query, operator='AND')
424 425 426 427 428 429 430 431 432 433
          else:
            omit_query = output_query

        if omit_asset_increase:
          asset_price_query = ComplexQuery(
                        ComplexQuery(
                            Query(**{'%s.total_price' % query_table: '<0'}),
                            Query(**{'%s.is_cancellation' % query_table: 0}),
                            operator='AND'),
                        ComplexQuery(
434
                            Query(**{'%s.total_price' % query_table: '>=0'}),
435 436 437 438 439 440 441 442 443 444 445 446 447
                            Query(**{'%s.is_cancellation' % query_table: 1}),
                            operator='AND'),
                        operator='OR')
          asset_increase_query = ComplexQuery(asset_price_query, condition_expression,
                                              operator='AND')
          if omit_query is not None:
            omit_query = ComplexQuery(omit_query, asset_increase_query, operator='AND')
          else:
            omit_query = asset_increase_query

        if omit_asset_decrease:
          asset_price_query = ComplexQuery(
                        ComplexQuery(
448
                            Query(**{'%s.total_price' % query_table: '>=0'}),
449 450 451 452 453 454 455 456 457 458 459 460 461
                            Query(**{'%s.is_cancellation' % query_table: 0}),
                            operator='AND'),
                        ComplexQuery(
                            Query(**{'%s.total_price' % query_table: '<0'}),
                            Query(**{'%s.is_cancellation' % query_table: 1}),
                            operator='AND'),
                        operator='OR')
          asset_decrease_query = ComplexQuery(asset_price_query, condition_expression,
                                              operator='AND')
          if omit_query is not None:
            omit_query = ComplexQuery(omit_query, asset_decrease_query, operator='AND')
          else:
            omit_query = asset_decrease_query
462 463 464

      return omit_query

465 466 467 468 469 470
    def _getOmitDict(self, omit_input=False, omit_output=False,
                     omit_asset_increase=False, omit_asset_decrease=False):
      return { 'input': omit_input,
               'output': omit_output,
               'asset_increase': omit_asset_increase,
               'asset_decrease': omit_asset_decrease, }
471

472
    def _generateSQLKeywordDict(self, table='stock', **kw):
473
        sql_kw, new_kw = self._generateKeywordDict(**kw)
474 475
        return self._generateSQLKeywordDictFromKeywordDict(table=table,
                 sql_kw=sql_kw, new_kw=new_kw)
476

477 478
    def _generateSQLKeywordDictFromKeywordDict(self, table='stock', sql_kw={},
                                               new_kw={}):
479
        ctool = getToolByName(self, 'portal_catalog')
480 481
        sql_kw = sql_kw.copy()
        new_kw = new_kw.copy()
482

483
        # Group-by expression  (eg. group_by=['node_uid'])
484
        group_by = new_kw.pop('group_by', [])
485

486
        # group by from stock table (eg. group_by_node=True)
487
        # prepend table name to avoid ambiguities.
488
        column_group_by = new_kw.pop('column_group_by', [])
489 490
        if column_group_by:
          group_by.extend(['%s.%s' % (table, x) for x in column_group_by])
491 492

        # group by from related keys columns (eg. group_by_node_category=True)
493 494 495
        related_key_group_by = new_kw.pop('related_key_group_by', [])
        if related_key_group_by:
          group_by.extend(['%s_%s' % (table, x) for x in related_key_group_by])
Vincent Pelletier's avatar
Vincent Pelletier committed
496

497 498 499
        # group by involving a related key (eg. group_by=['product_line_uid'])
        related_key_dict_passthrough_group_by = new_kw.get(
                'related_key_dict_passthrough', dict()).pop('group_by', [])
500 501 502
        if isinstance(related_key_dict_passthrough_group_by, basestring):
          related_key_dict_passthrough_group_by = (
                related_key_dict_passthrough_group_by,)
503 504
        group_by.extend(related_key_dict_passthrough_group_by)

505 506 507
        if group_by:
          new_kw['group_by'] = group_by

508 509 510 511
        # select expression
        select_dict = new_kw.get('select_dict', dict())
        related_key_select_expression_list = new_kw.pop(
                'related_key_select_expression_list', [])
512 513 514
        for related_key_select in related_key_select_expression_list:
          select_dict[related_key_select] = '%s_%s' % (table,
                                                       related_key_select)
515 516
        new_kw['select_dict'] = select_dict

517
        # Column values
518 519 520
        column_value_dict = new_kw.pop('column_value_dict', {})
        for key, value in column_value_dict.iteritems():
          new_kw['%s.%s' % (table, key)] = value
521
        # Related keys
522 523 524 525 526 527
        # First, the passthrough (acts as default values)
        for key, value in new_kw.pop('related_key_dict_passthrough', {})\
            .iteritems():
          new_kw[key] = value
        # Second, calculated values
        for key, value in new_kw.pop('related_key_dict', {}).iteritems():
528
          new_kw['%s_%s' % (table, key)] = value
529 530 531 532 533 534 535 536 537
        # Simulation states matched with input and output omission
        def joinQueriesIfNeeded(query_a, query_b, operator):
          if None not in (query_a, query_b):
            return ComplexQuery(query_a, query_b, operator=operator)
          elif query_a is not None:
            return query_a
          return query_b
        def getSimulationQuery(simulation_dict, omit_dict):
          simulation_query = self._buildSimulationStateQuery(
538 539
                               simulation_state_dict=simulation_dict,
                               table=table)
540 541 542 543 544 545 546 547 548 549 550 551 552 553 554 555 556 557
          omit_query = self._buildOmitQuery(query_table=table,
                                            omit_dict=omit_dict)
          return joinQueriesIfNeeded(query_a=simulation_query,
                                     query_b=omit_query, operator='AND')
        regular_query = getSimulationQuery(new_kw.pop('simulation_dict', {}),
                                           new_kw.pop('omit_dict', {}))
        reserved_kw = new_kw.pop('reserved_kw', None)
        if reserved_kw is not None:
          reserved_query = getSimulationQuery(
            reserved_kw.pop('simulation_dict', {}),
            reserved_kw.pop('omit_dict', {}))
          simulation_query = joinQueriesIfNeeded(query_a=regular_query,
                                                 query_b=reserved_query,
                                                 operator='OR')
        else:
          simulation_query = regular_query
        if simulation_query is not None:
          new_kw['query'] = simulation_query
558

559 560
        # Sort on
        if 'sort_on' in new_kw:
561 562
          table_column_list = ctool.getSQLCatalog()._getCatalogSchema(
                                                              table=table)
563 564 565
          sort_on = new_kw['sort_on']
          new_sort_on = []
          for column_id, sort_direction in sort_on:
566
            if column_id in table_column_list:
567 568 569
              column_id = '%s.%s' % (table, column_id)
            new_sort_on.append((column_id, sort_direction))
          new_kw['sort_on'] = tuple(new_sort_on)
Vincent Pelletier's avatar
Vincent Pelletier committed
570

571 572 573 574 575
        # Remove some internal parameters that does not have any meaning for
        # catalog
        new_kw.pop('ignore_group_by', None)
        new_kw.pop('movement_list_mode', None)

576
        sql_kw.update(ctool.buildSQLQuery(**new_kw))
577 578
        return sql_kw

579
    def _generateKeywordDict(self,
580
        # dates
581
        from_date=None, to_date=None, at_date=None,
582
        omit_mirror_date=1,
583
        # instances
Alexandre Boeglin's avatar
Alexandre Boeglin committed
584
        resource=None, node=None, payment=None,
585
        section=None, mirror_section=None, item=None,
586 587
        function=None, project=None, funding=None, payment_request=None,
        transformed_resource=None,
588 589 590
        # used for tracking
        input=0, output=0,
        # categories
Alexandre Boeglin's avatar
Alexandre Boeglin committed
591
        resource_category=None, node_category=None, payment_category=None,
592
        section_category=None, mirror_section_category=None,
593
        function_category=None, project_category=None, funding_category=None,
594
        payment_request_category=None,
595 596 597 598 599 600
        # categories with strict membership
        resource_category_strict_membership=None,
        node_category_strict_membership=None,
        payment_category_strict_membership=None,
        section_category_strict_membership=None,
        mirror_section_category_strict_membership=None,
601 602
        function_category_strict_membership=None,
        project_category_strict_membership=None,
603
        funding_category_strict_membership=None,
604
        payment_request_category_strict_membership=None,
605
        # simulation_state
606
        strict_simulation_state=0,
Alexandre Boeglin's avatar
Alexandre Boeglin committed
607
        simulation_state=None, transit_simulation_state = None, omit_transit=0,
608
        input_simulation_state=None, output_simulation_state=None,
609
        reserved_kw=None,
610
        # variations
611
        variation_text=None, sub_variation_text=None,
612
        variation_category=None,
613
        transformed_variation_text=None,
614
        # uids
615
        resource_uid=None, node_uid=None, section_uid=None, payment_uid=None,
616
        mirror_node_uid=None, mirror_section_uid=None, function_uid=None,
617
        project_uid=None, funding_uid=None, payment_request_uid=None,
618 619 620
        # omit input and output
        omit_input=0,
        omit_output=0,
621 622
        omit_asset_increase=0,
        omit_asset_decrease=0,
623 624
        # group by
        group_by_node=0,
625 626
        group_by_node_category=0,
        group_by_node_category_strict_membership=0,
627
        group_by_mirror_node=0,
628 629
        group_by_mirror_node_category=0,
        group_by_mirror_node_category_strict_membership=0,
630
        group_by_section=0,
631 632
        group_by_section_category=0,
        group_by_section_category_strict_membership=0,
633
        group_by_mirror_section=0,
634 635
        group_by_mirror_section_category=0,
        group_by_mirror_section_category_strict_membership=0,
636
        group_by_payment=0,
637 638
        group_by_payment_category=0,
        group_by_payment_category_strict_membership=0,
639 640 641
        group_by_sub_variation=0,
        group_by_variation=0,
        group_by_movement=0,
642
        group_by_resource=0,
643
        group_by_project=0,
644 645
        group_by_project_category=0,
        group_by_project_category_strict_membership=0,
646 647 648
        group_by_funding=0,
        group_by_funding_category=0,
        group_by_funding_category_strict_membership=0,
649 650 651
        group_by_payment_request=0,
        group_by_payment_request_category=0,
        group_by_payment_request_category_strict_membership=0,
652
        group_by_function=0,
653 654
        group_by_function_category=0,
        group_by_function_category_strict_membership=0,
655
        group_by_date=0,
656 657
        # sort_on
        sort_on=None,
658 659
        # keywords for related keys
        **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
660
      """
661 662 663 664
      Generates keywords and calls buildSQLQuery

      - omit_mirror_date: normally, date's parameters are only based on date
        column. If 0, it also used the mirror_date column.
Alexandre Boeglin's avatar
Alexandre Boeglin committed
665 666 667 668
      """
      new_kw = {}
      sql_kw = {}

669 670 671
      # input and output are used by getTrackingList
      sql_kw['input'] = input
      sql_kw['output'] = output
672 673 674
      # Add sort_on parameter if defined
      if sort_on is not None:
        new_kw['sort_on'] = sort_on
675

676 677
      class DictMixIn(dict):
        def set(dictionary, key, value):
678
          result = bool(value)
679 680 681 682 683 684 685 686 687
          if result:
            dictionary[key] = value
          return result

        def setUIDList(dictionary, key, value, as_text=0):
          uid_list = self._generatePropertyUidList(value, as_text=as_text)
          return dictionary.set(key, uid_list)

      column_value_dict = DictMixIn()
688 689

      if omit_mirror_date:
690
        date_dict = {}
691
        if from_date :
692
          date_dict.setdefault('query', []).append(from_date)
693 694
          date_dict['range'] = 'min'
          if to_date :
695
            date_dict.setdefault('query', []).append(to_date)
696 697
            date_dict['range'] = 'minmax'
          elif at_date :
698
            date_dict.setdefault('query', []).append(at_date)
699 700
            date_dict['range'] = 'minngt'
        elif to_date :
701
          date_dict.setdefault('query', []).append(to_date)
702
          date_dict['range'] = 'max'
703
        elif at_date :
704
          date_dict.setdefault('query', []).append(at_date)
705
          date_dict['range'] = 'ngt'
706 707
        if date_dict:
          column_value_dict['date'] = date_dict
708
      else:
709 710
        column_value_dict['date'] = {'query': [to_date], 'range': 'ngt'}
        column_value_dict['mirror_date'] = {'query': [from_date], 'range': 'nlt'}
711

712
      column_value_dict.set('resource_uid', resource_uid)
713
      column_value_dict.set('payment_uid', payment_uid)
714
      column_value_dict.set('project_uid', project_uid)
715
      column_value_dict.set('funding_uid', funding_uid)
716
      column_value_dict.set('payment_request_uid', payment_request_uid)
717
      column_value_dict.set('function_uid', function_uid)
718
      if column_value_dict.set('section_uid', section_uid):
719
        sql_kw['section_filtered'] = 1
720
      column_value_dict.set('node_uid', node_uid)
721 722
      column_value_dict.set('mirror_node_uid', mirror_node_uid)
      column_value_dict.set('mirror_section_uid', mirror_section_uid)
723 724 725 726
      column_value_dict.setUIDList('resource_uid', resource)
      column_value_dict.setUIDList('aggregate_uid', item)
      column_value_dict.setUIDList('node_uid', node)
      column_value_dict.setUIDList('payment_uid', payment)
727
      column_value_dict.setUIDList('project_uid', project)
728
      column_value_dict.setUIDList('funding_uid', funding)
729
      column_value_dict.setUIDList('payment_request_uid', payment_request)
730
      column_value_dict.setUIDList('function_uid', function)
731 732 733

      sql_kw['transformed_uid'] = self._generatePropertyUidList(transformed_resource)

734
      if column_value_dict.setUIDList('section_uid', section):
735
        sql_kw['section_filtered'] = 1
736 737 738 739 740 741 742 743
      column_value_dict.setUIDList('mirror_section_uid', mirror_section)
      column_value_dict.setUIDList('variation_text', variation_text,
                                   as_text=1)
      column_value_dict.setUIDList('sub_variation_text', sub_variation_text,
                                   as_text=1)
      new_kw['column_value_dict'] = column_value_dict.copy()

      related_key_dict = DictMixIn()
744
      # category membership
745 746
      related_key_dict.setUIDList('resource_category_uid', resource_category)
      related_key_dict.setUIDList('node_category_uid', node_category)
747
      related_key_dict.setUIDList('project_category_uid', project_category)
748
      related_key_dict.setUIDList('funding_category_uid', funding_category)
749
      related_key_dict.setUIDList('payment_request_category_uid', payment_request_category)
750
      related_key_dict.setUIDList('function_category_uid', function_category)
751 752 753
      related_key_dict.setUIDList('payment_category_uid', payment_category)
      if related_key_dict.setUIDList('section_category_uid',
                                     section_category):
754
        sql_kw['section_filtered'] = 1
755 756
      related_key_dict.setUIDList('mirror_section_category_uid',
                                  mirror_section_category)
757
      # category strict membership
758 759 760 761
      related_key_dict.setUIDList('resource_category_strict_membership_uid',
                                  resource_category_strict_membership)
      related_key_dict.setUIDList('node_category_strict_membership_uid',
                                  node_category_strict_membership)
762 763
      related_key_dict.setUIDList('project_category_strict_membership_uid',
                                  project_category_strict_membership)
764 765
      related_key_dict.setUIDList('funding_category_strict_membership_uid',
                                  funding_category_strict_membership)
766 767
      related_key_dict.setUIDList('payment_request_category_strict_membership_uid',
                                  payment_request_category_strict_membership)
768 769
      related_key_dict.setUIDList('function_category_strict_membership_uid',
                                  function_category_strict_membership)
770 771 772 773
      related_key_dict.setUIDList('payment_category_strict_membership_uid',
                                  payment_category_strict_membership)
      if related_key_dict.setUIDList('section_category_strict_membership_uid',
                                     section_category_strict_membership):
774
        sql_kw['section_filtered'] = 1
775 776 777
      related_key_dict.setUIDList(
        'mirror_section_category_strict_membership_uid',
        mirror_section_category_strict_membership)
Vincent Pelletier's avatar
Vincent Pelletier committed
778

779
      new_kw['related_key_dict'] = related_key_dict.copy()
780
      new_kw['related_key_dict_passthrough'] = kw
781

782 783 784
      #variation_category_uid_list = self._generatePropertyUidList(variation_category)
      #if len(variation_category_uid_list) :
      #  new_kw['variationCategory'] = variation_category_uid_list
Vincent Pelletier's avatar
Vincent Pelletier committed
785

786
      simulation_dict =  self._getSimulationStateDict(
Vincent Pelletier's avatar
Vincent Pelletier committed
787
                                simulation_state=simulation_state,
788 789 790 791 792
                                omit_transit=omit_transit,
                                input_simulation_state=input_simulation_state,
                                output_simulation_state=output_simulation_state,
                                transit_simulation_state=transit_simulation_state,
                                strict_simulation_state=strict_simulation_state)
793 794
      new_kw['simulation_dict'] = simulation_dict
      omit_dict = self._getOmitDict(omit_input=omit_input,
795 796 797
                                    omit_output=omit_output,
                                    omit_asset_increase=omit_asset_increase,
                                    omit_asset_decrease=omit_asset_decrease)
798
      new_kw['omit_dict'] = omit_dict
799
      if reserved_kw is not None:
800
        if not isinstance(reserved_kw, dict):
Vincent Pelletier's avatar
Vincent Pelletier committed
801
          # Not a dict when taken from URL, so, cast is needed
802 803
          # to make pop method available
          reserved_kw = dict(reserved_kw)
804
        new_reserved_kw = {}
805 806
        reserved_omit_input = reserved_kw.pop('omit_input',0)
        reserved_omit_output = reserved_kw.pop('omit_output',0)
807 808 809 810 811
        new_reserved_kw['omit_dict'] = self._getOmitDict(
                                         omit_input=reserved_omit_input,
                                         omit_output=reserved_omit_output)
        new_reserved_kw['simulation_dict'] = self._getSimulationStateDict(**reserved_kw)
        new_kw['reserved_kw'] = new_reserved_kw
812 813 814 815 816 817 818
      # It is necessary to use here another SQL query (or at least a subquery)
      # to get _DISTINCT_ uid from predicate_category table.
      # Otherwise, by using a where_expression, cells which fit conditions
      # more than one time are counted more than one time, and the resulting
      # inventory is false
      # XXX Perhaps is there a better solution
      add_kw = {}
819
      if variation_category is not None and variation_category:
820 821
        where_expression = self.getPortalObject().portal_categories\
          .buildSQLSelector(
822 823 824 825 826 827 828 829 830 831
            category_list = variation_category,
            query_table = 'predicate_category')
        if where_expression != '':
          add_kw['where_expression'] = where_expression
          add_kw['predicate_category.uid'] = '!=NULL'
          add_kw['group_by_expression'] = 'uid'
          add_query = self.portal_catalog(**add_kw)
          uid_list = []
          for line in add_query:
            uid_list.append(line.uid)
832 833
          new_kw['where_expression'] = '( %s )' % ' OR '.join(
                      ['catalog.uid=%s' % uid for uid in uid_list])
Alexandre Boeglin's avatar
Alexandre Boeglin committed
834

Sebastien Robin's avatar
Sebastien Robin committed
835
      # build the group by expression
836 837 838 839 840 841 842 843 844 845 846 847
      # if we group by a criterion, we also add this criterion to the select
      # expression, unless it is already selected in Resource_zGetInventoryList
      # the caller can also pass select_dict or select_list. select_expression,
      # which is deprecated in ZSQLCatalog is not supported here.
      select_dict = kw.get('select_dict', {})
      # we support select_list, if passed
      select_list = kw.get('select_list', [])
      for select_key in kw.get('select_list', []):
        select_dict[select_key] = None
      new_kw['select_dict'] = select_dict
      related_key_select_expression_list = []

848 849
      column_group_by_expression_list = []
      related_key_group_by_expression_list = []
850
      if group_by_node:
851
        column_group_by_expression_list.append('node_uid')
852
      if group_by_mirror_node:
853
        column_group_by_expression_list.append('mirror_node_uid')
854
      if group_by_section:
855
        column_group_by_expression_list.append('section_uid')
856
      if group_by_mirror_section:
857
        column_group_by_expression_list.append('mirror_section_uid')
858
      if group_by_payment:
859
        column_group_by_expression_list.append('payment_uid')
860
      if group_by_sub_variation:
861
        column_group_by_expression_list.append('sub_variation_text')
862
      if group_by_variation:
863
        column_group_by_expression_list.append('variation_text')
864
      if group_by_movement:
865
        column_group_by_expression_list.append('uid')
866
      if group_by_resource:
867 868 869
        column_group_by_expression_list.append('resource_uid')
      if group_by_project:
        column_group_by_expression_list.append('project_uid')
870 871
      if group_by_funding:
        column_group_by_expression_list.append('funding_uid')
872 873
      if group_by_payment_request:
        column_group_by_expression_list.append('payment_request_uid')
874 875
      if group_by_function:
        column_group_by_expression_list.append('function_uid')
876
      if group_by_date:
877 878 879 880 881 882 883
        column_group_by_expression_list.append('date')

      if column_group_by_expression_list:
        new_kw['column_group_by'] = column_group_by_expression_list

      if group_by_section_category:
        related_key_group_by_expression_list.append('section_category_uid')
884
        related_key_select_expression_list.append('section_category_uid')
885 886 887
      if group_by_section_category_strict_membership:
        related_key_group_by_expression_list.append(
            'section_category_strict_membership_uid')
888 889
        related_key_select_expression_list.append(
            'section_category_strict_membership_uid')
890 891
      if group_by_mirror_section_category:
        related_key_group_by_expression_list.append('mirror_section_category_uid')
892
        related_key_select_expression_list.append('mirror_section_category_uid')
893 894 895
      if group_by_mirror_section_category_strict_membership:
        related_key_group_by_expression_list.append(
            'mirror_section_category_strict_membership_uid')
896 897
        related_key_select_expression_list.append(
            'mirror_section_category_strict_membership_uid')
898 899
      if group_by_node_category:
        related_key_group_by_expression_list.append('node_category_uid')
900
        related_key_select_expression_list.append('node_category_uid')
901 902 903
      if group_by_node_category_strict_membership:
        related_key_group_by_expression_list.append(
            'node_category_strict_membership_uid')
904 905
        related_key_select_expression_list.append(
            'node_category_strict_membership_uid')
906 907 908 909 910
      if group_by_mirror_node_category:
        related_key_group_by_expression_list.append('mirror_node_category_uid')
      if group_by_mirror_node_category_strict_membership:
        related_key_group_by_expression_list.append(
            'mirror_node_category_strict_membership_uid')
911 912
        related_key_select_expression_list.append(
            'mirror_node_category_strict_membership_uid')
913 914
      if group_by_payment_category:
        related_key_group_by_expression_list.append('payment_category_uid')
915
        related_key_select_expression_list.append('payment_category_uid')
916 917 918
      if group_by_payment_category_strict_membership:
        related_key_group_by_expression_list.append(
            'payment_category_strict_membership_uid')
919 920
        related_key_select_expression_list.append(
            'payment_category_strict_membership_uid')
921 922
      if group_by_function_category:
        related_key_group_by_expression_list.append('function_category_uid')
923
        related_key_select_expression_list.append('function_category_uid')
924 925 926
      if group_by_function_category_strict_membership:
        related_key_group_by_expression_list.append(
            'function_category_strict_membership_uid')
927
        related_key_select_expression_list.append(
928
            'function_category_strict_membership_uid')
929 930 931 932 933 934 935 936
      if group_by_project_category:
        related_key_group_by_expression_list.append('project_category_uid')
        related_key_select_expression_list.append('project_category_uid')
      if group_by_project_category_strict_membership:
        related_key_group_by_expression_list.append(
            'project_category_strict_membership_uid')
        related_key_select_expression_list.append(
            'project_category_strict_membership_uid')
937 938 939 940 941 942 943 944
      if group_by_funding_category:
        related_key_group_by_expression_list.append('funding_category_uid')
        related_key_select_expression_list.append('funding_category_uid')
      if group_by_funding_category_strict_membership:
        related_key_group_by_expression_list.append(
            'funding_category_strict_membership_uid')
        related_key_select_expression_list.append(
            'funding_category_strict_membership_uid')
945 946 947 948 949 950 951 952
      if group_by_payment_category:
        related_key_group_by_expression_list.append('payment_request_category_uid')
        related_key_select_expression_list.append('payment_request_category_uid')
      if group_by_payment_request_category_strict_membership:
        related_key_group_by_expression_list.append(
            'payment_request_category_strict_membership_uid')
        related_key_select_expression_list.append(
            'payment_request_category_strict_membership_uid')
953

954 955
      if related_key_group_by_expression_list:
        new_kw['related_key_group_by'] = related_key_group_by_expression_list
956 957 958
      if related_key_select_expression_list:
        new_kw['related_key_select_expression_list'] =\
                related_key_select_expression_list
959 960

      return sql_kw, new_kw
961

Jean-Paul Smets's avatar
Jean-Paul Smets committed
962
    #######################################################
963 964
    # Inventory management
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
965
                              'getInventory')
966
    def getInventory(self, src__=0, simulation_period='', **kw):
967
      """
968 969
      Returns an inventory of a single or multiple resources on a single or
      multiple nodes as a single float value
970

971
      from_date (>=) - only take rows which date is >= from_date
972

973
      to_date   (<)  - only take rows which date is < to_date
974 975 976 977 978

      at_date   (<=) - only take rows which date is <= at_date

      resource (only in generic API in simulation)

979 980
      node           -  only take rows in stock table which node_uid is
                        equivalent to node
981

982 983
      payment        -  only take rows in stock table which payment_uid is
                        equivalent to payment
984

985 986
      section        -  only take rows in stock table which section_uid is
                        equivalent to section
987

988 989
      mirror_section -  only take rows in stock table which mirror_section_uid is
                        mirror_section
990

991 992
      resource_category  -  only take rows in stock table which
                        resource_uid is member of resource_category
993

994 995
      node_category   - only take rows in stock table which node_uid is
                        member of section_category
996

997 998
      payment_category   -  only take rows in stock table which payment_uid
                            is in section_category
999

1000
      section_category -  only take rows in stock table which section_uid is
1001 1002
                          member of section_category

Vincent Pelletier's avatar
Vincent Pelletier committed
1003
      mirror_section_category - only take rows in stock table which
1004
                                mirror_section_uid is member of
1005
                                mirror_section_category
1006 1007 1008 1009 1010 1011

      node_filter     - only take rows in stock table which node_uid
                        matches node_filter

      payment_filter  - only take rows in stock table which payment_uid
                        matches payment_filter
1012

1013 1014 1015
      section_filter  - only take rows in stock table which section_uid
                        matches section_filter

1016 1017
      mirror_section_filter - only take rows in stock table which
                              mirror_section_uid matches mirror_section_filter
1018

1019 1020 1021 1022 1023 1024
      variation_text -  only take rows in stock table with specified
                        variation_text.
                        This needs to be extended with some kind of
                        variation_category ?
                        XXX this way of implementing variation selection is far
                        from perfect
1025

1026 1027
      sub_variation_text - only take rows in stock table with specified
                        variation_text
1028

1029 1030
      variation_category - variation or list of possible variations (it is not
                        a cross-search ; SQL query uses OR)
1031 1032 1033

      simulation_state - only take rows with specified simulation_state

1034
      transit_simulation_state - specifies which states are transit states
1035

1036
      omit_transit   -  do not evaluate transit_simulation_state
1037

1038 1039
      input_simulation_state - only take rows with specified simulation_state
                        and quantity > 0
1040

1041 1042
      output_simulation_state - only take rows with specified simulation_state
                        and quantity < 0
1043

1044 1045 1046
      ignore_variation -  do not take into account variation in inventory
                        calculation (useless on getInventory, but useful on
                        getInventoryList)
1047

1048 1049
      standardise    -  provide a standard quantity rather than an SKU (XXX
                        not implemented yet)
1050

1051 1052
      omit_simulation - doesn't take into account simulation movements

1053 1054 1055 1056
      only_accountable - Only take into account accountable movements. By
                        default, only movements for which isAccountable() is
                        true will be taken into account.

1057 1058 1059 1060 1061
      omit_input     -  doesn't take into account movement with quantity > 0

      omit_output    -  doesn't take into account movement with quantity < 0

      omit_asset_increase - doesn't take into account movement with asset_price > 0
1062

1063
      omit_asset_decrease - doesn't take into account movement with asset_price < 0
1064 1065 1066

      selection_domain, selection_report - see ListBox

1067 1068
      group_by_variation - (useless on getInventory, but useful on
                        getInventoryList)
Sebastien Robin's avatar
Sebastien Robin committed
1069

1070 1071
      group_by_node  -  (useless on getInventory, but useful on
                        getInventoryList)
Sebastien Robin's avatar
Sebastien Robin committed
1072

1073 1074
      group_by_mirror_node - (useless on getInventory, but useful on
                        getInventoryList)
Sebastien Robin's avatar
Sebastien Robin committed
1075

1076 1077
      group_by_sub_variation - (useless on getInventory, but useful on
                        getInventoryList)
1078

1079 1080 1081
      group_by_movement - (useless on getInventory, but useful on
                        getInventoryList)

1082 1083
      precision - the precision used to round quantities and prices.

1084 1085 1086 1087
      metric_type   - convert the results to a specific metric_type

      quantity_unit - display results using this specific quantity unit

1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098
      transformed_resource - one, or several resources. list resources that can
                             be produced using those resources as input in a
                             transformation.
                             relative_resource_url for each returned line will
                             point to the transformed resource, while the stock
                             will be the stock of the produced resource,
                             expressed in number of transformed resources.
      transformed_variation_text - to be used with transformed_resource, to
                                   to refine the transformation selection only
                                   to those using variated resources as input.

1099 1100
      **kw           -  if we want extended selection with more keywords (but
                        bad performance) check what we can do with
1101
                        buildSQLQuery
1102

1103 1104
      NOTE: we may want to define a parameter so that we can select the kind of
      inventory statistics we want to display (ex. sum, average, cost, etc.)
1105
      """
1106 1107 1108 1109 1110 1111 1112 1113
      # JPS: this is a hint for implementation of xxx_filter arguments
      # node_uid_count = portal_catalog.countResults(**node_filter)
      # if node_uid_count not too big:
      #   node_uid_list = cache(portal_catalog(**node_filter))
      #   pass this list to ZSQL method
      # else:
      #   build a table in MySQL
      #   and join that table with the stock table
1114
      method = getattr(self,'get%sInventoryList' % simulation_period)
1115 1116
      kw['ignore_group_by'] = 1
      result = method(inventory_list=0, src__=src__, **kw)
1117
      if src__:
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1118 1119
        return result

1120 1121
      total_result = 0.0
      if len(result) > 0:
1122 1123
        if len(result) != 1:
          raise ValueError, 'Sorry we must have only one'
1124 1125 1126 1127 1128 1129 1130 1131
        result = result[0]

        if hasattr(result, "converted_quantity"):
          total_result = result.converted_quantity
        else:
          inventory = result.total_quantity
          if inventory is not None:
            total_result = inventory
1132 1133

      return total_result
Jean-Paul Smets's avatar
Jean-Paul Smets committed
1134

1135
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1136
                              'getCurrentInventory')
1137
    def getCurrentInventory(self, **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1138 1139 1140
      """
      Returns current inventory
      """
1141
      return self.getInventory(simulation_period='Current', **kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1142

1143
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1144
                              'getAvailableInventory')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1145 1146 1147
    def getAvailableInventory(self, **kw):
      """
      Returns available inventory
1148
      (current inventory - reserved_inventory)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1149
      """
1150
      return self.getInventory(simulation_period='Available', **kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1151

1152
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1153
                              'getFutureInventory')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1154 1155 1156 1157
    def getFutureInventory(self, **kw):
      """
      Returns future inventory
      """
1158
      return self.getInventory(simulation_period='Future', **kw)
Vincent Pelletier's avatar
Vincent Pelletier committed
1159 1160

    def _getDefaultGroupByParameters(self, ignore_group_by=0,
1161 1162
        group_by_node=0, group_by_mirror_node=0,
        group_by_section=0, group_by_mirror_section=0,
1163
        group_by_payment=0, group_by_project=0, group_by_funding=0,
1164
        group_by_function=0,
1165
        group_by_variation=0, group_by_sub_variation=0,
1166
        group_by_movement=0, group_by_date=0,
1167 1168
        group_by_section_category=0,
        group_by_section_category_strict_membership=0,
1169
        group_by_resource=None,
1170
        movement_list_mode=0,
1171
        group_by=None,
1172
        **ignored):
1173 1174
      """
      Set defaults group_by parameters
1175 1176 1177 1178 1179

      If ignore_group_by is true, this function returns an empty dict.

      If any group-by is provided, automatically group by resource aswell
      unless group_by_resource is explicitely set to false.
1180 1181 1182
      If no group by is provided, use the default group by: movement, node and
      resource, unless movement_list_mode is true, in that case, group by
      movement, node, resource and date (this is historically the default in
1183 1184 1185 1186
      getMovementHistoryList), section, mirror_section and payment (this is to
      make sure two lines will appear when we are, for instance both source and
      destination, implementation might not be optimal, because it uses lots of
      group by statements in SQL).
1187
      """
1188
      new_group_by_dict = {}
1189
      if not ignore_group_by and group_by is None:
1190
        if group_by_node or group_by_mirror_node or group_by_section or \
1191
           group_by_project or group_by_funding or group_by_function or \
1192
           group_by_mirror_section or group_by_payment or \
1193
           group_by_sub_variation or group_by_variation or \
1194 1195
           group_by_movement or group_by_date or group_by_section_category or\
           group_by_section_category_strict_membership:
1196 1197
          if group_by_resource is None:
            group_by_resource = 1
1198
          new_group_by_dict['group_by_resource'] = group_by_resource
1199
        elif group_by_resource is None:
1200 1201 1202
          new_group_by_dict['group_by_movement'] = 1
          new_group_by_dict['group_by_node'] = 1
          new_group_by_dict['group_by_resource'] = 1
1203 1204
          if movement_list_mode:
            new_group_by_dict['group_by_date'] = 1
1205 1206 1207 1208
            new_group_by_dict['group_by_mirror_node'] = 1
            new_group_by_dict['group_by_section'] = 1
            new_group_by_dict['group_by_mirror_section'] = 1
            new_group_by_dict['group_by_payment'] = 1
1209
      return new_group_by_dict
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1210

1211
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1212
                              'getInventoryList')
1213 1214
    def getInventoryList(self, src__=0, optimisation__=True,
                         ignore_variation=0, standardise=0,
Vincent Pelletier's avatar
Vincent Pelletier committed
1215
                         omit_simulation=0,
1216
                         only_accountable=True,
1217
                         default_stock_table='stock',
1218
                         selection_domain=None, selection_report=None,
Vincent Pelletier's avatar
Vincent Pelletier committed
1219
                         statistic=0, inventory_list=1,
1220
                         precision=None, connection_id=None,
1221
                         **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1222
      """
1223 1224 1225
        Returns a list of inventories for a single or multiple
        resources on a single or multiple nodes, grouped by resource,
        node, section, etc. Every line defines an inventory value for
Romain Courteaud's avatar
Romain Courteaud committed
1226
        a given group of resource, node, section.
1227 1228
        NOTE: we may want to define a parameter so that we can select
        the kind of inventory statistics we want to display (ex. sum,
Romain Courteaud's avatar
Romain Courteaud committed
1229
        average, cost, etc.)
1230 1231 1232 1233 1234 1235 1236 1237 1238 1239

        Optimisation queries.
        Optimisation of a stock lookup is done to avoid a table scan
        of all lines corresponding to a given node, section or payment,
        because they grow with time and query time should not.
        First query: Fetch fitting full inventory dates.
          For each node, section or payment, find the first anterior full
          inventory.
        Second query: Fetch full inventory amounts.
          Fetch values of inventory identified in the first query.
1240
        Third query: Classic stock table read.
1241 1242 1243 1244 1245 1246 1247 1248 1249 1250 1251 1252 1253
          Fetch all rows in stock table which are posterior to the inventory.
        Final result
          Add results of the second and third queries, and return it.

        Missing optimisations:
         - In a getInventory case where everything is specified for the
           resource, it's not required for the inventory to be full, it
           just need to be done for the right resource.
           If the resource isn't completely defined, we require inventory
           to be full, which is implemented.
         - Querying multiple nodes/categories/payments in one call prevents
           from using optimisation, it should be equivalent to multiple calls
           on individual nodes/categories/payments.
Vincent Pelletier's avatar
Vincent Pelletier committed
1254
         -
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1255
      """
1256 1257 1258 1259
      getCategory = self.getPortalObject().portal_categories.getCategoryUid

      result_column_id_dict = {}

1260
      metric_type = kw.pop('metric_type', None)
1261 1262
      quantity_unit = kw.pop('quantity_unit', None)
      quantity_unit_uid = None
1263

1264 1265 1266 1267 1268 1269
      if quantity_unit is not None:

        if isinstance(quantity_unit, str):
          quantity_unit_uid = getCategory(quantity_unit, 'quantity_unit')
          if quantity_unit_uid is not None:
            result_column_id_dict['converted_quantity'] = None
1270 1271 1272
            if metric_type is None:
              # use the default metric type
              metric_type = quantity_unit.split("/", 1)[0]
Kazuhiko Shiozaki's avatar
Kazuhiko Shiozaki committed
1273
        elif isinstance(quantity_unit, (int, float)):
1274 1275 1276 1277 1278 1279 1280 1281 1282 1283 1284 1285 1286 1287
          # quantity_unit used to be a numerical parameter..
          raise ValueError('Numeric values for quantity_unit are not supported')


      convert_quantity_result = False
      if metric_type is not None:
        metric_type_uid = getCategory(metric_type, 'metric_type')

        if metric_type_uid is not None:
          convert_quantity_result = True
          kw['metric_type_uid'] = Query(
                                    metric_type_uid=metric_type_uid,
                                    table_alias_list=(("measure", "measure"),))

1288 1289
      if src__:
        sql_source_list = []
1290
      # If no group at all, give a default sort group by
1291
      kw.update(self._getDefaultGroupByParameters(**kw))
1292 1293 1294 1295 1296 1297 1298 1299 1300 1301 1302 1303 1304 1305 1306 1307 1308 1309
      base_inventory_kw = {
        'stock_table_id': default_stock_table,
        'src__': src__,
        'ignore_variation': ignore_variation,
        'standardise': standardise,
        'omit_simulation': omit_simulation,
        'only_accountable': only_accountable,
        'selection_domain': selection_domain,
        'selection_report': selection_report,
        'precision': precision,
        'inventory_list': inventory_list,
        'connection_id': connection_id,
        'statistic': statistic,
        'convert_quantity_result': convert_quantity_result,
        'quantity_unit_uid': quantity_unit_uid,
      }
      # Get cached data
      if getattr(self, "Resource_zGetInventoryCacheResult", None) is not None and \
1310 1311
              optimisation__ and (not kw.get('from_date')) and \
              (bool(kw.get("at_date")) ^ bool(kw.get("to_date"))) and \
1312 1313 1314 1315 1316 1317 1318 1319 1320 1321 1322 1323 1324 1325 1326 1327 1328 1329 1330 1331 1332 1333 1334 1335 1336 1337 1338 1339 1340 1341
              'transformed_resource' not in kw:
        # Here is the different kind of date
        # from_date : >=
        # to_date   : <
        # at_date   : <=
        # As we just have from_date, it means that we must use
        # the to_date for the cache in order to avoid double computation
        # of the same line
        at_date = kw.pop("at_date", None)
        if at_date is None:
          to_date = kw.pop("to_date")
        else:
          # add one second so that we can use to_date
          to_date = at_date + MYSQL_MIN_DATETIME_RESOLUTION
        try:
          cached_result, cached_date = self._getCachedInventoryList(
              to_date=to_date,
              sql_kw=kw,
              **base_inventory_kw)
        except StockOptimisationError:
          cached_result = []
          kw['to_date'] = to_date
        else:
          if src__:
            sql_source_list.extend(cached_result)
          # Now must generate query for date diff
          kw['to_date'] = to_date
          kw['from_date'] = cached_date
      else:
        cached_result = []
1342
      sql_kw, new_kw = self._generateKeywordDict(**kw)
1343 1344 1345 1346 1347 1348 1349 1350 1351
      # Copy kw content as _generateSQLKeywordDictFromKeywordDict
      # remove some values from it
      try:
        new_kw_copy = deepcopy(new_kw)
      except TypeError:
        # new_kw contains wrong parameters
        # as optimisation has already been disable we
        # do not care about the deepcopy
        new_kw_copy = new_kw
1352
      stock_sql_kw = self._generateSQLKeywordDictFromKeywordDict(
1353 1354 1355 1356
          table=default_stock_table, sql_kw=sql_kw, new_kw=new_kw_copy)
      stock_sql_kw.update(base_inventory_kw)
      delta_result = self.Resource_zGetInventoryList(
          **stock_sql_kw)
1357
      if src__:
1358
        sql_source_list.append(delta_result)
1359
        result = ';\n-- NEXT QUERY\n'.join(sql_source_list)
1360 1361 1362 1363 1364
      else:
        if cached_result:
            result = self._addBrainResults(delta_result, cached_result, new_kw)
        else:
            result = delta_result
1365
      return result
Romain Courteaud's avatar
Romain Courteaud committed
1366

1367 1368 1369 1370 1371 1372 1373 1374 1375 1376 1377 1378 1379 1380 1381 1382 1383 1384 1385 1386 1387 1388 1389 1390 1391 1392 1393 1394 1395 1396 1397 1398 1399 1400 1401 1402 1403 1404 1405 1406 1407 1408 1409 1410 1411 1412 1413 1414 1415 1416 1417 1418 1419 1420 1421 1422 1423 1424 1425 1426 1427 1428 1429 1430 1431 1432 1433 1434 1435 1436 1437 1438 1439 1440 1441 1442 1443 1444 1445 1446 1447 1448 1449 1450 1451 1452 1453 1454 1455 1456 1457 1458 1459 1460 1461 1462 1463 1464 1465 1466 1467 1468 1469 1470 1471 1472 1473 1474 1475 1476 1477 1478 1479 1480 1481 1482 1483 1484 1485 1486 1487 1488 1489 1490 1491 1492 1493 1494 1495 1496 1497 1498 1499 1500 1501 1502 1503 1504 1505 1506 1507 1508 1509 1510 1511 1512 1513 1514 1515 1516 1517 1518 1519 1520 1521 1522 1523 1524 1525 1526 1527 1528 1529 1530 1531 1532 1533 1534 1535 1536 1537 1538 1539 1540 1541 1542 1543 1544 1545 1546 1547 1548 1549 1550 1551 1552 1553 1554 1555 1556 1557 1558 1559 1560 1561 1562 1563 1564 1565 1566 1567 1568 1569 1570
    def getInventoryCacheLag(self):
      """
      Returns a duration, in days, for stock cache management.
      If data in stock cache is older than lag compared to query's date
      (at_date or to_date), then it becomes a "soft miss": use found value,
      but add a new entry to cache at query's date minus half the lag.
      So this value should be:
      - Small enough that few enough rows need to be table-scanned for
        average queries (probably queries against current date).
      - Large enough that few enough documents get modified past that date,
        otherwise cache entries would be removed from cache all the time.
      """
      return self.SimulationTool_getInventoryCacheLag()

    def _getCachedInventoryList(self, to_date, sql_kw, stock_table_id, src__=False, **kw):
      """
      Try to get a cached inventory list result
      If not existing, fill the cache
      """
      Resource_zGetInventoryList = self.Resource_zGetInventoryList
      # Generate the SQL source without date parameter
      # This will be the cache key
      try:
          no_date_kw = deepcopy(sql_kw)
      except TypeError:
          LOG("SimulationTool._getCachedInventoryList", WARNING,
              "Failed copying sql_kw, disabling stock cache",
              error=exc_info())
          raise StockOptimisationError
      no_date_sql_kw, no_date_new_kw = self._generateKeywordDict(**no_date_kw)
      no_date_stock_sql_kw = self._generateSQLKeywordDictFromKeywordDict(
        table=stock_table_id, sql_kw=no_date_sql_kw,
        new_kw=no_date_new_kw)
      kw.update(no_date_stock_sql_kw)
      if src__:
        sql_source_list = []
      # Generate the cache key (md5 of query source)
      sql_text_hash = md5(Resource_zGetInventoryList(
        stock_table_id=stock_table_id,
        src__=1,
        **kw)).digest()
      # Try to get result from cache
      Resource_zGetInventoryCacheResult = self.Resource_zGetInventoryCacheResult
      inventory_cache_kw = {
        'query': sql_text_hash,
        'date': to_date,
      }
      try:
          cached_sql_result = Resource_zGetInventoryCacheResult(**inventory_cache_kw)
      except ProgrammingError:
          # First use of the optimisation, we need to create the table
          LOG("SimulationTool._getCachedInventoryList", INFO,
              "Creating inventory cache stock")
          if src__:
              sql_source_list.append(self.SimulationTool_zCreateInventoryCache(src__=1))
          else:
              self.SimulationTool_zCreateInventoryCache()
          cached_sql_result = None

      if src__:
        sql_source_list.append(Resource_zGetInventoryCacheResult(src__=1, **inventory_cache_kw))
      if cached_sql_result:
        brain_result = loads(cached_sql_result[0].result)
        # Rebuild the brains
        cached_result = Results(
          (brain_result['items'], brain_result['data']),
          brains=getBrain(
            Resource_zGetInventoryList.class_file_,
            Resource_zGetInventoryList.class_name_,
          ),
          parent=self,
        )
      else:
        cached_result = []
      cache_lag = self.getInventoryCacheLag()
      if cached_sql_result and to_date - DateTime(cached_sql_result[0].date) < cache_lag:
        cached_date = DateTime(cached_sql_result[0].date)
        result = cached_result
      else:
        # Cache miss, or hit with old data: store a new entry in cache.
        # Don't store it at to_date, as it risks being flushed soon (ie, when
        # any document older than to_date gets reindexed in stock table).
        # Don't store it at to_date - cache_lag, as it would risk expiring
        # soon as we store it (except if to_date is fixed for many queries,
        # which we cannot tell here).
        # So store it at half the cache_lag before to_date.
        cached_date = to_date - cache_lag / 2
        new_cache_kw = deepcopy(sql_kw)
        if cached_result:
          # We can use cached result to generate new cache result
          new_cache_kw['from_date'] = DateTime(cached_sql_result[0].date)
        sql_kw, new_kw = self._generateKeywordDict(
          to_date=cached_date,
          **new_cache_kw)
        kw.update(self._generateSQLKeywordDictFromKeywordDict(
            table=stock_table_id,
            sql_kw=sql_kw,
            new_kw=new_kw,
          )
        )
        new_result = Resource_zGetInventoryList(
          stock_table_id=stock_table_id,
          src__=src__,
          **kw)
        if src__:
          sql_source_list.append(new_result)
        else:
          result = self._addBrainResults(new_result, cached_result, new_kw)
          self.Resource_zInsertInventoryCacheResult(
            query=sql_text_hash,
            date=cached_date,
            result=dumps({
              'items': result.__items__,
              'data': result._data,
            }),
          )
      if src__:
        result = sql_source_list
      return result, cached_date

    def _addBrainResults(self, first_result, second_result, new_kw):
      """
      Build a Results which is the addition of two other result
      """
      # This part defined key to group lines from different Results
      group_by_id_list = []
      group_by_id_list_append = group_by_id_list.append

      for group_by_id in new_kw.get('column_group_by', []):
        if group_by_id == 'uid':
          group_by_id_list_append('stock_uid')
        else:
          group_by_id_list_append(group_by_id)
      # Add related key group by
      if 'select_list' in new_kw.get("related_key_dict_passthrough", []):
        for group_by_id in new_kw["related_key_dict_passthrough"]['group_by']:
          if group_by_id in new_kw["related_key_dict_passthrough"]["select_list"]:
            group_by_id_list_append(group_by_id)
          else:
            # XXX-Aurel : to review & change, must prevent coming here before
            raise ValueError, "Impossible to group by %s" %(group_by_id)
      elif "group_by" in new_kw.get("related_key_dict_passthrough", []):
        raise ValueError, "Impossible to group by %s" %(new_kw["related_key_dict_passthrough"]['group_by'],)

      if len(group_by_id_list):
        def getInventoryListKey(line):
          """
          Generate a key based on values used in SQL group_by
          """
          return tuple([line[x] for x in group_by_id_list])

      else:
        def getInventoryListKey(line):
          """
          Return a dummy key, all line will be summed
          """
          return "dummy"
      result_column_id_dict = {
        'inventory': None,
        'total_quantity': None,
        'total_price': None
      }
      def addLineValues(line_a=None, line_b=None):
        """
        Add columns of 2 lines and return a line with same
        schema. If one of the parameters is None, returns the
        other parameters.

        Arithmetic modifications on additions:
        None + x = x
        None + None = None
        """
        if line_a is None:
          return line_b
        if line_b is None:
          return line_a
        # Create a new Shared.DC.ZRDB.Results.Results.__class__
        # instance to add the line values.
        # the logic for the 5 lines below is taken from
        # Shared.DC.ZRDB.Results.Results.__getitem__
        Result = line_a.__class__
        parent = line_a.aq_parent
        result = Result((), parent)
        try:
          # We must copy the path so that getObject works
          setattr(result, 'path', line_a.path)
        except ValueError: # XXX: ValueError ? really ?
          # getInventory return no object, so no path available
          pass
        if parent is not None:
          result = result.__of__(parent)
        for key in line_a.__record_schema__:
          value = line_a[key]
          if key in result_column_id_dict:
            value_b = line_b[key]
            if None not in (value, value_b):
              result[key] = value + value_b
            elif value is not None:
              result[key] = value
            else:
              result[key] = value_b
          elif line_a[key] == line_b[key]:
            result[key] = line_a[key]
          elif key not in ('date', 'stock_uid', 'path'):
Vincent Pelletier's avatar
Vincent Pelletier committed
1571 1572 1573 1574 1575 1576 1577 1578 1579 1580 1581 1582 1583
            # There are 2 possible reasons to end up here:
            # - key corresponds to a projected column for which are neither
            #   known aggregated columns (in which case they should be in
            #   result_column_id_dict) nor part of grouping columns, and the
            #   result happens to be unstable. There are cases in ERP5 where
            #   such result is suposed to be stable, for example
            #   group_by=('xxx_uid'), selection_list=('xxx_path') because the
            #   relation is bijective (although the database doesn't know it).
            #   These should result in stable results (but don't necessarily
            #   do, ex: xxx_title when object title has been changed between
            #   cache fill and cache lookup).
            # - line_a and line_b are indeed mismatched, and code calling us
            #   has a bug.
1584 1585 1586 1587 1588 1589 1590 1591 1592 1593 1594 1595 1596 1597 1598 1599 1600 1601 1602 1603 1604 1605 1606 1607 1608 1609 1610 1611 1612 1613 1614 1615 1616 1617 1618 1619 1620 1621 1622 1623 1624 1625 1626 1627 1628 1629 1630 1631 1632 1633 1634 1635 1636
            LOG('InventoryTool.getInventoryList.addLineValues',
              PROBLEM,
              'mismatch for %s column: %s and %s' % (
                key, line_a[key], line_b[key]))
        return result
      # Add lines
      inventory_list_dict = {}
      for line_list in (first_result, second_result):
        for line in line_list:
          line_key = getInventoryListKey(line)
          line_a = inventory_list_dict.get(line_key)
          inventory_list_dict[line_key] = addLineValues(line_a, line)
      sorted_inventory_list = inventory_list_dict.values()
      # Sort results manually when required
      sort_on = new_kw.get('sort_on')
      if sort_on:
        def cmp_inventory_line(line_a, line_b):
          """
            Compare 2 inventory lines and sort them according to
            sort_on parameter.
          """
          result = 0
          for key, sort_direction in sort_on:
            try:
              result = cmp(line_a[key], line_b[key])
            except KeyError:
              raise Exception('Impossible to sort result since columns sort '
                'happens on are not available in result: %r' % (key, ))
            if result:
              if not sort_direction.upper().startswith('A'):
                # Default sort is ascending, if a sort is given and
                # it does not start with an 'A' then reverse sort.
                # Tedious syntax checking is MySQL's job, and
                # happened when queries were executed.
                result *= -1
              break
          return result
        sorted_inventory_list.sort(cmp_inventory_line)
      # Brain is rebuild properly using tuple not r instance
      column_list = first_result._searchable_result_columns()
      column_name_list = [x['name'] for x in column_list]
      # Rebuild a result object based on added results
      Resource_zGetInventoryList = self.Resource_zGetInventoryList
      return Results(
        (column_list, tuple([tuple([getattr(y, x) for x in column_name_list]) \
          for y in sorted_inventory_list])),
        parent=self,
        brains=getBrain(
          Resource_zGetInventoryList.class_file_,
          Resource_zGetInventoryList.class_name_,
        ),
      )

1637 1638
    security.declareProtected(Permissions.AccessContentsInformation,
                              'getConvertedInventoryList')
1639
    def getConvertedInventoryList(self, simulation_period='', **kw):
1640 1641 1642 1643 1644
      """
      Return list of inventory with a 'converted_quantity' additional column,
      which contains the sum of measurements for the specified metric type,
      expressed in the 'quantity_unit' unit.

1645 1646
      metric_type   - category
      quantity_unit - category
1647 1648
      """

1649 1650
      warn('getConvertedInventoryList is Deprecated, use ' \
           'getInventory instead.', DeprecationWarning)
1651 1652

      method = getattr(self,'get%sInventoryList' % simulation_period)
1653 1654

      return method(**kw)
1655 1656 1657 1658 1659 1660 1661 1662 1663 1664 1665 1666 1667 1668 1669 1670 1671 1672 1673 1674 1675 1676 1677 1678 1679 1680 1681 1682 1683 1684 1685 1686 1687 1688 1689

    security.declareProtected(Permissions.AccessContentsInformation,
                              'getAllInventoryList')
    def getAllInventoryList(self, src__=0, **kw):
      """
      Returns list of inventory, for all periods.
      Performs 1 SQL request for each simulation state, and merge the results.
      Rename relevant columns with a '${simulation}_' prefix
      (ex: 'total_price' -> 'current_total_price').
      """
      columns = ('total_quantity', 'total_price', 'converted_quantity')

      # Guess the columns to use to identify each row, by looking at the GROUP
      # clause. Note that the call to 'mergeZRDBResults' will crash if the GROUP
      # clause contains a column not requested in the SELECT clause.
      kw.update(self._getDefaultGroupByParameters(**kw), ignore_group_by=1)
      group_by_list = self._generateKeywordDict(**kw)[1].get('group_by', ())

      results = []
      edit_result = {}
      get_false_value = lambda row, column_name: row.get(column_name) or 0

      for simulation in 'current', 'available', 'future':
        method = getattr(self, 'get%sInventoryList' % simulation.capitalize())
        rename = {'inventory': None} # inventory column is deprecated
        for column in columns:
          rename[column] = new_name = '%s_%s' % (simulation, column)
          edit_result[new_name] = get_false_value
        results += (method(src__=src__, **kw), rename),

      if src__:
        return ';\n-- NEXT QUERY\n'.join(r[0] for r in results)
      return mergeZRDBResults(results, group_by_list, edit_result)


1690
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1691
                              'getCurrentInventoryList')
Vincent Pelletier's avatar
Vincent Pelletier committed
1692
    def getCurrentInventoryList(self, omit_transit=1,
1693
                                transit_simulation_state=None, **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1694
      """
Romain Courteaud's avatar
Romain Courteaud committed
1695
        Returns list of current inventory grouped by section or site
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1696
      """
1697 1698 1699
      portal = self.getPortalObject()
      kw['simulation_state'] = portal.getPortalCurrentInventoryStateList() + \
                               portal.getPortalTransitInventoryStateList()
1700
      if transit_simulation_state is None:
1701
        transit_simulation_state = portal.getPortalTransitInventoryStateList()
1702 1703 1704 1705 1706

      return self.getInventoryList(
                            omit_transit=omit_transit,
                            transit_simulation_state=transit_simulation_state,
                            **kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1707

1708
    security.declareProtected(Permissions.AccessContentsInformation,
1709
                              'getAvailableInventoryList')
1710
    def getAvailableInventoryList(self, omit_transit=1, transit_simulation_state=None, **kw):
1711 1712 1713
      """
        Returns list of current inventory grouped by section or site
      """
1714
      portal = self.getPortalObject()
1715
      if transit_simulation_state is None:
1716 1717 1718 1719
        transit_simulation_state = portal.getPortalTransitInventoryStateList()
      kw['simulation_state'] = portal.getPortalCurrentInventoryStateList() + \
                               portal.getPortalTransitInventoryStateList()
      reserved_kw = {'simulation_state': portal.getPortalReservedInventoryStateList(),
1720 1721 1722 1723
                     'transit_simulation_state': transit_simulation_state,
                     'omit_input': 1}
      return self.getInventoryList(reserved_kw=reserved_kw, omit_transit=omit_transit,
                     transit_simulation_state=transit_simulation_state, **kw)
1724 1725

    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1726
                              'getFutureInventoryList')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1727 1728
    def getFutureInventoryList(self, **kw):
      """
Romain Courteaud's avatar
Romain Courteaud committed
1729
        Returns list of future inventory grouped by section or site
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1730
      """
1731 1732 1733 1734 1735
      portal = self.getPortalObject()
      kw['simulation_state'] = portal.getPortalFutureInventoryStateList() + \
                               portal.getPortalTransitInventoryStateList() + \
                               portal.getPortalReservedInventoryStateList() + \
                               portal.getPortalCurrentInventoryStateList()
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1736 1737
      return self.getInventoryList(**kw)

1738
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1739
                              'getInventoryStat')
1740
    def getInventoryStat(self, simulation_period='', **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1741
      """
1742
      getInventoryStat is the pending to getInventoryList in order to
Romain Courteaud's avatar
Romain Courteaud committed
1743
      provide statistics on getInventoryList lines in ListBox such as:
1744
      total of inventories, number of variations, number of different
Romain Courteaud's avatar
Romain Courteaud committed
1745
      nodes, etc.
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1746
      """
Romain Courteaud's avatar
Romain Courteaud committed
1747
      kw['group_by_variation'] = 0
1748
      method = getattr(self,'get%sInventoryList' % simulation_period)
1749
      return method(statistic=1, inventory_list=0, optimisation__=False,
1750
                                   ignore_group_by=1, **kw)
1751 1752

    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1753
                              'getCurrentInventoryStat')
1754
    def getCurrentInventoryStat(self, **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1755 1756 1757
      """
      Returns statistics of current inventory grouped by section or site
      """
1758
      return self.getInventoryStat(simulation_period='Current', **kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1759

1760
    security.declareProtected(Permissions.AccessContentsInformation,
1761 1762 1763
                              'getAvailableInventoryStat')
    def getAvailableInventoryStat(self, **kw):
      """
1764
      Returns statistics of available inventory grouped by section or site
1765
      """
1766
      return self.getInventoryStat(simulation_period='Available', **kw)
1767 1768

    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1769
                              'getFutureInventoryStat')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1770 1771 1772 1773
    def getFutureInventoryStat(self, **kw):
      """
      Returns statistics of future inventory grouped by section or site
      """
1774
      return self.getInventoryStat(simulation_period='Future', **kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1775

1776
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1777
                              'getInventoryChart')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1778
    def getInventoryChart(self, src__=0, **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1779
      """
1780 1781
      Returns a list of couples derived from getInventoryList in order
      to feed a chart renderer. Each couple consist of a label
1782
      (node, section, payment, combination of node & section, etc.)
Romain Courteaud's avatar
Romain Courteaud committed
1783
      and an inventory value.
1784

1785
      Mostly useful for charts in ERP5 forms.
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1786
      """
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1787
      result = self.getInventoryList(src__=src__, **kw)
1788
      if src__ :
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1789
        return result
1790

1791
      return map(lambda r: (r.node_title, r.total_quantity), result)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1792

1793
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1794
                              'getCurrentInventoryChart')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1795 1796 1797 1798
    def getCurrentInventoryChart(self, **kw):
      """
      Returns list of current inventory grouped by section or site
      """
1799 1800
      kw['simulation_state'] = self.getPortalObject()\
        .getPortalCurrentInventoryStateList()
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1801 1802
      return self.getInventoryChart(**kw)

1803
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1804
                              'getFutureInventoryChart')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1805 1806 1807 1808
    def getFutureInventoryChart(self, **kw):
      """
      Returns list of future inventory grouped by section or site
      """
1809 1810
      portal = self.getPortalObject()
      kw['simulation_state'] = portal.getPortalFutureInventoryStateList() + \
1811
                               portal.getPortalTransitInventoryStateList() + \
1812 1813
                               portal.getPortalReservedInventoryStateList() + \
                               portal.getPortalCurrentInventoryStateList()
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1814 1815
      return self.getInventoryChart(**kw)

1816
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1817
                              'getInventoryAssetPrice')
Vincent Pelletier's avatar
Vincent Pelletier committed
1818
    def getInventoryAssetPrice(self, src__=0,
1819 1820 1821
                               simulation_period='',
                               valuation_method=None,
                               **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1822
      """
1823
      Same thing as getInventory but returns an asset
1824
      price rather than an inventory.
1825 1826 1827 1828 1829 1830 1831 1832 1833 1834 1835

      If valuation method is None, returns the sum of total prices.

      Else it should be a string, in:
        Filo
        Fifo
        WeightedAverage
        MonthlyWeightedAverage
        MovingAverage
      When using a specific valuation method, a resource_uid is expected
      as well as one of (section_uid or node_uid).
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1836
      """
1837 1838 1839 1840 1841 1842
      if valuation_method is None:
        method = getattr(self,'get%sInventoryList' % simulation_period)
        kw['ignore_group_by'] = 1
        result = method( src__=src__, inventory_list=0, **kw)
        if src__ :
          return result
1843

1844 1845 1846 1847
        if len(result) == 0:
          return 0.0

        total_result = 0.0
1848
        for result_line in result:
1849
          if result_line.total_price is not None:
1850
            total_result += result_line.total_price
1851

1852 1853 1854 1855 1856 1857 1858 1859 1860 1861 1862 1863
        return total_result

      if valuation_method not in ('Fifo', 'Filo', 'WeightedAverage',
        'MonthlyWeightedAverage', 'MovingAverage'):
        raise ValueError("Invalid valuation method: %s" % valuation_method)

      assert 'node_uid' in kw or 'section_uid' in kw
      sql_kw = self._generateSQLKeywordDict(**kw)

      if 'section_uid' in kw:
        # ignore internal movements
        sql_kw['where_expression'] += ' AND ' \
1864
          'NOT(stock.section_uid<=>stock.mirror_section_uid)'
1865 1866 1867

      result = self.Resource_zGetAssetPrice(
          valuation_method=valuation_method,
Aurel's avatar
Aurel committed
1868
          src__=src__,
1869 1870
          **sql_kw)

Aurel's avatar
Aurel committed
1871 1872 1873
      if src__ :
        return result

1874 1875
      if len(result) > 0:
        return result[-1].total_asset_price
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1876

1877
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1878
                              'getCurrentInventoryAssetPrice')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1879 1880 1881 1882
    def getCurrentInventoryAssetPrice(self, **kw):
      """
      Returns list of current inventory grouped by section or site
      """
1883
      kw['simulation_state'] = self.getPortalCurrentInventoryStateList()
1884
      return self.getInventoryAssetPrice(simulation_period='Current',**kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1885

1886
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1887
                              'getAvailableInventoryAssetPrice')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1888 1889 1890 1891 1892
    def getAvailableInventoryAssetPrice(self, **kw):
      """
      Returns list of available inventory grouped by section or site
      (current inventory - deliverable)
      """
1893 1894 1895
      portal = self.getPortalObject()
      kw['simulation_state'] = portal.getPortalReservedInventoryStateList() + \
                               portal.getPortalCurrentInventoryStateList()
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1896 1897
      return self.getInventoryAssetPrice(**kw)

1898
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1899
                              'getFutureInventoryAssetPrice')
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1900 1901 1902 1903
    def getFutureInventoryAssetPrice(self, **kw):
      """
      Returns list of future inventory grouped by section or site
      """
1904 1905 1906 1907
      portal = self.getPortalObject()
      kw['simulation_state'] = portal.getPortalFutureInventoryStateList() + \
                               portal.getPortalReservedInventoryStateList() + \
                               portal.getPortalCurrentInventoryStateList()
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1908 1909
      return self.getInventoryAssetPrice(**kw)

1910
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1911
                              'getInventoryHistoryList')
1912
    def getInventoryHistoryList(self, src__=0, ignore_variation=0,
1913 1914
                                standardise=0, omit_simulation=0,
                                only_accountable=True, omit_input=0,
1915
                                omit_output=0, selection_domain=None,
1916
                                selection_report=None, precision=None, **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1917
      """
1918 1919 1920
      Returns a time based serie of inventory values
      for a single or a group of resource, node, section, etc. This is useful
      to list the evolution with time of inventory values (quantity, asset price).
1921 1922 1923

      TODO:
        - make sure getInventoryHistoryList can return
1924
          cumulative values calculated by SQL (JPS)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1925
      """
1926
      sql_kw = self._generateSQLKeywordDict(**kw)
Romain Courteaud's avatar
Romain Courteaud committed
1927
      return self.Resource_getInventoryHistoryList(
1928
                      src__=src__, ignore_variation=ignore_variation,
1929
                      standardise=standardise, omit_simulation=omit_simulation,
1930
                      only_accountable=only_accountable,
Romain Courteaud's avatar
Romain Courteaud committed
1931
                      omit_input=omit_input, omit_output=omit_output,
1932
                      selection_domain=selection_domain,
1933 1934
                      selection_report=selection_report, precision=precision,
                      **sql_kw)
1935

1936
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1937
                              'getInventoryHistoryChart')
1938
    def getInventoryHistoryChart(self, src__=0, ignore_variation=0,
1939
                                 standardise=0, omit_simulation=0,
1940
                                 only_accountable=True,
Romain Courteaud's avatar
Romain Courteaud committed
1941
                                 omit_input=0, omit_output=0,
1942
                                 selection_domain=None,
1943
                                 selection_report=None, precision=None, **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1944
      """
1945 1946 1947 1948 1949
      getInventoryHistoryChart is the pensing to getInventoryHistoryList
      to ease the rendering of time based graphs which show the evolution
      of one ore more inventories. Each item in the serie consists of
      time, value and "colour" (multiple graphs can be drawn for example
      for each variation of a resource)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1950
      """
1951 1952
      sql_kw = self._generateSQLKeywordDict(**kw)

Romain Courteaud's avatar
Romain Courteaud committed
1953
      return self.Resource_getInventoryHistoryChart(
1954
                    src__=src__, ignore_variation=ignore_variation,
1955
                    standardise=standardise, omit_simulation=omit_simulation,
1956
                    only_accountable=only_accountable,
Romain Courteaud's avatar
Romain Courteaud committed
1957
                    omit_input=omit_input, omit_output=omit_output,
1958
                    selection_domain=selection_domain,
1959 1960
                    selection_report=selection_report, precision=precision,
                    **sql_kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1961

1962
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
1963
                              'getMovementHistoryList')
1964
    def getMovementHistoryList(self, src__=0, ignore_variation=0,
1965
                               standardise=0, omit_simulation=0,
1966
                               omit_input=0, omit_output=0,
1967
                               only_accountable=True,
1968
                               omit_asset_increase=0, omit_asset_decrease=0,
1969
                               selection_domain=None, selection_report=None,
1970
                               initial_running_total_quantity=0,
1971
                               initial_running_total_price=0, precision=None,
Romain Courteaud's avatar
Romain Courteaud committed
1972
                               **kw):
1973
      """Returns a list of movements which modify the inventory
1974
      for a single or a group of resource, node, section, etc.
1975 1976 1977
      A running total quantity and a running total price are available on
      brains. The initial values can be passed, in case you want to have an
      "initial summary line".
Alexandre Boeglin's avatar
Alexandre Boeglin committed
1978
      """
1979 1980
      kw['movement_list_mode'] = 1
      kw.update(self._getDefaultGroupByParameters(**kw))
1981
      sql_kw = self._generateSQLKeywordDict(**kw)
1982

Romain Courteaud's avatar
Romain Courteaud committed
1983
      return self.Resource_zGetMovementHistoryList(
1984 1985
                         src__=src__, ignore_variation=ignore_variation,
                         standardise=standardise,
1986
                         omit_simulation=omit_simulation,
1987
                         only_accountable=only_accountable,
Romain Courteaud's avatar
Romain Courteaud committed
1988
                         omit_input=omit_input, omit_output=omit_output,
1989 1990
                         omit_asset_increase=omit_asset_increase,
                         omit_asset_decrease=omit_asset_decrease,
1991
                         selection_domain=selection_domain,
1992 1993 1994 1995 1996
                         selection_report=selection_report,
                         initial_running_total_quantity=
                                  initial_running_total_quantity,
                         initial_running_total_price=
                                  initial_running_total_price,
1997
                         precision=precision, **sql_kw)
1998

1999
    security.declareProtected(Permissions.AccessContentsInformation,
Romain Courteaud's avatar
Romain Courteaud committed
2000
                              'getMovementHistoryStat')
2001
    def getMovementHistoryStat(self, src__=0, ignore_variation=0,
2002 2003
                               standardise=0, omit_simulation=0,
                               only_accountable=True, omit_input=0,
2004
                               omit_output=0, selection_domain=None,
2005
                               selection_report=None, precision=None, **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
2006
      """
Romain Courteaud's avatar
Romain Courteaud committed
2007 2008
      getMovementHistoryStat is the pending to getMovementHistoryList
      for ListBox stat
Alexandre Boeglin's avatar
Alexandre Boeglin committed
2009
      """
2010
      sql_kw = self._generateSQLKeywordDict(**kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
2011
      return self.Resource_zGetInventory(src__=src__,
2012
          ignore_variation=ignore_variation, standardise=standardise,
2013 2014 2015 2016
          omit_simulation=omit_simulation, only_accountable=only_accountable,
          omit_input=omit_input, omit_output=omit_output,
          selection_domain=selection_domain, selection_report=selection_report,
          precision=precision, **sql_kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
2017

Vincent Pelletier's avatar
Vincent Pelletier committed
2018
    security.declareProtected(Permissions.AccessContentsInformation,
2019
                              'getNextNegativeInventoryDate')
2020
    def getNextNegativeInventoryDate(self, src__=0, **kw):
Alexandre Boeglin's avatar
Alexandre Boeglin committed
2021 2022 2023
      """
      Returns statistics of inventory grouped by section or site
      """
2024 2025 2026
      #sql_kw = self._generateSQLKeywordDict(order_by_expression='stock.date', **kw)
      #sql_kw['group_by_expression'] = 'stock.uid'
      #sql_kw['order_by_expression'] = 'stock.date'
2027

2028
      result = self.getInventoryList(src__=src__,
2029
          sort_on = (('date', 'ascending'),), group_by_movement=1, **kw)
Alexandre Boeglin's avatar
Alexandre Boeglin committed
2030 2031
      if src__ :
        return result
2032

2033
      total_inventory = 0.
2034
      for inventory in result:
2035 2036 2037 2038 2039
        if inventory['inventory'] is not None:
          total_inventory += inventory['inventory']
          if total_inventory < 0:
            return inventory['date']

2040
      return None
Alexandre Boeglin's avatar
Alexandre Boeglin committed
2041

Jean-Paul Smets's avatar
Jean-Paul Smets committed
2042
    #######################################################
2043
    # Traceability management
2044
    security.declareProtected(Permissions.AccessContentsInformation, 'getTrackingList')
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2045
    def getTrackingList(self, src__=0,
2046
        selection_domain=None, selection_report=None,
2047
        strict_simulation_state=1, history=0, **kw) :
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2048
      """
2049
      Returns a list of items in the form
2050

Jean-Paul Smets's avatar
Jean-Paul Smets committed
2051 2052 2053 2054 2055 2056
        uid (of item)
        date
        node_uid
        section_uid
        resource_uid
        variation_text
2057
        delivery_uid
2058

2059 2060 2061 2062 2063 2064 2065 2066
      If at_date is provided, returns the a list which answers
      to the question "where are those items at this date" or
      "which are those items which are there a this date"

      If at_date is not provided, returns a history of positions
      which answers the question "where have those items been
      between this time and this time". This will be handled by
      something like getTrackingHistoryList
2067 2068

      This method is only suitable for singleton items (an item which can
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2069 2070
      only be at a single place at a given time). Such items include
      containers, serial numbers (ex. for engine), rolls with subrolls,
2071 2072

      This method is not suitable for batches (ex. a coloring batch).
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2073
      For such items, standard getInventoryList method is appropriate
2074

Jean-Paul Smets's avatar
Jean-Paul Smets committed
2075
      Parameters are the same as for getInventory.
2076

2077 2078 2079 2080 2081 2082 2083 2084 2085
      Default sort orders is based on dates, reverse.


      from_date (>=) -

      to_date   (<)  -

      at_date   (<=) - only take rows which date is <= at_date

2086 2087
      history (boolean) - keep history variations

2088 2089 2090 2091 2092 2093 2094 2095 2096 2097 2098 2099 2100 2101 2102 2103 2104 2105 2106 2107 2108 2109 2110
      resource (only in generic API in simulation)

      node        -  only take rows in stock table which node_uid is equivalent to node

      section        -  only take rows in stock table which section_uid is equivalent to section

      resource_category        -  only take rows in stock table which resource_uid is in resource_category

      node_category        -  only take rows in stock table which node_uid is in section_category

      section_category        -  only take rows in stock table which section_uid is in section_category

      variation_text - only take rows in stock table with specified variation_text
                       this needs to be extended with some kind of variation_category ?
                       XXX this way of implementing variation selection is far from perfect

      variation_category - variation or list of possible variations

      simulation_state - only take rows with specified simulation_state

      selection_domain, selection_report - see ListBox

      **kw  - if we want extended selection with more keywords (but bad performance)
2111
              check what we can do with buildSQLQuery
2112 2113 2114 2115

      Extra parameters for getTrackingList :

      item
2116

2117 2118 2119 2120 2121
      input - if set, answers to the question "which are those items which have been
              delivered for the first time after from_date". Cannot be used with output

      output - if set, answers to the question "which are those items which have been
               delivered for the last time before at_date or to_date". Cannot be used with input
2122

Jean-Paul Smets's avatar
Jean-Paul Smets committed
2123
      """
2124
      new_kw = self._generateSQLKeywordDict(table='item',strict_simulation_state=strict_simulation_state,**kw)
2125 2126 2127 2128 2129 2130
      for key in ('at_date', 'to_date'):
        value = kw.get(key, None)
        if value is not None:
          if isinstance(value, DateTime):
            value = value.toZone('UTC').ISO()
          value = '%s' % (value, )
Vincent Pelletier's avatar
Vincent Pelletier committed
2131
        # Do not remove dates in new_kw, they are required in
2132 2133
        # order to do a "select item left join item on date"
        new_kw[key] = value
2134

2135
      # Extra parameters for the SQL Method
2136
      new_kw['join_on_item'] = not history and (new_kw.get('at_date') or \
2137
                               new_kw.get('to_date') or \
2138
                               new_kw.get('input') or \
2139
                               new_kw.get('output'))
2140
      new_kw['date_condition_in_join'] = not (new_kw.get('input') or new_kw.get('output'))
2141

2142 2143
      # Pass simulation state to request
      if kw.has_key('item.simulation_state'):
2144
        new_kw['simulation_state_list'] = kw['item.simulation_state']
2145
      else:
2146
        new_kw['simulation_state_list'] =  None
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2147

2148
      return self.Resource_zGetTrackingList(src__=src__,
2149 2150 2151
                                            selection_domain=selection_domain,
                                            selection_report=selection_report,
                                            **new_kw)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2152 2153 2154 2155 2156 2157

    security.declareProtected(Permissions.AccessContentsInformation, 'getCurrentTrackingList')
    def getCurrentTrackingList(self, **kw):
      """
      Returns list of current inventory grouped by section or site
      """
2158 2159
      kw['item.simulation_state'] = self.getPortalObject()\
        .getPortalCurrentInventoryStateList()
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2160
      return self.getTrackingList(**kw)
Vincent Pelletier's avatar
Vincent Pelletier committed
2161

2162 2163 2164 2165 2166
    security.declareProtected(Permissions.AccessContentsInformation, 'getCurrentTrackingHistoryList')
    def getCurrentTrackingHistoryList(self, **kw):
      """
      Returns list of current inventory grouped by section or site
      """
2167 2168
      kw['item.simulation_state'] = self.getPortalObject()\
        .getPortalCurrentInventoryStateList()
2169
      return self.getTrackingHistoryList(**kw)
Vincent Pelletier's avatar
Vincent Pelletier committed
2170

2171 2172 2173 2174 2175 2176 2177
    security.declareProtected(Permissions.AccessContentsInformation, 'getTrackingHistoryList')
    def getTrackingHistoryList(self, **kw):
      """
      Returns history list of inventory grouped by section or site
      """
      kw['history'] = 1
      return self.getTrackingList(**kw)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2178 2179 2180 2181 2182 2183

    security.declareProtected(Permissions.AccessContentsInformation, 'getFutureTrackingList')
    def getFutureTrackingList(self, **kw):
      """
      Returns list of future inventory grouped by section or site
      """
2184 2185 2186 2187 2188
      portal = self.getPortalObject()
      kw['item.simulation_state'] = portal.getPortalFutureInventoryStateList() + \
                                    portal.getPortalTransitInventoryStateList() + \
                                    portal.getPortalReservedInventoryStateList() + \
                                    portal.getPortalCurrentInventoryStateList()
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2189 2190
      return self.getTrackingList(**kw)

Jean-Paul Smets's avatar
Jean-Paul Smets committed
2191 2192 2193 2194 2195 2196 2197 2198 2199 2200 2201 2202 2203 2204 2205 2206 2207 2208 2209 2210 2211 2212 2213 2214 2215 2216 2217
    #######################################################
    # Capacity Management
    security.declareProtected( Permissions.ModifyPortalContent, 'updateCapacity' )
    def updateCapacity(self, node):
      capacity_item_list = []
      for o in node.contentValues():
        if o.isCapacity():
          # Do whatever is needed
          capacity_item_list += o.asCapacityItemList()
          pass
      # Do whatever with capacity_item_list
      # and store the resulting new capacity in node
      node._capacity_item_list = capacity_item_list

    security.declareProtected( Permissions.ModifyPortalContent, 'isMovementInsideCapacity' )
    def isMovementInsideCapacity(self, movement):
      """
        Purpose: provide answer to customer for the question "can you do it ?"

        movement:
          date
          source destination (2 nodes)
          source_section ...
      """
      # Get nodes and dat
      source_node = movement.getSourceValue()
      destination_node = movement.getDestinationValue()
2218 2219
      start_date = movement.getStartDate()
      stop_date = movement.getStopDate()
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2220 2221 2222 2223 2224 2225 2226 2227 2228 2229 2230 2231 2232 2233 2234 2235 2236 2237 2238 2239 2240 2241 2242 2243 2244 2245 2246 2247 2248 2249 2250 2251 2252 2253 2254 2255 2256 2257 2258 2259 2260 2261 2262 2263 2264 2265 2266 2267 2268 2269 2270 2271 2272 2273 2274 2275 2276 2277 2278 2279 2280 2281 2282 2283 2284 2285 2286 2287 2288 2289 2290 2291 2292 2293 2294 2295 2296 2297 2298 2299 2300 2301 2302 2303 2304 2305 2306 2307 2308 2309 2310 2311 2312 2313 2314 2315 2316 2317 2318 2319 2320 2321 2322 2323 2324 2325 2326 2327 2328 2329 2330 2331 2332 2333 2334 2335 2336 2337 2338 2339 2340 2341 2342 2343 2344 2345 2346 2347 2348 2349 2350 2351 2352 2353 2354 2355 2356
      # Return result
      return self.isNodeInsideCapacity(source_node, start_date, additional_movement=movement, sign=1) and self.isNodeInsideCapacity(destination_node, stop_date, additional_movement=movement, sign=-1)

    security.declareProtected( Permissions.ModifyPortalContent, 'isNodeInsideCapacity' )
    def isNodeInsideCapacity(self, node, date, simulation_state=None, additional_movement=None, sign=1):
      """
        Purpose: decide if a node is consistent with its capacity definitions
        at a certain date (ie. considreing the stock / production history
      """
      # First get the current inventory situation for this node
      inventory_list = node.getInventoryList(XXXXX)
      # Add additional movement
      if additional_movement:
          inventory_list = inventory_list + sign * additional_movement # needs to be implemented
      # Return answer
      return self.isAmountListInsideCapacity(node, inventory_list)

    security.declareProtected( Permissions.ModifyPortalContent, 'isAmountListInsideCapacity' )
    def isAmountListInsideCapacity(self, node, amount_list,
         resource_aggregation_base_category=None, resource_aggregation_depth=None):
      """
        Purpose: decide if a list of amounts is consistent with the capacity of a node

        If any resource in amount_list is missing in the capacity of the node, resource
        aggregation is performed, based on resource_aggregation_base_category. If the
        base category is not specified, it is an error (should guess instead?). The resource
        aggregation is done at the level of resource_aggregation_depth in the tree
        of categories. If resource_aggregation_depth is not specified, it's an error.

        Assumptions: amount_list is an association list, like ((R1 V1) (R2 V2)).
                     node has an attribute '_capacity_item_list' which is a list of association lists.
                     resource_aggregation_base_category is a Base Category object or a list of Base
                     Category objects or None.
                     resource_aggregation_depth is a strictly positive integer or None.
      """
      # Make a copy of the attribute _capacity_item_list, because it may be necessary
      # to modify it for resource aggregation.
      capacity_item_list = node._capacity_item_list[:]

      # Make a mapping between resources and its indices.
      resource_map = {}
      index = 0
      for alist in capacity_item_list:
        for pair in alist:
          resource = pair[0]
#          LOG('isAmountListInsideCapacity', 0,
#              "resource is %s" % repr(resource))
          if resource not in resource_map:
            resource_map[resource] = index
            index += 1

      # Build a point from the amount list.
      point = zeros(index, 'd') # Fill up zeros for safety.
      mask_map = {}     # This is used to skip items in amount_list.
      for amount in amount_list:
        if amount[0] in mask_map:
          continue
        # This will fail, if amount_list has any different resource from the capacity.
        # If it has any different point, then we should ......
        #
        # There would be two possible different solutions:
        # 1) If a missing resource is a meta-resource of resources supported by the capacity,
        #    it is possible to add the resource into the capacity by aggregation.
        # 2) If a missing resource has a meta-resource as a parent and the capacity supports
        #    the meta-resource directly or indirectly (`indirectly' means `by aggregation'),
        #    it is possible to convert the missing resource into the meta-resource.
        #
        # However, another way has been implemented here. This does the following, if the resource
        # is not present in the capacity:
        # 1) If the value is zero, just ignore the resource, because zero is always acceptable.
        # 2) Attempt to aggregate resources both of the capacity and of the amount list. This aggregation
        #    is performed at the depth of 'resource_aggregation_depth' under the base category
        #    'resource_aggregation_base_category'.
        #
        resource = amount[0]
        if resource in resource_map:
          point[resource_map[amount[0]]] = amount[1]
        else:
          if amount[1] == 0:
            # If the value is zero, no need to consider.
            pass
          elif resource_aggregation_base_category is None or resource_aggregation_depth is None:
            # XXX use an appropriate error class
            # XXX should guess a base category instead of emitting an exception
            raise RuntimeError, "The resource '%s' is not found in the capacity, and the argument 'resource_aggregation_base_category' or the argument 'resource_aggregation_depth' is not specified" % resource
          else:
            # It is necessary to aggregate resources, to guess the capacity of this resource.

            def getAggregationResourceUrl(url, depth):
              # Return a partial url of the argument 'url'.
              # If 'url' is '/foo/bar/baz' and 'depth' is 2, return '/foo/bar'.
              pos = 0
              for i in range(resource_aggregation_depth):
                pos = url.find('/', pos+1)
                if pos < 0:
                  break
              if pos < 0:
                return None
              pos = url.find('/', pos+1)
              if pos < 0:
                pos = len(url)
              return url[:pos]

            def getAggregatedResourceList(aggregation_url, category, resource_list):
              # Return a list of resources which should be aggregated. 'aggregation_url' is used
              # for a top url of those resources. 'category' is a base category for the aggregation.
              aggregated_resource_list = []
              for resource in resource_list:
                for url in resource.getCategoryMembershipList(category, base=1):
                  if url.startswith(aggregation_url):
                    aggregated_resource_list.append(resource)
              return aggregated_resource_list

            def getAggregatedItemList(item_list, resource_list, aggregation_resource):
              # Return a list of association lists, which is a result of an aggregation.
              # 'resource_list' is a list of resources which should be aggregated.
              # 'aggregation_resource' is a category object which is a new resource created by
              # this aggregation.
              # 'item_list' is a list of association lists.
              new_item_list = []
              for alist in item_list:
                new_val = 0
                new_alist = []
                # If a resource is not a aggregated, then add it to the new alist as it is.
                # Otherwise, aggregate it to a single value.
                for pair in alist:
                  if pair[0] in resource_list:
                    new_val += pair[1]
                  else:
                    new_alist.append(pair)
                # If it is zero, ignore this alist, as it is nonsense.
                if new_val != 0:
                  new_alist.append([aggregation_resource, new_val])
                  new_item_list.append(new_alist)
              return new_item_list

            # Convert this to a string if necessary, for convenience.
2357
            if not isinstance(resource_aggregation_base_category, (tuple, list)):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2358 2359 2360 2361 2362 2363 2364 2365 2366 2367 2368 2369 2370 2371 2372 2373 2374 2375 2376 2377 2378 2379 2380 2381 2382 2383 2384 2385 2386 2387 2388 2389 2390 2391 2392 2393 2394 2395 2396 2397 2398 2399 2400 2401 2402 2403 2404 2405 2406 2407 2408 2409 2410 2411
              resource_aggregation_base_category = (resource_aggregation_base_category,)

            done = 0
#            LOG('isAmountListInsideCapacity', 0,
#                "resource_aggregation_base_category is %s" % repr(resource_aggregation_base_category))
            for category in resource_aggregation_base_category:
              for resource_url in resource.getCategoryMembershipList(category, base=1):
                aggregation_url = getAggregationResourceUrl(resource_url,
                                                            resource_aggregation_depth)
                if aggregation_url is None:
                  continue
                aggregated_resource_list = getAggregatedResourceList (aggregation_url,
                                                                      category,
                                                                      resource_map.keys())
                # If any, do the aggregation.
                if len(aggregated_resource_list) > 0:
                  aggregation_resource = self.portal_categories.resolveCategory(aggregation_url)
                  # Add the resource to the mapping.
 #                 LOG('aggregation_resource', 0, str(aggregation_resource))
                  resource_map[aggregation_resource] = index
                  index += 1
                  # Add the resource to the point.
                  point = resize(point, (index,))
                  val = 0
                  for aggregated_amount in amount_list:
                    for url in aggregated_amount[0].getCategoryMembershipList(category, base=1):
                      if url.startswith(aggregation_url):
                        val += aggregated_amount[1]
                        mask_map[aggregated_amount[0]] = None
                        break
                  point[index-1] = val
                  # Add capacity definitions of the resource into the capacity.
                  capacity_item_list += getAggregatedItemList(capacity_item_list,
                                                              aggregated_resource_list,
                                                              aggregation_resource)
                  done = 1
                  break
              if done:
                break
            if not done:
              raise RuntimeError, "Aggregation failed"

      # Build a matrix from the capacity item list.
#      LOG('resource_map', 0, str(resource_map))
      matrix = zeros((len(capacity_item_list)+1, index), 'd')
      for index in range(len(capacity_item_list)):
        for pair in capacity_item_list[index]:
          matrix[index,resource_map[pair[0]]] = pair[1]

#      LOG('isAmountListInsideCapacity', 0,
#          "matrix = %s, point = %s, capacity_item_list = %s" % (str(matrix), str(point), str(capacity_item_list)))
      return solve(matrix, point)


Jean-Paul Smets's avatar
Jean-Paul Smets committed
2412 2413
    # Asset Price Calculation
    def updateAssetPrice(self, resource, variation_text, section_category, node_category,
2414 2415 2416
                         strict_membership=0, simulation_state=None):
      if simulation_state is None:
        simulation_state = self.getPortalCurrentInventoryStateList()
2417 2418 2419
      category_tool = getToolByName(self, 'portal_categories')
      section_value = category_tool.resolveCategory(section_category)
      node_value = category_tool.resolveCategory(node_category)
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2420 2421 2422 2423
      # Initialize price
      current_asset_price = 0.0 # Missing: initial inventory price !!!
      current_inventory = 0.0
      # Parse each movement
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2424
      brain_list = self.Resource_zGetMovementHistoryList(resource=[resource],
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2425 2426 2427 2428
                             variation_text=variation_text,
                             section_category=section_category,
                             node_category=node_category,
                             strict_membership=strict_membership,
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2429 2430 2431 2432
                             simulation_state=simulation_state) # strict_membership not taken into account
                             # We select movements related to certain nodes (ex. Stock) and sections (ex.Coramy Group)
      result = []
      for b in brain_list:
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2433 2434
        m = b.getObject()
        if m is not None:
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2435 2436 2437 2438 2439 2440 2441 2442 2443 2444 2445 2446 2447 2448 2449 2450 2451 2452
          previous_inventory = current_inventory
          inventory_quantity = b.quantity # We should use the aggregated quantity provided by Resource_zGetMovementHistoryList
          quantity = m.getQuantity() # The movement quantity is important to determine the meaning of source and destination
          # Maybe we should take care of target qty in delired deliveries
          if quantity is None:
            quantity = 0.0
          if m.getSourceValue() is None:
            # This is a production movement or an inventory movement
            # Use Industrial Price
            current_inventory += inventory_quantity # Update inventory
            if m.getPortalType() in ('Inventory Line', 'Inventory Cell'): # XX should be replaced by isInventory ???
              asset_price = m.getPrice()
              if asset_price in (0.0, None):
                asset_price = current_asset_price # Use current price if no price defined
            else: # this is a production
              asset_price = m.getIndustrialPrice()
              if asset_price is None: asset_price = current_asset_price  # Use current price if no price defined
            result.append((m.getRelativeUrl(), m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2453
                          m.getQuantity(), 'Production or Inventory', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2454 2455 2456 2457 2458 2459
                        ))
          elif m.getDestinationValue() is None:
            # This is a consumption movement or an inventory movement
            current_inventory += inventory_quantity # Update inventory
            asset_price = current_asset_price
            result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2460
                          m.getQuantity(), 'Consumption or Inventory', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2461
                        ))
2462
          elif m.getSourceValue().isAcquiredMemberOf(node_category) and m.getDestinationValue().isAcquiredMemberOf(node_category):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2463 2464 2465 2466
            # This is an internal movement
            current_inventory += inventory_quantity # Update inventory
            asset_price = current_asset_price
            result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2467
                          m.getQuantity(), 'Internal', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2468
                        ))
2469
          elif m.getSourceValue().isAcquiredMemberOf(node_category) and quantity < 0:
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2470 2471 2472 2473
            # This is a physically inbound movement - try to use commercial price
            if m.getSourceSectionValue() is None:
              # No meaning
              current_inventory += inventory_quantity # Update inventory
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2474
              asset_price = current_asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2475
              result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2476
                            m.getQuantity(), 'Error', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2477 2478 2479 2480
                          ))
            elif m.getDestinationSectionValue() is None:
              # No meaning
              current_inventory += inventory_quantity # Update inventory
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2481
              asset_price = current_asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2482
              result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2483
                            m.getQuantity(), 'Error', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2484
                          ))
2485
            elif m.getDestinationSectionValue().isAcquiredMemberOf(section_category):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2486
              current_inventory += inventory_quantity # Update inventory
2487
              if m.getDestinationValue().isAcquiredMemberOf('site/Piquage'):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2488 2489 2490 2491
                # Production
                asset_price = m.getIndustrialPrice()
                if asset_price is None: asset_price = current_asset_price  # Use current price if no price defined
                result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2492
                              m.getQuantity(), 'Production', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2493
                            ))
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2494 2495 2496
              else:
                # Inbound from same section
                asset_price = current_asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2497
                result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2498
                              m.getQuantity(), 'Inbound same section', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2499
                            ))
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2500
            else:
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2501 2502 2503
              current_inventory += inventory_quantity # Update inventory
              asset_price = m.getPrice()
              result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2504
                            m.getQuantity(), 'Inbound different section', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2505
                          ))
2506
          elif m.getDestinationValue().isAcquiredMemberOf(node_category) and quantity > 0:
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2507 2508 2509 2510 2511 2512
            # This is a physically inbound movement - try to use commercial price
            if m.getSourceSectionValue() is None:
              # No meaning
              current_inventory += inventory_quantity # Update inventory
              asset_price = current_asset_price
              result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2513
                            m.getQuantity(), 'Error', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2514 2515 2516 2517
                          ))
            elif m.getDestinationSectionValue() is None:
              # No meaning
              current_inventory += inventory_quantity # Update inventory
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2518
              asset_price = current_asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2519
              result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2520
                            m.getQuantity(), 'Error', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2521
                          ))
2522
            elif m.getSourceSectionValue().isAcquiredMemberOf(section_category):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2523
              current_inventory += inventory_quantity # Update inventory
2524
              if m.getSourceValue().isAcquiredMemberOf('site/Piquage'):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2525 2526 2527 2528
                # Production
                asset_price = m.getIndustrialPrice()
                if asset_price is None: asset_price = current_asset_price  # Use current price if no price defined
                result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2529
                              m.getQuantity(), 'Production', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2530
                            ))
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2531
              else:
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2532 2533 2534
                # Inbound from same section
                asset_price = current_asset_price
                result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2535
                            m.getQuantity(), 'Inbound same section', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2536
                          ))
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2537
            else:
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2538 2539 2540
              current_inventory += inventory_quantity # Update inventory
              asset_price = m.getPrice()
              result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2541
                            m.getQuantity(), 'Inbound different section', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2542 2543 2544 2545 2546 2547
                          ))
          else:
            # Outbound movement
            current_inventory += inventory_quantity # Update inventory
            asset_price = current_asset_price
            result.append((m.getRelativeUrl(),m.getStartDate(), m.getSource(), m.getSourceSection(), m.getDestination(), m.getDestinationSection(),
2548
                            m.getQuantity(), 'Outbound', 'Price: %s' % asset_price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2549 2550 2551 2552 2553 2554 2555 2556 2557 2558 2559 2560 2561 2562
                          ))

          # Update asset_price
          if current_inventory > 0:
            if inventory_quantity is not None:
              # Update price with an average of incoming goods and current goods
              current_asset_price = ( current_asset_price * previous_inventory + asset_price * inventory_quantity ) / float(current_inventory)
          else:
            # New price is the price of incoming goods - negative stock has no meaning for asset calculation
            current_asset_price = asset_price

          result.append(('###New Asset Price', current_asset_price, 'New Inventory', current_inventory))

          # Update Asset Price on the right side
2563
          if m.getSourceSectionValue() is not None and m.getSourceSectionValue().isAcquiredMemberOf(section_category):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2564 2565
            # for each movement, source section is member of one and one only accounting category
            # therefore there is only one and one only source asset price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2566 2567 2568 2569 2570 2571 2572
            m._setSourceAssetPrice(current_asset_price)
            #quantity = m.getInventoriatedQuantity()
            #if quantity:
            #  #total_asset_price = - current_asset_price * quantity
            #  #m.Movement_zSetSourceTotalAssetPrice(uid=m.getUid(), total_asset_price = total_asset_price)
            #  m._setSourceAssetPrice(current_asset_price)
          if m.getDestinationSectionValue() is not None and m.getDestinationSectionValue().isMemberOf(section_category):
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2573 2574
            # for each movement, destination section is member of one and one only accounting category
            # therefore there is only one and one only destination asset price
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2575 2576 2577 2578 2579
            m._setDestinationAssetPrice(current_asset_price)
            #quantity = m.getInventoriatedQuantity()
            #if quantity:
            #  total_asset_price = current_asset_price * quantity
            #  m.Movement_zSetDestinationTotalAssetPrice(uid=m.getUid(), total_asset_price = total_asset_price)
2580 2581 2582
          # Global reindexing required afterwards in any case: so let us do it now
          # Until we get faster methods (->reindexObject())
          #m.immediateReindexObject()
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2583
          m.reindexObject()
2584
          #m.activate(priority=7).immediateReindexObject() # Too slow
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2585 2586

      return result
Jean-Paul Smets's avatar
Jean-Paul Smets committed
2587

2588 2589 2590
    # Used for mergeDeliveryList.
    class MergeDeliveryListError(Exception): pass

2591 2592 2593 2594 2595 2596 2597 2598 2599 2600 2601
    security.declareProtected( Permissions.ModifyPortalContent, 'mergeDeliveryList' )
    def mergeDeliveryList(self, delivery_list):
      """
        Merge multiple deliveries into one delivery.
        All delivery lines are merged into the first one.
        The first one is therefore called main_delivery here.
        The others are cancelled.
        Return the main delivery.
      """
      # Sanity checks.
      if len(delivery_list) == 0:
2602
        raise self.MergeDeliveryListError, "No delivery is passed"
2603
      elif len(delivery_list) == 1:
2604
        raise self.MergeDeliveryListError, "Only one delivery is passed"
2605 2606 2607 2608

      main_delivery = delivery_list[0]
      delivery_list = delivery_list[1:]

2609
      # Another sanity check. It is necessary for them to be identical in some attributes.
2610 2611 2612 2613 2614 2615 2616 2617 2618 2619
      for delivery in delivery_list:
        for attr in ('portal_type', 'simulation_state',
                     'source', 'destination',
                     'source_section', 'destination_section',
                     'source_decision', 'destination_decision',
                     'source_administration', 'destination_administration',
                     'source_payment', 'destination_payment'):
          main_value = main_delivery.getProperty(attr)
          value = delivery.getProperty(attr)
          if  main_value != value:
2620 2621 2622 2623
            raise self.MergeDeliveryListError, \
              "%s is not the same between %s and %s (%s and %s)" % (attr, delivery.getId(), main_delivery.getId(), value, main_value)

      # One more sanity check. Check if discounts are the same, if any.
2624
      main_discount_list = main_delivery.contentValues(filter = {'portal_type': self.getPortalDiscountTypeList()})
2625
      for delivery in delivery_list:
2626
        discount_list = delivery.contentValues(filter = {'portal_type': self.getPortalDiscountTypeList()})
2627 2628 2629 2630 2631 2632 2633 2634 2635 2636 2637 2638 2639
        if len(main_discount_list) != len(discount_list):
          raise self.MergeDeliveryListError, "Discount is not the same between %s and %s" % (delivery.getId(), main_delivery.getId())
        for discount in discount_list:
          for main_discount in main_discount_list:
            if discount.getDiscount() == main_discount.getDiscount() \
               and discount.getDiscountRatio() == main_discount.getDiscountRatio() \
               and discount.getDiscountType() == main_discount.getDiscountType() \
               and discount.getImmediateDiscount() == main_discount.getImmediateDiscount():
              break
          else:
            raise self.MergeDeliveryListError, "Discount is not the same between %s and %s" % (delivery.getId(), main_delivery.getId())

      # One more sanity check. Check if payment conditions are the same, if any.
2640
      main_payment_condition_list = main_delivery.contentValues(filter = {'portal_type': self.getPortalPaymentConditionTypeList()})
2641
      for delivery in delivery_list:
2642
        payment_condition_list = delivery.contentValues(filter = {'portal_type': self.getPortalPaymentConditionTypeList()})
2643 2644 2645 2646 2647 2648 2649 2650 2651 2652 2653 2654 2655
        if len(main_payment_condition_list) != len(payment_condition_list):
          raise self.MergeDeliveryListError, "Payment Condition is not the same between %s and %s" % (delivery.getId(), main_delivery.getId())
        for condition in payment_condition_list:
          for main_condition in main_payment_condition_list:
            if condition.getPaymentMode() == main_condition.getPaymentMode() \
               and condition.getPaymentAdditionalTerm() == main_condition.getPaymentAdditionalTerm() \
               and condition.getPaymentAmount() == main_condition.getPaymentAmount() \
               and condition.getPaymentEndOfMonth() == main_condition.getPaymentEndOfMonth() \
               and condition.getPaymentRatio() == main_condition.getPaymentRatio() \
               and condition.getPaymentTerm() == main_condition.getPaymentTerm():
              break
          else:
            raise self.MergeDeliveryListError, "Payment Condition is not the same between %s and %s" % (delivery.getId(), main_delivery.getId())
2656 2657 2658

      # Make sure that all activities are flushed, to get simulation movements from delivery cells.
      for delivery in delivery_list:
2659
        for order in delivery.getCausalityValueList(portal_type = self.getPortalOrderTypeList()):
2660 2661
          for applied_rule in order.getCausalityRelatedValueList(portal_type = 'Applied Rule'):
            applied_rule.flushActivity(invoke = 1)
2662
        for causality_related_delivery in delivery.getCausalityValueList(portal_type = self.getPortalDeliveryTypeList()):
2663 2664
          for applied_rule in causality_related_delivery.getCausalityRelatedValueList(portal_type = 'Applied Rule'):
            applied_rule.flushActivity(invoke = 1)
2665

2666 2667 2668 2669 2670
      # Get a list of simulated movements and invoice movements.
      main_simulated_movement_list = main_delivery.getSimulatedMovementList()
      main_invoice_movement_list = main_delivery.getInvoiceMovementList()
      simulated_movement_list = main_simulated_movement_list[:]
      invoice_movement_list = main_invoice_movement_list[:]
2671
      for delivery in delivery_list:
2672 2673 2674
        simulated_movement_list.extend(delivery.getSimulatedMovementList())
        invoice_movement_list.extend(delivery.getInvoiceMovementList())

2675 2676 2677 2678
      #for movement in simulated_movement_list + invoice_movement_list:
      #  parent = movement.aq_parent
      #  LOG('mergeDeliveryList', 0, 'movement = %s, parent = %s, movement.getPortalType() = %s, parent.getPortalType() = %s' % (repr(movement), repr(parent), repr(movement.getPortalType()), repr(parent.getPortalType())))

2679 2680 2681 2682 2683 2684 2685 2686 2687 2688 2689 2690
      LOG('mergeDeliveryList', 0, 'simulated_movement_list = %s, invoice_movement_list = %s' % (str(simulated_movement_list), str(invoice_movement_list)))
      for main_movement_list, movement_list in \
        ((main_simulated_movement_list, simulated_movement_list),
         (main_invoice_movement_list, invoice_movement_list)):
        root_group = self.collectMovement(movement_list,
                                          check_order = 0,
                                          check_path = 0,
                                          check_date = 0,
                                          check_criterion = 1,
                                          check_resource = 1,
                                          check_base_variant = 1,
                                          check_variant = 1)
2691 2692 2693 2694 2695 2696 2697 2698 2699
        for criterion_group in root_group.group_list:
          LOG('mergeDeliveryList dump tree', 0, 'criterion = %s, movement_list = %s, group_list = %s' % (repr(criterion_group.criterion), repr(criterion_group.movement_list), repr(criterion_group.group_list)))
          for resource_group in criterion_group.group_list:
            LOG('mergeDeliveryList dump tree', 0, 'resource = %s, movement_list = %s, group_list = %s' % (repr(resource_group.resource), repr(resource_group.movement_list), repr(resource_group.group_list)))
            for base_variant_group in resource_group.group_list:
              LOG('mergeDeliveryList dump tree', 0, 'base_category_list = %s, movement_list = %s, group_list = %s' % (repr(base_variant_group.base_category_list), repr(base_variant_group.movement_list), repr(base_variant_group.group_list)))
              for variant_group in base_variant_group.group_list:
                LOG('mergeDeliveryList dump tree', 0, 'category_list = %s, movement_list = %s, group_list = %s' % (repr(variant_group.category_list), repr(variant_group.movement_list), repr(variant_group.group_list)))

2700 2701 2702 2703 2704 2705 2706 2707 2708 2709 2710 2711 2712 2713
        for criterion_group in root_group.group_list:
          for resource_group in criterion_group.group_list:
            for base_variant_group in resource_group.group_list:
              # Get a list of categories.
              category_dict = {}
              for variant_group in base_variant_group.group_list:
                for category in variant_group.category_list:
                  category_dict[category] = 1
              category_list = category_dict.keys()

              # Try to find a delivery line.
              delivery_line = None
              for movement in base_variant_group.movement_list:
                if movement in main_movement_list:
2714 2715
                  if movement.aq_parent.getPortalType() in self.getPortalSimulatedMovementTypeList() \
                    or movement.aq_parent.getPortalType() in self.getPortalInvoiceMovementTypeList():
2716 2717 2718 2719 2720
                    delivery_line = movement.aq_parent
                  else:
                    delivery_line = movement
                  LOG('mergeDeliveryList', 0, 'delivery_line %s is found: criterion = %s, resource = %s, base_category_list = %s' % (repr(delivery_line), repr(criterion_group.criterion), repr(resource_group.resource), repr(base_variant_group.base_category_list)))
                  break
2721

2722 2723 2724
              if delivery_line is None:
                # Not found. So create a new delivery line.
                movement = base_variant_group.movement_list[0]
2725 2726
                if movement.aq_parent.getPortalType() in self.getPortalSimulatedMovementTypeList() \
                  or movement.aq_parent.getPortalType() in self.getPortalInvoiceMovementTypeList():
2727
                  delivery_line_type = movement.aq_parent.getPortalType()
2728
                else:
2729 2730 2731 2732 2733 2734 2735 2736 2737 2738
                  delivery_line_type = movement.getPortalType()
                delivery_line = main_delivery.newContent(portal_type = delivery_line_type,
                                                         resource = resource_group.resource)
                LOG('mergeDeliveryList', 0, 'New delivery_line %s is created: criterion = %s, resource = %s, base_category_list = %s' % (repr(delivery_line), repr(criterion_group.criterion), repr(resource_group.resource), repr(base_variant_group.base_category_list)))

              # Update the base categories and categories.
              #LOG('mergeDeliveryList', 0, 'base_category_list = %s, category_list = %s' % (repr(base_category_list), repr(category_list)))
              delivery_line.setVariationBaseCategoryList(base_variant_group.base_category_list)
              delivery_line.setVariationCategoryList(category_list)

2739
              object_to_update = None
2740 2741 2742 2743 2744 2745
              for variant_group in base_variant_group.group_list:
                if len(variant_group.category_list) == 0:
                  object_to_update = delivery_line
                else:
                  for delivery_cell in delivery_line.contentValues():
                    predicate_value_list = delivery_cell.getPredicateValueList()
2746
                    LOG('mergeDeliveryList', 0, 'delivery_cell = %s, predicate_value_list = %s, variant_group.category_list = %s' % (repr(delivery_cell), repr(predicate_value_list), repr(variant_group.category_list)))
2747 2748 2749 2750 2751 2752 2753
                    if len(predicate_value_list) == len(variant_group.category_list):
                      for category in variant_group.category_list:
                        if category not in predicate_value_list:
                          break
                      else:
                        object_to_update = delivery_cell
                        break
2754

2755
                #LOG('mergeDeliveryList', 0, 'object_to_update = %s' % repr(object_to_update))
2756
                if object_to_update is not None:
2757
                  cell_price = object_to_update.getPrice() or 0.0
2758
                  cell_quantity = object_to_update.getQuantity() or 0.0
2759
                  cell_target_quantity = object_to_update.getNetConvertedTargetQuantity() or 0.0 # XXX What to do ?
2760
                  cell_total_price = cell_target_quantity * cell_price
2761 2762 2763 2764 2765 2766 2767 2768 2769 2770
                  cell_category_list = list(object_to_update.getCategoryList())

                  for movement in variant_group.movement_list:
                    if movement in main_movement_list:
                      continue
                    LOG('mergeDeliveryList', 0, 'movement = %s' % repr(movement))
                    cell_quantity += movement.getQuantity()
                    cell_target_quantity += movement.getNetConvertedTargetQuantity()
                    try:
                      # XXX WARNING - ADD PRICED QUANTITY
2771 2772
                      cell_price = movement.getPrice()
                      cell_total_price += movement.getNetConvertedTargetQuantity() * cell_price
Yoshinori Okuji's avatar
Yoshinori Okuji committed
2773
                    except TypeError:
2774 2775 2776 2777 2778 2779 2780 2781 2782 2783 2784 2785 2786 2787 2788 2789 2790 2791 2792 2793 2794 2795
                      cell_total_price = None
                    for category in movement.getCategoryList():
                      if category not in cell_category_list:
                        cell_category_list.append(category)
                    # Make sure that simulation movements point to an appropriate delivery line or
                    # delivery cell.
                    if hasattr(movement, 'getDeliveryRelatedValueList'):
                      for simulation_movement in \
                        movement.getDeliveryRelatedValueList(portal_type = 'Simulation Movement'):
                        simulation_movement.setDeliveryValue(object_to_update)
                        #simulation_movement.reindexObject()
                    if hasattr(movement, 'getOrderRelatedValueList'):
                      for simulation_movement in \
                        movement.getOrderRelatedValueList(portal_type = 'Simulation Movement'):
                        simulation_movement.setOrderValue(object_to_update)
                        #simulation_movement.reindexObject()

                  if cell_target_quantity != 0 and cell_total_price is not None:
                    average_price = cell_total_price / cell_target_quantity
                  else:
                    average_price = 0

2796
                  LOG('mergeDeliveryList', 0, 'object_to_update = %s, cell_category_list = %s, cell_target_quantity = %s, cell_quantity = %s, average_price = %s' % (repr(object_to_update), repr(cell_category_list), repr(cell_target_quantity), repr(cell_quantity), repr(average_price)))
2797
                  object_to_update.setCategoryList(cell_category_list)
2798
                  if object_to_update.getPortalType() in self.getPortalSimulatedMovementTypeList():
2799 2800 2801 2802
                    object_to_update.edit(target_quantity = cell_target_quantity,
                                          quantity = cell_quantity,
                                          price = average_price,
                                          )
2803
                  elif object_to_update.getPortalType() in self.getPortalInvoiceMovementTypeList():
2804 2805 2806 2807 2808 2809 2810
                    # Invoices do not have target quantities, and the price never change.
                    object_to_update.edit(quantity = cell_quantity,
                                          price = cell_price,
                                          )
                  else:
                    raise self.MergeDeliveryListError, "Unknown portal type %s" % str(object_to_update.getPortalType())
                  #object_to_update.immediateReindexObject()
2811 2812 2813 2814 2815
                else:
                  raise self.MergeDeliveryListError, "No object to update"

      # Merge containers. Just copy them from other deliveries into the main.
      for delivery in delivery_list:
2816
        container_id_list = delivery.contentIds(filter = {'portal_type': self.getPortalContainerTypeList()})
2817 2818 2819
        if len(container_id_list) > 0:
          copy_data = delivery.manage_copyObjects(ids = container_id_list)
          new_id_list = main_delivery.manage_pasteObjects(copy_data)
2820 2821 2822 2823 2824 2825 2826 2827 2828 2829 2830 2831 2832 2833 2834 2835 2836 2837 2838 2839

      # Unify the list of causality.
      causality_list = main_delivery.getCausalityValueList()
      for delivery in delivery_list:
        for causality in delivery.getCausalityValueList():
          if causality not in causality_list:
            causality_list.append(causality)
      LOG("mergeDeliveryList", 0, "causality_list = %s" % str(causality_list))
      main_delivery.setCausalityValueList(causality_list)

      # Cancel deliveries.
      for delivery in delivery_list:
        LOG("mergeDeliveryList", 0, "cancelling %s" % repr(delivery))
        delivery.cancel()

      # Reindex the main delivery.
      main_delivery.reindexObject()

      return main_delivery

2840 2841
    #######################################################
    # Sequence
Vincent Pelletier's avatar
Vincent Pelletier committed
2842
    security.declareProtected(Permissions.AccessContentsInformation,
2843 2844 2845 2846 2847 2848 2849 2850 2851 2852
                              'getSequence')
    def getSequence(self, **kw):
      """
      getSequence is take the same parameters as Sequence constructor,
      and return a Sequence.
      """
      return Sequence(**kw)

    #######################################################
    # Time Management
Vincent Pelletier's avatar
Vincent Pelletier committed
2853
    security.declareProtected(Permissions.AccessContentsInformation,
2854
                              'getAvailableTime')
Vincent Pelletier's avatar
Vincent Pelletier committed
2855 2856
    def getAvailableTime(self, from_date=None, to_date=None,
                         portal_type=[], node=[],
2857
                         resource=[], src__=0, **kw):
2858 2859 2860 2861 2862 2863 2864 2865 2866 2867 2868 2869
      """
      Calculate available time for a node
      Returns an inventory of a single or multiple resources on a single
      node as a single float value

      from_date (>=) - only take rows which mirror_date is >= from_date

      to_date   (<)  - only take rows which date is < to_date

      node           - only take rows in stock table which node_uid is
                       equivalent to node

2870 2871 2872
      resource       - only take rows in stock table which resource_uid is
                       equivalent to resource

2873 2874 2875 2876 2877 2878 2879
      portal_type    - only take rows in stock table which portal_type
                       is in portal_type parameter
      """
      # XXX For now, consider that from_date and to_date are required
      if (from_date is None) or (to_date is None):
        raise NotImplementedError, \
              "getAvailableTime does not managed yet None values"
2880
      portal = self.getPortalObject()
2881 2882
      # Calculate portal_type
      if portal_type == []:
2883
        portal_type = portal.getPortalCalendarPeriodTypeList()
2884

2885 2886 2887
      simulation_state = portal.getPortalCurrentInventoryStateList() + \
                         portal.getPortalTransitInventoryStateList() + \
                         portal.getPortalReservedInventoryStateList()
2888

2889
      sql_result = portal.Person_zGetAvailableTime(
2890 2891 2892 2893
                          from_date=from_date,
                          to_date=to_date,
                          portal_type=portal_type,
                          node=node,
2894
                          resource=resource,
2895
                          simulation_state=simulation_state,
2896 2897
                          src__=src__, **kw)
      if not src__:
2898
        result = 0
2899 2900 2901 2902
        if len(sql_result) == 1:
          result = sql_result[0].total_quantity
      else:
        result = sql_result
2903 2904
      return result

Vincent Pelletier's avatar
Vincent Pelletier committed
2905
    security.declareProtected(Permissions.AccessContentsInformation,
2906
                              'getAvailableTimeSequence')
Vincent Pelletier's avatar
Vincent Pelletier committed
2907
    def getAvailableTimeSequence(self, from_date, to_date,
2908
                                 portal_type=[], node=[],
Vincent Pelletier's avatar
Vincent Pelletier committed
2909
                                 resource=[],
2910 2911 2912 2913 2914 2915 2916 2917 2918 2919 2920 2921
                                 src__=0,
                                 **kw):
      """
      Calculate available time for a node in multiple period of time.
      Each row is the available time for a specific period

      node           - only take rows in stock table which node_uid is
                       equivalent to node

      portal_type    - only take rows in stock table which portal_type
                       is in portal_type parameter

2922 2923 2924
      resource       - only take rows in stock table which resource_uid is
                       equivalent to resource

2925 2926 2927 2928 2929 2930 2931 2932
      from_date (>=) - return period which start >= from_date

      to_date   (<)  - return period which start < to_date

      second, minute,
      hour, day,
      month, year   - duration of each time period (cumulative)
      """
2933
      portal = self.getPortalObject()
2934 2935
      # Calculate portal_type
      if portal_type == []:
2936
        portal_type = portal.getPortalCalendarPeriodTypeList()
2937

2938 2939 2940 2941 2942 2943 2944 2945 2946 2947 2948
      sequence = Sequence(from_date, to_date, **kw)
      for sequence_item in sequence:
        setattr(sequence_item, 'total_quantity',
                self.getAvailableTime(
                          from_date=sequence_item.from_date,
                          to_date=sequence_item.to_date,
                          portal_type=portal_type,
                          node=node,
                          resource=resource,
                          src__=src__))
      return sequence
2949

2950 2951 2952 2953 2954 2955 2956 2957 2958 2959 2960 2961
    def _checkExpandAll(self, activate_kw={}):
      """Check all simulation trees using AppliedRule._checkExpand
      """
      portal = self.getPortalObject()
      active_process = portal.portal_activities.newActiveProcess().getPath()
      kw = dict(priority=3, tag='checkExpand')
      kw.update(group_method_cost=1, max_retry=0,
                active_process=active_process, **activate_kw)
      self._recurseCallMethod('_checkExpand', min_depth=1, max_depth=1,
                              activate_kw=kw)
      return active_process

2962
from Products.ERP5Type.DateUtils import addToDate
2963 2964 2965 2966 2967 2968 2969 2970 2971 2972

class SequenceItem:
  """
  SequenceItem define a time period.
  period.
  """
  def __init__(self, from_date, to_date):
    self.from_date = from_date
    self.to_date = to_date

2973 2974 2975 2976 2977
class Sequence:
  """
  Sequence is a iterable object, which calculate a range of time
  period.
  """
Vincent Pelletier's avatar
Vincent Pelletier committed
2978
  def __init__(self, from_date, to_date,
2979 2980 2981 2982 2983 2984 2985 2986 2987 2988 2989 2990 2991 2992 2993 2994 2995
               second=None, minute=None, hour=None,
               day=None, month=None, year=None):
    """
    Calculate a list of time period.
    Time period is a 2-tuple of 2 DateTime, which represent the from date
    and to date of the period.

    The start date of a period is calculated with the rule
        start_date of the previous + period duration

    from_date (>=) - return period which start >= from_date

    to_date   (<)  - return period which start < to_date

    second, minute,
    hour, day,
    month, year   - duration of each time period (cumulative)
2996
                    at least one of those parameters must be specified.
2997
    """
2998 2999 3000
    if not (second or minute or hour or day or month or year):
      raise ValueError('Period duration must be specified')

3001 3002 3003 3004
    self.item_list = []
    # Calculate all time period
    current_from_date = from_date
    while current_from_date < to_date:
Vincent Pelletier's avatar
Vincent Pelletier committed
3005
      current_to_date = addToDate(current_from_date,
3006 3007 3008 3009 3010 3011
                                  second=second,
                                  minute=minute,
                                  hour=hour,
                                  day=day,
                                  month=month,
                                  year=year)
Vincent Pelletier's avatar
Vincent Pelletier committed
3012
      self.item_list.append(SequenceItem(current_from_date,
3013
                                         current_to_date))
3014 3015 3016 3017 3018 3019 3020 3021 3022 3023 3024 3025 3026 3027 3028 3029 3030
      current_from_date = current_to_date

  def __len__(self):
    return len(self.item_list)

  def __getitem__(self, key):
    return self.item_list[key]

  def __contains__(self, value):
    return (value in self.item_list)

  def __iter__(self):
    for x in self.item_list:
      yield x

InitializeClass(Sequence)
allow_class(Sequence)
3031 3032
InitializeClass(SequenceItem)
allow_class(SequenceItem)
3033

3034
InitializeClass(SimulationTool)