NodeBudgetVariation.py 12.8 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
##############################################################################
#
# Copyright (c) 2008 Nexedi SA and Contributors. All Rights Reserved.
#
# 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.
#
##############################################################################

from AccessControl import ClassSecurityInfo

from AccessControl.ZopeGuards import guarded_getattr
31
from Products.ERP5Type import Permissions, PropertySheet
32
from Products.ERP5.Document.BudgetVariation import BudgetVariation
33
from Products.ZSQLCatalog.SQLCatalog import Query, NegatedQuery, ComplexQuery
34 35


36 37
class NodeBudgetVariation(BudgetVariation):
  """ A budget variation for node
38 39

  A script will return the list of possible nodes, or they will be configured
40 41
  explicitly on the budget variation. It is also possible to include a virtual
  node for all others not selected nodes.
42 43 44 45 46 47 48 49
  """
  # Default Properties
  property_sheets = ( PropertySheet.Base
                    , PropertySheet.XMLObject
                    , PropertySheet.SimpleItem
                    , PropertySheet.SortIndex
                    , PropertySheet.Path
                    , PropertySheet.Predicate
50
                    , PropertySheet.BudgetVariation
51 52 53 54 55 56 57 58 59 60 61
                    )

  # CMF Type Definition
  meta_type = 'ERP5 Node Budget Variation'
  portal_type = 'Node Budget Variation'
  add_permission = Permissions.AddPortalContent

  # Declarative security
  security = ClassSecurityInfo()
  security.declareObjectProtected(Permissions.AccessContentsInformation)

62
  # zope.interface.implements(BudgetVariation, )
63 64 65 66 67

  def asBudgetPredicate(self):
    """This budget variation in a predicate
    """

68
  def _getNodeList(self, context):
69 70 71 72
    """Returns the list of possible nodes
    """
    node_select_method_id = self.getProperty('node_select_method_id')
    if node_select_method_id:
73
      return guarded_getattr(context, node_select_method_id)()
74

75
    # no script defined, used the explicitly selected values
76 77 78 79
    node_list = self.getAggregateValueList()
    portal_categories = self.getPortalObject().portal_categories
    if self.getProperty('include_virtual_none_node'):
      node_list.append(portal_categories.budget_special_node.none)
80
    if self.getProperty('include_virtual_other_node'):
81 82
      node_list.append(portal_categories.budget_special_node.all_other)
    return node_list
83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99

  def _getNodeTitle(self, node):
    """Returns the title of a node
    """
    node_title_method_id = self.getProperty('node_title_method_id', 'getTitle')
    return guarded_getattr(node, node_title_method_id)()

  def getCellRangeForBudgetLine(self, budget_line, matrixbox=0):
    """The cell range added by this variation
    """
    base_category = self.getProperty('variation_base_category')
    prefix = ''
    if base_category:
      prefix = '%s/' % base_category

    node_item_list = [('%s%s' % (prefix, node.getRelativeUrl()),
                       self._getNodeTitle(node))
100
                           for node in self._getNodeList(budget_line)]
101 102 103 104 105 106 107 108
    variation_category_list = budget_line.getVariationCategoryList()
    if matrixbox:
      return [[i for i in node_item_list if i[0] in variation_category_list]]
    return [[i[0] for i in node_item_list if i[0] in variation_category_list]]

  def getInventoryQueryDict(self, budget_cell):
    """ Query dict to pass to simulation query
    """
Jérome Perrin's avatar
Jérome Perrin committed
109
    query_dict = dict()
110 111
    axis = self.getInventoryAxis()
    if not axis:
Jérome Perrin's avatar
Jérome Perrin committed
112
      return query_dict
113 114
    base_category = self.getProperty('variation_base_category')
    if not base_category:
Jérome Perrin's avatar
Jérome Perrin committed
115
      return query_dict
116
    budget_line = budget_cell.getParentValue()
117 118 119 120 121 122

    context = budget_cell
    if self.isMemberOf('budget_variation/budget'):
      context = budget_line.getParentValue()
    elif self.isMemberOf('budget_variation/budget_line'):
      context = budget_line
