github
/
jumpserver
mirror of https://github.com/jumpserver/jumpserver
1
0
Fork 0
Code Issues Releases Wiki Activity

perf: 添加生成 platform

pull/8873/head
ibuler 2022-08-22 15:23:28 +08:00
parent dd0b8e988c
commit f0c9c2b1ad
2 changed files with 6 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
utils/generate_fake_data/generate.py
View File

@ -12,7 +12,7 @@ sys.path.insert(0, APPS_DIR)
os.environ.setdefault("DJANGO_SETTINGS_MODULE", "jumpserver.settings") os.environ.setdefault("DJANGO_SETTINGS_MODULE", "jumpserver.settings")
django.setup() django.setup()
from resources.assets import AssetsGenerator, NodesGenerator from resources.assets import AssetsGenerator, NodesGenerator, PlatformGenerator
from resources.users import UserGroupGenerator, UserGenerator from resources.users import UserGroupGenerator, UserGenerator
from resources.perms import AssetPermissionGenerator from resources.perms import AssetPermissionGenerator
from resources.terminal import CommandGenerator, SessionGenerator from resources.terminal import CommandGenerator, SessionGenerator
@ -20,6 +20,7 @@ from resources.terminal import CommandGenerator, SessionGenerator
resource_generator_mapper = { resource_generator_mapper = {
'asset': AssetsGenerator, 'asset': AssetsGenerator,
'platform': PlatformGenerator,
'node': NodesGenerator, 'node': NodesGenerator,
'user': UserGenerator, 'user': UserGenerator,
'user_group': UserGroupGenerator, 'user_group': UserGroupGenerator,

5
utils/generate_fake_data/resources/assets.py
View File

@ -30,9 +30,12 @@ class PlatformGenerator(FakeDataGenerator):
def do_generate(self, batch, batch_size): def do_generate(self, batch, batch_size):
platforms = [] platforms = []
for i in batch: for i in batch:
category = choice(self.categories)
tp = choice(self.category_type[category].choices)
data = { data = {
'name': forgery_py.name.company_name(), 'name': forgery_py.name.company_name(),
'category': '' 'category': choice(self.categories),
'type': tp
} }
platforms.append(Platform(**data)) platforms.append(Platform(**data))
Platform.objects.bulk_create(platforms, ignore_conflicts=True) Platform.objects.bulk_create(platforms, ignore_conflicts=True)