Antony Lesuisse
3ced0ff61b
Keeping the attachments widget on form views and document indexation (currently limited to pdf/odt).
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
import logging
|
|
import zipfile
|
|
import xml.dom.minidom
|
|
from StringIO import StringIO
|
|
|
|
import pyPdf
|
|
|
|
import openerp
|
|
from openerp.osv import fields, osv
|
|
|
|
_logger = logging.getLogger(__name__)
|
|
|
|
class IrAttachment(osv.osv):
|
|
_inherit = 'ir.attachment'
|
|
|
|
def _index_odt(self, bin_data):
|
|
buf = u""
|
|
f = StringIO(bin_data)
|
|
if zipfile.is_zipfile(f):
|
|
try:
|
|
zf = zipfile.ZipFile(f)
|
|
self.content = xml.dom.minidom.parseString(zf.read("content.xml"))
|
|
for val in ["text:p", "text:h", "text:list"]:
|
|
for element in self.content.getElementsByTagName(val) :
|
|
for node in element.childNodes :
|
|
if node.nodeType == xml.dom.Node.TEXT_NODE :
|
|
buf += node.nodeValue
|
|
elif node.nodeType == xml.dom.Node.ELEMENT_NODE :
|
|
buf += self.textToString(node)
|
|
buf += "\n"
|
|
except Exception:
|
|
pass
|
|
return buf
|
|
|
|
def _index_pdf(self, bin_data):
|
|
buf = u""
|
|
if bin_data.startswith('%PDF-'):
|
|
f = StringIO(bin_data)
|
|
try:
|
|
pdf = pyPdf.PdfFileReader(f)
|
|
for page in pdf.pages:
|
|
buf += page.extractText()
|
|
except Exception:
|
|
pass
|
|
return buf
|
|
|
|
def _index(self, cr, uid, bin_data, datas_fname, mimetype):
|
|
# try to index odt content
|
|
buf = self._index_odt(bin_data)
|
|
if buf:
|
|
return buf
|
|
# try to index pdf content
|
|
buf = self._index_pdf(bin_data)
|
|
if buf:
|
|
return buf
|
|
|
|
return super(IrAttachment, self)._index(cr, uid, bin_data, datas_fname, mimetype)
|
|
|