add a clamp for values that can get really high

kaansoral · Feb 21, 2024 · 9250c3e · 9250c3e
1 parent 1e43904
commit 9250c3e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/model_validators.py b/model_validators.py
@@ -0,0 +1,4 @@
+import sys
+
+def clamp64BitInteger(prop, value):
+    return max(0, min(sys.maxsize, value))
diff --git a/models.py b/models.py
@@ -1,5 +1,6 @@
 from config import *
 from functions import *
+from model_validators import clamp64BitInteger
 
 class User(ndb.Expando):
 	name=ndb.StringProperty()
@@ -51,7 +52,7 @@ class Character(ndb.Expando):
 	type=ndb.StringProperty()
 	level=ndb.IntegerProperty(default=1)
 	worth=ndb.IntegerProperty(default=0) #total gold worth
-	xp=ndb.IntegerProperty(default=0)
+	xp=ndb.IntegerProperty(default=0, validator=clamp64BitInteger)
 	owner=ndb.StringProperty()
 	referrer=ndb.StringProperty()
 	created=ndb.DateTimeProperty(auto_now_add=True)