mirror of
https://github.com/OCA/bank-statement-import.git
synced 2025-01-20 12:37:43 +02:00
[ADD] migration script
and a module that takes care of saving imported files (+migration)
This commit is contained in:
22
account_bank_statement_import_save_file/models/__init__.py
Normal file
22
account_bank_statement_import_save_file/models/__init__.py
Normal file
@@ -0,0 +1,22 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
##############################################################################
|
||||
#
|
||||
# OpenERP, Open Source Management Solution
|
||||
# This module copyright (C) 2015 Therp BV <http://therp.nl>.
|
||||
#
|
||||
# This program is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Affero General Public License as
|
||||
# published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Affero General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
##############################################################################
|
||||
from . import account_bank_statement
|
||||
from . import account_bank_statement_import
|
||||
@@ -0,0 +1,34 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
##############################################################################
|
||||
#
|
||||
# OpenERP, Open Source Management Solution
|
||||
# This module copyright (C) 2015 Therp BV (<http://therp.nl>).
|
||||
#
|
||||
# This program is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Affero General Public License as
|
||||
# published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Affero General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
##############################################################################
|
||||
from openerp import models, fields, api
|
||||
|
||||
|
||||
class AccountBankStatement(models.Model):
|
||||
_inherit = 'account.bank.statement'
|
||||
|
||||
import_file = fields.Many2one(
|
||||
'ir.attachment', 'Import file', readonly=True)
|
||||
import_date = fields.Datetime(
|
||||
related=['import_file', 'create_date'], readonly=True)
|
||||
import_user = fields.Many2one(
|
||||
related=['import_file', 'create_uid'], readonly=True)
|
||||
import_log = fields.Text(
|
||||
related=['import_file', 'description'], readonly=True)
|
||||
@@ -0,0 +1,60 @@
|
||||
# -*- coding: utf-8 -*-
|
||||
##############################################################################
|
||||
#
|
||||
# OpenERP, Open Source Management Solution
|
||||
# This module copyright (C) 2015 Therp BV (<http://therp.nl>).
|
||||
#
|
||||
# This program is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU Affero General Public License as
|
||||
# published by the Free Software Foundation, either version 3 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 Affero General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU Affero General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
#
|
||||
##############################################################################
|
||||
import base64
|
||||
import inspect
|
||||
from openerp import models, fields, api
|
||||
|
||||
|
||||
class AccountBankStatementImport(models.Model):
|
||||
_inherit = 'account.bank.statement.import'
|
||||
|
||||
@api.model
|
||||
def _import_statement(self, statement):
|
||||
(statement_id, notifications) = \
|
||||
super(AccountBankStatementImport, self)._import_statement(
|
||||
statement)
|
||||
if statement_id:
|
||||
# get raw file data from the stack
|
||||
def get_data_file(frame):
|
||||
if 'data_file' in frame.f_locals:
|
||||
return frame.f_locals['data_file']
|
||||
if frame.f_back:
|
||||
return get_data_file(frame.f_back)
|
||||
return None
|
||||
data_file = get_data_file(inspect.currentframe())
|
||||
self.env['account.bank.statement'].browse([statement_id]).write({
|
||||
'import_file': self.env['ir.attachment'].create(
|
||||
self._create_import_file_attachment_data(
|
||||
data_file, statement_id, notifications)).id,
|
||||
})
|
||||
return (statement_id, notifications)
|
||||
|
||||
@api.model
|
||||
def _create_import_file_attachment_data(self, data_file, statement_id,
|
||||
notifications):
|
||||
return {
|
||||
'name': '<unknown>',
|
||||
'res_model': 'account.bank.statement',
|
||||
'res_id': statement_id,
|
||||
'type': 'binary',
|
||||
'datas': base64.b64encode(data_file),
|
||||
'description': notifications,
|
||||
}
|
||||
Reference in New Issue
Block a user