mirror of https://github.com/jumpserver/jumpserver
36 lines
891 B
Python
36 lines
891 B
Python
#!/usr/bin/env python
|
|
# -*- coding: utf-8 -*-
|
|
#
|
|
|
|
from django.core.exceptions import ValidationError
|
|
from common.utils import validate_ssh_private_key
|
|
|
|
|
|
__all__ = ['init_model', 'generate_fake']
|
|
|
|
|
|
def init_model():
|
|
from . import Cluster, SystemUser, AdminUser, AssetGroup, Asset
|
|
for cls in [Cluster, SystemUser, AdminUser, AssetGroup, Asset]:
|
|
if hasattr(cls, 'initial'):
|
|
cls.initial()
|
|
|
|
|
|
def generate_fake():
|
|
from . import Cluster, SystemUser, AdminUser, AssetGroup, Asset
|
|
for cls in [Cluster, SystemUser, AdminUser, AssetGroup, Asset]:
|
|
if hasattr(cls, 'generate_fake'):
|
|
cls.generate_fake()
|
|
|
|
|
|
def private_key_validator(value):
|
|
if not validate_ssh_private_key(value):
|
|
raise ValidationError(
|
|
_('%(value)s is not an even number'),
|
|
params={'value': value},
|
|
)
|
|
|
|
|
|
if __name__ == '__main__':
|
|
pass
|