addons/account_budget/account_budget.py

   1 # -*- coding: utf-8 -*-
   2 ##############################################################################
   3 #
   4 #    OpenERP, Open Source Management Solution
   5 #    Copyright (C) 2004-2010 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 datetime
  23
  24 from osv import osv, fields
  25 from tools.translate import _
  26
  27 import decimal_precision as dp
  28
  29 def strToDate(dt):
  30         dt_date=datetime.date(int(dt[0:4]),int(dt[5:7]),int(dt[8:10]))
  31         return dt_date
  32
  33 # ---------------------------------------------------------
  34 # Budgets
  35 # ---------------------------------------------------------
  36 class account_budget_post(osv.osv):
  37     _name = "account.budget.post"
  38     _description = "Budgetary Position"
  39     _columns = {
  40         'code': fields.char('Code', size=64, required=True),
  41         'name': fields.char('Name', size=256, required=True),
  42         'account_ids': fields.many2many('account.account', 'account_budget_rel', 'budget_id', 'account_id', 'Accounts'),
  43         'crossovered_budget_line': fields.one2many('crossovered.budget.lines', 'general_budget_id', 'Budget Lines'),
  44         'company_id': fields.many2one('res.company', 'Company', required=True),
  45     }
  46     _defaults = {
  47         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'account.budget.post', context=c)
  48     }
  49     _order = "name"
  50
  51 account_budget_post()
  52
  53
  54 class crossovered_budget(osv.osv):
  55     _name = "crossovered.budget"
  56     _description = "Budget"
  57
  58     _columns = {
  59         'name': fields.char('Name', size=64, required=True, states={'done':[('readonly',True)]}),
  60         'code': fields.char('Code', size=16, required=True, states={'done':[('readonly',True)]}),
  61         'creating_user_id': fields.many2one('res.users', 'Responsible User'),
  62         'validating_user_id': fields.many2one('res.users', 'Validate User', readonly=True),
  63         'date_from': fields.date('Start Date', required=True, states={'done':[('readonly',True)]}),
  64         'date_to': fields.date('End Date', required=True, states={'done':[('readonly',True)]}),
  65         'state' : fields.selection([('draft','Draft'),('confirm','Confirmed'),('validate','Validated'),('done','Done'),('cancel', 'Cancelled')], 'Status', select=True, required=True, readonly=True),
  66         'crossovered_budget_line': fields.one2many('crossovered.budget.lines', 'crossovered_budget_id', 'Budget Lines', states={'done':[('readonly',True)]}),
  67         'company_id': fields.many2one('res.company', 'Company', required=True),
  68     }
  69
  70     _defaults = {
  71         'state': 'draft',
  72         'creating_user_id': lambda self, cr, uid, context: uid,
  73         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'account.budget.post', context=c)
  74     }
  75
  76     def budget_confirm(self, cr, uid, ids, *args):
  77         self.write(cr, uid, ids, {
  78             'state': 'confirm'
  79         })
  80         return True
  81
  82     def budget_draft(self, cr, uid, ids, *args):
  83         self.write(cr, uid, ids, {
  84             'state': 'draft'
  85         })
  86         return True
  87
  88     def budget_validate(self, cr, uid, ids, *args):
  89         self.write(cr, uid, ids, {
  90             'state': 'validate',
  91             'validating_user_id': uid,
  92         })
  93         return True
  94
  95     def budget_cancel(self, cr, uid, ids, *args):
  96         self.write(cr, uid, ids, {
  97             'state': 'cancel'
  98         })
  99         return True
 100
 101     def budget_done(self, cr, uid, ids, *args):
 102         self.write(cr, uid, ids, {
 103             'state': 'done'
 104         })
 105         return True
 106
 107 crossovered_budget()
 108
 109 class crossovered_budget_lines(osv.osv):
 110
 111     def _prac_amt(self, cr, uid, ids, context={}):
 112         res = {}
 113         result = 0.0
 114         for line in self.browse(cr, uid, ids):
 115             acc_ids = [x.id for x in line.general_budget_id.account_ids]
 116             if not acc_ids:
 117                 raise osv.except_osv(_('Error!'),_("The General Budget '%s' has no Accounts!") % str(line.general_budget_id.name))
 118             date_to = line.date_to
 119             date_from = line.date_from
 120             if context.has_key('wizard_date_from'):
 121                 date_from = context['wizard_date_from']
 122             if context.has_key('wizard_date_to'):
 123                 date_to = context['wizard_date_to']
 124             if line.analytic_account_id.id:
 125                 cr.execute("SELECT SUM(amount) FROM account_analytic_line WHERE account_id=%s AND (date "
 126                        "between to_date(%s,'yyyy-mm-dd') AND to_date(%s,'yyyy-mm-dd')) AND "
 127                        "general_account_id=ANY(%s)", (line.analytic_account_id.id, date_from, date_to,acc_ids,))
 128                 result = cr.fetchone()[0]
 129             if result is None:
 130                 result = 0.00
 131             res[line.id] = result
 132         return res
 133
 134     def _prac(self, cr, uid, ids, name, args, context):
 135         res={}
 136         for line in self.browse(cr, uid, ids):
 137             res[line.id] = self._prac_amt(cr, uid, [line.id], context=context)[line.id]
 138         return res
 139
 140     def _theo_amt(self, cr, uid, ids, context={}):
 141         res = {}
 142         for line in self.browse(cr, uid, ids):
 143             today = datetime.datetime.today()
 144             date_to = today.strftime("%Y-%m-%d")
 145             date_from = line.date_from
 146             if context.has_key('wizard_date_from'):
 147                 date_from = context['wizard_date_from']
 148             if context.has_key('wizard_date_to'):
 149                 date_to = context['wizard_date_to']
 150
 151             if line.paid_date:
 152                 if strToDate(line.date_to) <= strToDate(line.paid_date):
 153                     theo_amt = 0.00
 154                 else:
 155                     theo_amt = line.planned_amount
 156             else:
 157                 total = strToDate(line.date_to) - strToDate(line.date_from)
 158                 elapsed = min(strToDate(line.date_to),strToDate(date_to)) - max(strToDate(line.date_from),strToDate(date_from))
 159                 if strToDate(date_to) < strToDate(line.date_from):
 160                     elapsed = strToDate(date_to) - strToDate(date_to)
 161
 162                 if total.days:
 163                     theo_amt = float(elapsed.days / float(total.days)) * line.planned_amount
 164                 else:
 165                     theo_amt = line.planned_amount
 166
 167             res[line.id] = theo_amt
 168         return res
 169
 170     def _theo(self, cr, uid, ids, name, args, context):
 171         res = {}
 172         for line in self.browse(cr, uid, ids):
 173             res[line.id] = self._theo_amt(cr, uid, [line.id], context=context)[line.id]
 174         return res
 175
 176     def _perc(self, cr, uid, ids, name, args, context):
 177         res = {}
 178         for line in self.browse(cr, uid, ids):
 179             if line.theoritical_amount <> 0.00:
 180                 res[line.id] = float(line.practical_amount or 0.0 / line.theoritical_amount) * 100
 181             else:
 182                 res[line.id] = 0.00
 183         return res
 184
 185     _name = "crossovered.budget.lines"
 186     _description = "Budget Line"
 187     _columns = {
 188         'crossovered_budget_id': fields.many2one('crossovered.budget', 'Budget', ondelete='cascade', select=True, required=True),
 189         'analytic_account_id': fields.many2one('account.analytic.account', 'Analytic Account',required=False),
 190         'general_budget_id': fields.many2one('account.budget.post', 'Budgetary Position',required=True),
 191         'date_from': fields.date('Start Date', required=True),
 192         'date_to': fields.date('End Date', required=True),
 193         'paid_date': fields.date('Paid Date'),
 194         'planned_amount':fields.float('Planned Amount', required=True, digits_compute=dp.get_precision('Account')),
 195         'practical_amount':fields.function(_prac, method=True, string='Practical Amount', type='float', digits_compute=dp.get_precision('Account')),
 196         'theoritical_amount':fields.function(_theo, method=True, string='Theoritical Amount', type='float', digits_compute=dp.get_precision('Account')),
 197         'percentage':fields.function(_perc, method=True, string='Percentage', type='float'),
 198         'company_id': fields.related('crossovered_budget_id', 'company_id', type='many2one', relation='res.company', string='Company', store=True, readonly=True)
 199     }
 200
 201 crossovered_budget_lines()
 202
 203 class account_analytic_account(osv.osv):
 204     _inherit = "account.analytic.account"
 205
 206     _columns = {
 207         'crossovered_budget_line': fields.one2many('crossovered.budget.lines', 'analytic_account_id', 'Budget Lines'),
 208     }
 209
 210 account_analytic_account()
 211
 212 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: