class Category(db.Model):
"""Database model representing a Discord category channel."""
__tablename__ = "categories"
id = db.Column(db.String, primary_key=True)
name = db.Column(db.String, nullable=False)
class Channel(db.Model):
"""Database model representing a Discord channel."""
__tablename__ = "channels"
id = db.Column(db.String, primary_key=True)
name = db.Column(db.String, nullable=False)
category_id = db.Column(db.String,
db.ForeignKey("categories.id"),
nullable=True)
is_staff = db.Column(db.Boolean, nullable=False)
class Message(db.Model):
"""Database model representing a message sent in a Discord server."""
__tablename__ = "messages"
id = db.Column(db.String, primary_key=True)
channel_id = db.Column(
db.String,
db.ForeignKey("channels.id", ondelete="CASCADE"),
)
author_id = db.Column(db.String,
db.ForeignKey("users.id", ondelete="CASCADE"))
created_at = db.Column(db.DateTime, default=datetime.utcnow)
class Thread(db.Model):
"""Database model representing a Thread channel."""
__tablename__ = "threads"
id = db.Column(db.String, primary_key=True)
parent_channel_id = db.Column(db.String,
db.ForeignKey("channels.id",
ondelete="CASCADE"),
nullable=False)
created_at = db.Column(TZDateTime(), default=datetime.now(timezone.utc))
name = db.Column(db.String, nullable=False)
archived = db.Column(db.Boolean, default=False, nullable=False)
auto_archive_duration = db.Column(db.Integer, nullable=False)
locked = db.Column(db.Boolean, default=False, nullable=False)
type = db.Column(db.String, nullable=False, index=True)
class Message(db.Model):
"""Database model representing a message sent in a Discord server."""
__tablename__ = "messages"
id = db.Column(db.String, primary_key=True)
channel_id = db.Column(db.String,
db.ForeignKey("channels.id", ondelete="CASCADE"),
index=True)
thread_id = db.Column(db.String,
db.ForeignKey("threads.id", ondelete="CASCADE"),
index=True)
author_id = db.Column(db.String,
db.ForeignKey("users.id", ondelete="CASCADE"),
index=True)
created_at = db.Column(TZDateTime(), default=datetime.now(timezone.utc))
is_deleted = db.Column(db.Boolean, default=False)
class User(db.Model):
"""Database model representing a Discord user."""
__tablename__ = "users"
id = db.Column(db.String, primary_key=True)
name = db.Column(db.String, nullable=False)
avatar_hash = db.Column(db.String, nullable=True)
joined_at = db.Column(db.DateTime, nullable=False)
created_at = db.Column(db.DateTime, nullable=False)
is_staff = db.Column(db.Boolean, nullable=False)
opt_out = db.Column(db.Boolean, default=False)
bot = db.Column(db.Boolean, default=False)
in_guild = db.Column(db.Boolean, default=True)
is_verified = db.Column(db.Boolean, default=False)
public_flags = db.Column(db.JSON, default={})
verified_at = db.Column(db.DateTime, nullable=True)
@classmethod
def bulk_upsert(cls: type, users: List[Dict[str, Any]]) -> Any:
"""Perform a bulk insert/update of the database to sync the user table."""
qs = insert(cls.__table__).values(users)
update_cols = [
"name", "avatar_hash", "joined_at", "is_staff", "bot", "in_guild",
"is_verified", "public_flags"
]
return qs.on_conflict_do_update(
index_elements=[cls.id],
set_={k: getattr(qs.excluded, k)
for k in update_cols}).returning(cls.__table__).gino.all()