from peewee import TextField, IntegerField, DateTimeField
import datetime
import random

from base import BaseModel


class User(BaseModel):

    username = TextField()
    display_name = TextField()
    discord_id = TextField()
    level = IntegerField()
    exp = IntegerField()
    currency = IntegerField()
    message_count = IntegerField()
    joined_date = 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