You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.

43 lines
1.6 KiB

  1. # -*- coding: utf-8 -*-
  2. ###################################################################################
  3. #
  4. # Copyright (C) 2017 MuK IT GmbH
  5. #
  6. # This program is free software: you can redistribute it and/or modify
  7. # it under the terms of the GNU Affero General Public License as
  8. # published by the Free Software Foundation, either version 3 of the
  9. # License, or (at your option) any later version.
  10. #
  11. # This program is distributed in the hope that it will be useful,
  12. # but WITHOUT ANY WARRANTY; without even the implied warranty of
  13. # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14. # GNU Affero General Public License for more details.
  15. #
  16. # You should have received a copy of the GNU Affero General Public License
  17. # along with this program. If not, see <http://www.gnu.org/licenses/>.
  18. #
  19. ###################################################################################
  20. import base64
  21. import collections
  22. import logging
  23. import mimetypes
  24. import os.path
  25. from odoo import _
  26. from odoo import models, api, fields
  27. _logger = logging.getLogger(__name__)
  28. class IrAttachment(models.Model):
  29. _inherit = 'ir.attachment'
  30. extension = fields.Char(string='File Extension', compute='_compute_extension')
  31. @api.depends('mimetype', 'datas_fname')
  32. def _compute_extension(self):
  33. for attachment in self:
  34. if attachment.datas_fname:
  35. attachment.extension = os.path.splitext(attachment.datas_fname)[1]
  36. elif attachment.mimetype:
  37. attachment.extension = mimetypes.guess_extension(attachment.mimetype, strict=False)