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 openerp.osv import fields, osv
  25 from openerp.tools.translate import _
  26
  27 import openerp.addons.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'),('cancel', 'Cancelled'),('confirm','Confirmed'),('validate','Validated'),('done','Done')], '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=None):
 112         res = {}
 113         result = 0.0
 114         if context is None:
 115             context = {}
 116         for line in self.browse(cr, uid, ids, context=context):
 117             acc_ids = [x.id for x in line.general_budget_id.account_ids]
 118             if not acc_ids:
 119                 raise osv.except_osv(_('Error!'),_("The Budget '%s' has no accounts!") % str(line.general_budget_id.name))
 120             date_to = line.date_to
 121             date_from = line.date_from
 122             if context.has_key('wizard_date_from'):
 123                 date_from = context['wizard_date_from']
 124             if context.has_key('wizard_date_to'):
 125                 date_to = context['wizard_date_to']
 126             if line.analytic_account_id.id:
 127                 cr.execute("SELECT SUM(amount) FROM account_analytic_line WHERE account_id=%s AND (date "
 128                        "between to_date(%s,'yyyy-mm-dd') AND to_date(%s,'yyyy-mm-dd')) AND "
 129                        "general_account_id=ANY(%s)", (line.analytic_account_id.id, date_from, date_to,acc_ids,))
 130                 result = cr.fetchone()[0]
 131             if result is None:
 132                 result = 0.00
 133             res[line.id] = result
 134         return res
 135
 136     def _prac(self, cr, uid, ids, name, args, context=None):
 137         res={}
 138         for line in self.browse(cr, uid, ids, context=context):
 139             res[line.id] = self._prac_amt(cr, uid, [line.id], context=context)[line.id]
 140         return res
 141
 142     def _theo_amt(self, cr, uid, ids, context=None):
 143         res = {}
 144         if context is None:
 145             context = {}
 146         for line in self.browse(cr, uid, ids, context=context):
 147             today = datetime.datetime.today()
 148             date_to = today.strftime("%Y-%m-%d")
 149             date_from = line.date_from
 150             if context.has_key('wizard_date_from'):
 151                 date_from = context['wizard_date_from']
 152             if context.has_key('wizard_date_to'):
 153                 date_to = context['wizard_date_to']
 154
 155             if line.paid_date:
 156                 if strToDate(line.date_to) <= strToDate(line.paid_date):
 157                     theo_amt = 0.00
 158                 else:
 159                     theo_amt = line.planned_amount
 160             else:
 161                 total = strToDate(line.date_to) - strToDate(line.date_from)
 162                 elapsed = min(strToDate(line.date_to),strToDate(date_to)) - max(strToDate(line.date_from),strToDate(date_from))
 163                 if strToDate(date_to) < strToDate(line.date_from):
 164                     elapsed = strToDate(date_to) - strToDate(date_to)
 165
 166                 if total.days:
 167                     theo_amt = float(elapsed.days / float(total.days)) * line.planned_amount
 168                 else:
 169                     theo_amt = line.planned_amount
 170
 171             res[line.id] = theo_amt
 172         return res
 173
 174     def _theo(self, cr, uid, ids, name, args, context=None):
 175         res = {}
 176         for line in self.browse(cr, uid, ids, context=context):
 177             res[line.id] = self._theo_amt(cr, uid, [line.id], context=context)[line.id]
 178         return res
 179
 180     def _perc(self, cr, uid, ids, name, args, context=None):
 181         res = {}
 182         for line in self.browse(cr, uid, ids, context=context):
 183             if line.theoritical_amount <> 0.00:
 184                 res[line.id] = float((line.practical_amount or 0.0) / line.theoritical_amount) * 100
 185             else:
 186                 res[line.id] = 0.00
 187         return res
 188
 189     _name = "crossovered.budget.lines"
 190     _description = "Budget Line"
 191     _columns = {
 192         'crossovered_budget_id': fields.many2one('crossovered.budget', 'Budget', ondelete='cascade', select=True, required=True),
 193         'analytic_account_id': fields.many2one('account.analytic.account', 'Analytic Account'),
 194         'general_budget_id': fields.many2one('account.budget.post', 'Budgetary Position',required=True),
 195         'date_from': fields.date('Start Date', required=True),
 196         'date_to': fields.date('End Date', required=True),
 197         'paid_date': fields.date('Paid Date'),
 198         'planned_amount':fields.float('Planned Amount', required=True, digits_compute=dp.get_precision('Account')),
 199         'practical_amount':fields.function(_prac, string='Practical Amount', type='float', digits_compute=dp.get_precision('Account')),
 200         'theoritical_amount':fields.function(_theo, string='Theoretical Amount', type='float', digits_compute=dp.get_precision('Account')),
 201         'percentage':fields.function(_perc, string='Percentage', type='float'),
 202         'company_id': fields.related('crossovered_budget_id', 'company_id', type='many2one', relation='res.company', string='Company', store=True, readonly=True)
 203     }
 204
 205 crossovered_budget_lines()
 206
 207 class account_analytic_account(osv.osv):
 208     _inherit = "account.analytic.account"
 209
 210     _columns = {
 211         'crossovered_budget_line': fields.one2many('crossovered.budget.lines', 'analytic_account_id', 'Budget Lines'),
 212     }
 213
 214 account_analytic_account()
 215
 216 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: