From 5b217e1b40c7c7eeecfa8031aa3fd81bd032955d Mon Sep 17 00:00:00 2001 From: Goffi Date: Sun, 19 Apr 2020 22:52:15 +0200 Subject: [PATCH] tools (common/data_objects): new `avatar_basename` property --- sat/tools/common/data_objects.py | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/sat/tools/common/data_objects.py b/sat/tools/common/data_objects.py index 8bac7299..350a1549 100644 --- a/sat/tools/common/data_objects.py +++ b/sat/tools/common/data_objects.py @@ -21,6 +21,7 @@ from sat.core.constants import Const as C from sat.tools.common import data_format +from os.path import basename try: from jinja2 import Markup as safe @@ -361,6 +362,13 @@ class Identity(object): self.jid_str = jid_str self.data = data if data is not None else {} + @property + def avatar_basename(self): + try: + return basename(self.data['avatar']['path']) + except (TypeError, KeyError): + return None + def __getitem__(self, key): return self.data[key]