117 lines
4.7 KiB
Python
117 lines
4.7 KiB
Python
import hashlib
|
|
import os
|
|
|
|
from PyQt4 import QtGui
|
|
|
|
import state
|
|
from addresses import addBMIfNotPresent
|
|
from bmconfigparser import BMConfigParser
|
|
|
|
str_broadcast_subscribers = '[Broadcast subscribers]'
|
|
str_chan = '[chan]'
|
|
|
|
|
|
def identiconize(address):
|
|
size = 48
|
|
|
|
if not BMConfigParser().getboolean('bitmessagesettings', 'useidenticons'):
|
|
return QtGui.QIcon()
|
|
|
|
# If you include another identicon library, please generate an
|
|
# example identicon with the following md5 hash:
|
|
# 3fd4bf901b9d4ea1394f0fb358725b28
|
|
|
|
identicon_lib = BMConfigParser().safeGet(
|
|
'bitmessagesettings', 'identiconlib', 'qidenticon_two_x')
|
|
|
|
# As an 'identiconsuffix' you could put "@bitmessage.ch" or "@bm.addr"
|
|
# to make it compatible with other identicon generators. (Note however,
|
|
# that E-Mail programs might convert the BM-address to lowercase first.)
|
|
# It can be used as a pseudo-password to salt the generation of
|
|
# the identicons to decrease the risk of attacks where someone creates
|
|
# an address to mimic someone else's identicon.
|
|
identiconsuffix = BMConfigParser().get('bitmessagesettings', 'identiconsuffix')
|
|
if identicon_lib[:len('qidenticon')] == 'qidenticon':
|
|
# originally by:
|
|
# :Author:Shin Adachi <shn@glucose.jp>
|
|
# Licensed under FreeBSD License.
|
|
# stripped from PIL and uses QT instead (by sendiulo, same license)
|
|
import qidenticon
|
|
icon_hash = hashlib.md5(
|
|
addBMIfNotPresent(address) + identiconsuffix).hexdigest()
|
|
use_two_colors = identicon_lib[:len('qidenticon_two')] == 'qidenticon_two'
|
|
opacity = int(
|
|
identicon_lib not in (
|
|
'qidenticon_x', 'qidenticon_two_x',
|
|
'qidenticon_b', 'qidenticon_two_b'
|
|
)) * 255
|
|
penwidth = 0
|
|
image = qidenticon.render_identicon(
|
|
int(icon_hash, 16), size, use_two_colors, opacity, penwidth)
|
|
# filename = './images/identicons/'+hash+'.png'
|
|
# image.save(filename)
|
|
idcon = QtGui.QIcon()
|
|
idcon.addPixmap(image, QtGui.QIcon.Normal, QtGui.QIcon.Off)
|
|
return idcon
|
|
elif identicon_lib == 'pydenticon':
|
|
# Here you could load pydenticon.py
|
|
# (just put it in the "src" folder of your Bitmessage source)
|
|
from pydenticon import Pydenticon
|
|
# It is not included in the source, because it is licensed under GPLv3
|
|
# GPLv3 is a copyleft license that would influence our licensing
|
|
# Find the source here:
|
|
# https://github.com/azaghal/pydenticon
|
|
# note that it requires pillow (or PIL) to be installed:
|
|
# https://python-pillow.org/
|
|
idcon_render = Pydenticon(
|
|
addBMIfNotPresent(address) + identiconsuffix, size * 3)
|
|
rendering = idcon_render._render()
|
|
data = rendering.convert("RGBA").tostring("raw", "RGBA")
|
|
qim = QtGui.QImage(data, size, size, QtGui.QImage.Format_ARGB32)
|
|
pix = QtGui.QPixmap.fromImage(qim)
|
|
idcon = QtGui.QIcon()
|
|
idcon.addPixmap(pix, QtGui.QIcon.Normal, QtGui.QIcon.Off)
|
|
return idcon
|
|
|
|
|
|
def avatarize(address):
|
|
"""
|
|
Loads a supported image for the given address' hash form 'avatars' folder
|
|
falls back to default avatar if 'default.*' file exists
|
|
falls back to identiconize(address)
|
|
"""
|
|
idcon = QtGui.QIcon()
|
|
icon_hash = hashlib.md5(addBMIfNotPresent(address)).hexdigest()
|
|
if address == str_broadcast_subscribers:
|
|
# don't hash [Broadcast subscribers]
|
|
icon_hash = address
|
|
# https://www.riverbankcomputing.com/static/Docs/PyQt4/qimagereader.html#supportedImageFormats
|
|
# QImageReader.supportedImageFormats ()
|
|
extensions = [
|
|
'PNG', 'GIF', 'JPG', 'JPEG', 'SVG', 'BMP', 'MNG', 'PBM', 'PGM', 'PPM',
|
|
'TIFF', 'XBM', 'XPM', 'TGA']
|
|
# try to find a specific avatar
|
|
for ext in extensions:
|
|
lower_hash = state.appdata + 'avatars/' + icon_hash + '.' + ext.lower()
|
|
upper_hash = state.appdata + 'avatars/' + icon_hash + '.' + ext.upper()
|
|
if os.path.isfile(lower_hash):
|
|
idcon.addFile(lower_hash)
|
|
return idcon
|
|
elif os.path.isfile(upper_hash):
|
|
idcon.addFile(upper_hash)
|
|
return idcon
|
|
# if we haven't found any, try to find a default avatar
|
|
for ext in extensions:
|
|
lower_default = state.appdata + 'avatars/' + 'default.' + ext.lower()
|
|
upper_default = state.appdata + 'avatars/' + 'default.' + ext.upper()
|
|
if os.path.isfile(lower_default):
|
|
default = lower_default
|
|
idcon.addFile(lower_default)
|
|
return idcon
|
|
elif os.path.isfile(upper_default):
|
|
default = upper_default
|
|
idcon.addFile(upper_default)
|
|
return idcon
|
|
# If no avatar is found
|
|
return identiconize(address)
|