80 lines
3.3 KiB
Python
80 lines
3.3 KiB
Python
# -*- coding: utf-8 -*-
|
|
# Part of Odoo. See LICENSE file for full copyright and licensing details.
|
|
|
|
from base64 import b64encode
|
|
from hashlib import sha512
|
|
from odoo import models, fields, api
|
|
from odoo.tools import html_escape, file_open
|
|
|
|
|
|
def get_hsl_from_seed(seed):
|
|
hashed_seed = sha512(seed.encode()).hexdigest()
|
|
# full range of colors, in degree
|
|
hue = int(hashed_seed[0:2], 16) * 360 / 255
|
|
# colorful result but not too flashy, in percent
|
|
sat = int(hashed_seed[2:4], 16) * ((70 - 40) / 255) + 40
|
|
# not too bright and not too dark, in percent
|
|
lig = 45
|
|
return f'hsl({hue:.0f}, {sat:.0f}%, {lig:.0f}%)'
|
|
|
|
|
|
class AvatarMixin(models.AbstractModel):
|
|
_name = 'avatar.mixin'
|
|
_inherit = ['image.mixin']
|
|
_description = "Avatar Mixin"
|
|
_avatar_name_field = "name"
|
|
|
|
# all image fields are base64 encoded and PIL-supported
|
|
avatar_1920 = fields.Image("Avatar", compute="_compute_avatar_1920")
|
|
avatar_1024 = fields.Image("Avatar 1024", compute="_compute_avatar_1024")
|
|
avatar_512 = fields.Image("Avatar 512", compute="_compute_avatar_512")
|
|
avatar_256 = fields.Image("Avatar 256", compute="_compute_avatar_256")
|
|
avatar_128 = fields.Image("Avatar 128", compute="_compute_avatar_128")
|
|
|
|
def _compute_avatar(self, avatar_field, image_field):
|
|
for record in self:
|
|
avatar = record[image_field]
|
|
if not avatar:
|
|
if record.id and record[record._avatar_name_field]:
|
|
avatar = record._avatar_generate_svg()
|
|
else:
|
|
avatar = b64encode(record._avatar_get_placeholder())
|
|
record[avatar_field] = avatar
|
|
|
|
@api.depends(lambda self: [self._avatar_name_field, 'image_1920'])
|
|
def _compute_avatar_1920(self):
|
|
self._compute_avatar('avatar_1920', 'image_1920')
|
|
|
|
@api.depends(lambda self: [self._avatar_name_field, 'image_1024'])
|
|
def _compute_avatar_1024(self):
|
|
self._compute_avatar('avatar_1024', 'image_1024')
|
|
|
|
@api.depends(lambda self: [self._avatar_name_field, 'image_512'])
|
|
def _compute_avatar_512(self):
|
|
self._compute_avatar('avatar_512', 'image_512')
|
|
|
|
@api.depends(lambda self: [self._avatar_name_field, 'image_256'])
|
|
def _compute_avatar_256(self):
|
|
self._compute_avatar('avatar_256', 'image_256')
|
|
|
|
@api.depends(lambda self: [self._avatar_name_field, 'image_128'])
|
|
def _compute_avatar_128(self):
|
|
self._compute_avatar('avatar_128', 'image_128')
|
|
|
|
def _avatar_generate_svg(self):
|
|
initial = html_escape(self[self._avatar_name_field][0].upper())
|
|
bgcolor = get_hsl_from_seed(self[self._avatar_name_field] + str(self.create_date.timestamp() if self.create_date else ""))
|
|
return b64encode((
|
|
"<?xml version='1.0' encoding='UTF-8' ?>"
|
|
"<svg height='180' width='180' xmlns='http://www.w3.org/2000/svg' xmlns:xlink='http://www.w3.org/1999/xlink'>"
|
|
f"<rect fill='{bgcolor}' height='180' width='180'/>"
|
|
f"<text fill='#ffffff' font-size='96' text-anchor='middle' x='90' y='125' font-family='sans-serif'>{initial}</text>"
|
|
"</svg>"
|
|
).encode())
|
|
|
|
def _avatar_get_placeholder_path(self):
|
|
return "base/static/img/avatar_grey.png"
|
|
|
|
def _avatar_get_placeholder(self):
|
|
return file_open(self._avatar_get_placeholder_path(), 'rb').read()
|