perf: 添加生成 platform

2025-09-13 22:09:17 +00:00 · 2022-08-22 15:23:28 +08:00
parent dd0b8e988c
commit f0c9c2b1ad
2 changed files with 6 additions and 2 deletions
--- a/utils/generate_fake_data/resources/assets.py
+++ b/utils/generate_fake_data/resources/assets.py
@@ -30,9 +30,12 @@ class PlatformGenerator(FakeDataGenerator):
    def do_generate(self, batch, batch_size):
        platforms = []
        for i in batch:
+            category = choice(self.categories)
+            tp = choice(self.category_type[category].choices)
            data = {
                'name': forgery_py.name.company_name(),
-                'category': ''
+                'category': choice(self.categories),
+                'type': tp
            }
            platforms.append(Platform(**data))
        Platform.objects.bulk_create(platforms, ignore_conflicts=True)