jumpserver/apps/users/models/utils.py

#!/usr/bin/env python
# -*- coding: utf-8 -*-
# 

from . import User, UserGroup


def init_model():
    for cls in [User, UserGroup]:
        if getattr(cls, 'initial'):
            cls.initial()


def generate_fake():
    for cls in [User, UserGroup]:
        if getattr(cls, 'generate_fake'):
            cls.generate_fake()