366 lines
18 KiB
Python
366 lines
18 KiB
Python
# -*- coding: utf-8 -*-
|
|
##############################################################################
|
|
#
|
|
# OpenERP, Open Source Management Solution
|
|
# Copyright (C) 2004-2010 Tiny SPRL (<http://tiny.be>).
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU Affero General Public License as
|
|
# published by the Free Software Foundation, either version 3 of the
|
|
# License, or (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
#
|
|
##############################################################################
|
|
|
|
import time
|
|
from datetime import datetime
|
|
|
|
from openerp.osv import fields, osv
|
|
from openerp import tools
|
|
from openerp.tools.translate import _
|
|
import openerp.addons.decimal_precision as dp
|
|
|
|
class account_analytic_account(osv.osv):
|
|
_name = 'account.analytic.account'
|
|
_inherit = ['mail.thread']
|
|
_description = 'Analytic Account'
|
|
_track = {
|
|
'state': {
|
|
'analytic.mt_account_pending': lambda self, cr, uid, obj, ctx=None: obj.state == 'pending',
|
|
'analytic.mt_account_closed': lambda self, cr, uid, obj, ctx=None: obj.state == 'close',
|
|
'analytic.mt_account_opened': lambda self, cr, uid, obj, ctx=None: obj.state == 'open',
|
|
},
|
|
}
|
|
|
|
def _compute_level_tree(self, cr, uid, ids, child_ids, res, field_names, context=None):
|
|
currency_obj = self.pool.get('res.currency')
|
|
recres = {}
|
|
def recursive_computation(account):
|
|
result2 = res[account.id].copy()
|
|
for son in account.child_ids:
|
|
result = recursive_computation(son)
|
|
for field in field_names:
|
|
if (account.currency_id.id != son.currency_id.id) and (field!='quantity'):
|
|
result[field] = currency_obj.compute(cr, uid, son.currency_id.id, account.currency_id.id, result[field], context=context)
|
|
result2[field] += result[field]
|
|
return result2
|
|
for account in self.browse(cr, uid, ids, context=context):
|
|
if account.id not in child_ids:
|
|
continue
|
|
recres[account.id] = recursive_computation(account)
|
|
return recres
|
|
|
|
def _debit_credit_bal_qtty(self, cr, uid, ids, fields, arg, context=None):
|
|
res = {}
|
|
if context is None:
|
|
context = {}
|
|
child_ids = tuple(self.search(cr, uid, [('parent_id', 'child_of', ids)]))
|
|
for i in child_ids:
|
|
res[i] = {}
|
|
for n in fields:
|
|
res[i][n] = 0.0
|
|
|
|
if not child_ids:
|
|
return res
|
|
|
|
where_date = ''
|
|
where_clause_args = [tuple(child_ids)]
|
|
if context.get('from_date', False):
|
|
where_date += " AND l.date >= %s"
|
|
where_clause_args += [context['from_date']]
|
|
if context.get('to_date', False):
|
|
where_date += " AND l.date <= %s"
|
|
where_clause_args += [context['to_date']]
|
|
cr.execute("""
|
|
SELECT a.id,
|
|
sum(
|
|
CASE WHEN l.amount > 0
|
|
THEN l.amount
|
|
ELSE 0.0
|
|
END
|
|
) as debit,
|
|
sum(
|
|
CASE WHEN l.amount < 0
|
|
THEN -l.amount
|
|
ELSE 0.0
|
|
END
|
|
) as credit,
|
|
COALESCE(SUM(l.amount),0) AS balance,
|
|
COALESCE(SUM(l.unit_amount),0) AS quantity
|
|
FROM account_analytic_account a
|
|
LEFT JOIN account_analytic_line l ON (a.id = l.account_id)
|
|
WHERE a.id IN %s
|
|
""" + where_date + """
|
|
GROUP BY a.id""", where_clause_args)
|
|
for row in cr.dictfetchall():
|
|
res[row['id']] = {}
|
|
for field in fields:
|
|
res[row['id']][field] = row[field]
|
|
return self._compute_level_tree(cr, uid, ids, child_ids, res, fields, context)
|
|
|
|
def name_get(self, cr, uid, ids, context=None):
|
|
res = []
|
|
if not ids:
|
|
return res
|
|
if isinstance(ids, (int, long)):
|
|
ids = [ids]
|
|
for id in ids:
|
|
elmt = self.browse(cr, uid, id, context=context)
|
|
res.append((id, self._get_one_full_name(elmt)))
|
|
return res
|
|
|
|
def _get_full_name(self, cr, uid, ids, name=None, args=None, context=None):
|
|
if context == None:
|
|
context = {}
|
|
res = {}
|
|
for elmt in self.browse(cr, uid, ids, context=context):
|
|
res[elmt.id] = self._get_one_full_name(elmt)
|
|
return res
|
|
|
|
def _get_one_full_name(self, elmt, level=6):
|
|
if level<=0:
|
|
return '...'
|
|
if elmt.parent_id and not elmt.type == 'template':
|
|
parent_path = self._get_one_full_name(elmt.parent_id, level-1) + " / "
|
|
else:
|
|
parent_path = ''
|
|
return parent_path + elmt.name
|
|
|
|
def _child_compute(self, cr, uid, ids, name, arg, context=None):
|
|
result = {}
|
|
if context is None:
|
|
context = {}
|
|
|
|
for account in self.browse(cr, uid, ids, context=context):
|
|
result[account.id] = map(lambda x: x.id, [child for child in account.child_ids if child.state != 'template'])
|
|
|
|
return result
|
|
|
|
def _get_analytic_account(self, cr, uid, ids, context=None):
|
|
company_obj = self.pool.get('res.company')
|
|
analytic_obj = self.pool.get('account.analytic.account')
|
|
accounts = []
|
|
for company in company_obj.browse(cr, uid, ids, context=context):
|
|
accounts += analytic_obj.search(cr, uid, [('company_id', '=', company.id)])
|
|
return accounts
|
|
|
|
def _set_company_currency(self, cr, uid, ids, name, value, arg, context=None):
|
|
if isinstance(ids, (int, long)):
|
|
ids=[ids]
|
|
for account in self.browse(cr, uid, ids, context=context):
|
|
if account.company_id:
|
|
if account.company_id.currency_id.id != value:
|
|
raise osv.except_osv(_('Error!'), _("If you set a company, the currency selected has to be the same as it's currency. \nYou can remove the company belonging, and thus change the currency, only on analytic account of type 'view'. This can be really useful for consolidation purposes of several companies charts with different currencies, for example."))
|
|
if value:
|
|
cr.execute("""update account_analytic_account set currency_id=%s where id=%s""", (value, account.id))
|
|
self.invalidate_cache(cr, uid, ['currency_id'], [account.id], context=context)
|
|
|
|
def _currency(self, cr, uid, ids, field_name, arg, context=None):
|
|
result = {}
|
|
for rec in self.browse(cr, uid, ids, context=context):
|
|
if rec.company_id:
|
|
result[rec.id] = rec.company_id.currency_id.id
|
|
else:
|
|
result[rec.id] = rec.currency_id.id
|
|
return result
|
|
|
|
_columns = {
|
|
'name': fields.char('Account/Contract Name', required=True, track_visibility='onchange'),
|
|
'complete_name': fields.function(_get_full_name, type='char', string='Full Name'),
|
|
'code': fields.char('Reference', select=True, track_visibility='onchange', copy=False),
|
|
'type': fields.selection([('view','Analytic View'), ('normal','Analytic Account'),('contract','Contract or Project'),('template','Template of Contract')], 'Type of Account', required=True,
|
|
help="If you select the View Type, it means you won\'t allow to create journal entries using that account.\n"\
|
|
"The type 'Analytic account' stands for usual accounts that you only want to use in accounting.\n"\
|
|
"If you select Contract or Project, it offers you the possibility to manage the validity and the invoicing options for this account.\n"\
|
|
"The special type 'Template of Contract' allows you to define a template with default data that you can reuse easily."),
|
|
'template_id': fields.many2one('account.analytic.account', 'Template of Contract'),
|
|
'description': fields.text('Description'),
|
|
'parent_id': fields.many2one('account.analytic.account', 'Parent Analytic Account', select=2),
|
|
'child_ids': fields.one2many('account.analytic.account', 'parent_id', 'Child Accounts', copy=True),
|
|
'child_complete_ids': fields.function(_child_compute, relation='account.analytic.account', string="Account Hierarchy", type='many2many'),
|
|
'line_ids': fields.one2many('account.analytic.line', 'account_id', 'Analytic Entries', copy=False),
|
|
'balance': fields.function(_debit_credit_bal_qtty, type='float', string='Balance', multi='debit_credit_bal_qtty', digits_compute=dp.get_precision('Account')),
|
|
'debit': fields.function(_debit_credit_bal_qtty, type='float', string='Debit', multi='debit_credit_bal_qtty', digits_compute=dp.get_precision('Account')),
|
|
'credit': fields.function(_debit_credit_bal_qtty, type='float', string='Credit', multi='debit_credit_bal_qtty', digits_compute=dp.get_precision('Account')),
|
|
'quantity': fields.function(_debit_credit_bal_qtty, type='float', string='Quantity', multi='debit_credit_bal_qtty'),
|
|
'quantity_max': fields.float('Prepaid Service Units', help='Sets the higher limit of time to work on the contract, based on the timesheet. (for instance, number of hours in a limited support contract.)'),
|
|
'partner_id': fields.many2one('res.partner', 'Customer'),
|
|
'user_id': fields.many2one('res.users', 'Project Manager', track_visibility='onchange'),
|
|
'manager_id': fields.many2one('res.users', 'Account Manager', track_visibility='onchange'),
|
|
'date_start': fields.date('Start Date'),
|
|
'date': fields.date('Expiration Date', select=True, track_visibility='onchange'),
|
|
'company_id': fields.many2one('res.company', 'Company', required=False), #not required because we want to allow different companies to use the same chart of account, except for leaf accounts.
|
|
'state': fields.selection([('template', 'Template'),
|
|
('draft','New'),
|
|
('open','In Progress'),
|
|
('pending','To Renew'),
|
|
('close','Closed'),
|
|
('cancelled', 'Cancelled')],
|
|
'Status', required=True,
|
|
track_visibility='onchange', copy=False),
|
|
'currency_id': fields.function(_currency, fnct_inv=_set_company_currency, #the currency_id field is readonly except if it's a view account and if there is no company
|
|
store = {
|
|
'res.company': (_get_analytic_account, ['currency_id'], 10),
|
|
}, string='Currency', type='many2one', relation='res.currency'),
|
|
}
|
|
|
|
def on_change_template(self, cr, uid, ids, template_id, date_start=False, context=None):
|
|
if not template_id:
|
|
return {}
|
|
res = {'value':{}}
|
|
template = self.browse(cr, uid, template_id, context=context)
|
|
if template.date_start and template.date:
|
|
from_dt = datetime.strptime(template.date_start, tools.DEFAULT_SERVER_DATE_FORMAT)
|
|
to_dt = datetime.strptime(template.date, tools.DEFAULT_SERVER_DATE_FORMAT)
|
|
timedelta = to_dt - from_dt
|
|
res['value']['date'] = datetime.strftime(datetime.now() + timedelta, tools.DEFAULT_SERVER_DATE_FORMAT)
|
|
if not date_start:
|
|
res['value']['date_start'] = fields.date.today()
|
|
res['value']['quantity_max'] = template.quantity_max
|
|
res['value']['parent_id'] = template.parent_id and template.parent_id.id or False
|
|
res['value']['description'] = template.description
|
|
return res
|
|
|
|
def on_change_partner_id(self, cr, uid, ids,partner_id, name, context=None):
|
|
res={}
|
|
if partner_id:
|
|
partner = self.pool.get('res.partner').browse(cr, uid, partner_id, context=context)
|
|
if partner.user_id:
|
|
res['manager_id'] = partner.user_id.id
|
|
if not name:
|
|
res['name'] = _('Contract: ') + partner.name
|
|
return {'value': res}
|
|
|
|
def _default_company(self, cr, uid, context=None):
|
|
user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
|
|
if user.company_id:
|
|
return user.company_id.id
|
|
return self.pool.get('res.company').search(cr, uid, [('parent_id', '=', False)])[0]
|
|
|
|
def _get_default_currency(self, cr, uid, context=None):
|
|
user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
|
|
return user.company_id.currency_id.id
|
|
|
|
_defaults = {
|
|
'type': 'normal',
|
|
'company_id': _default_company,
|
|
'code' : lambda obj, cr, uid, context: obj.pool.get('ir.sequence').get(cr, uid, 'account.analytic.account', context=context),
|
|
'state': 'open',
|
|
'user_id': lambda self, cr, uid, ctx: uid,
|
|
'partner_id': lambda self, cr, uid, ctx: ctx.get('partner_id', False),
|
|
'manager_id': lambda self, cr, uid, ctx: ctx.get('manager_id', False),
|
|
'date_start': lambda *a: time.strftime('%Y-%m-%d'),
|
|
'currency_id': _get_default_currency,
|
|
}
|
|
|
|
def check_recursion(self, cr, uid, ids, context=None, parent=None):
|
|
return super(account_analytic_account, self)._check_recursion(cr, uid, ids, context=context, parent=parent)
|
|
|
|
_order = 'code, name asc'
|
|
_constraints = [
|
|
(check_recursion, 'Error! You cannot create recursive analytic accounts.', ['parent_id']),
|
|
]
|
|
|
|
def name_create(self, cr, uid, name, context=None):
|
|
raise osv.except_osv(_('Warning'), _("Quick account creation disallowed."))
|
|
|
|
def copy(self, cr, uid, id, default=None, context=None):
|
|
""" executed only on the toplevel copied object of the hierarchy.
|
|
Subobject are actually copied with copy_data"""
|
|
if not default:
|
|
default = {}
|
|
analytic = self.browse(cr, uid, id, context=context)
|
|
default['name'] = _("%s (copy)") % analytic['name']
|
|
return super(account_analytic_account, self).copy(cr, uid, id, default, context=context)
|
|
|
|
def on_change_company(self, cr, uid, id, company_id):
|
|
if not company_id:
|
|
return {}
|
|
currency = self.pool.get('res.company').read(cr, uid, [company_id], ['currency_id'])[0]['currency_id']
|
|
return {'value': {'currency_id': currency}}
|
|
|
|
def on_change_parent(self, cr, uid, id, parent_id):
|
|
if not parent_id:
|
|
return {}
|
|
parent = self.read(cr, uid, [parent_id], ['partner_id','code'])[0]
|
|
if parent['partner_id']:
|
|
partner = parent['partner_id'][0]
|
|
else:
|
|
partner = False
|
|
res = {'value': {}}
|
|
if partner:
|
|
res['value']['partner_id'] = partner
|
|
return res
|
|
|
|
def name_search(self, cr, uid, name, args=None, operator='ilike', context=None, limit=100):
|
|
if not args:
|
|
args=[]
|
|
if context is None:
|
|
context={}
|
|
account_ids = []
|
|
if name:
|
|
account_ids = self.search(cr, uid, [('code', '=', name)] + args, limit=limit, context=context)
|
|
if not account_ids:
|
|
dom = []
|
|
if '/' in name:
|
|
for name2 in name.split('/'):
|
|
# intermediate search without limit and args - could be expensive for large tables if `name` is not selective
|
|
account_ids = self.search(cr, uid, dom + [('name', operator, name2.strip())], limit=None, context=context)
|
|
if not account_ids: break
|
|
dom = [('parent_id','in',account_ids)]
|
|
if account_ids and args:
|
|
# final filtering according to domain (args)4
|
|
account_ids = self.search(cr, uid, [('id', 'in', account_ids)] + args, limit=limit, context=context)
|
|
if not account_ids:
|
|
return super(account_analytic_account, self).name_search(cr, uid, name, args, operator=operator, context=context, limit=limit)
|
|
return self.name_get(cr, uid, account_ids, context=context)
|
|
|
|
class account_analytic_line(osv.osv):
|
|
_name = 'account.analytic.line'
|
|
_description = 'Analytic Line'
|
|
|
|
_columns = {
|
|
'name': fields.char('Description', required=True),
|
|
'date': fields.date('Date', required=True, select=True),
|
|
'amount': fields.float('Amount', required=True, help='Calculated by multiplying the quantity and the price given in the Product\'s cost price. Always expressed in the company main currency.', digits_compute=dp.get_precision('Account')),
|
|
'unit_amount': fields.float('Quantity', help='Specifies the amount of quantity to count.'),
|
|
'account_id': fields.many2one('account.analytic.account', 'Analytic Account', required=True, ondelete='restrict', select=True, domain=[('type','<>','view')]),
|
|
'user_id': fields.many2one('res.users', 'User'),
|
|
'company_id': fields.related('account_id', 'company_id', type='many2one', relation='res.company', string='Company', store=True, readonly=True),
|
|
|
|
}
|
|
|
|
def _get_default_date(self, cr, uid, context=None):
|
|
return fields.date.context_today(self, cr, uid, context=context)
|
|
|
|
def __get_default_date(self, cr, uid, context=None):
|
|
return self._get_default_date(cr, uid, context=context)
|
|
|
|
_defaults = {
|
|
'date': __get_default_date,
|
|
'company_id': lambda self,cr,uid,c: self.pool.get('res.company')._company_default_get(cr, uid, 'account.analytic.line', context=c),
|
|
'amount': 0.00
|
|
}
|
|
|
|
_order = 'date desc'
|
|
|
|
def _check_no_view(self, cr, uid, ids, context=None):
|
|
analytic_lines = self.browse(cr, uid, ids, context=context)
|
|
for line in analytic_lines:
|
|
if line.account_id.type == 'view':
|
|
return False
|
|
return True
|
|
|
|
_constraints = [
|
|
(_check_no_view, 'You cannot create analytic line on view account.', ['account_id']),
|
|
]
|
|
|
|
# vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4:
|