summaryrefslogtreecommitdiffstats
path: root/alternativ_nick.py
blob: 61385cb6fda0ef600d71e772216dfae483252891 (plain)
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
#!/usr/bin/env python
"""
alternativ_nick.py - Use an alternativ nick, if your choosen nick is already taken
Copyright 2016, Alexander Sulfrian <alex@spline.inf.fu-berlin.de>
Licensed under the Eiffel Forum License 2.

http://inamidst.com/phenny/
"""


def _set_nick(wrapper, nick):
    # The phenny argument of the event handler is only a wrapper to enable the
    # reply and say methods. If we try to change the 'nick' attribute directly,
    # the modification is lost. So we need to get the real phenny instance
    # before.
    phenny = wrapper.__getattribute__('bot')
    phenny.nick = nick


def _build_nick(phenny):
    # We have different ways to generate an alternativ_nick:
    # * First we try to use the config option.
    # * If it is not set or it does not work, we append an underscore after the
    #   default nick.
    # * If even this did not work, we append a number and count upwards until a
    #   valid nick is found.

    alternativ_nick = getattr(phenny.config, 'alternativ_nick', None)
    if alternativ_nick is not None and phenny.nick != alternativ_nick:
        return alternativ_nick
    else:
        if phenny.data.get('alternativ_nick.count') is not None:
            count = phenny.data['alternativ_nick.count']
            count += 1
            phenny.data['alternativ_nick.count'] = count
            return ('%s_%d' % (phenny.config.nick, count))
        else:
            phenny.data['alternativ_nick.count'] = 0
            return ('%s_' % phenny.config.nick)


def nickname_in_use(phenny, input):
    new_nick = _build_nick(phenny)
    _set_nick(phenny, new_nick)
    phenny.write(('NICK', new_nick))
nickname_in_use.rule = r'(.*)'
nickname_in_use.event = '433'
nickname_in_use.thread = False