NeoAnd/knowledge
1
0
Fork 0
mirror of https://github.com/OCA/knowledge.git synced 2025-07-17 04:32:55 -06:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[ADD] use libmagic or mimetypes to detect a file's type

Browse Source
This commit is contained in:
Holger Brunn 2014-09-10 12:10:15 +02:00
parent 7a7c7d6274
commit 9ff3f16430
2 changed files with 18 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
attachment_preview/__openerp__.py
View File

@ -29,6 +29,9 @@ This addons allows to preview attachments supported by http://viewerjs.org.
Currently, that's most Libreoffice files and PDFs. Currently, that's most Libreoffice files and PDFs.
For filetype recognition, you'll get the best results by installing
``python-magic``.
Acknowledgements Acknowledgements
================ ================

19
attachment_preview/model/ir_attachment.py
View File

@ -20,6 +20,8 @@
############################################################################## ##############################################################################
import collections import collections
import os.path import os.path
import mimetypes
import base64
from openerp.osv.orm import Model from openerp.osv.orm import Model
@ -35,9 +37,18 @@ class IrAttachment(Model):
extension = '' extension = ''
if this.datas_fname: if this.datas_fname:
filename, extension = os.path.splitext(this.datas_fname) filename, extension = os.path.splitext(this.datas_fname)
extension = extension.lstrip('.')
if not extension: if not extension:
# TODO: lookup the extension via mimetype try:
pass import magic
result[this.id] = extension ms = magic.open(magic.MAGIC_MIME_TYPE)
ms.load()
mimetype = ms.buffer(
base64.b64decode(this.datas))
except ImportError:
(mimetype, encoding) = mimetypes.guess_type(
'data:;base64,' + this.datas, strict=False)
extension = mimetypes.guess_extension(
mimetype, strict=False)
result[this.id] = (extension or '').lstrip('.').lower()
return result if isinstance(ids, collections.Iterable) else result[ids] return result if isinstance(ids, collections.Iterable) else result[ids]