odoo/ext/custom-addons/dp_custom/models/res_company.py

56 lines
2.0 KiB
Python

# -*- coding: utf-8 -*-
##############################################################################
#
# datenpol gmbh
# Copyright (C) 2013-TODAY datenpol gmbh (<http://www.datenpol.at/>)
#
# 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 os
import base64
from odoo import models, api
class Company(models.Model):
_inherit = 'res.company'
@api.model
def set_company_logo(self, company, logo):
filename = os.path.join(os.path.dirname(os.path.realpath(__file__)), logo)
fi = open(filename, 'rb')
content = base64.b64encode(fi.read())
content = content.decode()
fi.close()
self.browse(company).write({'logo': content})
@api.model
def set_company_supplier(self, company):
self.browse(company).partner_id.supplier = True
@api.model
def set_company_customer(self, company):
self.browse(company).partner_id.customer = True
@api.model
def set_company_favicon(self, company, logo):
filename = os.path.join(os.path.dirname(os.path.realpath(__file__)), logo)
fi = open(filename, 'rb')
content = base64.b64encode(fi.read())
content = content.decode()
fi.close()
self.browse(company).write(
{'favicon_backend': content, 'favicon_backend_mimetype': 'image/x-icon'})