[FIX] account_asset : Asset in multi company: impossible to create/post items for...
[odoo/odoo.git] / addons / account_asset / account_asset.py
1 # -*- encoding: 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 time
23 from datetime import datetime
24 from dateutil.relativedelta import relativedelta
25
26 from osv import osv, fields
27 import decimal_precision as dp
28
29 class account_asset_category(osv.osv):
30     _name = 'account.asset.category'
31     _description = 'Asset category'
32
33     _columns = {
34         'name': fields.char('Name', size=64, required=True, select=1),
35         'note': fields.text('Note'),
36         'account_analytic_id': fields.many2one('account.analytic.account', 'Analytic account'),
37         'account_asset_id': fields.many2one('account.account', 'Asset Account', required=True),
38         'account_depreciation_id': fields.many2one('account.account', 'Depreciation Account', required=True),
39         'account_expense_depreciation_id': fields.many2one('account.account', 'Depr. Expense Account', required=True),
40         'journal_id': fields.many2one('account.journal', 'Journal', required=True),
41         'company_id': fields.many2one('res.company', 'Company', required=True),
42         'method': fields.selection([('linear','Linear'),('degressive','Degressive')], 'Computation Method', required=True, help="Choose the method to use to compute the amount of depreciation lines.\n"\
43             "  * Linear: Calculated on basis of: Gross Value / Number of Depreciations\n" \
44             "  * Degressive: Calculated on basis of: Remaining Value * Degressive Factor"),
45         'method_number': fields.integer('Number of Depreciations'),
46         'method_period': fields.integer('Period Length', help="State here the time between 2 depreciations, in months", required=True),
47         'method_progress_factor': fields.float('Degressive Factor'),
48         'method_time': fields.selection([('number','Number of Depreciations'),('end','Ending Date')], 'Time Method', required=True,
49                                   help="Choose the method to use to compute the dates and number of depreciation lines.\n"\
50                                        "  * Number of Depreciations: Fix the number of depreciation lines and the time between 2 depreciations.\n" \
51                                        "  * Ending Date: Choose the time between 2 depreciations and the date the depreciations won't go beyond."),
52         'method_end': fields.date('Ending date'),
53         'prorata':fields.boolean('Prorata Temporis', help='Indicates that the first depreciation entry for this asset have to be done from the purchase date instead of the first January'),
54         'open_asset': fields.boolean('Skip Draft State', help="Check this if you want to automatically confirm the assets of this category when created by invoices."),
55     }
56
57     _defaults = {
58         'company_id': lambda self, cr, uid, context: self.pool.get('res.company')._company_default_get(cr, uid, 'account.asset.category', context=context),
59         'method': 'linear',
60         'method_number': 5,
61         'method_time': 'number',
62         'method_period': 12,
63         'method_progress_factor': 0.3,
64     }
65
66     def onchange_account_asset(self, cr, uid, ids, account_asset_id, context=None):
67         res = {'value':{}}
68         if account_asset_id:
69            res['value'] = {'account_depreciation_id': account_asset_id}
70         return res
71
72 account_asset_category()
73
74 class account_asset_asset(osv.osv):
75     _name = 'account.asset.asset'
76     _description = 'Asset'
77
78     def _get_period(self, cr, uid, context={}):
79         periods = self.pool.get('account.period').find(cr, uid)
80         if periods:
81             return periods[0]
82         else:
83             return False
84
85     def _get_last_depreciation_date(self, cr, uid, ids, context=None):
86         """
87         @param id: ids of a account.asset.asset objects
88         @return: Returns a dictionary of the effective dates of the last depreciation entry made for given asset ids. If there isn't any, return the purchase date of this asset
89         """
90         cr.execute("""
91             SELECT a.id as id, COALESCE(MAX(l.date),a.purchase_date) AS date
92             FROM account_asset_asset a
93             LEFT JOIN account_move_line l ON (l.asset_id = a.id)
94             WHERE a.id IN %s
95             GROUP BY a.id, a.purchase_date """, (tuple(ids),))
96         return dict(cr.fetchall())
97
98     def _compute_board_amount(self, cr, uid, asset, i, residual_amount, amount_to_depr, undone_dotation_number, posted_depreciation_line_ids, total_days, depreciation_date, context=None):
99         #by default amount = 0
100         amount = 0
101         if i == undone_dotation_number:
102             amount = residual_amount
103         else:
104             if asset.method == 'linear':
105                 amount = amount_to_depr / (undone_dotation_number - len(posted_depreciation_line_ids))
106                 if asset.prorata:
107                     amount = amount_to_depr / asset.method_number
108                     days = total_days - float(depreciation_date.strftime('%j'))
109                     if i == 1:
110                         amount = (amount_to_depr / asset.method_number) / total_days * days
111                     elif i == undone_dotation_number:
112                         amount = (amount_to_depr / asset.method_number) / total_days * (total_days - days)
113             elif asset.method == 'degressive':
114                 amount = residual_amount * asset.method_progress_factor
115                 if asset.prorata:
116                     days = total_days - float(depreciation_date.strftime('%j'))
117                     if i == 1:
118                         amount = (residual_amount * asset.method_progress_factor) / total_days * days
119                     elif i == undone_dotation_number:
120                         amount = (residual_amount * asset.method_progress_factor) / total_days * (total_days - days)
121         return amount
122
123     def _compute_board_undone_dotation_nb(self, cr, uid, asset, depreciation_date, total_days, context=None):
124         undone_dotation_number = asset.method_number
125         if asset.method_time == 'end':
126             end_date = datetime.strptime(asset.method_end, '%Y-%m-%d')
127             undone_dotation_number = 0
128             while depreciation_date <= end_date:
129                 depreciation_date = (datetime(depreciation_date.year, depreciation_date.month, depreciation_date.day) + relativedelta(months=+asset.method_period))
130                 undone_dotation_number += 1
131         if asset.prorata:
132             undone_dotation_number += 1
133         return undone_dotation_number
134
135     def compute_depreciation_board(self, cr, uid, ids, context=None):
136         depreciation_lin_obj = self.pool.get('account.asset.depreciation.line')
137         for asset in self.browse(cr, uid, ids, context=context):
138             if asset.value_residual == 0.0:
139                 continue
140             posted_depreciation_line_ids = depreciation_lin_obj.search(cr, uid, [('asset_id', '=', asset.id), ('move_check', '=', True)])
141             old_depreciation_line_ids = depreciation_lin_obj.search(cr, uid, [('asset_id', '=', asset.id), ('move_id', '=', False)])
142             if old_depreciation_line_ids:
143                 depreciation_lin_obj.unlink(cr, uid, old_depreciation_line_ids, context=context)
144
145             amount_to_depr = residual_amount = asset.value_residual
146             if asset.prorata:
147                 depreciation_date = datetime.strptime(self._get_last_depreciation_date(cr, uid, [asset.id], context)[asset.id], '%Y-%m-%d')
148             else:
149                 # depreciation_date = 1st January of purchase year
150                 purchase_date = datetime.strptime(asset.purchase_date, '%Y-%m-%d')
151                 depreciation_date = datetime(purchase_date.year, 1, 1)
152             day = depreciation_date.day
153             month = depreciation_date.month
154             year = depreciation_date.year
155             total_days = (year % 4) and 365 or 366
156
157             undone_dotation_number = self._compute_board_undone_dotation_nb(cr, uid, asset, depreciation_date, total_days, context=context)
158             for x in range(len(posted_depreciation_line_ids), undone_dotation_number):
159                 i = x + 1
160                 amount = self._compute_board_amount(cr, uid, asset, i, residual_amount, amount_to_depr, undone_dotation_number, posted_depreciation_line_ids, total_days, depreciation_date, context=context)
161                 residual_amount -= amount
162                 vals = {
163                      'amount': amount,
164                      'asset_id': asset.id,
165                      'sequence': i,
166                      'name': str(asset.id) +'/' + str(i),
167                      'remaining_value': residual_amount,
168                      'depreciated_value': (asset.purchase_value - asset.salvage_value) - (residual_amount + amount),
169                      'depreciation_date': depreciation_date.strftime('%Y-%m-%d'),
170                 }
171                 depreciation_lin_obj.create(cr, uid, vals, context=context)
172                 # Considering Depr. Period as months
173                 depreciation_date = (datetime(year, month, day) + relativedelta(months=+asset.method_period))
174                 day = depreciation_date.day
175                 month = depreciation_date.month
176                 year = depreciation_date.year
177         return True
178
179     def validate(self, cr, uid, ids, context={}):
180         return self.write(cr, uid, ids, {
181             'state':'open'
182         }, context)
183
184     def set_to_close(self, cr, uid, ids, context=None):
185         return self.write(cr, uid, ids, {'state': 'close'}, context=context)
186
187     def _amount_residual(self, cr, uid, ids, name, args, context=None):
188         cr.execute("""SELECT
189                 l.asset_id as id, round(SUM(abs(l.debit-l.credit))) AS amount
190             FROM
191                 account_move_line l
192             WHERE
193                 l.asset_id IN %s GROUP BY l.asset_id """, (tuple(ids),))
194         res=dict(cr.fetchall())
195         for asset in self.browse(cr, uid, ids, context):
196             res[asset.id] = asset.purchase_value - res.get(asset.id, 0.0) - asset.salvage_value
197         for id in ids:
198             res.setdefault(id, 0.0)
199         return res
200
201     def onchange_company_id(self, cr, uid, ids, company_id, context=None):
202         val = {}
203         if company_id:
204             company = self.pool.get('res.company').browse(cr, uid, company_id, context=context)
205             if company.currency_id.company_id and company.currency_id.company_id.id != company_id:
206                 val['currency_id'] = False
207             else:
208                 val['currency_id'] = company.currency_id.id
209         return {'value': val}
210
211     _columns = {
212         'account_move_line_ids': fields.one2many('account.move.line', 'asset_id', 'Entries', readonly=True, states={'draft':[('readonly',False)]}),
213         'name': fields.char('Asset', size=64, required=True, readonly=True, states={'draft':[('readonly',False)]}),
214         'code': fields.char('Reference', size=32, readonly=True, states={'draft':[('readonly',False)]}),
215         'purchase_value': fields.float('Gross value ', required=True, readonly=True, states={'draft':[('readonly',False)]}),
216         'currency_id': fields.many2one('res.currency','Currency',required=True, readonly=True, states={'draft':[('readonly',False)]}),
217         'company_id': fields.many2one('res.company', 'Company', required=True, readonly=True, states={'draft':[('readonly',False)]}),
218         'note': fields.text('Note'),
219         'category_id': fields.many2one('account.asset.category', 'Asset category', required=True, change_default=True, readonly=True, states={'draft':[('readonly',False)]}),
220         'parent_id': fields.many2one('account.asset.asset', 'Parent Asset', readonly=True, states={'draft':[('readonly',False)]}),
221         'child_ids': fields.one2many('account.asset.asset', 'parent_id', 'Children Assets'),
222         'purchase_date': fields.date('Purchase Date', required=True, readonly=True, states={'draft':[('readonly',False)]}),
223         'state': fields.selection([('draft','Draft'),('open','Running'),('close','Close')], 'State', required=True,
224                                   help="When an asset is created, the state is 'Draft'.\n" \
225                                        "If the asset is confirmed, the state goes in 'Running' and the depreciation lines can be posted in the accounting.\n" \
226                                        "You can manually close an asset when the depreciation is over. If the last line of depreciation is posted, the asset automatically goes in that state."),
227         'active': fields.boolean('Active'),
228         'partner_id': fields.many2one('res.partner', 'Partner', readonly=True, states={'draft':[('readonly',False)]}),
229         'method': fields.selection([('linear','Linear'),('degressive','Degressive')], 'Computation Method', required=True, readonly=True, states={'draft':[('readonly',False)]}, help="Choose the method to use to compute the amount of depreciation lines.\n"\
230             "  * Linear: Calculated on basis of: Gross Value / Number of Depreciations\n" \
231             "  * Degressive: Calculated on basis of: Remaining Value * Degressive Factor"),
232         'method_number': fields.integer('Number of Depreciations', readonly=True, states={'draft':[('readonly',False)]}, help="Calculates Depreciation within specified interval"),
233         'method_period': fields.integer('Period Length', required=True, readonly=True, states={'draft':[('readonly',False)]}, help="State here the time during 2 depreciations, in months"),
234         'method_end': fields.date('Ending Date', readonly=True, states={'draft':[('readonly',False)]}),
235         'method_progress_factor': fields.float('Degressive Factor', readonly=True, states={'draft':[('readonly',False)]}),
236         'value_residual': fields.function(_amount_residual, method=True, digits_compute=dp.get_precision('Account'), string='Residual Value'),
237         'method_time': fields.selection([('number','Number of Depreciations'),('end','Ending Date')], 'Time Method', required=True, readonly=True, states={'draft':[('readonly',False)]},
238                                   help="Choose the method to use to compute the dates and number of depreciation lines.\n"\
239                                        "  * Number of Depreciations: Fix the number of depreciation lines and the time between 2 depreciations.\n" \
240                                        "  * Ending Date: Choose the time between 2 depreciations and the date the depreciations won't go beyond."),
241         'prorata':fields.boolean('Prorata Temporis', readonly=True, states={'draft':[('readonly',False)]}, help='Indicates that the first depreciation entry for this asset have to be done from the purchase date instead of the first January'),
242         'history_ids': fields.one2many('account.asset.history', 'asset_id', 'History', readonly=True),
243         'depreciation_line_ids': fields.one2many('account.asset.depreciation.line', 'asset_id', 'Depreciation Lines', readonly=True, states={'draft':[('readonly',False)],'open':[('readonly',False)]}),
244         'salvage_value': fields.float('Salvage Value', digits_compute=dp.get_precision('Account'), help="It is the amount you plan to have that you cannot depreciate.", readonly=True, states={'draft':[('readonly',False)]}),
245     }
246     _defaults = {
247         'code': lambda obj, cr, uid, context: obj.pool.get('ir.sequence').get(cr, uid, 'account.asset.code'),
248         'purchase_date': lambda obj, cr, uid, context: time.strftime('%Y-%m-%d'),
249         'active': True,
250         'state': 'draft',
251         'method': 'linear',
252         'method_number': 5,
253         'method_time': 'number',
254         'method_period': 12,
255         'method_progress_factor': 0.3,
256         'currency_id': lambda self,cr,uid,c: self.pool.get('res.users').browse(cr, uid, uid, c).company_id.currency_id.id,
257         'company_id': lambda self, cr, uid, context: self.pool.get('res.company')._company_default_get(cr, uid, 'account.asset.asset',context=context),
258     }
259
260     def _check_recursion(self, cr, uid, ids, context=None, parent=None):
261         return super(account_asset_asset, self)._check_recursion(cr, uid, ids, context=context, parent=parent)
262
263     def _check_prorata(self, cr, uid, ids, context=None):
264         for asset in self.browse(cr, uid, ids, context=context):
265             if asset.prorata and asset.method_time != 'number':
266                 return False
267         return True
268
269     _constraints = [
270         (_check_recursion, 'Error ! You can not create recursive assets.', ['parent_id']),
271         (_check_prorata, 'Prorata temporis can be applied only for time method "number of depreciations".', ['prorata']),
272     ]
273
274     def onchange_category_id(self, cr, uid, ids, category_id, context=None):
275         res = {'value':{}}
276         asset_categ_obj = self.pool.get('account.asset.category')
277         if category_id:
278             category_obj = asset_categ_obj.browse(cr, uid, category_id, context=context)
279             res['value'] = {
280                             'method': category_obj.method,
281                             'method_number': category_obj.method_number,
282                             'method_time': category_obj.method_time,
283                             'method_period': category_obj.method_period,
284                             'method_progress_factor': category_obj.method_progress_factor,
285                             'method_end': category_obj.method_end,
286                             'prorata': category_obj.prorata,
287             }
288         return res
289
290     def onchange_method_time(self, cr, uid, ids, method_time='number', context=None):
291         res = {'value': {}}
292         if method_time != 'number':
293             res['value'] = {'prorata': False}
294         return res
295
296     def copy(self, cr, uid, id, default=None, context=None):
297         if default is None:
298             default = {}
299         if context is None:
300             context = {}
301         default.update({'depreciation_line_ids': [], 'state': 'draft'})
302         return super(account_asset_asset, self).copy(cr, uid, id, default, context=context)
303
304     def _compute_entries(self, cr, uid, ids, period_id, context={}):
305         result = []
306         period_obj = self.pool.get('account.period')
307         depreciation_obj = self.pool.get('account.asset.depreciation.line')
308         period = period_obj.browse(cr, uid, period_id, context=context)
309         depreciation_ids = depreciation_obj.search(cr, uid, [('asset_id', 'in', ids), ('depreciation_date', '<', period.date_stop), ('depreciation_date', '>', period.date_start), ('move_check', '=', False)], context=context)
310         return depreciation_obj.create_move(cr, uid, depreciation_ids, context=context)
311
312     def create(self, cr, uid, vals, context=None):
313         asset_id = super(account_asset_asset, self).create(cr, uid, vals, context=context)
314         self.compute_depreciation_board(cr, uid, [asset_id], context=context)
315         return asset_id
316
317 account_asset_asset()
318
319 class account_asset_depreciation_line(osv.osv):
320     _name = 'account.asset.depreciation.line'
321     _description = 'Asset depreciation line'
322
323     def _get_move_check(self, cr, uid, ids, name, args, context=None):
324         res = {}
325         for line in self.browse(cr, uid, ids, context=context):
326             res[line.id] = bool(line.move_id)
327         return res
328
329     _columns = {
330         'name': fields.char('Depreciation Name', size=64, required=True, select=1),
331         'sequence': fields.integer('Sequence of the depreciation', required=True),
332         'asset_id': fields.many2one('account.asset.asset', 'Asset', required=True),
333         'parent_state': fields.related('asset_id', 'state', type='char', string='State of Asset'),
334         'amount': fields.float('Depreciation Amount', required=True),
335         'remaining_value': fields.float('Amount to Depreciate', required=True),
336         'depreciated_value': fields.float('Amount Already Depreciated', required=True),
337         'depreciation_date': fields.char('Depreciation Date', size=64, select=1),
338         'move_id': fields.many2one('account.move', 'Depreciation Entry'),
339         'move_check': fields.function(_get_move_check, method=True, type='boolean', string='Posted', store=True)
340     }
341
342     def create_move(self, cr, uid, ids, context=None):
343         can_close = False
344         if context is None:
345             context = {}
346         asset_obj = self.pool.get('account.asset.asset')
347         period_obj = self.pool.get('account.period')
348         move_obj = self.pool.get('account.move')
349         move_line_obj = self.pool.get('account.move.line')
350         currency_obj = self.pool.get('res.currency')
351         created_move_ids = []
352         for line in self.browse(cr, uid, ids, context=context):
353             if currency_obj.is_zero(cr, uid, line.asset_id.currency_id, line.remaining_value):
354                 can_close = True
355             depreciation_date = line.asset_id.prorata and line.asset_id.purchase_date or time.strftime('%Y-%m-%d')
356             period_ids = period_obj.find(cr, uid, depreciation_date, context=context)
357             company_currency = line.asset_id.company_id.currency_id.id
358             current_currency = line.asset_id.currency_id.id
359             context.update({'date': depreciation_date})
360             amount = currency_obj.compute(cr, uid, current_currency, company_currency, line.amount, context=context)
361             sign = line.asset_id.category_id.journal_id.type = 'purchase' and 1 or -1
362             asset_name = line.asset_id.name
363             reference = line.name
364             move_vals = {
365                 'name': asset_name,
366                 'date': depreciation_date,
367                 'ref': reference,
368                 'period_id': period_ids and period_ids[0] or False,
369                 'journal_id': line.asset_id.category_id.journal_id.id,
370                 }
371             move_id = move_obj.create(cr, uid, move_vals, context=context)
372             journal_id = line.asset_id.category_id.journal_id.id
373             partner_id = line.asset_id.partner_id.id
374             move_line_obj.create(cr, uid, {
375                 'name': asset_name,
376                 'ref': reference,
377                 'move_id': move_id,
378                 'account_id': line.asset_id.category_id.account_depreciation_id.id,
379                 'debit': 0.0,
380                 'credit': amount,
381                 'period_id': period_ids and period_ids[0] or False,
382                 'journal_id': journal_id,
383                 'partner_id': partner_id,
384                 'currency_id': company_currency <> current_currency and  current_currency or False,
385                 'amount_currency': company_currency <> current_currency and - sign * line.amount or 0.0,
386                 'date': depreciation_date,
387             })
388             move_line_obj.create(cr, uid, {
389                 'name': asset_name,
390                 'ref': reference,
391                 'move_id': move_id,
392                 'account_id': line.asset_id.category_id.account_expense_depreciation_id.id,
393                 'credit': 0.0,
394                 'debit': amount,
395                 'period_id': period_ids and period_ids[0] or False,
396                 'journal_id': journal_id,
397                 'partner_id': partner_id,
398                 'currency_id': company_currency <> current_currency and  current_currency or False,
399                 'amount_currency': company_currency <> current_currency and sign * line.amount or 0.0,
400                 'analytic_account_id': line.asset_id.category_id.account_analytic_id.id,
401                 'date': depreciation_date,
402                 'asset_id': line.asset_id.id
403             })
404             self.write(cr, uid, line.id, {'move_id': move_id}, context=context)
405             created_move_ids.append(move_id)
406             if can_close:
407                 asset_obj.write(cr, uid, [line.asset_id.id], {'state': 'close'}, context=context)
408         return created_move_ids
409
410 account_asset_depreciation_line()
411
412 class account_move_line(osv.osv):
413     _inherit = 'account.move.line'
414     _columns = {
415         'asset_id': fields.many2one('account.asset.asset', 'Asset'),
416         'entry_ids': fields.one2many('account.move.line', 'asset_id', 'Entries', readonly=True, states={'draft':[('readonly',False)]}),
417
418     }
419 account_move_line()
420
421 class account_asset_history(osv.osv):
422     _name = 'account.asset.history'
423     _description = 'Asset history'
424     _columns = {
425         'name': fields.char('History name', size=64, select=1),
426         'user_id': fields.many2one('res.users', 'User', required=True),
427         'date': fields.date('Date', required=True),
428         'asset_id': fields.many2one('account.asset.asset', 'Asset', required=True),
429         'method_time': fields.selection([('number','Number of Depreciations'),('end','Ending Date')], 'Time Method', required=True,
430                                   help="The method to use to compute the dates and number of depreciation lines.\n"\
431                                        "Number of Depreciations: Fix the number of depreciation lines and the time between 2 depreciations.\n" \
432                                        "Ending Date: Choose the time between 2 depreciations and the date the depreciations won't go beyond."),
433         'method_number': fields.integer('Number of Depreciations'),
434         'method_period': fields.integer('Period Length', help="Time in month between two depreciations"),
435         'method_end': fields.date('Ending date'),
436         'note': fields.text('Note'),
437     }
438     _order = 'date desc'
439     _defaults = {
440         'date': lambda *args: time.strftime('%Y-%m-%d'),
441         'user_id': lambda self, cr, uid, ctx: uid
442     }
443
444 account_asset_history()
445
446 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: