mirror of
https://github.com/TitanEmbeds/Titan.git
synced 2024-11-14 10:01:21 +01:00
17 lines
578 B
Python
17 lines
578 B
Python
from titanembeds.database import db
|
|
|
|
class TitanTokens(db.Model):
|
|
__tablename__ = "titan_tokens"
|
|
user_id = db.Column(db.BigInteger, nullable=False, primary_key=True) # Discord user id of user
|
|
tokens = db.Column(db.Integer, nullable=False, default=0) # Token amount
|
|
|
|
def __init__(self, user_id, tokens):
|
|
self.user_id = user_id
|
|
self.tokens = tokens
|
|
|
|
def get_titan_token(user_id):
|
|
q = db.session.query(TitanTokens).filter(TitanTokens.user_id == user_id).first()
|
|
if q:
|
|
return q.tokens
|
|
else:
|
|
return -1 |