perf: 修改生成假数据

This commit is contained in:
ibuler
2022-05-30 16:01:57 +08:00
committed by 老广
parent 15423291cc
commit 196663f205
4 changed files with 48 additions and 7 deletions

View File

@@ -24,7 +24,6 @@ class UserGenerator(FakeDataGenerator):
group_ids: list
def pre_generate(self):
self.roles = list(dict(User.ROLE.choices).keys())
self.group_ids = list(UserGroup.objects.all().values_list('id', flat=True))
def set_org(self, users):
@@ -53,7 +52,6 @@ class UserGenerator(FakeDataGenerator):
username=username,
email=email,
name=username.title(),
role=choice(self.roles),
created_by='Faker'
)
users.append(u)