File: rainbow.py

package info (click to toggle)
poezio 0.15.1-2
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid
  • size: 6,108 kB
  • sloc: python: 24,816; xml: 987; ansic: 329; makefile: 200; sh: 20; javascript: 2
file content (47 lines) | stat: -rw-r--r-- 1,231 bytes parent folder | download | duplicates (3)
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
"""
This plugin colors each character of a message with a random color.

.. note:: As ticket `#3273`_ puts it, the final output is closer to vomit than a rainbow.

Usage
-----

.. glossary::

    /rainbow

        Say something in a Chat tab.

.. note:: Can create fun things when used with :ref:`The figlet plugin <figlet-plugin>`.

.. _#3273: https://lab.louiz.org/poezio/poezio/-/issues/3273
"""
from poezio.plugin import BasePlugin
from poezio import xhtml
import random

possible_colors = list(range(256))
# remove the colors that are almost white or almost black
for col in [
        16, 232, 233, 234, 235, 236, 237, 15, 231, 255, 254, 253, 252, 251
]:
    possible_colors.remove(col)


def rand_color():
    return '\x19%s}' % (random.choice(possible_colors), )


class Plugin(BasePlugin):
    def init(self):
        self.api.add_event_handler('muc_say', self.rainbowize)
        self.api.add_event_handler('private_say', self.rainbowize)
        self.api.add_event_handler('conversation_say', self.rainbowize)

    def rainbowize(self, msg, tab):
        msg['body'] = ''.join([
            '%s%s' % (
                rand_color(),
                char,
            ) for char in xhtml.clean_text(msg['body'])
        ])