From a5e3dfbae81712e295f4917d4499278b7a6ed8e3 Mon Sep 17 00:00:00 2001 From: Mustafa Yontar Date: Sun, 31 Jan 2021 02:14:22 +0300 Subject: [PATCH] create initial user --- models/Union.py | 17 +++++++++++++++-- 1 file changed, 15 insertions(+), 2 deletions(-) diff --git a/models/Union.py b/models/Union.py index 8101fcf..c4b7b89 100644 --- a/models/Union.py +++ b/models/Union.py @@ -1,18 +1,31 @@ from mongoengine import * +from werkzeug.security import generate_password_hash +from string import printable +from random import choices class Union(Document): name = StringField() - logo = ImageField(thumbnail_size=(120,120)) + logo = ImageField(thumbnail_size=(120, 120)) description = StringField() legal_registration_number = StringField() headquarter = StringField() + email = StringField() - def save(self,*args, **kwargs): + def save(self, *args, **kwargs): super(Union, self).save(*args, **kwargs) from .Group import Group + from .User import User group = Group() group.union = self group.name = 'Root' group.rights = ["*::*::{}}/*".format(self.id)] group.save() + user = User() + user.username = "{}@root".format(self.legal_registration_number) + user.user_group = group + user.union = self + password = choices(printable, k=10) + user.password = generate_password_hash(password) + # TODO: send mail or etc + user.save()