BudgetVariation.py 4.84 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
##############################################################################
#
# 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

30
from Products.ERP5Type import Permissions, PropertySheet
31 32 33 34
from Products.ERP5.Document.Predicate import Predicate

class BudgetVariation(Predicate):
  """Base class for budget variations.
35

36 37 38 39 40 41
  """

  # Default Properties
  property_sheets = ( PropertySheet.Base
                    , PropertySheet.XMLObject
                    , PropertySheet.SimpleItem
42
                    , PropertySheet.CategoryCore
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62
                    , PropertySheet.SortIndex
                    , PropertySheet.Path
                    , PropertySheet.Predicate
                    , PropertySheet.BudgetVariation
                    )

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

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

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

  def initializeBudgetLine(self, budget_line):
63 64 65 66 67 68 69 70 71
    """Initialize a budget line.
  
    Called when a new budget line is created.
    """

  def initializeBudget(self, budget):
    """Initialize a budget.

    Called when a budget is associated to a budget model.
72 73 74 75 76 77 78
    """

  def getBudgetLineVariationRangeCategoryList(self, budget_line):
    """Returns the variation range categories for this budget line
    """
    return []

79 80 81 82 83
  def getBudgetVariationRangeCategoryList(self, budget):
    """Returns the variation range categories for this budget
    """
    return []

84
  def getCellRangeForBudgetLine(self, budget_line, matrixbox=0):
85
    """Return the cell range to use for the budget line
86 87 88 89 90 91 92 93
    """
    return []

  def getInventoryQueryDict(self, budget_cell):
    """Returns the query dict to pass to simulation query
    """
    return {}

94 95 96 97 98
  def getInventoryListQueryDict(self, budget_line):
    """Returns the query dict to pass to simulation query for a budget line
    """
    return {}

99 100
  def _getCellKeyFromInventoryListBrain(self, brain, budget_line,
                                        cell_key_cache=None):
101 102 103
    """Compute the cell key from an inventory brain.
    The cell key can be used to retrieve the budget cell in the corresponding
    budget line using budget_line.getCell
104
    A dictionnary can be passed as "cell_key_cache" to cache catalog lookups
105 106 107 108 109 110 111 112 113 114
    """
    if not self.isMemberOf('budget_variation/budget_cell'):
      return None

    axis = self.getInventoryAxis()
    if not axis:
      return None
    base_category = self.getProperty('variation_base_category')
    if not base_category:
      return None
115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131
    
    getObject = self.getPortalObject().portal_catalog.getObject
    def getUrlFromUidNoCache(uid):
      relative_url = getObject(uid).getRelativeUrl()
      if relative_url.startswith('%s/' % base_category):
        return relative_url
      return '%s/%s' % (base_category, relative_url)

    if cell_key_cache is not None:
      def getUrlFromUidWithCache(uid):
        try:
          return cell_key_cache[uid]
        except KeyError:
          relative_url = getUrlFromUidNoCache(uid)
          cell_key_cache[uid] = relative_url
          return relative_url
      getUrlFromUid = getUrlFromUidWithCache
132
    else:
133
      getUrlFromUid = getUrlFromUidNoCache
134

135 136 137 138 139 140
    if axis == 'movement':
      return getUrlFromUid(getattr(brain, 'default_%s_uid' % base_category))
    elif axis == 'movement_strict_membership':
      return getUrlFromUid(getattr(brain,
                                   'default_strict_%s_uid' % base_category))
    return getUrlFromUid(getattr(brain, '%s_uid' % axis))
141