25 lines
871 B
Python
25 lines
871 B
Python
from discord import Embed
|
|
|
|
from db import Base
|
|
from sqlalchemy import Column, Integer, Text, Boolean, BigInteger
|
|
|
|
|
|
class Greetings(Base):
|
|
__tablename__ = "greetings"
|
|
id = Column(Integer, primary_key=True)
|
|
join_message = Column(Text, nullable=False, default="")
|
|
join_enable = Column(Boolean, nullable=False, default=False)
|
|
leave_message = Column(Text, nullable=False, default="")
|
|
leave_enable = Column(Boolean, nullable=False, default=False)
|
|
guild = Column(BigInteger, nullable=False, unique=True)
|
|
|
|
def __init__(self, guild: int):
|
|
self.guild = guild
|
|
|
|
def join_embed(self, guild_name: str, user: str):
|
|
embed = Embed()
|
|
embed.add_field(name=f"Welcome to {guild_name} !", value=self.join_message.format(user))
|
|
return embed
|
|
|
|
def leave_msg(self, user: str):
|
|
return self.leave_message.format(user)
|