mirror of
https://github.com/RYGhub/royalnet.git
synced 2024-11-23 11:34:18 +00:00
Handle events differently
This commit is contained in:
parent
62016e6fd3
commit
a0ac85b213
GPG key ID:
896A80F55F7C97F0
2 changed files with 39 additions and 94 deletions
35
poetry.lock
generated
35
poetry.lock
generated
|
|
@ -198,20 +198,16 @@ description = "A Python wrapper for the Discord API"
|
|||
name = "discord.py"
|
||||
optional = true
|
||||
python-versions = ">=3.5.3"
|
||||
version = "1.4.0a2339+gcbdf660"
|
||||
version = "1.3.3"
|
||||
|
||||
[package.dependencies]
|
||||
aiohttp = ">=3.6.0,<3.7.0"
|
||||
websockets = ">=6.0,<7.0 || >7.0,<8.0 || >8.0,<8.0.1 || >8.0.1,<9.0"
|
||||
|
||||
[package.extras]
|
||||
docs = ["sphinx (1.8.5)", "sphinxcontrib_trio (1.1.1)", "sphinxcontrib-websupport"]
|
||||
docs = ["sphinx (1.8.5)", "sphinxcontrib-trio (1.1.1)", "sphinxcontrib-websupport"]
|
||||
voice = ["PyNaCl (1.3.0)"]
|
||||
|
||||
[package.source]
|
||||
reference = "cbdf660ddcee8b3d25d69b3504c2975900b6da91"
|
||||
type = "git"
|
||||
url = "https://github.com/Rapptz/discord.py/"
|
||||
[[package]]
|
||||
category = "dev"
|
||||
description = "Docutils -- Python Documentation Utilities"
|
||||
|
|
@ -560,7 +556,7 @@ description = "pytest: simple powerful testing with Python"
|
|||
name = "pytest"
|
||||
optional = false
|
||||
python-versions = ">=3.5"
|
||||
version = "5.4.2"
|
||||
version = "5.4.3"
|
||||
|
||||
[package.dependencies]
|
||||
atomicwrites = ">=1.0"
|
||||
|
|
@ -911,11 +907,11 @@ version = "0.14.0"
|
|||
|
||||
[[package]]
|
||||
category = "dev"
|
||||
description = "Measures number of Terminal column cells of wide-character codes"
|
||||
description = "Measures the displayed width of unicode strings in a terminal"
|
||||
name = "wcwidth"
|
||||
optional = false
|
||||
python-versions = "*"
|
||||
version = "0.1.9"
|
||||
version = "0.2.3"
|
||||
|
||||
[[package]]
|
||||
category = "main"
|
||||
|
|
@ -943,7 +939,7 @@ description = "YouTube video downloader"
|
|||
name = "youtube-dl"
|
||||
optional = true
|
||||
python-versions = "*"
|
||||
version = "2020.5.29"
|
||||
version = "2020.6.6"
|
||||
|
||||
[extras]
|
||||
alchemy_easy = ["sqlalchemy", "psycopg2_binary", "bcrypt"]
|
||||
|
|
@ -958,7 +954,7 @@ sentry = ["sentry_sdk"]
|
|||
telegram = ["python_telegram_bot"]
|
||||
|
||||
[metadata]
|
||||
content-hash = "d5db7520e6b434b76360627fd06e5b414d1941b833fe08007b9a8c9391b7d5e5"
|
||||
content-hash = "218f4a253a7ef17bb871abf541ae7182f1626cd43d55417de12f9561c58ca0e9"
|
||||
python-versions = "^3.8"
|
||||
|
||||
[metadata.files]
|
||||
|
|
@ -1099,7 +1095,10 @@ deprecation = [
|
|||
{file = "deprecation-2.1.0-py2.py3-none-any.whl", hash = "sha256:a10811591210e1fb0e768a8c25517cabeabcba6f0bf96564f8ff45189f90b14a"},
|
||||
{file = "deprecation-2.1.0.tar.gz", hash = "sha256:72b3bde64e5d778694b0cf68178aed03d15e15477116add3fb773e581f9518ff"},
|
||||
]
|
||||
"discord.py" = []
|
||||
"discord.py" = [
|
||||
{file = "discord.py-1.3.3-py3-none-any.whl", hash = "sha256:406871b06d86c3dc49fba63238519f28628dac946fef8a0e22988ff58ec05580"},
|
||||
{file = "discord.py-1.3.3.tar.gz", hash = "sha256:ad00e34c72d2faa8db2157b651d05f3c415d7d05078e7e41dc9e8dc240051beb"},
|
||||
]
|
||||
docutils = [
|
||||
{file = "docutils-0.16-py2.py3-none-any.whl", hash = "sha256:0c5b78adfbf7762415433f5515cd5c9e762339e23369dbe8000d84a4bf4ab3af"},
|
||||
{file = "docutils-0.16.tar.gz", hash = "sha256:c2de3a60e9e7d07be26b7f2b00ca0309c207e06c100f9cc2a94931fc75a478fc"},
|
||||
|
|
@ -1361,8 +1360,8 @@ pyrsistent = [
|
|||
{file = "pyrsistent-0.16.0.tar.gz", hash = "sha256:28669905fe725965daa16184933676547c5bb40a5153055a8dee2a4bd7933ad3"},
|
||||
]
|
||||
pytest = [
|
||||
{file = "pytest-5.4.2-py3-none-any.whl", hash = "sha256:95c710d0a72d91c13fae35dce195633c929c3792f54125919847fdcdf7caa0d3"},
|
||||
{file = "pytest-5.4.2.tar.gz", hash = "sha256:eb2b5e935f6a019317e455b6da83dd8650ac9ffd2ee73a7b657a30873d67a698"},
|
||||
{file = "pytest-5.4.3-py3-none-any.whl", hash = "sha256:5c0db86b698e8f170ba4582a492248919255fcd4c79b1ee64ace34301fb589a1"},
|
||||
{file = "pytest-5.4.3.tar.gz", hash = "sha256:7979331bfcba207414f5e1263b5a0f8f521d0f457318836a7355531ed1a4c7d8"},
|
||||
]
|
||||
python-dateutil = [
|
||||
{file = "python-dateutil-2.8.1.tar.gz", hash = "sha256:73ebfe9dbf22e832286dafa60473e4cd239f8592f699aa5adaf10050e6e1823c"},
|
||||
|
|
@ -1526,8 +1525,8 @@ uvloop = [
|
|||
{file = "uvloop-0.14.0.tar.gz", hash = "sha256:123ac9c0c7dd71464f58f1b4ee0bbd81285d96cdda8bc3519281b8973e3a461e"},
|
||||
]
|
||||
wcwidth = [
|
||||
{file = "wcwidth-0.1.9-py2.py3-none-any.whl", hash = "sha256:cafe2186b3c009a04067022ce1dcd79cb38d8d65ee4f4791b8888d6599d1bbe1"},
|
||||
{file = "wcwidth-0.1.9.tar.gz", hash = "sha256:ee73862862a156bf77ff92b09034fc4825dd3af9cf81bc5b360668d425f3c5f1"},
|
||||
{file = "wcwidth-0.2.3-py2.py3-none-any.whl", hash = "sha256:980fbf4f3c196c0f329cdcd1e84c554d6a211f18e252e525a0cf4223154a41d6"},
|
||||
{file = "wcwidth-0.2.3.tar.gz", hash = "sha256:edbc2b718b4db6cdf393eefe3a420183947d6aa312505ce6754516f458ff8830"},
|
||||
]
|
||||
websockets = [
|
||||
{file = "websockets-8.1-cp36-cp36m-macosx_10_6_intel.whl", hash = "sha256:3762791ab8b38948f0c4d281c8b2ddfa99b7e510e46bd8dfa942a5fff621068c"},
|
||||
|
|
@ -1573,6 +1572,6 @@ yarl = [
|
|||
{file = "yarl-1.4.2.tar.gz", hash = "sha256:58cd9c469eced558cd81aa3f484b2924e8897049e06889e8ff2510435b7ef74b"},
|
||||
]
|
||||
youtube-dl = [
|
||||
{file = "youtube_dl-2020.5.29-py2.py3-none-any.whl", hash = "sha256:2d45840772ecc57e151b0be78dd89e9772b6aa29295746be38abb9c30dad5bb3"},
|
||||
{file = "youtube_dl-2020.5.29.tar.gz", hash = "sha256:1a3d84afa851dce2fccc2dfc0f9ffa0e22314ffba6d528b34b4a7fe3e0cf2264"},
|
||||
{file = "youtube_dl-2020.6.6-py2.py3-none-any.whl", hash = "sha256:813310fb7ab265c379e9aebb2bd6cde6995f9a8f22a06d2ff81c05dfab7da864"},
|
||||
{file = "youtube_dl-2020.6.6.tar.gz", hash = "sha256:74e6cc7395060fc39f0b8e21c1e4707486da904c96145bd875187bda2da83b04"},
|
||||
]
|
||||
|
|
|
|||
|
|
@ -2,10 +2,11 @@ import asyncio as aio
|
|||
import logging
|
||||
import warnings
|
||||
import io
|
||||
import sys
|
||||
from typing import *
|
||||
import royalnet.backpack.tables as rbt
|
||||
import royalnet.commands as rc
|
||||
from royalnet.utils import asyncify
|
||||
from royalnet.utils import asyncify, sentry_exc
|
||||
from royalnet.serf import Serf
|
||||
from .escape import escape
|
||||
from .voiceplayer import VoicePlayer
|
||||
|
|
@ -46,7 +47,7 @@ class DiscordSerf(Serf):
|
|||
self.Client = self.client_factory()
|
||||
"""The custom :class:`discord.Client` class that will be instantiated later."""
|
||||
|
||||
self.client = self.Client()
|
||||
self.client = self.Client(status=discord.Status.do_not_disturb)
|
||||
"""The custom :class:`discord.Client` instance."""
|
||||
|
||||
self.voice_players: List[VoicePlayer] = []
|
||||
|
|
@ -140,92 +141,37 @@ class DiscordSerf(Serf):
|
|||
"""Create a custom class inheriting from :py:class:`discord.Client`."""
|
||||
# noinspection PyMethodParameters
|
||||
class DiscordClient(discord.Client):
|
||||
async def on_message(cli, message: "discord.Message"):
|
||||
async def on_message(cli, message: "discord.Message") -> None:
|
||||
"""Handle messages received by passing them to the handle_message method of the bot."""
|
||||
# TODO: keep reference to these tasks somewhere
|
||||
self.loop.create_task(self.handle_message(message))
|
||||
|
||||
async def on_ready(cli) -> None:
|
||||
"""Change the bot presence to ``online`` when the bot is ready."""
|
||||
await cli.change_presence(status=discord.Status.online)
|
||||
log.debug("Discord client is ready!")
|
||||
await cli.change_presence(status=discord.Status.online, activity=None)
|
||||
|
||||
async def on_resume(cli) -> None:
|
||||
log.debug("Discord client resumed connection.")
|
||||
|
||||
async def on_error(self, event_method, *args, **kwargs):
|
||||
exc_type, exc_obj, exc_tb = sys.exc_info()
|
||||
sentry_exc(exc_obj)
|
||||
|
||||
return DiscordClient
|
||||
|
||||
async def run(self):
|
||||
await super().run()
|
||||
await self.client.login(self.token)
|
||||
await self.client.connect()
|
||||
|
||||
def find_channel(self,
|
||||
channel_type: Optional[Type["discord.abc.GuildChannel"]] = None,
|
||||
name: Optional[str] = None,
|
||||
guild: Optional["discord.Guild"] = None,
|
||||
accessible_to: List["discord.User"] = None,
|
||||
required_permissions: List[str] = None) -> Optional["discord.abc.GuildChannel"]:
|
||||
"""Find the best channel matching all requests.
|
||||
|
||||
In case multiple channels match all requests, return the one with the most members connected.
|
||||
|
||||
Args:
|
||||
channel_type: Filter channels by type (select only :class:`discord.VoiceChannel`,
|
||||
:class:`discord.TextChannel`, ...).
|
||||
name: Filter channels by name starting with ``name`` (using :meth:`str.startswith`).
|
||||
Note that some channel types don't have names; this check will be skipped for them.
|
||||
guild: Filter channels by guild, keep only channels inside this one.
|
||||
accessible_to: Filter channels by permissions, keeping only channels where *all* these users have
|
||||
the required permissions.
|
||||
required_permissions: Filter channels by permissions, keeping only channels where the users have *all* these
|
||||
:class:`discord.Permissions`.
|
||||
|
||||
Returns:
|
||||
Either a :class:`~discord.abc.GuildChannel`, or :const:`None` if no channels were found."""
|
||||
warnings.warn("This function will be removed soon.", category=DeprecationWarning)
|
||||
if accessible_to is None:
|
||||
accessible_to = []
|
||||
if required_permissions is None:
|
||||
required_permissions = []
|
||||
channels: List[discord.abc.GuildChannel] = []
|
||||
for ch in self.client.get_all_channels():
|
||||
if channel_type is not None and not isinstance(ch, channel_type):
|
||||
continue
|
||||
|
||||
if name is not None:
|
||||
while True:
|
||||
try:
|
||||
ch_name: str = ch.name
|
||||
if not ch_name.startswith(name):
|
||||
continue
|
||||
except AttributeError:
|
||||
pass
|
||||
|
||||
ch_guild: "discord.Guild" = ch.guild
|
||||
if guild is not None and guild != ch_guild:
|
||||
continue
|
||||
|
||||
for user in accessible_to:
|
||||
member: "discord.Member" = ch.guild.get_member(user.id)
|
||||
if member is None:
|
||||
continue
|
||||
permissions: "discord.Permissions" = ch.permissions_for(member)
|
||||
missing_perms = False
|
||||
for permission in required_permissions:
|
||||
if not permissions.__getattribute__(permission):
|
||||
missing_perms = True
|
||||
break
|
||||
if missing_perms:
|
||||
continue
|
||||
|
||||
channels.append(ch)
|
||||
|
||||
if len(channels) == 0:
|
||||
return None
|
||||
else:
|
||||
# Give priority to channels with the most people
|
||||
def people_count(c: "discord.VoiceChannel"):
|
||||
return len(c.members)
|
||||
|
||||
channels.sort(key=people_count, reverse=True)
|
||||
|
||||
return channels[0]
|
||||
await self.client.connect(reconnect=False)
|
||||
except discord.GatewayNotFound:
|
||||
log.error("Discord Gateway not found! Retrying in 60 seconds...")
|
||||
await aio.sleep(60)
|
||||
except discord.ConnectionClosed:
|
||||
log.error("Discord connection was closed! Retrying in 15 seconds...")
|
||||
await aio.sleep(60)
|
||||
|
||||
def find_voice_players(self, guild: "discord.Guild") -> List[VoicePlayer]:
|
||||
candidate_players: List[VoicePlayer] = []
|
||||
|
|
|
|||
Loading…
Reference in a new issue