github/jumpserver
1
0
Fork 0
mirror of https://github.com/jumpserver/jumpserver.git synced 2025-09-10 11:49:10 +00:00
Code Issues Projects Releases Wiki Activity

perf: 添加生成 platform

Browse Source
This commit is contained in:
ibuler
2022-08-22 15:23:28 +08:00
parent dd0b8e988c
commit f0c9c2b1ad
2 changed files with 6 additions and 2 deletions
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)