Add commands

1 files changed, 2 insertions, 6 deletions

diff --git a/models/user.py b/models/user.py
index dcf35ab..c38135d 100644
--- a/models/user.py
+++ b/models/user.py
@@ -1,6 +1,5 @@
 from peewee import TextField, IntegerField, DateTimeField
 import datetime
-import random
 
 from base import BaseModel
 
@@ -15,14 +14,11 @@ class User(BaseModel):
     currency = IntegerField()
     message_count = IntegerField()
     joined_date = DateTimeField(default=datetime.datetime.now)
+    exp_gained_time = DateTimeField(default=datetime.datetime.now)
+    daily_gained_time = DateTimeField(default=datetime.datetime.now)
 
     def add_coins(self, additional_coins: int):
         self.currency = self.currency + additional_coins
 
     def remove_coins(self, lost_coins: int):
         self.currency = self.currency - lost_coins
-
-    def add_message(self):
-        self.message_count += 1
-        gained_exp = random.randrange(10, 15)
-        self.exp += gained_exp