263 lines
12 KiB
Python
263 lines
12 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
|
||
|
import operator
|
||
|
|
||
|
from osv import fields
|
||
|
from osv import osv
|
||
|
|
||
|
#
|
||
|
# Object definition
|
||
|
#
|
||
|
|
||
|
class account_analytic_account(osv.osv):
|
||
|
_name = 'account.analytic.account'
|
||
|
_description = 'Analytic Accounts'
|
||
|
|
||
|
def _credit_calc(self, cr, uid, ids, name, arg, context={}):
|
||
|
|
||
|
where_date = ''
|
||
|
if context.get('from_date',False):
|
||
|
where_date += " AND l.date >= '" + context['from_date'] + "'"
|
||
|
if context.get('to_date',False):
|
||
|
where_date += " AND l.date <= '" + context['to_date'] + "'"
|
||
|
|
||
|
cr.execute("SELECT a.id, COALESCE(SUM(l.amount),0) FROM account_analytic_account a LEFT JOIN account_analytic_line l ON (a.id=l.account_id "+where_date+") WHERE l.amount<0 and a.id =ANY(%s) GROUP BY a.id",(ids,))
|
||
|
r = dict(cr.fetchall())
|
||
|
for i in ids:
|
||
|
r.setdefault(i,0.0)
|
||
|
return r
|
||
|
|
||
|
def _debit_calc(self, cr, uid, ids, name, arg, context={}):
|
||
|
|
||
|
where_date = ''
|
||
|
if context.get('from_date',False):
|
||
|
where_date += " AND l.date >= '" + context['from_date'] + "'"
|
||
|
if context.get('to_date',False):
|
||
|
where_date += " AND l.date <= '" + context['to_date'] + "'"
|
||
|
|
||
|
cr.execute("SELECT a.id, COALESCE(SUM(l.amount),0) FROM account_analytic_account a LEFT JOIN account_analytic_line l ON (a.id=l.account_id "+where_date+") WHERE l.amount>0 and a.id =ANY(%s) GROUP BY a.id" ,(ids,))
|
||
|
r= dict(cr.fetchall())
|
||
|
for i in ids:
|
||
|
r.setdefault(i,0.0)
|
||
|
return r
|
||
|
|
||
|
def _balance_calc(self, cr, uid, ids, name, arg, context={}):
|
||
|
res = {}
|
||
|
ids2 = self.search(cr, uid, [('parent_id', 'child_of', ids)])
|
||
|
for i in ids:
|
||
|
res.setdefault(i,0.0)
|
||
|
if not ids2:
|
||
|
return res
|
||
|
|
||
|
where_date = ''
|
||
|
if context.get('from_date',False):
|
||
|
where_date += " AND l.date >= '" + context['from_date'] + "'"
|
||
|
if context.get('to_date',False):
|
||
|
where_date += " AND l.date <= '" + context['to_date'] + "'"
|
||
|
|
||
|
cr.execute("SELECT a.id, COALESCE(SUM(l.amount),0) FROM account_analytic_account a LEFT JOIN account_analytic_line l ON (a.id=l.account_id "+where_date+") WHERE a.id =ANY(%s) GROUP BY a.id",(ids2,))
|
||
|
|
||
|
for account_id, sum in cr.fetchall():
|
||
|
res[account_id] = sum
|
||
|
|
||
|
cr.execute("SELECT a.id, r.currency_id FROM account_analytic_account a INNER JOIN res_company r ON (a.company_id = r.id) where a.id =ANY(%s)",(ids2,))
|
||
|
|
||
|
currency= dict(cr.fetchall())
|
||
|
|
||
|
res_currency= self.pool.get('res.currency')
|
||
|
for id in ids:
|
||
|
if id not in ids2:
|
||
|
continue
|
||
|
for child in self.search(cr, uid, [('parent_id', 'child_of', [id])]):
|
||
|
if child <> id:
|
||
|
res.setdefault(id, 0.0)
|
||
|
if currency[child]<>currency[id]:
|
||
|
res[id] += res_currency.compute(cr, uid, currency[child], currency[id], res.get(child, 0.0), context=context)
|
||
|
else:
|
||
|
res[id] += res.get(child, 0.0)
|
||
|
|
||
|
cur_obj = res_currency.browse(cr,uid,currency.values(),context)
|
||
|
cur_obj = dict([(o.id, o) for o in cur_obj])
|
||
|
for id in ids:
|
||
|
if id in ids2:
|
||
|
res[id] = res_currency.round(cr,uid,cur_obj[currency[id]],res.get(id,0.0))
|
||
|
|
||
|
return dict([(i, res[i]) for i in ids ])
|
||
|
|
||
|
def _quantity_calc(self, cr, uid, ids, name, arg, context={}):
|
||
|
#XXX must convert into one uom
|
||
|
res = {}
|
||
|
ids2 = self.search(cr, uid, [('parent_id', 'child_of', ids)])
|
||
|
acc_set = ",".join(map(str, ids2))
|
||
|
|
||
|
for i in ids:
|
||
|
res.setdefault(i,0.0)
|
||
|
|
||
|
if not acc_set:
|
||
|
return res
|
||
|
|
||
|
where_date = ''
|
||
|
if context.get('from_date',False):
|
||
|
where_date += " AND l.date >= '" + context['from_date'] + "'"
|
||
|
if context.get('to_date',False):
|
||
|
where_date += " AND l.date <= '" + context['to_date'] + "'"
|
||
|
|
||
|
cr.execute('SELECT a.id, COALESCE(SUM(l.unit_amount), 0) \
|
||
|
FROM account_analytic_account a \
|
||
|
LEFT JOIN account_analytic_line l ON (a.id = l.account_id ' + where_date + ') \
|
||
|
WHERE a.id =ANY(%s) GROUP BY a.id',(ids2,))
|
||
|
|
||
|
for account_id, sum in cr.fetchall():
|
||
|
res[account_id] = sum
|
||
|
|
||
|
for id in ids:
|
||
|
if id not in ids2:
|
||
|
continue
|
||
|
for child in self.search(cr, uid, [('parent_id', 'child_of', [id])]):
|
||
|
if child <> id:
|
||
|
res.setdefault(id, 0.0)
|
||
|
res[id] += res.get(child, 0.0)
|
||
|
return dict([(i, res[i]) for i in ids])
|
||
|
|
||
|
def name_get(self, cr, uid, ids, context={}):
|
||
|
if not len(ids):
|
||
|
return []
|
||
|
reads = self.read(cr, uid, ids, ['name','parent_id'], context)
|
||
|
res = []
|
||
|
for record in reads:
|
||
|
name = record['name']
|
||
|
if record['parent_id']:
|
||
|
name = record['parent_id'][1]+' / '+name
|
||
|
res.append((record['id'], name))
|
||
|
return res
|
||
|
|
||
|
def _complete_name_calc(self, cr, uid, ids, prop, unknow_none, unknow_dict):
|
||
|
res = self.name_get(cr, uid, ids)
|
||
|
return dict(res)
|
||
|
|
||
|
def _get_company_currency(self, cr, uid, ids, field_name, arg, context={}):
|
||
|
result = {}
|
||
|
for rec in self.browse(cr, uid, ids, context):
|
||
|
result[rec.id] = (rec.company_id.currency_id.id,rec.company_id.currency_id.code) or False
|
||
|
return result
|
||
|
|
||
|
_columns = {
|
||
|
'name' : fields.char('Account Name', size=128, required=True),
|
||
|
'complete_name': fields.function(_complete_name_calc, method=True, type='char', string='Full Account Name'),
|
||
|
'code' : fields.char('Account Code', size=24),
|
||
|
# 'active' : fields.boolean('Active', help="If the active field is set to true, it will allow you to hide the analytic account without removing it."),
|
||
|
'type': fields.selection([('view','View'), ('normal','Normal')], 'Account Type'),
|
||
|
'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'),
|
||
|
'line_ids': fields.one2many('account.analytic.line', 'account_id', 'Analytic Entries'),
|
||
|
'balance' : fields.function(_balance_calc, method=True, type='float', string='Balance'),
|
||
|
'debit' : fields.function(_debit_calc, method=True, type='float', string='Debit'),
|
||
|
'credit' : fields.function(_credit_calc, method=True, type='float', string='Credit'),
|
||
|
'quantity': fields.function(_quantity_calc, method=True, type='float', string='Quantity'),
|
||
|
'quantity_max': fields.float('Maximum Quantity', help='Sets the higher limit of quantity of hours.'),
|
||
|
'partner_id' : fields.many2one('res.partner', 'Associated Partner'),
|
||
|
'contact_id' : fields.many2one('res.partner.address', 'Contact'),
|
||
|
'user_id' : fields.many2one('res.users', 'Account Manager'),
|
||
|
'date_start': fields.date('Date Start'),
|
||
|
'date': fields.date('Date End'),
|
||
|
'company_id': fields.many2one('res.company', 'Company', required=True),
|
||
|
'company_currency_id': fields.function(_get_company_currency, method=True, type='many2one', relation='res.currency', string='Currency'),
|
||
|
'state': fields.selection([('draft','Draft'),('open','Open'), ('pending','Pending'),('cancelled', 'Cancelled'),('close','Close'),('template', 'Template')], 'State', required=True,readonly=True,
|
||
|
help='* When an account is created its in \'Draft\' state.\
|
||
|
\n* If any associated partner is there, it can be in \'Open\' state.\
|
||
|
\n* If any pending balance is there it can be in \'Pending\'. \
|
||
|
\n* And finally when all the transactions are over, it can be in \'Close\' state. \
|
||
|
\n* The project can be in either if the states \'Template\' and \'Running\'.\n If it is template then we can make projects based on the template projects. If its in \'Running\' state it is a normal project.\
|
||
|
\n If it is to be reviewed then the state is \'Pending\'.\n When the project is completed the state is set to \'Done\'.'),
|
||
|
}
|
||
|
|
||
|
def _default_company(self, cr, uid, context={}):
|
||
|
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]
|
||
|
_defaults = {
|
||
|
# 'active' : lambda *a : True,
|
||
|
'type' : lambda *a : 'normal',
|
||
|
'company_id': _default_company,
|
||
|
'state' : lambda *a : 'open',
|
||
|
'user_id' : lambda self,cr,uid,ctx : uid,
|
||
|
'partner_id': lambda self,cr, uid, ctx: ctx.get('partner_id', False),
|
||
|
'contact_id': lambda self,cr, uid, ctx: ctx.get('contact_id', False),
|
||
|
'date_start': lambda *a: time.strftime('%Y-%m-%d')
|
||
|
}
|
||
|
|
||
|
def check_recursion(self, cr, uid, ids, parent=None):
|
||
|
return super(account_analytic_account, self).check_recursion(cr, uid, ids, parent=parent)
|
||
|
|
||
|
_order = 'parent_id desc,code'
|
||
|
_constraints = [
|
||
|
(check_recursion, 'Error! You can not create recursive analytic accounts.', ['parent_id'])
|
||
|
]
|
||
|
|
||
|
def create(self, cr, uid, vals, context=None):
|
||
|
parent_id = vals.get('parent_id', 0)
|
||
|
if ('code' not in vals or not vals['code']) and not parent_id:
|
||
|
vals['code'] = self.pool.get('ir.sequence').get(cr, uid, 'account.analytic.account')
|
||
|
return super(account_analytic_account, self).create(cr, uid, vals, context=context)
|
||
|
|
||
|
def copy(self, cr, uid, id, default=None, context={}):
|
||
|
if not default:
|
||
|
default = {}
|
||
|
default['code'] = False
|
||
|
default['line_ids'] = []
|
||
|
return super(account_analytic_account, self).copy(cr, uid, id, default, context=context)
|
||
|
|
||
|
|
||
|
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]
|
||
|
childs = self.search(cr, uid, [('parent_id', '=', parent_id), ('active', 'in', [True, False])])
|
||
|
numchild = len(childs)
|
||
|
if parent['partner_id']:
|
||
|
partner = parent['partner_id'][0]
|
||
|
else:
|
||
|
partner = False
|
||
|
res = {'value' : {'code' : '%s - %03d' % (parent['code'] or '', numchild + 1),}}
|
||
|
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 not context:
|
||
|
context={}
|
||
|
account = self.search(cr, uid, [('code', '=', name)]+args, limit=limit, context=context)
|
||
|
if not account:
|
||
|
account = self.search(cr, uid, [('name', 'ilike', '%%%s%%' % name)]+args, limit=limit, context=context)
|
||
|
newacc = account
|
||
|
while newacc:
|
||
|
newacc = self.search(cr, uid, [('parent_id', 'in', newacc)]+args, limit=limit, context=context)
|
||
|
account+=newacc
|
||
|
return self.name_get(cr, uid, account, context=context)
|
||
|
|
||
|
account_analytic_account()
|