magnet2/plugin_command_rpg.py at master · grompe/magnet2 · 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
# Magnet2 by Grom PE. Public domain.
import time, random, md5
from magnet_api import *

def command_roll(bot, room, nick, access_level, parameters, message):
  p = parameters.find('d')
  x = parameters.find('+')
  s = parameters.find(' ')
  if s != -1: parameters = parameters[:s]
  number = 0
  sides = 0
  add = 0
  if p == -1: p = len(parameters)
  try:
    if x == -1:
      sides = int(parameters[p+1:])
    else:
      sides = int(parameters[p+1:x])
      add = int(parameters[x+1:])
    number = int(parameters[:p])
  except:
    pass
  if number < 1: number = 1
  if number > 10: number = 10
  if sides < 2: sides = 6
  if sides > 100: sides = 100
  if add < -1000: add = -1000
  if add > 1000: add = 1000
  rolls = []
  for i in range(number): rolls.append(random.randint(1, sides))
  if add: rolls.append(add)
  output = '%s, %dd%d+%d roll: %s'%(nick, number, sides, add, '+'.join([str(i) for i in rolls]))
  if len(rolls) > 1: output += '=' + str(sum(rolls))
  return output

def command_flip(bot, room, nick, access_level, parameters, message):
  return nick+', '+random.choice(['Heads!', 'Tails!'])

def load(bot):
  bot.add_command('roll', command_roll, LEVEL_GUEST, 'rpg')
  bot.add_command('flip', command_flip, LEVEL_GUEST, 'rpg')

def unload(bot):
  pass

def info(bot):
  return 'RPG plugin v1.0'