addons/account_analytic_plans/account_analytic_plans.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 time
  23 from lxml import etree
  24
  25 from osv import fields, osv
  26 import tools
  27 from tools.translate import _
  28
  29 class one2many_mod2(fields.one2many):
  30     def get(self, cr, obj, ids, name, user=None, offset=0, context=None, values=None):
  31         if context is None:
  32             context = {}
  33         res = {}
  34         for id in ids:
  35             res[id] = []
  36         ids2 = None
  37         if 'journal_id' in context:
  38             journal = obj.pool.get('account.journal').browse(cr, user, context['journal_id'], context=context)
  39             pnum = int(name[7]) -1
  40             plan = journal.plan_id
  41             if plan and len(plan.plan_ids) > pnum:
  42                 acc_id = plan.plan_ids[pnum].root_analytic_id.id
  43                 ids2 = obj.pool.get(self._obj).search(cr, user, [(self._fields_id,'in',ids),('analytic_account_id','child_of',[acc_id])], limit=self._limit)
  44         if ids2 is None:
  45             ids2 = obj.pool.get(self._obj).search(cr, user, [(self._fields_id,'in',ids)], limit=self._limit)
  46         for r in obj.pool.get(self._obj)._read_flat(cr, user, ids2, [self._fields_id], context=context, load='_classic_write'):
  47             res[r[self._fields_id]].append( r['id'] )
  48         return res
  49
  50 class account_analytic_line(osv.osv):
  51     _inherit = 'account.analytic.line'
  52     _description = 'Analytic Line'
  53
  54     def _get_amount(self, cr, uid, ids, name, args, context=None):
  55         res = {}
  56         for id in ids:
  57             res.setdefault(id, 0.0)
  58         for line in self.browse(cr, uid, ids, context=context):
  59             amount = line.move_id and line.move_id.amount_currency * (line.percentage / 100) or 0.0
  60             res[line.id] = amount
  61         return res
  62
  63     _columns = {
  64         'amount_currency': fields.function(_get_amount, string="Amount Currency", type="float", store=True, help="The amount expressed in the related account currency if not equal to the company one.", readonly=True),
  65         'percentage': fields.float('Percentage')
  66     }
  67
  68 account_analytic_line()
  69
  70 class account_analytic_plan(osv.osv):
  71     _name = "account.analytic.plan"
  72     _description = "Analytic Plan"
  73     _columns = {
  74         'name': fields.char('Analytic Plan', size=64, required=True, select=True),
  75         'plan_ids': fields.one2many('account.analytic.plan.line', 'plan_id', 'Analytic Plans'),
  76     }
  77
  78 account_analytic_plan()
  79
  80 class account_analytic_plan_line(osv.osv):
  81     _name = "account.analytic.plan.line"
  82     _description = "Analytic Plan Line"
  83     _order = "sequence, id"
  84     _columns = {
  85         'plan_id': fields.many2one('account.analytic.plan','Analytic Plan',required=True),
  86         'name': fields.char('Plan Name', size=64, required=True, select=True),
  87         'sequence': fields.integer('Sequence'),
  88         'root_analytic_id': fields.many2one('account.analytic.account', 'Root Account', help="Root account of this plan.", required=False),
  89         'min_required': fields.float('Minimum Allowed (%)'),
  90         'max_required': fields.float('Maximum Allowed (%)'),
  91     }
  92     _defaults = {
  93         'min_required': 100.0,
  94         'max_required': 100.0,
  95     }
  96
  97 account_analytic_plan_line()
  98
  99 class account_analytic_plan_instance(osv.osv):
 100     _name = "account.analytic.plan.instance"
 101     _description = "Analytic Plan Instance"
 102     _columns = {
 103         'name': fields.char('Analytic Distribution', size=64),
 104         'code': fields.char('Distribution Code', size=16),
 105         'journal_id': fields.many2one('account.analytic.journal', 'Analytic Journal' ),
 106         'account_ids': fields.one2many('account.analytic.plan.instance.line', 'plan_id', 'Account Id'),
 107         'account1_ids': one2many_mod2('account.analytic.plan.instance.line', 'plan_id', 'Account1 Id'),
 108         'account2_ids': one2many_mod2('account.analytic.plan.instance.line', 'plan_id', 'Account2 Id'),
 109         'account3_ids': one2many_mod2('account.analytic.plan.instance.line', 'plan_id', 'Account3 Id'),
 110         'account4_ids': one2many_mod2('account.analytic.plan.instance.line', 'plan_id', 'Account4 Id'),
 111         'account5_ids': one2many_mod2('account.analytic.plan.instance.line', 'plan_id', 'Account5 Id'),
 112         'account6_ids': one2many_mod2('account.analytic.plan.instance.line', 'plan_id', 'Account6 Id'),
 113         'plan_id': fields.many2one('account.analytic.plan', "Model's Plan"),
 114     }
 115
 116     def search(self, cr, user, args, offset=0, limit=None, order=None, context=None, count=False):
 117         if context is None:
 118             context = {}
 119         journal_obj = self.pool.get('account.journal')
 120         if context.get('journal_id', False):
 121             journal = journal_obj.browse(cr, user, [context['journal_id']], context=context)[0]
 122             analytic_journal = journal.analytic_journal_id and journal.analytic_journal_id.id or False
 123             args.append('|')
 124             args.append(('journal_id', '=', analytic_journal))
 125             args.append(('journal_id', '=', False))
 126         res = super(account_analytic_plan_instance, self).search(cr, user, args, offset=offset, limit=limit, order=order,
 127                                                                  context=context, count=count)
 128         return res
 129
 130     def copy(self, cr, uid, id, default=None, context=None):
 131         if not default:
 132             default = {}
 133         default.update({'account1_ids':False, 'account2_ids':False, 'account3_ids':False,
 134                 'account4_ids':False, 'account5_ids':False, 'account6_ids':False})
 135         return super(account_analytic_plan_instance, self).copy(cr, uid, id, default, context=context)
 136
 137     def _default_journal(self, cr, uid, context=None):
 138         if context is None:
 139             context = {}
 140         journal_obj = self.pool.get('account.journal')
 141         if context.has_key('journal_id') and context['journal_id']:
 142             journal = journal_obj.browse(cr, uid, context['journal_id'], context=context)
 143             if journal.analytic_journal_id:
 144                 return journal.analytic_journal_id.id
 145         return False
 146
 147     _defaults = {
 148         'plan_id': False,
 149         'journal_id': _default_journal,
 150     }
 151     def name_get(self, cr, uid, ids, context=None):
 152         res = []
 153         for inst in self.browse(cr, uid, ids, context=context):
 154             name = inst.name or '/'
 155             if name and inst.code:
 156                 name=name+' ('+inst.code+')'
 157             res.append((inst.id, name))
 158         return res
 159
 160     def name_search(self, cr, uid, name, args=None, operator='ilike', context=None, limit=100):
 161         args = args or []
 162         if name:
 163             ids = self.search(cr, uid, [('code', '=', name)] + args, limit=limit, context=context or {})
 164             if not ids:
 165                 ids = self.search(cr, uid, [('name', operator, name)] + args, limit=limit, context=context or {})
 166         else:
 167             ids = self.search(cr, uid, args, limit=limit, context=context or {})
 168         return self.name_get(cr, uid, ids, context or {})
 169
 170     def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
 171         if context is None:
 172             context = {}
 173         wiz_id = self.pool.get('ir.actions.act_window').search(cr, uid, [("name","=","analytic.plan.create.model.action")], context=context)
 174         res = super(account_analytic_plan_instance,self).fields_view_get(cr, uid, view_id, view_type, context, toolbar=toolbar, submenu=submenu)
 175         journal_obj = self.pool.get('account.journal')
 176         analytic_plan_obj = self.pool.get('account.analytic.plan')
 177         if (res['type']=='form'):
 178             plan_id = False
 179             if context.get('journal_id', False):
 180                 plan_id = journal_obj.browse(cr, uid, int(context['journal_id']), context=context).plan_id
 181             elif context.get('plan_id', False):
 182                 plan_id = analytic_plan_obj.browse(cr, uid, int(context['plan_id']), context=context)
 183
 184             if plan_id:
 185                 i=1
 186                 res['arch'] = """<form string="%s">
 187     <field name="name"/>
 188     <field name="code"/>
 189     <field name="journal_id"/>
 190     <button name="%d" string="Save This Distribution as a Model" type="action" colspan="2"/>
 191     """% (tools.to_xml(plan_id.name), wiz_id[0])
 192                 for line in plan_id.plan_ids:
 193                     res['arch']+="""
 194                     <field name="account%d_ids" string="%s" nolabel="1" colspan="4">
 195                     <tree string="%s" editable="bottom">
 196                         <field name="rate"/>
 197                         <field name="analytic_account_id" domain="[('parent_id','child_of',[%d])]" groups="analytic.group_analytic_accounting"/>
 198                     </tree>
 199                 </field>
 200                 <newline/>"""%(i,tools.to_xml(line.name),tools.to_xml(line.name),line.root_analytic_id and line.root_analytic_id.id or 0)
 201                     i+=1
 202                 res['arch'] += "</form>"
 203                 doc = etree.fromstring(res['arch'].encode('utf8'))
 204                 xarch, xfields = self._view_look_dom_arch(cr, uid, doc, view_id, context=context)
 205                 res['arch'] = xarch
 206                 res['fields'] = xfields
 207             return res
 208         else:
 209             return res
 210
 211     def create(self, cr, uid, vals, context=None):
 212         journal_obj = self.pool.get('account.journal')
 213         ana_plan_instance_obj = self.pool.get('account.analytic.plan.instance')
 214         acct_anal_acct = self.pool.get('account.analytic.account')
 215         acct_anal_plan_line_obj = self.pool.get('account.analytic.plan.line')
 216         if context and 'journal_id' in context:
 217             journal = journal_obj.browse(cr, uid, context['journal_id'], context=context)
 218
 219             pids = ana_plan_instance_obj.search(cr, uid, [('name','=',vals['name']), ('code','=',vals['code']), ('plan_id','<>',False)], context=context)
 220             if pids:
 221                 raise osv.except_osv(_('Error!'), _('A model with this name and code already exists.'))
 222
 223             res = acct_anal_plan_line_obj.search(cr, uid, [('plan_id','=',journal.plan_id.id)], context=context)
 224             for i in res:
 225                 total_per_plan = 0
 226                 item = acct_anal_plan_line_obj.browse(cr, uid, i, context=context)
 227                 temp_list = ['account1_ids','account2_ids','account3_ids','account4_ids','account5_ids','account6_ids']
 228                 for l in temp_list:
 229                     if vals.has_key(l):
 230                         for tempo in vals[l]:
 231                             if acct_anal_acct.search(cr, uid, [('parent_id', 'child_of', [item.root_analytic_id.id]), ('id', '=', tempo[2]['analytic_account_id'])], context=context):
 232                                 total_per_plan += tempo[2]['rate']
 233                 if total_per_plan < item.min_required or total_per_plan > item.max_required:
 234                     raise osv.except_osv(_('Error!'),_('The total should be between %s and %s.') % (str(item.min_required), str(item.max_required)))
 235
 236         return super(account_analytic_plan_instance, self).create(cr, uid, vals, context=context)
 237
 238     def write(self, cr, uid, ids, vals, context=None, check=True, update_check=True):
 239         if context is None:
 240             context = {}
 241         this = self.browse(cr, uid, ids[0], context=context)
 242         invoice_line_obj = self.pool.get('account.invoice.line')
 243         if this.plan_id and not vals.has_key('plan_id'):
 244             #this instance is a model, so we have to create a new plan instance instead of modifying it
 245             #copy the existing model
 246             temp_id = self.copy(cr, uid, this.id, None, context=context)
 247             #get the list of the invoice line that were linked to the model
 248             lists = invoice_line_obj.search(cr, uid, [('analytics_id','=',this.id)], context=context)
 249             #make them link to the copy
 250             invoice_line_obj.write(cr, uid, lists, {'analytics_id':temp_id}, context=context)
 251
 252             #and finally modify the old model to be not a model anymore
 253             vals['plan_id'] = False
 254             if not vals.has_key('name'):
 255                 vals['name'] = this.name and (str(this.name)+'*') or "*"
 256             if not vals.has_key('code'):
 257                 vals['code'] = this.code and (str(this.code)+'*') or "*"
 258         return super(account_analytic_plan_instance, self).write(cr, uid, ids, vals, context=context)
 259
 260 account_analytic_plan_instance()
 261
 262 class account_analytic_plan_instance_line(osv.osv):
 263     _name = "account.analytic.plan.instance.line"
 264     _description = "Analytic Instance Line"
 265     _rec_name = "analytic_account_id"
 266     _columns = {
 267         'plan_id': fields.many2one('account.analytic.plan.instance', 'Plan Id'),
 268         'analytic_account_id': fields.many2one('account.analytic.account', 'Analytic Account', required=True, domain=[('type','<>','view')]),
 269         'rate': fields.float('Rate (%)', required=True),
 270     }
 271     _defaults = {
 272         'rate': 100.0
 273     }
 274     def name_get(self, cr, uid, ids, context=None):
 275         if not ids:
 276             return []
 277         reads = self.read(cr, uid, ids, ['analytic_account_id'], context=context)
 278         res = []
 279         for record in reads:
 280             res.append((record['id'], record['analytic_account_id']))
 281         return res
 282
 283 account_analytic_plan_instance_line()
 284
 285 class account_journal(osv.osv):
 286     _inherit = "account.journal"
 287     _name = "account.journal"
 288     _columns = {
 289         'plan_id': fields.many2one('account.analytic.plan', 'Analytic Plans'),
 290     }
 291
 292 account_journal()
 293
 294 class account_invoice_line(osv.osv):
 295     _inherit = "account.invoice.line"
 296     _name = "account.invoice.line"
 297     _columns = {
 298         'analytics_id': fields.many2one('account.analytic.plan.instance', 'Analytic Distribution'),
 299     }
 300
 301     def create(self, cr, uid, vals, context=None):
 302         if 'analytics_id' in vals and isinstance(vals['analytics_id'], tuple):
 303             vals['analytics_id'] = vals['analytics_id'][0]
 304         return super(account_invoice_line, self).create(cr, uid, vals, context=context)
 305
 306     def move_line_get_item(self, cr, uid, line, context=None):
 307         res = super(account_invoice_line, self).move_line_get_item(cr, uid, line, context=context)
 308         res ['analytics_id'] = line.analytics_id and line.analytics_id.id or False
 309         return res
 310
 311     def product_id_change(self, cr, uid, ids, product, uom, qty=0, name='', type='out_invoice', partner_id=False, fposition_id=False, price_unit=False, currency_id=False, context=None, company_id=None):
 312         res_prod = super(account_invoice_line, self).product_id_change(cr, uid, ids, product, uom, qty, name, type, partner_id, fposition_id, price_unit, currency_id, context=context, company_id=company_id)
 313         rec = self.pool.get('account.analytic.default').account_get(cr, uid, product, partner_id, uid, time.strftime('%Y-%m-%d'), context=context)
 314         if rec and rec.analytics_id:
 315             res_prod['value'].update({'analytics_id': rec.analytics_id.id})
 316         return res_prod
 317
 318 account_invoice_line()
 319
 320 class account_move_line(osv.osv):
 321
 322     _inherit = "account.move.line"
 323     _name = "account.move.line"
 324     _columns = {
 325         'analytics_id':fields.many2one('account.analytic.plan.instance', 'Analytic Distribution'),
 326     }
 327
 328     def _default_get_move_form_hook(self, cursor, user, data):
 329         data = super(account_move_line, self)._default_get_move_form_hook(cursor, user, data)
 330         if data.has_key('analytics_id'):
 331             del(data['analytics_id'])
 332         return data
 333
 334     def create_analytic_lines(self, cr, uid, ids, context=None):
 335         if context is None:
 336             context = {}
 337         super(account_move_line, self).create_analytic_lines(cr, uid, ids, context=context)
 338         analytic_line_obj = self.pool.get('account.analytic.line')
 339         for line in self.browse(cr, uid, ids, context=context):
 340            if line.analytics_id:
 341                if not line.journal_id.analytic_journal_id:
 342                    raise osv.except_osv(_('No Analytic Journal !'),_("You have to define an analytic journal on the '%s' journal.") % (line.journal_id.name,))
 343
 344                toremove = analytic_line_obj.search(cr, uid, [('move_id','=',line.id)], context=context)
 345                if toremove:
 346                     analytic_line_obj.unlink(cr, uid, toremove, context=context)
 347                for line2 in line.analytics_id.account_ids:
 348                    val = (line.credit or  0.0) - (line.debit or 0.0)
 349                    amt=val * (line2.rate/100)
 350                    al_vals={
 351                        'name': line.name,
 352                        'date': line.date,
 353                        'account_id': line2.analytic_account_id.id,
 354                        'unit_amount': line.quantity,
 355                        'product_id': line.product_id and line.product_id.id or False,
 356                        'product_uom_id': line.product_uom_id and line.product_uom_id.id or False,
 357                        'amount': amt,
 358                        'general_account_id': line.account_id.id,
 359                        'move_id': line.id,
 360                        'journal_id': line.journal_id.analytic_journal_id.id,
 361                        'ref': line.ref,
 362                        'percentage': line2.rate
 363                    }
 364                    analytic_line_obj.create(cr, uid, al_vals, context=context)
 365         return True
 366
 367     def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
 368         if context is None:
 369             context = {}
 370         result = super(account_move_line, self).fields_view_get(cr, uid, view_id, view_type, context, toolbar=toolbar, submenu=submenu)
 371         return result
 372
 373 account_move_line()
 374
 375 class account_invoice(osv.osv):
 376     _name = "account.invoice"
 377     _inherit = "account.invoice"
 378
 379     def line_get_convert(self, cr, uid, x, part, date, context=None):
 380         res=super(account_invoice,self).line_get_convert(cr, uid, x, part, date, context=context)
 381         res['analytics_id'] = x.get('analytics_id', False)
 382         return res
 383
 384     def _get_analytic_lines(self, cr, uid, id, context=None):
 385         inv = self.browse(cr, uid, [id])[0]
 386         cur_obj = self.pool.get('res.currency')
 387         invoice_line_obj = self.pool.get('account.invoice.line')
 388         acct_ins_obj = self.pool.get('account.analytic.plan.instance')
 389         company_currency = inv.company_id.currency_id.id
 390         if inv.type in ('out_invoice', 'in_refund'):
 391             sign = 1
 392         else:
 393             sign = -1
 394
 395         iml = invoice_line_obj.move_line_get(cr, uid, inv.id, context=context)
 396
 397         for il in iml:
 398             if il.get('analytics_id', False):
 399
 400                 if inv.type in ('in_invoice', 'in_refund'):
 401                     ref = inv.reference
 402                 else:
 403                     ref = self._convert_ref(cr, uid, inv.number)
 404                 obj_move_line = acct_ins_obj.browse(cr, uid, il['analytics_id'], context=context)
 405                 ctx = context.copy()
 406                 ctx.update({'date': inv.date_invoice})
 407                 amount_calc = cur_obj.compute(cr, uid, inv.currency_id.id, company_currency, il['price'], context=ctx) * sign
 408                 qty = il['quantity']
 409                 il['analytic_lines'] = []
 410                 for line2 in obj_move_line.account_ids:
 411                     amt = amount_calc * (line2.rate/100)
 412                     qtty = qty* (line2.rate/100)
 413                     al_vals = {
 414                         'name': il['name'],
 415                         'date': inv['date_invoice'],
 416                         'unit_amount': qtty,
 417                         'product_id': il['product_id'],
 418                         'account_id': line2.analytic_account_id.id,
 419                         'amount': amt,
 420                         'product_uom_id': il['uos_id'],
 421                         'general_account_id': il['account_id'],
 422                         'journal_id': self._get_journal_analytic(cr, uid, inv.type),
 423                         'ref': ref,
 424                     }
 425                     il['analytic_lines'].append((0, 0, al_vals))
 426         return iml
 427
 428 account_invoice()
 429
 430 class account_analytic_plan(osv.osv):
 431     _inherit = "account.analytic.plan"
 432     _columns = {
 433         'default_instance_id': fields.many2one('account.analytic.plan.instance', 'Default Entries'),
 434     }
 435 account_analytic_plan()
 436
 437 class analytic_default(osv.osv):
 438     _inherit = "account.analytic.default"
 439     _columns = {
 440         'analytics_id': fields.many2one('account.analytic.plan.instance', 'Analytic Distribution'),
 441     }
 442
 443 analytic_default()
 444
 445 class sale_order_line(osv.osv):
 446     _inherit = "sale.order.line"
 447
 448     # Method overridden to set the analytic account by default on criterion match
 449     def invoice_line_create(self, cr, uid, ids, context=None):
 450         create_ids = super(sale_order_line,self).invoice_line_create(cr, uid, ids, context=context)
 451         inv_line_obj = self.pool.get('account.invoice.line')
 452         acct_anal_def_obj = self.pool.get('account.analytic.default')
 453         if ids:
 454             sale_line = self.browse(cr, uid, ids[0], context=context)
 455             for line in inv_line_obj.browse(cr, uid, create_ids, context=context):
 456                 rec = acct_anal_def_obj.account_get(cr, uid, line.product_id.id, sale_line.order_id.partner_id.id, uid, time.strftime('%Y-%m-%d'), context)
 457
 458                 if rec:
 459                     inv_line_obj.write(cr, uid, [line.id], {'analytics_id': rec.analytics_id.id}, context=context)
 460         return create_ids
 461
 462 sale_order_line()
 463
 464
 465 class account_bank_statement(osv.osv):
 466     _inherit = "account.bank.statement"
 467     _name = "account.bank.statement"
 468
 469     def create_move_from_st_line(self, cr, uid, st_line_id, company_currency_id, st_line_number, context=None):
 470         account_move_line_pool = self.pool.get('account.move.line')
 471         account_bank_statement_line_pool = self.pool.get('account.bank.statement.line')
 472         st_line = account_bank_statement_line_pool.browse(cr, uid, st_line_id, context=context)
 473         result = super(account_bank_statement,self).create_move_from_st_line(cr, uid, st_line_id, company_currency_id, st_line_number, context=context)
 474         move = st_line.move_ids and st_line.move_ids[0] or False
 475         if move:
 476             for line in move.line_id:
 477                 account_move_line_pool.write(cr, uid, [line.id], {'analytics_id':st_line.analytics_id.id}, context=context)
 478         return result
 479
 480     def button_confirm_bank(self, cr, uid, ids, context=None):
 481         super(account_bank_statement,self).button_confirm_bank(cr, uid, ids, context=context)
 482         for st in self.browse(cr, uid, ids, context=context):
 483             for st_line in st.line_ids:
 484                 if st_line.analytics_id:
 485                     if not st.journal_id.analytic_journal_id:
 486                         raise osv.except_osv(_('No Analytic Journal !'),_("You have to define an analytic journal on the '%s' journal.") % (st.journal_id.name,))
 487                 if not st_line.amount:
 488                     continue
 489         return True
 490
 491 account_bank_statement()
 492
 493
 494 class account_bank_statement_line(osv.osv):
 495     _inherit = "account.bank.statement.line"
 496     _name = "account.bank.statement.line"
 497     _columns = {
 498         'analytics_id': fields.many2one('account.analytic.plan.instance', 'Analytic Distribution'),
 499     }
 500 account_bank_statement_line()
 501
 502 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: