##############################################################################
#
# Copyright (c) 2004-2008 TINY SPRL. (http://tiny.be) All Rights Reserved.
#
# $Id$
#
# WARNING: This program as such is intended to be used by professional
# programmers who take the whole responsability of assessing all potential
# consequences resulting from its eventual inadequacies and bugs
# End users who are looking for a ready-to-use solution with commercial
# garantees and support are strongly adviced to contract a Free Software
# Service Company
#
# This program is Free Software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# 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 General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with this program; if not, write to the Free Software
# Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
#
##############################################################################

import time
import wizard
import pooler

dates_form = '''<?xml version="1.0"?>
<form string="Select period">
    <field name="company_id"/>
    <newline/>
    <field name="period_id"/>
    <newline/>
    <field name="based_on"/>
</form>'''

dates_fields = {
    'company_id': {'string': 'Company', 'type': 'many2one',
        'relation': 'res.company', 'required': True},
    'period_id': {'string':'Period', 'type':'many2one',
        'relation': 'account.period', 'required':True},
    'based_on':{'string':'Base on', 'type':'selection', 'selection':[
            ('invoices','Invoices'),
            ('payments','Payments'),
            ], 'required':True, 'default': lambda *a: 'invoices'},
}


class wizard_report(wizard.interface):

    def _get_defaults(self, cr, uid, data, context):
        pool = pooler.get_pool(cr.dbname)
        period_obj = pool.get('account.period')
        data['form']['period_id'] = period_obj.find(cr, uid)[0]

        user = pool.get('res.users').browse(cr, uid, uid, context=context)
        if user.company_id:
            company_id = user.company_id.id
        else:
            company_id = pool.get('res.company').search(cr, uid,
                    [('parent_id', '=', False)])[0]
        data['form']['company_id'] = company_id

        return data['form']

    states = {
        'init': {
            'actions': [_get_defaults],
            'result': {
                'type': 'form',
                'arch': dates_form,
                'fields': dates_fields,
                'state': [
                    ('end', 'Cancel'),
                    ('report', 'Print VAT Decl.')
                ]
            }
        },
        'report': {
            'actions': [],
            'result': {
                'type': 'print',
                'report': 'account.vat.declaration',
                'state':'end'
            }
        }
    }

wizard_report('account.vat.declaration')