From 197408a3555fa0142fe6c658f274ddb2ab2f4baa Mon Sep 17 00:00:00 2001
From: leiyong <1549684884@qq.com>
Date: Wed, 2 Dec 2020 09:20:55 +0800
Subject: [PATCH] * Fix bug.

---
 module/product/model.php   | 2 +-
 module/program/control.php | 7 ++++++-
 2 files changed, 7 insertions(+), 2 deletions(-)

diff --git a/module/product/model.php b/module/product/model.php
index 56cdbcf30c..d4fa9126e6 100644
--- a/module/product/model.php
+++ b/module/product/model.php
@@ -348,7 +348,7 @@ class productModel extends model
      * Get product id by project.
      *
      * @param  int    $projectID
-     * @param  int    $isFirst
+     * @param  bool   $isFirst
      * @access public
      * @return array
      */
diff --git a/module/program/control.php b/module/program/control.php
index 26d98c8890..b804497126 100644
--- a/module/program/control.php
+++ b/module/program/control.php
@@ -713,7 +713,12 @@ class program extends control
             }
             else
             {
-                if($model == 'waterfall') $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->createLink('programplan', 'create', "projectID=$projectID", '', '', $projectID)));
+                if($model == 'waterfall')
+                {
+                    $productID = $this->loadModel('product')->getProductIDByProject($projectID, true);
+                    $this->session->set('projectPlanList', $this->createLink('programplan', 'browse', "projectID=$projectID&productID=$productID&type=lists", '', '', $projectID));
+                    $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->createLink('programplan', 'create', "projectID=$projectID", '', '', $projectID)));
+                }
 
                 $this->send(array('result' => 'success', 'message' => $this->lang->saveSuccess, 'locate' => $this->createLink('project', 'create', '', '', '', $projectID)));
             }