Jérome Perrin's avatar
Jérome Perrin committed
123 124 125 126 127
    
    if axis == 'movement':
      axis = 'default_%s' % base_category
    if axis == 'movement_strict_membership':
      axis = 'default_strict_%s' % base_category
128 129 130
    
    uid_based_axis = False
    if axis in ('node', 'section', 'payment', 'function', 'project',
Jérome Perrin's avatar
Jérome Perrin committed
131
                'mirror_section', 'mirror_node', 'funding' ):
132 133
      axis = '%s_uid' % axis
      uid_based_axis = True
134

135
    query = None
136 137
    portal_categories = self.getPortalObject().portal_categories
    for criterion_category in context.getMembershipCriterionCategoryList():
138 139 140 141
      if '/' not in criterion_category: # safe ...
        continue
      criterion_base_category, node_url = criterion_category.split('/', 1)
      if criterion_base_category == base_category:
142 143
        if node_url == 'budget_special_node/none':
          # This is the "Nothing" virtual node
144 145
          query = Query(**{axis: None})
        elif node_url == 'budget_special_node/all_other':
146 147
          # This is the "All Other" virtual node
          other_uid_list = []
148
          none_node_selected = False
149 150 151
          for node in self._getNodeList(budget_line):
            if '%s/%s' % (base_category, node.getRelativeUrl()) in\
                                    budget_line.getVariationCategoryList():
152 153 154
              if node.getRelativeUrl() == 'budget_special_node/none':
                none_node_selected = True
              else:
155 156 157 158
                if uid_based_axis:
                  other_uid_list.append(node.getUid())
                else:
                  other_uid_list.append(node.getRelativeUrl())
159 160
          if none_node_selected:
            # in this case we don't want to include NULL in All others
161
            query = NegatedQuery(Query(**{axis: other_uid_list}))
162
          else:
163
            query = ComplexQuery(
164 165
                            NegatedQuery(Query(**{axis: other_uid_list})),
                            Query(**{axis: None}),
166
                            operator="OR")
167

168
        else:
169 170 171 172 173 174
          value = portal_categories.getCategoryValue(node_url,
                  base_category=criterion_base_category)
          if uid_based_axis:
            query_dict.setdefault(axis, []).append(value.getUid())
          else:
            query_dict.setdefault(axis, []).append(value.getRelativeUrl())
175

176 177 178 179 180 181 182 183 184 185
    if query:
      if axis in query_dict:
        query_dict[axis] = ComplexQuery(
              query,
              Query(**{axis: query_dict[axis]}),
              operator='OR')
      else:
        query_dict[axis] = query


Jérome Perrin's avatar
Jérome Perrin committed
186
    return query_dict
187

188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205
  def getInventoryListQueryDict(self, budget_line):
    """Returns the query dict to pass to simulation query for a budget line
    """
    axis = self.getInventoryAxis()
    if not axis:
      return dict()
    base_category = self.getProperty('variation_base_category')
    if not base_category:
      return dict()

    context = budget_line
    if self.isMemberOf('budget_variation/budget'):
      context = budget_line.getParentValue()

    portal_categories = self.getPortalObject().portal_categories
    query_dict = dict()
    if axis == 'movement':
      axis = 'default_%s_uid' % base_category
206
      query_dict['select_list'] = [axis]
207 208
      query_dict['group_by'] = [axis]
    elif axis == 'movement_strict_membership':
209 210
      axis = 'default_strict_%s_uid' % base_category
      query_dict['select_list'] = [axis]
211 212 213
      query_dict['group_by'] = [axis]
    else:
      query_dict['group_by_%s' % axis] = True
214

215 216
    uid_based_axis = False
    if axis in ('node', 'section', 'payment', 'function', 'project',
Jérome Perrin's avatar
Jérome Perrin committed
217
                'mirror_section', 'mirror_node', 'funding' ):
218 219
      axis = '%s_uid' % axis
      uid_based_axis = True
220 221 222 223

    # if we have a virtual "all others" node, we don't set a criterion here.
    if self.getProperty('include_virtual_other_node'):
      return query_dict
224
    
225
    found = False
226
    for node_url in context.getVariationCategoryList(
227
                          base_category_list=(base_category,)):
228
      if node_url != '%s/budget_special_node/none' % base_category:
