mirror of
https://github.com/TitanEmbeds/Titan.git
synced 2024-12-25 06:27:03 +01:00
Update and delete messages even if not in discordbot messages cache, and raise max_messages deque
This commit is contained in:
parent
5b578b01d1
commit
0cfcb42ac4
@ -7,13 +7,14 @@ import aiohttp
|
|||||||
import asyncio
|
import asyncio
|
||||||
import sys
|
import sys
|
||||||
import logging
|
import logging
|
||||||
|
import json
|
||||||
logging.basicConfig(filename='titanbot.log',level=logging.INFO,format='%(asctime)s %(message)s', datefmt='%m/%d/%Y %I:%M:%S %p')
|
logging.basicConfig(filename='titanbot.log',level=logging.INFO,format='%(asctime)s %(message)s', datefmt='%m/%d/%Y %I:%M:%S %p')
|
||||||
logging.getLogger('TitanBot')
|
logging.getLogger('TitanBot')
|
||||||
logging.getLogger('sqlalchemy')
|
logging.getLogger('sqlalchemy')
|
||||||
|
|
||||||
class Titan(discord.Client):
|
class Titan(discord.Client):
|
||||||
def __init__(self):
|
def __init__(self):
|
||||||
super().__init__()
|
super().__init__(max_messages=20000)
|
||||||
self.aiosession = aiohttp.ClientSession(loop=self.loop)
|
self.aiosession = aiohttp.ClientSession(loop=self.loop)
|
||||||
self.http.user_agent += ' TitanEmbeds-Bot'
|
self.http.user_agent += ' TitanEmbeds-Bot'
|
||||||
self.database = DatabaseInterface(self)
|
self.database = DatabaseInterface(self)
|
||||||
@ -244,3 +245,35 @@ class Titan(discord.Client):
|
|||||||
async def on_webhooks_update(self, server):
|
async def on_webhooks_update(self, server):
|
||||||
await self.wait_until_dbonline()
|
await self.wait_until_dbonline()
|
||||||
await self.database.update_guild(server)
|
await self.database.update_guild(server)
|
||||||
|
|
||||||
|
async def on_socket_raw_receive(self, msg):
|
||||||
|
if type(msg) is not str:
|
||||||
|
return
|
||||||
|
msg = json.loads(msg)
|
||||||
|
if msg["op"] != 0:
|
||||||
|
return
|
||||||
|
action = msg["t"]
|
||||||
|
await asyncio.sleep(1)
|
||||||
|
if action == "MESSAGE_UPDATE":
|
||||||
|
if not self.in_messages_cache(msg["d"]["id"]):
|
||||||
|
channel = self.get_channel(msg["d"]["channel_id"])
|
||||||
|
message = await self.get_message(channel, msg["d"]["id"])
|
||||||
|
await self.on_message_edit(None, message)
|
||||||
|
if action == "MESSAGE_DELETE":
|
||||||
|
if not self.in_messages_cache(msg["d"]["id"]):
|
||||||
|
await self.process_raw_message_delete(msg["d"]["id"], msg["d"]["channel_id"])
|
||||||
|
if action == "MESSAGE_DELETE_BULK":
|
||||||
|
for msgid in msg["d"]["ids"]:
|
||||||
|
if not self.in_messages_cache(msgid):
|
||||||
|
await self.process_raw_message_delete(msgid, msg["d"]["channel_id"])
|
||||||
|
|
||||||
|
async def process_raw_message_delete(self, msg_id, channel_id):
|
||||||
|
channel = self.get_channel(channel_id)
|
||||||
|
msg = discord.Message(channel=channel, reactions=[], id=msg_id, type=0, timestamp="2017-01-15T02:59:58", content="What fun is there in making sense?") # Procreate a fake message object
|
||||||
|
await self.on_message_delete(msg)
|
||||||
|
|
||||||
|
def in_messages_cache(self, msg_id):
|
||||||
|
for msg in self.messages:
|
||||||
|
if msg.id == msg_id:
|
||||||
|
return True
|
||||||
|
return False
|
Loading…
Reference in New Issue
Block a user