from mongoengine import * from werkzeug.security import generate_password_hash from string import ascii_letters, digits from random import choices class Union(Document): name = StringField() logo = ImageField(thumbnail_size=(120, 120)) description = StringField() legal_registration_number = StringField() headquarter = StringField() email = StringField() 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.gov_id = 'root' user.username = "{}@root".format(self.legal_registration_number) user.user_group = group user.union = self password = ''.join(choices(ascii_letters+digits, k=10)) print("New union password : {}".format(password)) user.password = generate_password_hash(password) # TODO: send password via mail or sms or nothing user.save()