create initial user

pull/5/head
Mustafa Yontar 3 years ago
parent 58c5d37a21
commit a5e3dfbae8
  1. 17
      models/Union.py

@ -1,18 +1,31 @@
from mongoengine import * from mongoengine import *
from werkzeug.security import generate_password_hash
from string import printable
from random import choices
class Union(Document): class Union(Document):
name = StringField() name = StringField()
logo = ImageField(thumbnail_size=(120,120)) logo = ImageField(thumbnail_size=(120, 120))
description = StringField() description = StringField()
legal_registration_number = StringField() legal_registration_number = StringField()
headquarter = StringField() headquarter = StringField()
email = StringField()
def save(self,*args, **kwargs): def save(self, *args, **kwargs):
super(Union, self).save(*args, **kwargs) super(Union, self).save(*args, **kwargs)
from .Group import Group from .Group import Group
from .User import User
group = Group() group = Group()
group.union = self group.union = self
group.name = 'Root' group.name = 'Root'
group.rights = ["*::*::{}}/*".format(self.id)] group.rights = ["*::*::{}}/*".format(self.id)]
group.save() 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()

Loading…
Cancel
Save