This commit is contained in:
Son NK 2019-07-12 10:38:36 +02:00 committed by Son NK
parent c4775bd66c
commit 2fc3108141
1 changed files with 4 additions and 2 deletions

View File

@ -118,11 +118,13 @@ class User(db.Model, ModelMixin, UserMixin):
@classmethod
def create(cls, email, name, password=None, **kwargs):
user = super(User, cls).create(email=email, name=name, **kwargs)
user: User = super(User, cls).create(email=email, name=name, **kwargs)
if not password:
# set a random password
user.set_password(random_string(20))
password = random_string(20)
user.set_password(password)
# by default new user will be trial period
user.plan = PlanEnum.trial