refactor: cr调整

TencentBlueKing · Nov 9, 2023 · 383f286 · 383f286
1 parent 0a2aac7
commit 383f286
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/bk-user/bkuser/apis/web/tenant_setting/views.py b/src/bk-user/bkuser/apis/web/tenant_setting/views.py
@@ -28,10 +28,13 @@
     TenantUserValidityPeriodConfig,
     UserBuiltinField,
 )
+from bkuser.apps.tenant.periodic_tasks import send_tenant_user_expiring_notification
 from bkuser.biz.tenant_setting import NotificationTemplate
 from bkuser.common.error_codes import error_codes
 from bkuser.common.views import ExcludePatchAPIViewMixin, ExcludePutAPIViewMixin
 
+send_tenant_user_expiring_notification()
+
 
 class TenantUserFieldListApi(CurrentUserTenantMixin, generics.ListAPIView):
     pagination_class = None
@@ -148,6 +151,9 @@ def get(self, request, *args, **kwargs):
     def put(self, request, *args, **kwargs):
         instance = self.get_queryset().first()
 
+        if not instance:
+            raise error_codes.OBJECT_NOT_FOUND.f(_("账户有效期配置丢失，请联系系统管理员"))
+
         # 边界限制: 当前租户的管理才可做更新操作
         operator = request.user.username
         if not TenantManager.objects.filter(tenant_id=instance.tenant_id, tenant_user_id=operator).exists():