mirror of
https://github.com/jumpserver/jumpserver.git
synced 2025-09-07 10:20:57 +00:00
perf(project): 优化命名的风格 (#5693)
perf: 修改错误的地 perf: 优化写错的几个 Co-authored-by: ibuler <ibuler@qq.com>
This commit is contained in:
@@ -21,11 +21,11 @@ class UserGroupGenerator(FakeDataGenerator):
|
||||
class UserGenerator(FakeDataGenerator):
|
||||
resource = 'user'
|
||||
roles: list
|
||||
groups_id: list
|
||||
group_ids: list
|
||||
|
||||
def pre_generate(self):
|
||||
self.roles = list(dict(User.ROLE.choices).keys())
|
||||
self.groups_id = list(UserGroup.objects.all().values_list('id', flat=True))
|
||||
self.group_ids = list(UserGroup.objects.all().values_list('id', flat=True))
|
||||
|
||||
def set_org(self, users):
|
||||
relations = []
|
||||
@@ -39,7 +39,7 @@ class UserGenerator(FakeDataGenerator):
|
||||
def set_groups(self, users):
|
||||
relations = []
|
||||
for i in users:
|
||||
groups_to_join = sample(self.groups_id, 3)
|
||||
groups_to_join = sample(self.group_ids, 3)
|
||||
_relations = [User.groups.through(user_id=i.id, usergroup_id=gid) for gid in groups_to_join]
|
||||
relations.extend(_relations)
|
||||
User.groups.through.objects.bulk_create(relations, ignore_conflicts=True)
|
||||
|
Reference in New Issue
Block a user