bank-statement-import/account_bank_statement_import_online/wizards/online_bank_statement_pull_wizard.py

# Copyright 2019 Brainbean Apps (https://brainbeanapps.com)
# Copyright 2019 Dataplug (https://dataplug.io)
# License AGPL-3.0 or later (https://www.gnu.org/licenses/agpl).

from odoo import fields, models, api


class OnlineBankStatementPullWizard(models.TransientModel):
    _name = 'online.bank.statement.pull.wizard'
    _description = 'Online Bank Statement Pull Wizard'

    date_since = fields.Datetime(
        string='Since',
        required=True,
        default=fields.Datetime.now,
    )
    date_until = fields.Datetime(
        string='Until',
        required=True,
        default=fields.Datetime.now,
    )
    provider_ids = fields.Many2many(
        string='Providers',
        comodel_name='online.bank.statement.provider',
        column1='wizard_id',
        column2='provider_id',
        relation='online_bank_statement_provider_pull_wizard_rel'
    )

    @api.multi
    def action_pull(self):
        self.ensure_one()
        self.provider_ids._pull(self.date_since, self.date_until)
        return {'type': 'ir.actions.act_window_close'}