26 lines
858 B
Python
26 lines
858 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)
|
||
|
|
||
|
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)
|