add taiwan stock kbar data (#272)

* add: kbar data * add: remove start and end data * add: remove unuse code --------- Co-authored-by: YenChen <yenchen0416@google.com>
FinMind · Oct 29, 2023 · 02bff12 · 02bff12
1 parent b6fb28b
commit 02bff12
Show file tree

Hide file tree

Showing 3 changed files with 48 additions and 0 deletions.
diff --git a/FinMind/data/data_loader.py b/FinMind/data/data_loader.py
@@ -1237,6 +1237,34 @@ def taiwan_stock_10year(
         )
         return tw_stock_10year
 
+    def taiwan_stock_bar(
+        self,
+        stock_id: str = "",
+        date: str = "",
+        timeout: int = None,
+    ) -> pd.DataFrame:
+        """get 台股分 K 資料表
+        :param timeout (int): timeout seconds, default None
+
+        :return: 台股分 K 資料表 TaiwanStockKBar
+        :rtype pd.DataFrame
+        :rtype column date (str)
+        :rtype column minute (str)
+        :rtype column stock_id (str)
+        :rtype column open (float)
+        :rtype column high (float)
+        :rtype column low (float)
+        :rtype column close (float)
+        :rtype column volume (int)
+        """
+        taiwan_stock_bar = self.get_data(
+            dataset=Dataset.TaiwanStockKBar,
+            data_id=stock_id,
+            start_date=date,
+            timeout=timeout,
+        )
+        return taiwan_stock_bar
+
     def us_stock_info(self, timeout: int = None) -> pd.DataFrame:
         """get 美國股票代碼總覽
         :param timeout (int): timeout seconds, default None

diff --git a/FinMind/schema/data.py b/FinMind/schema/data.py
@@ -75,6 +75,7 @@ class Dataset(str, Enum):
     TaiwanFuturesSnapshot = "taiwan_futures_snapshot"
     TaiwanOptionsSnapshot = "taiwan_options_snapshot"
     TaiwanStockPriceAdj = "TaiwanStockPriceAdj"
+    TaiwanStockKBar = "TaiwanStockKBar"
 
 
 class Version(str, Enum):

diff --git a/tests/data/test_data_loader.py b/tests/data/test_data_loader.py
@@ -619,6 +619,25 @@ def test_taiwan_stock_10year(data_loader):
     )
 
 
+def test_taiwan_stock_bar(data_loader):
+    data = data_loader.taiwan_stock_bar(
+        stock_id="2330", date="2023-01-05"
+    )
+    assert_data(
+        data,
+        [
+            "date",
+            "minute",
+            "stock_id",
+            "open",
+            "high",
+            "low",
+            "close",
+            "volume",
+        ],
+    )
+
+
 def test_us_stock_info(data_loader):
     data = data_loader.us_stock_info()
     assert_data(