gx

@@ -30,8 +30,8 @@ class VisualizationController extends BaseController
      * @method :post
      * @time   :2023/11/15 10:26
      */
-    public function info(Visualization $visualization){
-        $info = $visualization->read(['type'=>$this->map['type']],['main_html','type','id','project_id']);
+    public function info(VisualizationLogic $logic){
+        $info = $logic->getVisualizationInfo();
         if($info === false){
             $info = [];
         }

@@ -30,6 +30,26 @@ class VisualizationLogic extends BaseLogic
     }
 
     /**
+     * @remark :获取代码块
+     * @name   :getVisualizationInfo
+     * @author :lyh
+     * @method :post
+     * @time   :2023/11/17 14:44
+     */
+    public function getVisualizationInfo(){
+        $type = $this->getType($this->param['source'],$this->param['source_id']);
+        $typeArray = [1,3,5,7];//单页数据
+        if(in_array($type,$typeArray)){
+            $bTemplateModel = new BTemplate();
+            $info = $bTemplateModel->read(['source'=>$this->param['source'],'source_id'=>$this->param['source_id'],'template_id'=>0]);
+        }else{//模块数据
+            $bTemplateMainModel = new BTemplateMain();
+            $info = $bTemplateMainModel->read(['type'=>$type]);
+        }
+        return $this->success($info);
+    }
+
+    /**
      * @remark :保存定制html
      * @name   :saveHtml
      * @author :lyh

@@ -60,8 +60,9 @@ class ExtendLogic extends BaseLogic
         $info = $this->model->read(['key'=>$key.$i]);
         if($info !== false){
             return $this->getKey($key,$i+1);
+        }else{
+            return $key.$i;
         }
-        return $key;
     }
 
     /**