fix: make MARKET enum accept lowercase values

rqalpha/const.py

@@ -216,10 +216,23 @@ class TRADING_CALENDAR_TYPE(CustomEnum):
 # backward compatible
 TRADING_CALENDAR_TYPE.EXCHANGE = TRADING_CALENDAR_TYPE.CN_STOCK
 
-class MARKET(CustomEnum):
+class MarketEnumMeta(CustomEnumMeta):
+    def __getitem__(cls, item):
+        if isinstance(item, str):
+            item = item.upper()
+        return super(MarketEnumMeta, cls).__getitem__(item)
+
+
+class MARKET(CustomEnum, metaclass=MarketEnumMeta):
     CN = "CN"
     HK = "HK"
 
+    @classmethod
+    def _missing_(cls, value):
+        if isinstance(value, str):
+            return cls._value2member_map_.get(value.upper())
+        return None
+
 
 class TAX_TYPE(CustomEnum):
     CAPITAL_GAINS = "capital_gains"  # 资本收益税（含增值税、附加税等）

tests/unittest/test_const.py

@@ -0,0 +1,31 @@
+import pytest
+
+from rqalpha.const import MARKET
+
+
+@pytest.mark.parametrize(
+    "value, expected",
+    [
+        ("CN", MARKET.CN),
+        ("cn", MARKET.CN),
+        ("Cn", MARKET.CN),
+        ("HK", MARKET.HK),
+        ("hk", MARKET.HK),
+        ("Hk", MARKET.HK),
+    ],
+)
+def test_market_accepts_case_insensitive_values(value, expected):
+    assert MARKET(value) is expected
+
+
+@pytest.mark.parametrize(
+    "key, expected",
+    [
+        ("CN", MARKET.CN),
+        ("cn", MARKET.CN),
+        ("HK", MARKET.HK),
+        ("hk", MARKET.HK),
+    ],
+)
+def test_market_accepts_case_insensitive_keys(key, expected):
+    assert MARKET[key] is expected