diff --git a/cefi/main.py b/cefi/main.py index cc5f390d..d4650cfb 100644 --- a/cefi/main.py +++ b/cefi/main.py @@ -226,7 +226,7 @@ async def get_positions(self): _info.append(f"{client.name}:\n{await client.get_account_position()}") return "\n".join(_info) - async def get_pnls(self): + async def get_pnls(self, **kwargs): """ Return account pnl. @@ -236,9 +236,14 @@ async def get_pnls(self): Returns: pnl """ - _info = ["📊\n"] + _info = ["🏆\n"] for client in self.clients: - _info.append(f"{client.name}:\n{await client.get_account_pnl()}") + client_name = f"{client.name}:\n" + account_pnl = await client.get_account_pnl( + period=kwargs.get("period", None) + ) + client_info = f"{client_name}{account_pnl}" + _info.append(client_info) return "\n".join(_info) async def submit_order(self, order_params):