wushaodong пре 4 година
родитељ
комит
b2b7c79ea6
1 измењених фајлова са 7 додато и 1 уклоњено
  1. 7 1
      apps/wxapp/tenant/views.py

+ 7 - 1
apps/wxapp/tenant/views.py

@@ -34,6 +34,7 @@ class GetTenantView(APIView):
         emplayee = Employee.objects.filter(user=request.user).first()
         if emplayee:
             data = {
+                'company_no':emplayee.tenant.company_no,
                 'company_name':emplayee.tenant.company_name,
                 'position':emplayee.position,
                 'user_type_text':emplayee.get_type_display(),
@@ -60,10 +61,15 @@ class TenantViewSet(CustomModelViewSet):
             is_validity = Tenant.check_validity(pk)
             if not is_validity:
                 raise CustomError('该企业认证已过期,请联系管理员续费!')
+            user = self.request.user
+            if user.type == Employee.SUPER:
+                raise CustomError('您是企业超级管理员,不能绑定其他企业!')
             with transaction.atomic():
                 user = self.request.user
+                user.type = Employee.REPAIR
+                user.save()
                 tenant = Tenant.getByNo(pk)
-                Employee.objects.filter(user=user).update(tenant=tenant)
+                Employee.objects.filter(user=user).update(tenant=tenant, type=Employee.REPAIR,position='报修人')
         except CustomError as e:
             return response_error(e.get_error_msg())
         except Exception as e: