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
117
118
119
120
121
122
|
"""
Dice plugin: roll some dice
Usage of this plugin is not recommended.
Commands
--------
.. glossary::
/roll [number of dice] [duration of the roll]
Roll one or several unicode dice
Configuration
-------------
.. glossary::
:sorted:
refresh
**Default:** ``0.5``
Interval in seconds between each correction (the closest to 0 is the fastest)
default_duration
**Default:** ``5``
Total duration of the animation.
"""
import random
from typing import Optional
from poezio import tabs
from poezio.decorators import command_args_parser
from poezio.plugin import BasePlugin
DICE = '\u2680\u2681\u2682\u2683\u2684\u2685'
class DiceRoll:
__slots__ = [
'duration', 'total_duration', 'dice_number', 'msgtype', 'jid',
'msgid', 'increments'
]
def __init__(self, total_duration, dice_number, msgtype, jid, msgid, increments):
self.duration = 0
self.total_duration = total_duration
self.dice_number = dice_number
self.msgtype = msgtype
self.jid = jid
self.msgid = msgid
self.increments = increments
def reroll(self):
self.duration += self.increments
def is_finished(self):
return self.duration >= self.total_duration
def roll_dice(num_dice: int) -> str:
return ''.join(random.choice(DICE) for _ in range(num_dice))
class Plugin(BasePlugin):
default_config = {"dice": {"refresh": 0.75, "default_duration": 7.5}}
def init(self):
for tab_t in [tabs.MucTab, tabs.DynamicConversationTab, tabs.StaticConversationTab, tabs.PrivateTab]:
self.api.add_tab_command(
tab_t,
'roll',
self.command_dice,
help='Roll a die',
usage='[number] [duration]')
@command_args_parser.quoted(0, 2, ['', ''], True)
def command_dice(self, args):
tab = self.api.current_tab()
duration = self.config.get('default_duration')
num_dice = 1
try:
if args[0]:
num_dice = int(args[0])
if args[1]:
duration = float(args[1])
except ValueError:
self.core.command.help("roll")
return
else:
if num_dice <= 0 or duration < 0:
self.core.command.help("roll")
return
msgtype = 'groupchat' if isinstance(tab, tabs.MucTab) else 'chat'
message = self.core.xmpp.make_message(tab.jid)
message['type'] = msgtype
message['body'] = roll_dice(num_dice)
message.send()
increment = self.config.get('refresh')
msgid = message['id']
roll = DiceRoll(duration, num_dice, msgtype, tab.jid, msgid, increment)
event = self.api.create_delayed_event(increment, self.delayed_event,
roll)
self.api.add_timed_event(event)
def delayed_event(self, roll):
if roll.is_finished():
return
roll.reroll()
message = self.core.xmpp.make_message(roll.jid)
message["type"] = roll.msgtype
message["body"] = roll_dice(roll.dice_number)
message["replace"]["id"] = roll.msgid
message.send()
event = self.api.create_delayed_event(roll.increments,
self.delayed_event, roll)
self.api.add_timed_event(event)
|
