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