mirror of
https://github.com/TitanEmbeds/Titan.git
synced 2024-11-14 18:11:23 +01:00
27 lines
1.1 KiB
Python
27 lines
1.1 KiB
Python
from titanembeds.database import db
|
|
|
|
class Cosmetics(db.Model):
|
|
__tablename__ = "cosmetics"
|
|
id = db.Column(db.Integer, primary_key=True) # Auto increment id
|
|
user_id = db.Column(db.String(255), nullable=False) # Discord user id of user of cosmetics
|
|
css = db.Column(db.Boolean(), nullable=False) # If they can create/edit custom CSS
|
|
css_limit = db.Column(db.Integer, nullable=False, server_default="0") # Custom CSS Limit
|
|
webhook_icon = db.Column(db.Boolean(), nullable=False, server_default=db.false()) # If they can set the webhook icon for all guilds
|
|
|
|
def __init__(self, user_id, **kwargs):
|
|
self.user_id = user_id
|
|
|
|
if "css" in kwargs:
|
|
self.css = kwargs["css"]
|
|
else:
|
|
self.css = False
|
|
|
|
if "css_limit" in kwargs:
|
|
self.css_limit = kwargs["css_limit"]
|
|
else:
|
|
self.css_limit = 0
|
|
|
|
if "webhook_icon" in kwargs:
|
|
self.webhook_icon = kwargs["webhook_icon"]
|
|
else:
|
|
self.webhook_icon = False |