part fix zoom issue

mole · mole · commit 230d0db5b816 · 2023-05-10T23:28:01.000+08:00
diff --git a/finplotWindow.py b/finplotWindow.py
@@ -13,6 +13,8 @@
 import finplot as fplt
 
 import pandas as pd
+import numpy as np
+from datetime import datetime as dt
 import backtrader as bt
 from pyqtgraph import mkColor, mkBrush
 
@@ -379,13 +381,26 @@ def setIndicator(self, indicatorName, activated):
         self.updateChart()
 
         pass
-  
+
+    def _date_str2x(self, ax, date_str):
+        df = ax.getAxis('bottom').vb.datasrc.df
+        dftime = np.array(df.iloc[:, 0])
+        lsttime = dftime.tolist()
+        # print(dftime)
+        # print(lsttime)
+
+        xtime = dt.strptime(date_str, '%Y-%m-%d %H:%M:%S')
+        xint = int((xtime.timestamp()+8*3600)*1e9)
+        print(xint,lsttime[0])
+        x = lsttime.index(xint)
+        return [x]
+
     def zoomTo(self, dateStr1, dateStr2):
 
         for win in fplt.windows:
             for ax in win.axs:
-                x1 = fplt._dateStr2x(ax,dateStr1)
-                x2 = fplt._dateStr2x(ax,dateStr2)
+                x1 = self._date_str2x(ax, dateStr1)
+                x2 = self._date_str2x(ax, dateStr2)
 
                 # Do not zoom exactly on the trade, so take a little bit before & after
                 date1 = x1[0]
