mirror of
https://github.com/wassname/Open-Assistant.git
synced 2026-08-05 12:40:38 +08:00
Yannic Kilcher
6452bb860d
added playbook to deploy dev machine added playbook to deploy dev machine added next.js font module, updated fonts, updated login page replaced logos, changed logo on login and header added 404 and email confirmation pages, changed discord and github buttons to show conditionally also removed unused imports, fixed one spelling error, and minor styling changes Quick format to the authenticated user page, updated header with user profile, styling updates fixed html encoding added checkout for release re-vamped release config and ports
89 lines
2.4 KiB
Python
89 lines
2.4 KiB
Python
# -*- coding: utf-8 -*-
|
|
import asyncio
|
|
from abc import ABC, abstractmethod
|
|
from datetime import datetime
|
|
|
|
import discord
|
|
from loguru import logger
|
|
|
|
|
|
class ChannelExpiredException(Exception):
|
|
pass
|
|
|
|
|
|
class ChannelHandlerBase(ABC):
|
|
queue: asyncio.Queue
|
|
completed: bool = False
|
|
expiry_date: datetime
|
|
expired: bool = False
|
|
|
|
def __init__(self, *, expiry_date: datetime = None):
|
|
self.expiry_date = expiry_date
|
|
self.queue = asyncio.Queue()
|
|
|
|
async def read(self) -> discord.Message:
|
|
"""Call this method to read the next message from the user in the handler method."""
|
|
if self.expired:
|
|
raise ChannelExpiredException()
|
|
|
|
msg = await self.queue.get()
|
|
if msg is None:
|
|
if self.expired:
|
|
raise ChannelExpiredException()
|
|
else:
|
|
raise RuntimeError("Unexpected None message read")
|
|
return msg
|
|
|
|
def on_reply(self, message: discord.Message) -> None:
|
|
self.queue.put_nowait(message)
|
|
|
|
def on_expire(self) -> None:
|
|
logger.info("ChannelHandler: on_expire")
|
|
self.expired = True
|
|
self.queue.put_nowait(None)
|
|
|
|
def on_completed(self) -> None:
|
|
logger.info("ChannelHandler: on_completed")
|
|
self.completed = True
|
|
|
|
def tick(self, now: datetime):
|
|
if now > self.expiry_date and not self.expired:
|
|
self.on_expire()
|
|
|
|
@abstractmethod
|
|
async def handler_loop(self):
|
|
...
|
|
|
|
async def finalize(self):
|
|
pass
|
|
|
|
|
|
class AutoDestructThreadHandler(ChannelHandlerBase):
|
|
first_message: discord.Message = None
|
|
thread: discord.Thread = None
|
|
|
|
def __init__(self, *, expiry_date: datetime = None):
|
|
super().__init__(expiry_date=expiry_date)
|
|
|
|
async def read(self) -> discord.Message:
|
|
try:
|
|
return await super().read()
|
|
except ChannelExpiredException:
|
|
await self.cleanup()
|
|
raise
|
|
|
|
async def cleanup(self):
|
|
logger.debug("AutoDestructThreadHandler.cleanup")
|
|
if self.thread:
|
|
logger.debug(f"deleting thread: {self.thread.name}")
|
|
await self.thread.delete()
|
|
self.thread = None
|
|
if self.first_message:
|
|
logger.debug(f"deleting first_message: {self.first_message.content}")
|
|
await self.first_message.delete()
|
|
self.first_message = None
|
|
|
|
async def finalize(self):
|
|
await self.cleanup()
|
|
return await super().finalize()
|