Update run.py

Am-Delta · Nov 18, 2023 · e5cfce1 · e5cfce1
1 parent 1b1c2c4
commit e5cfce1
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/run.py b/run.py
@@ -191,6 +191,15 @@ def db_update():
         cur.execute("UPDATE Settings SET settings = ? WHERE ID =?", (str(settings), 1))
         conn.commit()
 
+    if settings.get("invitation_limit", None) is None:
+        add_dict = {
+            "invitation_limit": 10,
+            "buy_only_customers": "off"
+        }
+        settings.update(add_dict)
+        cur.execute("UPDATE Settings SET settings = ? WHERE ID =?", (str(settings), 1))
+        conn.commit()
+
     try:
         cur.execute("SELECT * FROM Redeem")
         records = cur.fetchall()