[FIX] regression that made the qweb modules be loaded 2 times
[odoo/odoo.git] / 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
52
53 class crossovered_budget(osv.osv):
54     _name = "crossovered.budget"
55     _description = "Budget"
56
57     _columns = {
58         'name': fields.char('Name', size=64, required=True, states={'done':[('readonly',True)]}),
59         'code': fields.char('Code', size=16, required=True, states={'done':[('readonly',True)]}),
60         'creating_user_id': fields.many2one('res.users', 'Responsible User'),
61         'validating_user_id': fields.many2one('res.users', 'Validate User', readonly=True),
62         'date_from': fields.date('Start Date', required=True, states={'done':[('readonly',True)]}),
63         'date_to': fields.date('End Date', required=True, states={'done':[('readonly',True)]}),
64         'state' : fields.selection([('draft','Draft'),('cancel', 'Cancelled'),('confirm','Confirmed'),('validate','Validated'),('done','Done')], 'Status', select=True, required=True, readonly=True),
65         'crossovered_budget_line': fields.one2many('crossovered.budget.lines', 'crossovered_budget_id', 'Budget Lines', states={'done':[('readonly',True)]}),
66         'company_id': fields.many2one('res.company', 'Company', required=True),
67     }
68
69     _defaults = {
70         'state': 'draft',
71         'creating_user_id': lambda self, cr, uid, context: uid,
72         'company_id': lambda self, cr, uid, c: self.pool.get('res.company')._company_default_get(cr, uid, 'account.budget.post', context=c)
73     }
74
75     def budget_confirm(self, cr, uid, ids, *args):
76         self.write(cr, uid, ids, {
77             'state': 'confirm'
78         })
79         return True
80
81     def budget_draft(self, cr, uid, ids, *args):
82         self.write(cr, uid, ids, {
83             'state': 'draft'
84         })
85         return True
86
87     def budget_validate(self, cr, uid, ids, *args):
88         self.write(cr, uid, ids, {
89             'state': 'validate',
90             'validating_user_id': uid,
91         })
92         return True
93
94     def budget_cancel(self, cr, uid, ids, *args):
95         self.write(cr, uid, ids, {
96             'state': 'cancel'
97         })
98         return True
99
100     def budget_done(self, cr, uid, ids, *args):
101         self.write(cr, uid, ids, {
102             'state': 'done'
103         })
104         return True
105
106
107 class crossovered_budget_lines(osv.osv):
108
109     def _prac_amt(self, cr, uid, ids, context=None):
110         res = {}
111         result = 0.0
112         if context is None: 
113             context = {}
114         for line in self.browse(cr, uid, ids, context=context):
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 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=None):
135         res={}
136         for line in self.browse(cr, uid, ids, context=context):
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=None):
141         res = {}
142         if context is None: 
143             context = {}
144         for line in self.browse(cr, uid, ids, context=context):
145             today = datetime.datetime.today()
146             date_to = today.strftime("%Y-%m-%d")
147             date_from = line.date_from
148             if context.has_key('wizard_date_from'):
149                 date_from = context['wizard_date_from']
150             if context.has_key('wizard_date_to'):
151                 date_to = context['wizard_date_to']
152
153             if line.paid_date:
154                 if strToDate(line.date_to) <= strToDate(line.paid_date):
155                     theo_amt = 0.00
156                 else:
157                     theo_amt = line.planned_amount
158             else:
159                 total = strToDate(line.date_to) - strToDate(line.date_from)
160                 elapsed = min(strToDate(line.date_to),strToDate(date_to)) - max(strToDate(line.date_from),strToDate(date_from))
161                 if strToDate(date_to) < strToDate(line.date_from):
162                     elapsed = strToDate(date_to) - strToDate(date_to)
163
164                 if total.days:
165                     theo_amt = float(elapsed.days / float(total.days)) * line.planned_amount
166                 else:
167                     theo_amt = line.planned_amount
168
169             res[line.id] = theo_amt
170         return res
171
172     def _theo(self, cr, uid, ids, name, args, context=None):
173         res = {}
174         for line in self.browse(cr, uid, ids, context=context):
175             res[line.id] = self._theo_amt(cr, uid, [line.id], context=context)[line.id]
176         return res
177
178     def _perc(self, cr, uid, ids, name, args, context=None):
179         res = {}
180         for line in self.browse(cr, uid, ids, context=context):
181             if line.theoritical_amount <> 0.00:
182                 res[line.id] = float((line.practical_amount or 0.0) / line.theoritical_amount) * 100
183             else:
184                 res[line.id] = 0.00
185         return res
186
187     _name = "crossovered.budget.lines"
188     _description = "Budget Line"
189     _columns = {
190         'crossovered_budget_id': fields.many2one('crossovered.budget', 'Budget', ondelete='cascade', select=True, required=True),
191         'analytic_account_id': fields.many2one('account.analytic.account', 'Analytic Account'),
192         'general_budget_id': fields.many2one('account.budget.post', 'Budgetary Position',required=True),
193         'date_from': fields.date('Start Date', required=True),
194         'date_to': fields.date('End Date', required=True),
195         'paid_date': fields.date('Paid Date'),
196         'planned_amount':fields.float('Planned Amount', required=True, digits_compute=dp.get_precision('Account')),
197         'practical_amount':fields.function(_prac, string='Practical Amount', type='float', digits_compute=dp.get_precision('Account')),
198         'theoritical_amount':fields.function(_theo, string='Theoretical Amount', type='float', digits_compute=dp.get_precision('Account')),
199         'percentage':fields.function(_perc, string='Percentage', type='float'),
200         'company_id': fields.related('crossovered_budget_id', 'company_id', type='many2one', relation='res.company', string='Company', store=True, readonly=True)
201     }
202
203
204 class account_analytic_account(osv.osv):
205     _inherit = "account.analytic.account"
206
207     _columns = {
208         'crossovered_budget_line': fields.one2many('crossovered.budget.lines', 'analytic_account_id', 'Budget Lines'),
209     }
210
211
212 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: