Browse Source

Add duration arg type, fix GuildMember.remove_role

pull/21/head
Andrei 8 years ago
parent
commit
a8793c869f
  1. 7
      disco/bot/parser.py
  2. 2
      disco/types/guild.py

7
disco/bot/parser.py

@ -2,7 +2,6 @@ import re
import six
import copy
# Regex which splits out argument parts
PARTS_RE = re.compile('(\<|\[|\{)((?:\w+|\:|\||\.\.\.| (?:[0-9]+))+)(?:\>|\]|\})')
@ -16,6 +15,12 @@ TYPE_MAP = {
'snowflake': lambda ctx, data: int(data),
}
try:
import dateparser
TYPE_MAP['duration'] = lambda ctx, data: dateparser.parse(data, settings={'TIMEZONE': 'UTC'})
except ImportError:
pass
def to_bool(ctx, data):
if data in BOOL_OPTS:

2
disco/types/guild.py

@ -207,7 +207,7 @@ class GuildMember(SlottedModel):
self.client.api.guilds_members_roles_add(self.guild.id, self.user.id, to_snowflake(role))
def remove_role(self, role):
self.clients.api.guilds_members_roles_remove(self.guild.id, self.user.id, to_snowflake(role))
self.client.api.guilds_members_roles_remove(self.guild.id, self.user.id, to_snowflake(role))
@cached_property
def owner(self):

Loading…
Cancel
Save