mirror of
https://github.com/TitanEmbeds/Titan.git
synced 2024-11-15 02:21:21 +01:00
44 lines
1.6 KiB
Python
44 lines
1.6 KiB
Python
from flask_sqlalchemy import SQLAlchemy
|
|
|
|
db = SQLAlchemy()
|
|
|
|
from .guilds import Guilds
|
|
from .unauthenticated_users import UnauthenticatedUsers
|
|
from .unauthenticated_bans import UnauthenticatedBans
|
|
from .authenticated_users import AuthenticatedUsers
|
|
from .cosmetics import Cosmetics, set_badges, get_badges, add_badge, remove_badge
|
|
from .user_css import UserCSS
|
|
from .administrators import Administrators, get_administrators_list
|
|
from .titan_tokens import TitanTokens, get_titan_token
|
|
from .token_transactions import TokenTransactions
|
|
from .patreon import Patreon
|
|
from .disabled_guilds import DisabledGuilds, list_disabled_guilds
|
|
from .discordbotsorg_transactions import DiscordBotsOrgTransactions
|
|
from .application_settings import ApplicationSettings
|
|
|
|
def set_titan_token(user_id, amt_change, action):
|
|
token_count = get_titan_token(user_id)
|
|
if token_count >= 0:
|
|
token_usr = db.session.query(TitanTokens).filter(TitanTokens.user_id == user_id).first()
|
|
else:
|
|
token_count = 0
|
|
token_usr = TitanTokens(user_id, 0)
|
|
new_token_count = token_count + amt_change
|
|
if new_token_count < 0:
|
|
return False
|
|
transact = TokenTransactions(user_id, action, amt_change, token_count, new_token_count)
|
|
db.session.add(transact)
|
|
token_usr.tokens = new_token_count
|
|
db.session.add(token_usr)
|
|
return True
|
|
|
|
def init_application_settings():
|
|
settings = db.session.query(ApplicationSettings).first()
|
|
if not settings:
|
|
settings = ApplicationSettings()
|
|
db.session.add(settings)
|
|
db.session.commit()
|
|
|
|
def get_application_settings():
|
|
return db.session.query(ApplicationSettings).first()
|