Merge pull request #13 from fugle-dev/feature/technical

feat: add REST endpoints for stock technical API

Author: kevinypfan

fugle_marketdata/rest/stock/client.py

@@ -1,6 +1,7 @@
 from .intraday import Intraday
 from .historical import Historical
 from .snapshot import Snapshot
+from .technical import Technical
 
 
 class RestStockClient:
@@ -19,3 +20,7 @@ def historical(self):
     @property
     def snapshot(self):
         return Snapshot(**self.config)
+    
+    @property
+    def technical(self):
+        return Technical(**self.config)

fugle_marketdata/rest/stock/technical.py

@@ -0,0 +1,22 @@
+from ..base_rest import BaseRest
+
+class Technical(BaseRest):
+    def sma(self, **params):
+        symbol = params.pop('symbol')
+        return self.request(f"technical/sma/{symbol}", **params)
+
+    def rsi(self, **params):
+        symbol = params.pop('symbol')
+        return self.request(f"technical/rsi/{symbol}", **params)
+
+    def kdj(self, **params):
+        symbol = params.pop('symbol')
+        return self.request(f"technical/kdj/{symbol}", **params)
+
+    def macd(self, **params):
+        symbol = params.pop('symbol')
+        return self.request(f"technical/macd/{symbol}", **params)
+
+    def bb(self, **params):
+        symbol = params.pop('symbol')
+        return self.request(f"technical/bb/{symbol}", **params)

tests/test_http_client.py

@@ -323,7 +323,7 @@ def test_intraday_volumes_api_key(self, mocker, api_key_client):
             'https://api.fugle.tw/marketdata/v1.0/futopt/intraday/volumes/2330',
             headers={'X-API-KEY': 'api-key'}
         )
-    
+
 class TestFutOptRestHistoricalClient:
     def test_futopt_historical(self, api_key_client):
         futopt = api_key_client.futopt
@@ -364,4 +364,104 @@ def test_historical_daily_bearer_token(self, bearer_client, mocker):
         mock_get.assert_called_once_with(
             'https://api.fugle.tw/marketdata/v1.0/futopt/historical/daily/2330',
             headers={'Authorization': 'Bearer bearer-token'}
-        )
+        )
+
+
+class TestStockRestTechnicalClient:
+    def test_stock_technical(self, api_key_client):
+        stock = api_key_client.stock
+        assert hasattr(stock.technical, 'sma')
+        assert hasattr(stock.technical, 'rsi')
+        assert hasattr(stock.technical, 'kdj')
+        assert hasattr(stock.technical, 'macd')
+        assert hasattr(stock.technical, 'bb')
+
+    def test_technical_sma_api_key(self, mocker, api_key_client):
+        stock = api_key_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.sma(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/sma/2330',
+            headers={'X-API-KEY': 'api-key'}
+        )
+
+    def test_technical_sma_bearer_token(self, mocker, bearer_client):
+        stock = bearer_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.sma(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/sma/2330',
+            headers={'Authorization': 'Bearer bearer-token'}
+        )
+
+    def test_technical_rsi_api_key(self, mocker, api_key_client):
+        stock = api_key_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.rsi(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/rsi/2330',
+            headers={'X-API-KEY': 'api-key'}
+        )
+
+    def test_technical_rsi_bearer_token(self, mocker, bearer_client):
+        stock = bearer_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.rsi(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/rsi/2330',
+            headers={'Authorization': 'Bearer bearer-token'}
+        )
+
+    def test_technical_kdj_api_key(self, mocker, api_key_client):
+        stock = api_key_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.kdj(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/kdj/2330',
+            headers={'X-API-KEY': 'api-key'}
+        )
+
+    def test_technical_kdj_bearer_token(self, mocker, bearer_client):
+        stock = bearer_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.kdj(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/kdj/2330',
+            headers={'Authorization': 'Bearer bearer-token'}
+        )
+
+    def test_technical_macd_api_key(self, mocker, api_key_client):
+        stock = api_key_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.macd(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/macd/2330',
+            headers={'X-API-KEY': 'api-key'}
+        )
+
+    def test_technical_macd_bearer_token(self, mocker, bearer_client):
+        stock = bearer_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.macd(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/macd/2330',
+            headers={'Authorization': 'Bearer bearer-token'}
+        )
+
+    def test_technical_bb_api_key(self, mocker, api_key_client):
+        stock = api_key_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.bb(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/bb/2330',
+            headers={'X-API-KEY': 'api-key'}
+        )
+
+    def test_technical_bb_bearer_token(self, mocker, bearer_client):
+        stock = bearer_client.stock
+        mock_get = mocker.patch('requests.get')
+        stock.technical.bb(symbol='2330')
+        mock_get.assert_called_once_with(
+            'https://api.fugle.tw/marketdata/v1.0/stock/technical/bb/2330',
+            headers={'Authorization': 'Bearer bearer-token'}
+        )