add c_order_percent for chinese market fee (#61)

src/pybroker/common.py

@@ -144,6 +144,7 @@ class FeeMode(Enum):
     """
 
     ORDER_PERCENT = "order_percent"
+    C_ORDER_PERCENT = "c_order_percent"
     PER_ORDER = "per_order"
     PER_SHARE = "per_share"
 

src/pybroker/portfolio.py

@@ -344,12 +344,22 @@ def __init__(
         self._entry_id: int = 0
         self._trade_id: int = 0
 
-    def _calculate_fees(self, fill_price: Decimal, shares: Decimal) -> Decimal:
+    def _calculate_fees(self, fill_price: Decimal, shares: Decimal, type: Literal["buy", "sell"]) -> Decimal:
         fees = Decimal()
         if self._fee_mode is None or self._fee_amount is None:
             return fees
         if self._fee_mode == FeeMode.ORDER_PERCENT:
             fees = self._fee_amount / Decimal(100) * fill_price * shares
+        elif self._fee_mode == FeeMode.C_ORDER_PERCENT:
+            if type == "buy":
+                fees = self._fee_amount / Decimal(100) * fill_price * shares
+                if fees < Decimal(5.0):
+                    fees = Decimal(5.0)
+            else:
+                fees = self._fee_amount / Decimal(100) * fill_price * shares
+                if fees < Decimal(5.0):
+                    fees = Decimal(5.0)
+                fees = fees + Decimal(0.0001) * fill_price * shares
         elif self._fee_mode == FeeMode.PER_ORDER:
             fees = self._fee_amount
         elif self._fee_mode == FeeMode.PER_SHARE:
@@ -402,7 +412,7 @@ def _add_order(
         shares: Decimal,
     ) -> Order:
         self._order_id += 1
-        fees = self._calculate_fees(fill_price, shares)
+        fees = self._calculate_fees(fill_price, shares, type)
         order = Order(
             id=self._order_id,
             date=date,