############################################################################## # # Copyright (c) 2009 Nexedi KK and Contributors. All Rights Reserved. # # WARNING: This program as such is intended to be used by professional # programmers who take the whole responsibility 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 # guarantees 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., 51 Franklin Street - Fifth Floor, Boston, MA # 02110-1301, USA. # ############################################################################## from AccessControl import ClassSecurityInfo from Products.ERP5Type import PropertySheet, Permissions from Products.ERP5.Document.Predicate import Predicate from Products.ERP5Type.Utils import UpperCase class RoundingModel(Predicate): """ A Rounding Model class which defines rounding rule. """ meta_type = 'ERP5 Rounding Model' portal_type = 'Rounding Model' add_permission = Permissions.AddPortalContent security = ClassSecurityInfo() property_sheets = (PropertySheet.Base, PropertySheet.SimpleItem, PropertySheet.XMLObject, PropertySheet.CategoryCore, PropertySheet.DublinCore, PropertySheet.Predicate, PropertySheet.SortIndex, PropertySheet.RoundingModel, ) security.declareProtected(Permissions.AccessContentsInformation, 'roundValue') def roundValue(self, value): """ Return rounded value. """ if self.getRoundingMethodId() is not None: rounding_method = getattr(self, self.getRoundingMethodId(), None) if rounding_method is None: raise ValueError, 'Rounding method (%s) was not found.' else: from decimal import Decimal from Products.ERP5.Tool.RoundingTool import ROUNDING_OPTION_DICT decimal_rounding_option = self.getDecimalRoundingOption() if (decimal_rounding_option is None or decimal_rounding_option not in ROUNDING_OPTION_DICT): raise ValueError, 'Decimal rounding option must be selected.' def rounding_method(value, decimal_exponent): return float(Decimal(str(value)).quantize(Decimal(decimal_exponent), rounding=decimal_rounding_option)) return rounding_method(value, self.getDecimalExponent()) security.declareProtected(Permissions.AccessContentsInformation, 'getRoundingProxy') def getRoundingProxy(self, document): """ Return a rounding proxy object which getter methods returns rounded value by following the rounding model definition. """ rounding_model = self rounded_property_getter_method_name_list = [] if isinstance(document, RoundingProxy): temp_document = document._getOriginalDocument() original_document = document else: from Products.ERP5Type import Document if document.__class__.__name__ == 'TempDocument': class_ = document.__class__.__bases__[0] else: class_ = document.__class__ constructor = getattr(Document, 'newTemp%s' % class_.__name__) temp_document = constructor(document.getParentValue(), 'id') temp_document.__dict__.update(document.__dict__) original_document = temp_document for property_id in rounding_model.getRoundedPropertyIdList(): getter_name = 'get%s' % UpperCase(property_id) getter = getattr(temp_document, getter_name, None) setter_name = 'set%s' % UpperCase(property_id) setter = getattr(temp_document, setter_name, None) if getter is not None and setter is not None: # round the property value itself setter(self.roundValue(getter())) else: # cannot round the property value so that the return value of getter # will be rounded rounded_property_getter_method_name_list.append(getter_name) class _RoundingProxy(RoundingProxy): def _getOriginalDocument(self): if isinstance(original_document, RoundingProxy): return original_document._editOriginalDocument() else: return original_document def __getattr__(self, name): attribute = getattr(original_document, name) if getattr(attribute, 'DUMMY_ROUNDING_METHOD_MARK', None) is DUMMY_ROUNDING_METHOD_MARK: return attribute if name in rounded_property_getter_method_name_list: def dummyMethod(*args, **kw): return rounding_model.roundValue(attribute(*args, **kw)) dummyMethod.DUMMY_ROUNDING_METHOD_MARK = DUMMY_ROUNDING_METHOD_MARK return dummyMethod else: return attribute return _RoundingProxy() DUMMY_ROUNDING_METHOD_MARK = object() class RoundingProxy(object): """Super class of _RoundingProxy class defined above. Use this class for isinstance method to check if object is a real instance or a rounding proxy instance. """