addons/decimal_precision/decimal_precision.py

   1 # -*- encoding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2009 Tiny SPRL (<http://tiny.be>).
   6 #
   7 #    This program is free software: you can redistribute it and/or modify
   8 #    it under the terms of the GNU Affero General Public License as
   9 #    published by the Free Software Foundation, either version 3 of the
  10 #    License, or (at your option) any later version.
  11 #
  12 #    This program is distributed in the hope that it will be useful,
  13 #    but WITHOUT ANY WARRANTY; without even the implied warranty of
  14 #    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  15 #    GNU Affero General Public License for more details.
  16 #
  17 #    You should have received a copy of the GNU Affero General Public License
  18 #    along with this program.  If not, see <http://www.gnu.org/licenses/>.
  19 #
  20 ##############################################################################
  21
  22 import openerp
  23 from openerp import SUPERUSER_ID
  24 from openerp import tools
  25 from openerp.osv import orm, fields
  26 from openerp.modules.registry import RegistryManager
  27
  28 class decimal_precision(orm.Model):
  29     _name = 'decimal.precision'
  30     _columns = {
  31         'name': fields.char('Usage', size=50, select=True, required=True),
  32         'digits': fields.integer('Digits', required=True),
  33     }
  34     _defaults = {
  35         'digits': 2,
  36     }
  37
  38     _sql_constraints = [
  39         ('name_uniq', 'unique (name)', """Only one value can be defined for each given usage!"""),
  40     ]
  41
  42     @tools.ormcache(skiparg=3)
  43     def precision_get(self, cr, uid, application):
  44         cr.execute('select digits from decimal_precision where name=%s', (application,))
  45         res = cr.fetchone()
  46         return res[0] if res else 2
  47
  48     def clear_cache(self, cr):
  49         """clear cache and update models. Notify other workers to restart their registry."""
  50         self.precision_get.clear_cache(self)
  51         for obj in self.pool.obj_list():
  52             for colname, col in self.pool.get(obj)._columns.items():
  53                 if hasattr(col, 'digits_change'):
  54                     col.digits_change(cr)
  55         RegistryManager.signal_caches_change(cr.dbname)
  56
  57     def create(self, cr, uid, data, context=None):
  58         res = super(decimal_precision, self).create(cr, uid, data, context=context)
  59         self.clear_cache(cr)
  60         return res
  61
  62     def unlink(self, cr, uid, ids, context=None):
  63         res = super(decimal_precision, self).unlink(cr, uid, ids, context=context)
  64         self.clear_cache(cr)
  65         return res
  66
  67     def write(self, cr, uid, ids, data, *args, **argv):
  68         res = super(decimal_precision, self).write(cr, uid, ids, data, *args, **argv)
  69         self.clear_cache(cr)
  70         return res
  71
  72
  73 def get_precision(application):
  74     def change_digit(cr):
  75         decimal_precision = openerp.registry(cr.dbname)['decimal.precision']
  76         res = decimal_precision.precision_get(cr, SUPERUSER_ID, application)
  77         return (16, res)
  78     return change_digit
  79
  80 class DecimalPrecisionFloat(orm.AbstractModel):
  81     """ Override qweb.field.float to add a `decimal_precision` domain option
  82     and use that instead of the column's own value if it is specified
  83     """
  84     _inherit = 'ir.qweb.field.float'
  85
  86
  87     def precision(self, cr, uid, column, options=None, context=None):
  88         dp = options and options.get('decimal_precision')
  89         if dp:
  90             return self.pool['decimal.precision'].precision_get(
  91                 cr, uid, dp)
  92
  93         return super(DecimalPrecisionFloat, self).precision(
  94             cr, uid, column, options=options, context=context)
  95
  96 class DecimalPrecisionTestModel(orm.Model):
  97     _name = 'decimal.precision.test'
  98
  99     _columns = {
 100         'float': fields.float(),
 101         'float_2': fields.float(digits=(16, 2)),
 102         'float_4': fields.float(digits=(16, 4)),
 103     }
 104
 105 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: