From 244c476e2a717aede76d6bf036975df8c7d31309 Mon Sep 17 00:00:00 2001
From: zjf <zhangjf@probim.com.cn>
Date: Mon, 06 Mar 2023 16:45:51 +0800
Subject: [PATCH] 企业编码更改逻辑

---
 IphoneBIMe/IphoneBIMe/Classes/Login/ViewControllers/PBLoginViewController.m |   17 +++++++++++++----
 1 files changed, 13 insertions(+), 4 deletions(-)

diff --git a/IphoneBIMe/IphoneBIMe/Classes/Login/ViewControllers/PBLoginViewController.m b/IphoneBIMe/IphoneBIMe/Classes/Login/ViewControllers/PBLoginViewController.m
index c890ab9..7d600f1 100644
--- a/IphoneBIMe/IphoneBIMe/Classes/Login/ViewControllers/PBLoginViewController.m
+++ b/IphoneBIMe/IphoneBIMe/Classes/Login/ViewControllers/PBLoginViewController.m
@@ -267,15 +267,24 @@
 #pragma mark - 编辑企业码
 - (void)codeBtnAction {
     PBEnterpriseCodeController *codeVC = [[PBEnterpriseCodeController alloc] init];
-    codeVC.SetCodeBlock = ^(NSDictionary * _Nonnull obj) {
+    codeVC.SetCodeBlock = ^(NSString * _Nonnull code, NSDictionary * _Nonnull obj) {
         self.customData = obj;
         NSMutableArray *arrM = [[NSMutableArray alloc] init];
         NSArray *arr = [NSString getApiArrData];
+        BOOL isHave = false;
         for (NSDictionary *dict in arr) {
-            [dict setValue:@"0" forKey:@"Check"];
-            [arrM addObject:dict];
+            NSString *haveCode = [dict valueForKey:@"Code"];
+            if ([code isEqualToString:haveCode]) {
+                [arrM addObject:obj];
+                isHave = true;
+            } else {
+                [dict setValue:@"0" forKey:@"Check"];
+                [arrM addObject:dict];
+            }
         }
-        [arrM addObject:self.customData];
+        if (!isHave) {
+            [arrM addObject:self.customData];
+        }
         [self saveApiArr:arrM.copy];
         [self storedLocallyApiWithData:self.customData];
         [self setApi];

--
Gitblit v1.9.3