Donation goal banner

webapp/titanembeds/database/__init__.py

@@ -14,6 +14,7 @@ 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)
@@ -29,4 +30,14 @@ def set_titan_token(user_id, amt_change, action):
     db.session.add(transact)
     token_usr.tokens = new_token_count
     db.session.add(token_usr)
-    return True
+    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()

webapp/titanembeds/database/application_settings.py

@@ -0,0 +1,13 @@
+from titanembeds.database import db
+
+class ApplicationSettings(db.Model):
+    __tablename__ = "application_settings"
+    id = db.Column(db.Integer, primary_key=True, nullable=False)                        # Auto increment id
+    donation_goal_progress = db.Column(db.Integer, nullable=False, server_default="0")  # Current progress towards donation goal
+    donation_goal_total = db.Column(db.Integer, nullable=False, server_default="0")     # Total donation required to hit goal. 0 to now show banners
+    donation_goal_end = db.Column(db.Date(), nullable=True)                         # When to end donation goal
+
+    def __init__(self):
+        self.donation_goal_progress = 0
+        self.donation_goal_total = 0
+        self.donation_goal_end = None