feat: Added trading calendar interfaces

Kerr-pixel · Kerr-pixel · commit 009980d56095 · 2023-02-10T12:10:23.000+08:00
diff --git a/webull-python-sdk-core/webullsdkcore/__init__.py b/webull-python-sdk-core/webullsdkcore/__init__.py
@@ -1,4 +1,4 @@
-__version__ = "0.1.4"
+__version__ = "0.1.5"
 
 import logging
 
diff --git a/webull-python-sdk-demos/setup.py b/webull-python-sdk-demos/setup.py
@@ -15,8 +15,8 @@
     LONG_DESCRIPTION = fp.read()
 
 requires = [
-    "webull-python-sdk-mdata==0.1.4",
-    "webull-python-sdk-trade==0.1.4"
+    "webull-python-sdk-mdata==0.1.5",
+    "webull-python-sdk-trade==0.1.5"
 ]
 
 setup_args = {
diff --git a/webull-python-sdk-demos/tests/trade/request/test_trade_calendar.py b/webull-python-sdk-demos/tests/trade/request/test_trade_calendar.py
@@ -0,0 +1,46 @@
+# Copyright 2022 Webull
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# 	http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+import unittest
+
+from webullsdkcore.client import ApiClient
+from webullsdkcore.exception.exceptions import ServerException
+from webullsdktrade.api import API
+from webullsdktrade.common.markets import Markets
+
+optional_api_endpoint = "</optional_api_endpoint>"
+your_app_key = "</your_app_key>"
+your_app_secret = "</your_app_secret>"
+
+# 'hk' or 'us'
+region_id = "<region_id>"
+api_client = ApiClient(your_app_key, your_app_secret, region_id)
+api_client.add_endpoint(region_id, optional_api_endpoint)
+
+
+class TestTradeCalendar(unittest.TestCase):
+
+    def test_trade_calendar(self):
+        api = API(api_client)
+        try:
+            response = api.trade_calendar.get_trade_calendar(Markets.US.name, "2023-01-20", "2023-02-03")
+            print(response.json())
+        except ServerException as se:
+            print(se)
+
+        try:
+            response = api.trade_calendar.get_trade_calendar(Markets.HK.name, "2023-01-20", "2023-02-03")
+            print(response.json())
+        except ServerException as se:
+            print(se)
diff --git a/webull-python-sdk-mdata/setup.py b/webull-python-sdk-mdata/setup.py
@@ -15,8 +15,8 @@
     LONG_DESCRIPTION = fp.read()
 
 requires = [
-    "webull-python-sdk-core==0.1.4",
-    "webull-python-sdk-quotes-core==0.1.4"
+    "webull-python-sdk-core==0.1.5",
+    "webull-python-sdk-quotes-core==0.1.5"
 ]
 
 setup_args = {
diff --git a/webull-python-sdk-mdata/webullsdkmdata/__init__.py b/webull-python-sdk-mdata/webullsdkmdata/__init__.py
@@ -1,3 +1,3 @@
 # coding=utf-8
 
-__version__ = '0.1.4'
+__version__ = '0.1.5'
diff --git a/webull-python-sdk-quotes-core/setup.py b/webull-python-sdk-quotes-core/setup.py
@@ -24,7 +24,7 @@
     "grpcio==1.51.1",
     "grpcio-tools==1.51.1",
     "protobuf==4.21.12",
-    "webull-python-sdk-core==0.1.4"
+    "webull-python-sdk-core==0.1.5"
 ]
 
 setup_args = {
diff --git a/webull-python-sdk-quotes-core/webullsdkquotescore/__init__.py b/webull-python-sdk-quotes-core/webullsdkquotescore/__init__.py
@@ -1 +1 @@
-__version__ = "0.1.4"
+__version__ = "0.1.5"
diff --git a/webull-python-sdk-trade-events-core/setup.py b/webull-python-sdk-trade-events-core/setup.py
@@ -23,7 +23,7 @@
     "grpcio==1.51.1",
     "grpcio-tools==1.51.1",
     "protobuf==4.21.12",
-    "webull-python-sdk-core==0.1.4"
+    "webull-python-sdk-core==0.1.5"
 ]
 
 setup_args = {
diff --git a/webull-python-sdk-trade-events-core/webullsdktradeeventscore/__init__.py b/webull-python-sdk-trade-events-core/webullsdktradeeventscore/__init__.py
@@ -1 +1 @@
-__version__ = "0.1.4"
+__version__ = "0.1.5"
diff --git a/webull-python-sdk-trade/setup.py b/webull-python-sdk-trade/setup.py
@@ -15,8 +15,8 @@
     LONG_DESCRIPTION = fp.read()
 
 requires = [
-    "webull-python-sdk-trade-events-core==0.1.4",
-    "webull-python-sdk-core==0.1.4"
+    "webull-python-sdk-trade-events-core==0.1.5",
+    "webull-python-sdk-core==0.1.5"
 ]
 
 setup_args = {
diff --git a/webull-python-sdk-trade/webullsdktrade/__init__.py b/webull-python-sdk-trade/webullsdktrade/__init__.py
@@ -1,2 +1,2 @@
-__version__ = "0.1.4"
+__version__ = "0.1.5"
 
diff --git a/webull-python-sdk-trade/webullsdktrade/api.py b/webull-python-sdk-trade/webullsdktrade/api.py
@@ -17,6 +17,7 @@
 from webullsdkmdata.quotes.instrument import Instrument
 from webullsdkmdata.quotes.market_data import MarketData
 from webullsdktrade.trade.trade_instrument import TradeInstrument
+from webullsdktrade.trade.trade_calendar import TradeCalendar
 
 
 class API:
@@ -26,3 +27,4 @@ def __init__(self, api_client):
         self.account = Account(api_client)
         self.order = OrderOperation(api_client)
         self.trade_instrument = TradeInstrument(api_client)
+        self.trade_calendar = TradeCalendar(api_client)
diff --git a/webull-python-sdk-trade/webullsdktrade/common/markets.py b/webull-python-sdk-trade/webullsdktrade/common/markets.py
@@ -0,0 +1,24 @@
+# Copyright 2022 Webull
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# 	http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from webullsdkcore.common.easy_enum import EasyEnum
+
+
+class Markets(EasyEnum):
+
+    # US market
+    US = (1, "US")
+
+    # HK market
+    HK = (2, "HK")
diff --git a/webull-python-sdk-trade/webullsdktrade/common/trading_date_type.py b/webull-python-sdk-trade/webullsdktrade/common/trading_date_type.py
@@ -0,0 +1,24 @@
+# Copyright 2022 Webull
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# 	http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+from webullsdkcore.common.easy_enum import EasyEnum
+
+
+class TradingDateType(EasyEnum):
+
+    # Full day
+    FULL_DAY = (1, "FULL_DAY")
+
+    # Half day
+    HALF_DAY = (2, "HALF_DAY")
diff --git a/webull-python-sdk-trade/webullsdktrade/request/get_trade_calendar_request.py b/webull-python-sdk-trade/webullsdktrade/request/get_trade_calendar_request.py
@@ -0,0 +1,30 @@
+# Copyright 2022 Webull
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# 	http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+# coding=utf-8
+from webullsdkcore.request import ApiRequest
+
+
+class TradeCalendarRequest(ApiRequest):
+    def __init__(self):
+        ApiRequest.__init__(self, "/trade/calendar", version='v1', method="GET", query_params={})
+
+    def set_market(self, market):
+        self.add_query_param("market", market)
+
+    def set_start(self, start):
+        self.add_query_param("start", start)
+
+    def set_end(self, end):
+        self.add_query_param("end", end)
diff --git a/webull-python-sdk-trade/webullsdktrade/trade/trade_calendar.py b/webull-python-sdk-trade/webullsdktrade/trade/trade_calendar.py
@@ -0,0 +1,37 @@
+# Copyright 2022 Webull
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+# 	http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+from webullsdktrade.request.get_trade_calendar_request import TradeCalendarRequest
+
+
+class TradeCalendar:
+
+    def __init__(self, api_client):
+        self.client = api_client
+
+    def get_trade_calendar(self, market, start, end):
+        """
+        Get the trading calendar for the specified market.
+        Trading days are obtained by excluding weekends and holidays, and the dates of temporary market closures are
+        not excluded.
+
+        :param market: Markets, enumeration
+        :param start: Start date
+        :param end: End date
+        """
+        trade_calendar_request = TradeCalendarRequest()
+        trade_calendar_request.set_market(market)
+        trade_calendar_request.set_start(start)
+        trade_calendar_request.set_end(end)
+        response = self.client.get_response(trade_calendar_request)
+        return response

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-__version__ = "0.1.4"`
	`1`	`+__version__ = "0.1.5"`
`2`	`2`
`3`	`3`	`import logging`
`4`	`4`
Original file line number	Diff line number	Diff line change
`@@ -15,8 +15,8 @@`
`15`	`15`	`LONG_DESCRIPTION = fp.read()`
`16`	`16`
`17`	`17`	`requires = [`
`18`		`- "webull-python-sdk-mdata==0.1.4",`
`19`		`- "webull-python-sdk-trade==0.1.4"`
	`18`	`+ "webull-python-sdk-mdata==0.1.5",`
	`19`	`+ "webull-python-sdk-trade==0.1.5"`
`20`	`20`	`]`
`21`	`21`
`22`	`22`	`setup_args = {`
Original file line number	Diff line number	Diff line change
`@@ -1,3 +1,3 @@`
`1`	`1`	`# coding=utf-8`
`2`	`2`
`3`		`-__version__ = '0.1.4'`
	`3`	`+__version__ = '0.1.5'`
Original file line number	Diff line number	Diff line change
`@@ -24,7 +24,7 @@`
`24`	`24`	`"grpcio==1.51.1",`
`25`	`25`	`"grpcio-tools==1.51.1",`
`26`	`26`	`"protobuf==4.21.12",`
`27`		`- "webull-python-sdk-core==0.1.4"`
	`27`	`+ "webull-python-sdk-core==0.1.5"`
`28`	`28`	`]`
`29`	`29`
`30`	`30`	`setup_args = {`
Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-__version__ = "0.1.4"`
	`1`	`+__version__ = "0.1.5"`
Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@`
`23`	`23`	`"grpcio==1.51.1",`
`24`	`24`	`"grpcio-tools==1.51.1",`
`25`	`25`	`"protobuf==4.21.12",`
`26`		`- "webull-python-sdk-core==0.1.4"`
	`26`	`+ "webull-python-sdk-core==0.1.5"`
`27`	`27`	`]`
`28`	`28`
`29`	`29`	`setup_args = {`