PyBitmessage/src/bitmessageqt/utils.py at bootstrap · Bitmessage/PyBitmessage · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
import hashlib
import os

from PyQt4 import QtGui

import state
from addresses import addBMIfNotPresent
from bmconfigparser import config

str_broadcast_subscribers = '[Broadcast subscribers]'
str_chan = '[chan]'


def identiconize(address):
    size = 48

    if not config.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 = config.safeGet(
        'bitmessagesettings', 'identiconlib', 'qidenticon_two_x')

    # As an 'identiconsuffix' you could put "@bitmessge.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 = config.get('bitmessagesettings', 'identiconsuffix')
    if identicon_lib[:len('qidenticon')] == 'qidenticon':
        # originally by:
        # :Author:Shin Adachi <shn@glucose.jp>
        # Licesensed 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)