From b219589d73f46bd98995bceb4caf16dadb56b869 Mon Sep 17 00:00:00 2001
From: Pu Zhibing <393733352@qq.com>
Date: 星期五, 12 九月 2025 20:30:13 +0800
Subject: [PATCH] 修改bug

---
 ManagementQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java |   69 +++++++++++++++++++++++-----------
 1 files changed, 46 insertions(+), 23 deletions(-)

diff --git a/ManagementQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java b/ManagementQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java
index c4a66e2..37bac06 100644
--- a/ManagementQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java
+++ b/ManagementQYTTravel/guns-admin/src/main/java/com/stylefeng/guns/modular/system/controller/general/TDriverController.java
@@ -168,7 +168,7 @@
      */
     @RequestMapping("/tDriver_add")
     public String tDriverAdd(Model model) {
-        List<TCompany> companyList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 2));
+        List<TCompany> companyList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 2).eq("state", 0).ne("flag", 3));
         model.addAttribute("companyList", companyList);
     
         Integer roleType = shiroExtUtil.getUser().getRoleType();
@@ -177,8 +177,7 @@
             List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>()
                     .eq("type", 3)
                     .eq("superiorId", shiroExtUtil.getUser().getObjectId())
-                    .notIn("state", 1)
-                    .notIn("flag", 3));
+                    .eq("state", 0).ne("flag", 3));
             model.addAttribute("franchiseeList", franchiseeList);
         } else {
             model.addAttribute("franchiseeList", null);
@@ -187,8 +186,15 @@
         model.addAttribute("objectName", tCompanyService.selectById(shiroExtUtil.getUser().getObjectId()).getName());
 
         //查询线路列表
-        List<TLine> lineList = itLineService.selectList(new EntityWrapper<TLine>().eq("state", 1));
-        model.addAttribute("lineList",lineList);
+        if(1 != roleType){
+            Integer objectId = shiroExtUtil.getUser().getObjectId();
+            List<TLine> lineList = itLineService.selectList(new EntityWrapper<TLine>().eq("state", 1).last(" and (id in (select lineId from t_line_company where companyId = " + objectId + ") or companyId = " + objectId + ")"));
+            model.addAttribute("lineList",lineList);
+        }else{
+            List<TLine> lineList = itLineService.selectList(new EntityWrapper<TLine>().eq("state", 1));
+            model.addAttribute("lineList",lineList);
+        }
+
         List<TRegion> provinceList = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", 0));
         model.addAttribute("provinceList",provinceList);
         return PREFIX + "tDriver_add.html";
@@ -261,19 +267,19 @@
         model.addAttribute("objectName", tCompanyService.selectById(shiroExtUtil.getUser().getObjectId()).getName());
     
         if (1 == roleType) {
-            List<TCompany> companyList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 2).ne("flag", 3));
+            List<TCompany> companyList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 2).eq("state", 0).ne("flag", 3));
             model.addAttribute("companyList", companyList);
-            List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 3).eq("superiorId", tDriver.getCompanyId()));
+            List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 3).eq("state", 0).ne("flag", 3).eq("superiorId", tDriver.getCompanyId()));
             model.addAttribute("franchiseeList", franchiseeList);
         } else if (2 == roleType) {
-            List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 3).eq("superiorId", shiroExtUtil.getUser().getObjectId()));
+            List<TCompany> franchiseeList = tCompanyService.selectList(new EntityWrapper<TCompany>().eq("type", 3).eq("state", 0).ne("flag", 3).eq("superiorId", shiroExtUtil.getUser().getObjectId()));
             model.addAttribute("franchiseeList", franchiseeList);
         }
 
         //查询平台ID
-        TCompany company = tCompanyService.selectOne(new EntityWrapper<TCompany>().eq("type", 1));
+        TCompany company = tCompanyService.selectOne(new EntityWrapper<TCompany>().eq("type", 1).eq("state", 0).ne("flag", 3));
         //判断是平台司机还是加盟司机
