feat: 增加业务信息表 (closed #378)

Author: noahchi

apps/gsekit/migrations/0016_alter_job_bk_app_code.py

@@ -0,0 +1,18 @@
+# Generated by Django 3.2.15 on 2025-04-14 04:05
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("gsekit", "0015_auto_20230714_1639"),
+    ]
+
+    operations = [
+        migrations.AlterField(
+            model_name="job",
+            name="bk_app_code",
+            field=models.CharField(default="gsekit2", max_length=32, verbose_name="蓝鲸应用ID"),
+        ),
+    ]

apps/gsekit/migrations/0017_business.py

@@ -0,0 +1,27 @@
+# Generated by Django 3.2.15 on 2025-04-14 04:09
+
+from django.db import migrations, models
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("gsekit", "0016_alter_job_bk_app_code"),
+    ]
+
+    operations = [
+        migrations.CreateModel(
+            name="Business",
+            fields=[
+                ("id", models.BigAutoField(auto_created=True, primary_key=True, serialize=False, verbose_name="ID")),
+                ("bk_biz_id", models.IntegerField(verbose_name="业务 ID")),
+                ("bk_biz_name", models.CharField(max_length=255, verbose_name="业务名称")),
+                ("created_at", models.DateTimeField(auto_now_add=True)),
+                ("updated_at", models.DateTimeField(auto_now=True)),
+            ],
+            options={
+                "verbose_name": "业务信息",
+                "verbose_name_plural": "业务信息",
+            },
+        ),
+    ]

apps/gsekit/periodic_tasks/sync_biz_info.py

@@ -0,0 +1,31 @@
+# -*- coding: utf-8 -*-
+"""
+Tencent is pleased to support the open source community by making 蓝鲸 (Blueking) available.
+Copyright (C) 2017-2021 THL A29 Limited, a Tencent company. All rights reserved.
+Licensed under the MIT License (the "License"); you may not use this file except in compliance with the License.
+You may obtain a copy of the License at https://opensource.org/licenses/MIT
+Unless required by applicable law or agreed to in writing, software distributed under the License is distributed on
+an "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and limitations under the License.
+"""
+from celery.task import periodic_task
+from celery.schedules import crontab
+from apps.api import CCApi
+
+from blueapps.utils.logger import logger
+
+from apps.gsekit.process.models import Business
+
+
+@periodic_task(run_every=crontab(hour=0, minute=0))
+def sync_business():
+    all_biz_list = CCApi.search_business({"fields": ["bk_biz_id", "bk_biz_name"]}, use_admin=True).get("info") or []
+
+    for item in all_biz_list:
+        bk_biz_id = item.get("bk_biz_id")
+        bk_biz_name = item.get("bk_biz_name")
+        try:
+            # 假设 bk_biz_id 是唯一标识，根据它来查找记录
+            Business.objects.update_or_create(bk_biz_id=bk_biz_id, defaults={"bk_biz_name": bk_biz_name})
+        except Exception as e:
+            logger.info(f"处理数据 {bk_biz_id} 时出现错误: {e}")

apps/gsekit/process/models.py

@@ -140,3 +140,19 @@ class Meta:
         ]
         verbose_name = _("进程实例（ProcessInst）")
         verbose_name_plural = _("进程实例（ProcessInst）")
+
+
+class Business(models.Model):
+    bk_biz_id = models.IntegerField(verbose_name="业务 ID")
+    bk_biz_name = models.CharField(max_length=255, verbose_name="业务名称")
+    # 记录对象的创建时间，在对象首次创建时自动设置
+    created_at = models.DateTimeField(auto_now_add=True)
+    # 记录对象的最后修改时间，每次对象保存时自动更新
+    updated_at = models.DateTimeField(auto_now=True)
+
+    def __str__(self):
+        return self.bk_biz_name
+
+    class Meta:
+        verbose_name = "业务信息"
+        verbose_name_plural = verbose_name