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

fix: 修复生成假数据脚本错误

pull/5693/head^2
xinwen 2021-03-05 14:29:54 +08:00 committed by 老广
parent 840e5e8863
commit 7f42e59714
2 changed files with 3 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -65,13 +65,9 @@ class AssetsGenerator(FakeDataGenerator):
self.nodes_id = list(Node.objects.all().values_list('id', flat=True)) self.nodes_id = list(Node.objects.all().values_list('id', flat=True))
def set_assets_nodes(self, assets): def set_assets_nodes(self, assets):
assets_id = [asset.id for asset in assets] for asset in assets:
objs = []
for asset_id in assets_id:
nodes_id_add_to = random.sample(self.nodes_id, 3) nodes_id_add_to = random.sample(self.nodes_id, 3)
objs_add = [Asset.nodes.through(asset_id=asset_id, node_id=nid) for nid in nodes_id_add_to] asset.nodes.add(*nodes_id_add_to)
objs.extend(objs_add)
Asset.nodes.through.objects.bulk_create(objs, ignore_conflicts=True)
def do_generate(self, batch, batch_size): def do_generate(self, batch, batch_size):
assets = [] assets = []

2
utils/generate_fake_data/resources/base.py
View File

@ -15,7 +15,7 @@ class FakeDataGenerator:
seed() seed()
def switch_org(self, org_id): def switch_org(self, org_id):
o = Organization.get_instance(org_id, default=True) o = Organization.get_instance(org_id, default=Organization.default())
if o: if o:
o.change_to() o.change_to()
print('Current org is: {}'.format(o)) print('Current org is: {}'.format(o))