Browse Source

Fix some Python3 type issues

pull/49/head
Ondřej Slabý 8 years ago
parent
commit
ef32f67e0f
  1. 4
      disco/bot/command.py
  2. 3
      disco/gateway/sharder.py
  3. 3
      disco/util/snowflake.py

4
disco/bot/command.py

@ -3,6 +3,8 @@ import argparse
from holster.enum import Enum
from six import integer_types
from disco.bot.parser import ArgumentSet, ArgumentError
from disco.util.functional import cached_property
@ -186,7 +188,7 @@ class Command(object):
return ctx.msg.client.state.users.select_one(username=uid[0], discriminator=uid[1])
def resolve_channel(ctx, cid):
if isinstance(cid, (int, long)):
if isinstance(cid, integer_types):
return ctx.msg.guild.channels.get(cid)
else:
return ctx.msg.guild.channels.select_one(name=cid)

3
disco/gateway/sharder.py

@ -6,6 +6,7 @@ import pickle
import logging
import marshal
from six import integer_types, string_types
from six.moves import range
from disco.client import Client
@ -83,7 +84,7 @@ class AutoSharder(object):
@staticmethod
def dumps(data):
if isinstance(data, (basestring, int, long, bool, list, set, dict)):
if isinstance(data, (string_types, integer_types, bool, list, set, dict)):
return '\x01' + marshal.dumps(data)
elif isinstance(data, object) and data.__class__.__name__ == 'code':
return '\x01' + marshal.dumps(data)

3
disco/util/snowflake.py

@ -26,6 +26,9 @@ def from_datetime(date):
def from_timestamp(ts):
if six.PY3:
return int(ts * 1000.0 - DISCORD_EPOCH) << 22
else:
return long(ts * 1000.0 - DISCORD_EPOCH) << 22

Loading…
Cancel
Save