229 230 231
        __traceback_info__ = (node_url, )
        if uid_based_axis:
          query_dict.setdefault(axis, []).append(
232 233
                portal_categories.getCategoryValue(node_url,
                      base_category=base_category).getUid())
234 235 236 237
        else:
          query_dict.setdefault(axis, []).append(
                portal_categories.getCategoryValue(node_url,
                      base_category=base_category).getRelativeUrl())
238 239
      found = True
    if found:
240
      if self.getProperty('include_virtual_none_node'):
241 242 243 244 245 246 247
        if axis in query_dict:
          query_dict[axis] = ComplexQuery(
                Query(**{axis: None}),
                Query(**{axis: query_dict[axis]}),
                operator="OR")
        else:
          query_dict[axis] = Query(**{axis: None})
248 249
      return query_dict
    return dict()
250
  
251 252
  def _getCellKeyFromInventoryListBrain(self, brain, budget_line,
                                         cell_key_cache=None):
253
    """Compute key from inventory brain, with support for virtual nodes.
254
    """
255 256 257
    cell_key_cache[None] = '%s/budget_special_node/none'\
                                 % self.getProperty('variation_base_category')
    
258
    key = BudgetVariation._getCellKeyFromInventoryListBrain(
259
                   self, brain, budget_line, cell_key_cache=cell_key_cache)
260
    if self.getProperty('include_virtual_other_node'):
261 262
      if key not in budget_line.getVariationCategoryList(
          base_category_list=(self.getProperty('variation_base_category'),)):
263 264
        key = '%s/budget_special_node/all_other' % (
            self.getProperty('variation_base_category'),)
265
    return key
266

267 268 269 270 271 272 273 274 275
  def getBudgetLineVariationRangeCategoryList(self, budget_line):
    """Returns the Variation Range Category List that can be applied to this
    budget line.
    """
    base_category = self.getProperty('variation_base_category')
    prefix = ''
    if base_category:
      prefix = '%s/' % base_category
    return [(self._getNodeTitle(node), '%s%s' % (prefix, node.getRelativeUrl()))
276
                for node in self._getNodeList(budget_line)]
277

278
  def getBudgetVariationRangeCategoryList(self, budget):
279
    """Returns the Variation Range Category List that can be applied to this
280 281 282 283 284 285 286 287 288
    budget.
    """
    base_category = self.getProperty('variation_base_category')
    prefix = ''
    if base_category:
      prefix = '%s/' % base_category
    return [(self._getNodeTitle(node), '%s%s' % (prefix, node.getRelativeUrl()))
                for node in self._getNodeList(budget)]

289 290 291 292 293
  def initializeBudgetLine(self, budget_line):
    """Initialize a budget line
    """
    budget_line_variation_category_list =\
       list(budget_line.getVariationBaseCategoryList() or [])
294 295
    budget_line_membership_criterion_base_category_list =\
       list(budget_line.getMembershipCriterionBaseCategoryList() or [])
296 297 298 299 300
    base_category = self.getProperty('variation_base_category')
    if base_category:
      budget_line_variation_category_list.append(base_category)
      budget_line.setVariationBaseCategoryList(
              budget_line_variation_category_list)
301 302 303 304
    if self.isMemberOf('budget_variation/budget_line'):
      budget_line_membership_criterion_base_category_list.append(base_category)
      budget_line.setMembershipCriterionBaseCategoryList(
          budget_line_membership_criterion_base_category_list)
305

306 307 308
  def initializeBudget(self, budget):
    """Initialize a budget.
    """
309
    budget_variation_base_category_list =\
310
       list(budget.getVariationBaseCategoryList() or [])
311 312
    budget_membership_criterion_base_category_list =\
       list(budget.getMembershipCriterionBaseCategoryList() or [])
313 314
    base_category = self.getProperty('variation_base_category')
    if base_category:
315 316 317 318
      if base_category not in budget_variation_base_category_list:
        budget_variation_base_category_list.append(base_category)
      if base_category not in budget_membership_criterion_base_category_list:
        budget_membership_criterion_base_category_list.append(base_category)
319
      budget.setVariationBaseCategoryList(
320 321 322
              budget_variation_base_category_list)
      budget.setMembershipCriterionBaseCategoryList(
              budget_membership_criterion_base_category_list)
323