-        if ((SinataUtil.isEmpty(tDriver.getCompanyId()) || tDriver.getCompanyId() == 0 || tDriver.getCompanyId() == company.getId()) && (SinataUtil.isEmpty(tDriver.getFranchiseeId()) || tDriver.getFranchiseeId() == 0)){
+        if ((SinataUtil.isEmpty(tDriver.getCompanyId()) || tDriver.getCompanyId() == 0 || tDriver.getCompanyId().equals(company.getId())) && (SinataUtil.isEmpty(tDriver.getFranchiseeId()) || tDriver.getFranchiseeId() == 0)){
             model.addAttribute("companyType",1);
         }else{
             model.addAttribute("companyType",2);
@@ -315,26 +321,43 @@
         model.addAttribute("six",six);
 
         //查询线路列表
-        List<TLine> lineList = itLineService.selectList(new EntityWrapper<TLine>().eq("state", 1));
-        model.addAttribute("lineList",lineList);
+        if(1 != roleType){
+            Integer objectId = shiroExtUtil.getUser().getObjectId();
+            List<TLine> lineList = itLineService.selectList(new EntityWrapper<TLine>().eq("state", 1).last(" and (id in (select lineId from t_line_company where companyId = " + objectId + ") or companyId = " + objectId + ")"));
+            model.addAttribute("lineList",lineList);
+        }else{
+            List<TLine> lineList = itLineService.selectList(new EntityWrapper<TLine>().eq("state", 1));
+            model.addAttribute("lineList",lineList);
+        }
         //查询驾驶员已添加的线路列表
         List<Map<String, Object>> driverLineList = tDriverLineService.getDriverLineListByDriverId(tDriverId);
         model.addAttribute("driverLineList",driverLineList);
         model.addAttribute("provinceId", 1);
+        model.addAttribute("cityId","");
+        model.addAttribute("areaId","");
+        model.addAttribute("provinceId", "");
+        model.addAttribute("areaList", new ArrayList<>());
+        model.addAttribute("cityList", new ArrayList<>());
         if(ToolUtil.isNotEmpty(tDriver.getPlaceOfEmployment())){
             TRegion area = tRegionService.selectOne(new EntityWrapper<TRegion>().eq("code", tDriver.getPlaceOfEmployment()));
-            model.addAttribute("areaId",area.getId());
-            TRegion city = tRegionService.selectOne(new EntityWrapper<TRegion>().eq("id", area.getParentId()));
-            model.addAttribute("cityId",city.getId());
-            TRegion province = tRegionService.selectOne(new EntityWrapper<TRegion>().eq("id", city.getParentId()));
-            model.addAttribute("provinceId",province.getId());
+            if(null != area) {
+                model.addAttribute("areaId", area.getId());
+                TRegion city = tRegionService.selectOne(new EntityWrapper<TRegion>().eq("id", area.getParentId()));
+                if (null != city) {
+                    model.addAttribute("cityId", city.getId());
+                    TRegion province = tRegionService.selectOne(new EntityWrapper<TRegion>().eq("id", city.getParentId()));
+                    if (null != province) {
+                        model.addAttribute("provinceId", province.getId());
+                    }
+                    //市
+                    List<TRegion> cityList = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", city.getParentId()));
+                    model.addAttribute("cityList", cityList);
+                }
 
-            //市
-            List<TRegion> cityList = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", city.getParentId()));
-            model.addAttribute("cityList",cityList);
-            //区
-            List<TRegion> areaList = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", area.getParentId()));
-            model.addAttribute("areaList",areaList);
+                //区
+                List<TRegion> areaList = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", area.getParentId()));
+                model.addAttribute("areaList",areaList);
+            }
         }
         //省
         List<TRegion> provinceList = tRegionService.selectList(new EntityWrapper<TRegion>().eq("parent_id", 0));

--
Gitblit v1.7.1