server-tools/document_extract_from_database/model/ir_attachment.py

# -*- coding: utf-8 -*-
###############################################################################
#
#    OpenERP, Open Source Management Solution
#    Copyright (C) 2012 Julius Network Solutions SARL <contact@julius.fr>
#
#    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 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 General Public License for more details.
#
#    You should have received a copy of the GNU General Public License
#    along with this program.  If not, see <http://www.gnu.org/licenses/>.
#
###############################################################################

import base64

from openerp.osv.orm import Model


class ir_attachment(Model):
    _inherit = 'ir.attachment'

    def _write_again(self, cr, uid, id, context=None):
        current_data = self.browse(cr, uid, id, context=context)
        location = self.pool['ir.config_parameter'].\
            get_param(cr, uid, 'ir_attachment.location')
        if current_data.db_datas and location:
            vals = {
                'datas': base64.encodestring(current_data.db_datas),
                'db_datas': False,
            }
            self.write(cr, uid, id, vals, context=context)
        return True

    def write_again(self, cr, uid, ids, context=None):
        if context is None:
            context = {}
        for document_id in ids:
            self._write_again(cr, uid, document_id, context=context)
        return True