mirror of https://github.com/jumpserver/jumpserver
33 lines
949 B
Python
33 lines
949 B
Python
import random
|
|
|
|
import forgery_py
|
|
|
|
from accounts.models import Account
|
|
from assets.models import Asset
|
|
from .base import FakeDataGenerator
|
|
|
|
|
|
class AccountGenerator(FakeDataGenerator):
|
|
resource = 'account'
|
|
|
|
def __init__(self, *args, **kwargs):
|
|
super().__init__(*args, **kwargs)
|
|
self.assets = list(list(Asset.objects.all()[:5000]))
|
|
|
|
def do_generate(self, batch, batch_size):
|
|
accounts = []
|
|
for i in batch:
|
|
asset = random.choice(self.assets)
|
|
name = forgery_py.internet.user_name(True) + '-' + str(i)
|
|
d = {
|
|
'username': name,
|
|
'name': name,
|
|
'asset': asset,
|
|
'secret': name,
|
|
'secret_type': 'password',
|
|
'is_active': True,
|
|
'privileged': False,
|
|
}
|
|
accounts.append(Account(**d))
|
|
Account.objects.bulk_create(accounts, ignore_conflicts=True)
|