addons/account/wizard/account_report_common.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 openerp.osv import fields, osv
  26 from openerp.osv.orm import setup_modifiers
  27 from openerp.tools.translate import _
  28
  29 class account_common_report(osv.osv_memory):
  30     _name = "account.common.report"
  31     _description = "Account Common Report"
  32
  33     def onchange_chart_id(self, cr, uid, ids, chart_account_id=False, context=None):
  34         res = {}
  35         if chart_account_id:
  36             company_id = self.pool.get('account.account').browse(cr, uid, chart_account_id, context=context).company_id.id
  37             now = time.strftime('%Y-%m-%d')
  38             domain = [('company_id', '=', company_id), ('date_start', '<', now), ('date_stop', '>', now)]
  39             fiscalyears = self.pool.get('account.fiscalyear').search(cr, uid, domain, limit=1)
  40             res['value'] = {'company_id': company_id, 'fiscalyear_id': fiscalyears and fiscalyears[0] or False}
  41         return res
  42
  43     _columns = {
  44         'chart_account_id': fields.many2one('account.account', 'Chart of Account', help='Select Charts of Accounts', required=True, domain = [('parent_id','=',False)]),
  45         'company_id': fields.related('chart_account_id', 'company_id', type='many2one', relation='res.company', string='Company', readonly=True),
  46         'fiscalyear_id': fields.many2one('account.fiscalyear', 'Fiscal Year', help='Keep empty for all open fiscal year'),
  47         'filter': fields.selection([('filter_no', 'No Filters'), ('filter_date', 'Date'), ('filter_period', 'Periods')], "Filter by", required=True),
  48         'period_from': fields.many2one('account.period', 'Start Period'),
  49         'period_to': fields.many2one('account.period', 'End Period'),
  50         'journal_ids': fields.many2many('account.journal', string='Journals', required=True),
  51         'date_from': fields.date("Start Date"),
  52         'date_to': fields.date("End Date"),
  53         'target_move': fields.selection([('posted', 'All Posted Entries'),
  54                                          ('all', 'All Entries'),
  55                                         ], 'Target Moves', required=True),
  56
  57         }
  58
  59     def _check_company_id(self, cr, uid, ids, context=None):
  60         for wiz in self.browse(cr, uid, ids, context=context):
  61             company_id = wiz.company_id.id
  62             if wiz.fiscalyear_id and company_id != wiz.fiscalyear_id.company_id.id:
  63                 return False
  64             if wiz.period_from and company_id != wiz.period_from.company_id.id:
  65                 return False
  66             if wiz.period_to and company_id != wiz.period_to.company_id.id:
  67                 return False
  68         return True
  69
  70     _constraints = [
  71         (_check_company_id, 'The fiscalyear, periods or chart of account chosen have to belong to the same company.', ['chart_account_id','fiscalyear_id','period_from','period_to']),
  72     ]
  73
  74     def fields_view_get(self, cr, uid, view_id=None, view_type='form', context=None, toolbar=False, submenu=False):
  75         if context is None:context = {}
  76         res = super(account_common_report, self).fields_view_get(cr, uid, view_id=view_id, view_type=view_type, context=context, toolbar=toolbar, submenu=False)
  77         if context.get('active_model', False) == 'account.account':
  78             doc = etree.XML(res['arch'])
  79             nodes = doc.xpath("//field[@name='chart_account_id']")
  80             for node in nodes:
  81                 node.set('readonly', '1')
  82                 node.set('help', 'If you print the report from Account list/form view it will not consider Charts of account')
  83                 setup_modifiers(node, res['fields']['chart_account_id'])
  84             res['arch'] = etree.tostring(doc)
  85         return res
  86
  87     def onchange_filter(self, cr, uid, ids, filter='filter_no', fiscalyear_id=False, context=None):
  88         res = {'value': {}}
  89         if filter == 'filter_no':
  90             res['value'] = {'period_from': False, 'period_to': False, 'date_from': False ,'date_to': False}
  91         if filter == 'filter_date':
  92             res['value'] = {'period_from': False, 'period_to': False, 'date_from': time.strftime('%Y-01-01'), 'date_to': time.strftime('%Y-%m-%d')}
  93         if filter == 'filter_period' and fiscalyear_id:
  94             start_period = end_period = False
  95             cr.execute('''
  96                 SELECT * FROM (SELECT p.id
  97                                FROM account_period p
  98                                LEFT JOIN account_fiscalyear f ON (p.fiscalyear_id = f.id)
  99                                WHERE f.id = %s
 100                                AND p.special = false
 101                                ORDER BY p.date_start ASC, p.special ASC
 102                                LIMIT 1) AS period_start
 103                 UNION ALL
 104                 SELECT * FROM (SELECT p.id
 105                                FROM account_period p
 106                                LEFT JOIN account_fiscalyear f ON (p.fiscalyear_id = f.id)
 107                                WHERE f.id = %s
 108                                AND p.date_start < NOW()
 109                                AND p.special = false
 110                                ORDER BY p.date_stop DESC
 111                                LIMIT 1) AS period_stop''', (fiscalyear_id, fiscalyear_id))
 112             periods =  [i[0] for i in cr.fetchall()]
 113             if periods and len(periods) > 1:
 114                 start_period = periods[0]
 115                 end_period = periods[1]
 116             res['value'] = {'period_from': start_period, 'period_to': end_period, 'date_from': False, 'date_to': False}
 117         return res
 118
 119     def _get_account(self, cr, uid, context=None):
 120         user = self.pool.get('res.users').browse(cr, uid, uid, context=context)
 121         accounts = self.pool.get('account.account').search(cr, uid, [('parent_id', '=', False), ('company_id', '=', user.company_id.id)], limit=1)
 122         return accounts and accounts[0] or False
 123
 124     def _get_fiscalyear(self, cr, uid, context=None):
 125         if context is None:
 126             context = {}
 127         now = time.strftime('%Y-%m-%d')
 128         company_id = False
 129         ids = context.get('active_ids', [])
 130         if ids and context.get('active_model') == 'account.account':
 131             company_id = self.pool.get('account.account').browse(cr, uid, ids[0], context=context).company_id.id
 132         else:  # use current company id
 133             company_id = self.pool.get('res.users').browse(cr, uid, uid, context=context).company_id.id
 134         domain = [('company_id', '=', company_id), ('date_start', '<', now), ('date_stop', '>', now)]
 135         fiscalyears = self.pool.get('account.fiscalyear').search(cr, uid, domain, limit=1)
 136         return fiscalyears and fiscalyears[0] or False
 137
 138     def _get_all_journal(self, cr, uid, context=None):
 139         return self.pool.get('account.journal').search(cr, uid ,[])
 140
 141     _defaults = {
 142             'fiscalyear_id': _get_fiscalyear,
 143             'company_id': lambda self,cr,uid,c: self.pool.get('res.company')._company_default_get(cr, uid, 'account.common.report',context=c),
 144             'journal_ids': _get_all_journal,
 145             'filter': 'filter_no',
 146             'chart_account_id': _get_account,
 147             'target_move': 'posted',
 148     }
 149
 150     def _build_contexts(self, cr, uid, ids, data, context=None):
 151         if context is None:
 152             context = {}
 153         result = {}
 154         result['fiscalyear'] = 'fiscalyear_id' in data['form'] and data['form']['fiscalyear_id'] or False
 155         result['journal_ids'] = 'journal_ids' in data['form'] and data['form']['journal_ids'] or False
 156         result['chart_account_id'] = 'chart_account_id' in data['form'] and data['form']['chart_account_id'] or False
 157         result['state'] = 'target_move' in data['form'] and data['form']['target_move'] or ''
 158         if data['form']['filter'] == 'filter_date':
 159             result['date_from'] = data['form']['date_from']
 160             result['date_to'] = data['form']['date_to']
 161         elif data['form']['filter'] == 'filter_period':
 162             if not data['form']['period_from'] or not data['form']['period_to']:
 163                 raise osv.except_osv(_('Error!'),_('Select a starting and an ending period.'))
 164             result['period_from'] = data['form']['period_from']
 165             result['period_to'] = data['form']['period_to']
 166         return result
 167
 168     def _print_report(self, cr, uid, ids, data, context=None):
 169         raise (_('Error!'), _('Not implemented.'))
 170
 171     def check_report(self, cr, uid, ids, context=None):
 172         if context is None:
 173             context = {}
 174         data = {}
 175         data['ids'] = context.get('active_ids', [])
 176         data['model'] = context.get('active_model', 'ir.ui.menu')
 177         data['form'] = self.read(cr, uid, ids, ['date_from',  'date_to',  'fiscalyear_id', 'journal_ids', 'period_from', 'period_to',  'filter',  'chart_account_id', 'target_move'], context=context)[0]
 178         for field in ['fiscalyear_id', 'chart_account_id', 'period_from', 'period_to']:
 179             if isinstance(data['form'][field], tuple):
 180                 data['form'][field] = data['form'][field][0]
 181         used_context = self._build_contexts(cr, uid, ids, data, context=context)
 182         data['form']['periods'] = used_context.get('periods', False) and used_context['periods'] or []
 183         data['form']['used_context'] = dict(used_context, lang=context.get('lang', 'en_US'))
 184         return self._print_report(cr, uid, ids, data, context=context)
 185
 186
 187 # vim:expandtab:smartindent:tabstop=4:softtabstop=4:shiftwidth=4: