From ef1c3311950b88627cb6d772fe5147f147930964 Mon Sep 17 00:00:00 2001
From: zhangjunjie <zhangjunjie1@xiaomi.com>
Date: Thu, 28 Nov 2019 20:08:30 +0800
Subject: [PATCH] update

---
 components/date-picker/DateRangePanel.js | 13 +++++++------
 components/date-picker/WeekRangePanel.js |  8 ++++----
 2 files changed, 11 insertions(+), 10 deletions(-)

diff --git a/components/date-picker/DateRangePanel.js b/components/date-picker/DateRangePanel.js
index 597b7a1c6..e5ca62984 100644
--- a/components/date-picker/DateRangePanel.js
+++ b/components/date-picker/DateRangePanel.js
@@ -149,10 +149,10 @@ class DateRangePanel extends Component {
    * @param {Number} year 当前年份
    * @param {Number} month 当前月份
    */
-  getHeaderCenterContent (year, month) {
+  getHeaderCenterContent (year, month, flag) {
     const { localeDatas, locale } = this.props
-    const {currentView} = this.state
-    if (currentView === 'year') {
+    const { layout } = this.state
+    if (layout[flag] === 'year') {
       return (year - 4) + '~' + (year + 7)
     }
     let arr = [localeDatas.datePicker.monthShort[month - 1]]
@@ -192,7 +192,7 @@ class DateRangePanel extends Component {
             this.setState({ layout })
           }}
         >
-          {this.getHeaderCenterContent(year, month)}
+          {this.getHeaderCenterContent(year, month, lr)}
         </span>
         {
           <div className='hi-datepicker__header-btns'>
@@ -283,7 +283,7 @@ class DateRangePanel extends Component {
   }
 
   getRangeDateStr () {
-    let {leftDate, rightDate, showMask} = this.state
+    let { range, showMask, leftDate, rightDate } = this.state
     let { format } = this.props
     format = format.substr(format.match(/[H|h]\s*/).index)
     const cls = classNames(
@@ -291,7 +291,7 @@ class DateRangePanel extends Component {
     )
     return (
       <span className={cls}>
-        {`${dateFormat(leftDate, format)} - ${dateFormat(rightDate, format)}`}
+        {`${dateFormat(range.startDate ? range.startDate : leftDate, format)} - ${dateFormat(range.endDate ? range.endDate : rightDate, format)}`}
       </span>
     )
   }
@@ -450,6 +450,7 @@ class DateRangePanel extends Component {
               date={date}
               onPick={(d, r) => {
                 this.setState({
+                  range: d,
                   leftDate: d.startDate,
                   rightDate: d.endDate
                 })
diff --git a/components/date-picker/WeekRangePanel.js b/components/date-picker/WeekRangePanel.js
index 4e389cc45..db08c2ef0 100644
--- a/components/date-picker/WeekRangePanel.js
+++ b/components/date-picker/WeekRangePanel.js
@@ -42,10 +42,10 @@ export default class WeekRangePanel extends Component {
    * @param {Number} year 当前年份
    * @param {Number} month 当前月份
    */
-  getHeaderCenterContent (year, month) {
+  getHeaderCenterContent (year, month, flag) {
     const { localeDatas, locale } = this.props
-    const {currentView} = this.state
-    if (currentView === 'year') {
+    const { layout } = this.state
+    if (layout[flag] === 'year') {
       return (year - 4) + '~' + (year + 7)
     }
     let arr = [localeDatas.datePicker.monthShort[month - 1]]
@@ -105,7 +105,7 @@ export default class WeekRangePanel extends Component {
             const layout = Object.assign({}, this.state.layout, {[lr]: 'year'})
             this.setState({ layout })
           }}>
-          {this.getHeaderCenterContent(year, month)}
+          {this.getHeaderCenterContent(year, month, lr)}
         </span>
         <div className='hi-datepicker__header-btns'>
           <span onClick={() => this.changeMonth(false, lr)} ><Icon name='right' /></span>