mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-09-13 22:09:17 +00:00
perf: remove system user
This commit is contained in:
@@ -5,45 +5,6 @@ import forgery_py
|
||||
from .base import FakeDataGenerator
|
||||
|
||||
from assets.models import *
|
||||
from assets.const import Protocol
|
||||
|
||||
|
||||
class AdminUsersGenerator(FakeDataGenerator):
|
||||
resource = 'admin_user'
|
||||
|
||||
def do_generate(self, batch, batch_size):
|
||||
admin_users = []
|
||||
for i in batch:
|
||||
username = forgery_py.internet.user_name(True)
|
||||
password = forgery_py.basic.password()
|
||||
admin_users.append(AdminUser(
|
||||
name=username.title(),
|
||||
username=username,
|
||||
password=password,
|
||||
org_id=self.org.id,
|
||||
created_by='Fake',
|
||||
))
|
||||
AdminUser.objects.bulk_create(admin_users, ignore_conflicts=True)
|
||||
|
||||
|
||||
class SystemUsersGenerator(FakeDataGenerator):
|
||||
def do_generate(self, batch, batch_size):
|
||||
system_users = []
|
||||
protocols = list(dict(Protocol.choices).keys())
|
||||
for i in batch:
|
||||
username = forgery_py.internet.user_name(True)
|
||||
protocol = random.choice(protocols)
|
||||
name = username.title()
|
||||
name = f'{name}-{protocol}'
|
||||
system_users.append(SystemUser(
|
||||
name=name,
|
||||
username=username,
|
||||
password=forgery_py.basic.password(),
|
||||
protocol=protocol,
|
||||
org_id=self.org.id,
|
||||
created_by='Fake',
|
||||
))
|
||||
SystemUser.objects.bulk_create(system_users, ignore_conflicts=True)
|
||||
|
||||
|
||||
class NodesGenerator(FakeDataGenerator):
|
||||
@@ -62,7 +23,6 @@ class AssetsGenerator(FakeDataGenerator):
|
||||
node_ids: list
|
||||
|
||||
def pre_generate(self):
|
||||
self.admin_user_ids = list(AdminUser.objects.all().values_list('id', flat=True))
|
||||
self.node_ids = list(Node.objects.all().values_list('id', flat=True))
|
||||
|
||||
def set_assets_nodes(self, assets):
|
||||
|
Reference in New Issue
Block a user