CTYPTO/freqtrade
1
0
Fork 0
mirror of https://github.com/freqtrade/freqtrade.git synced 2025-11-29 00:23:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: Use Rich for list-strategies command

Browse Source
This commit is contained in:
Matthias
2024-07-06 18:09:03 +02:00
parent 2e0372d247
commit 8d00e1d929
1 changed files with 21 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files

40
freqtrade/commands/list_commands.py
View File

@@ -59,7 +59,7 @@ def start_list_exchanges(args: Dict[str, Any]) -> None:
trade_modes = Text( trade_modes = Text(
", ".join( ", ".join(
(f"{a.get('margin_mode', '')} {a["trading_mode"]}").lstrip() (f"{a.get('margin_mode', '')} {a['trading_mode']}").lstrip()
for a in exchange["trade_modes"] for a in exchange["trade_modes"]
), ),
style="", style="",
@@ -94,27 +94,17 @@ def _print_rich_table(summary: str, headers: List[str], tabular_data: List[Dict[
def _print_objs_tabular(objs: List, print_colorized: bool) -> None: def _print_objs_tabular(objs: List, print_colorized: bool) -> None:
if print_colorized:
colorama_init(autoreset=True)
red = Fore.RED
yellow = Fore.YELLOW
reset = Style.RESET_ALL
else:
red = ""
yellow = ""
reset = ""
names = [s["name"] for s in objs] names = [s["name"] for s in objs]
objs_to_print = [ objs_to_print = [
{ {
"name": s["name"] if s["name"] else "--", "name": Text(s["name"] if s["name"] else "--"),
"location": s["location_rel"], "location": Text(s["location_rel"]),
"status": ( "status": (
red + "LOAD FAILED" + reset Text("LOAD FAILED", style="bold red")
if s["class"] is None if s["class"] is None
else "OK" else Text("OK", style="bold green")
if names.count(s["name"]) == 1 if names.count(s["name"]) == 1
else yellow + "DUPLICATE NAME" + reset else Text("DUPLICATE NAME", style="bold yellow")
), ),
} }
for s in objs for s in objs
@@ -124,11 +114,23 @@ def _print_objs_tabular(objs: List, print_colorized: bool) -> None:
objs_to_print[idx].update( objs_to_print[idx].update(
{ {
"hyperoptable": "Yes" if s["hyperoptable"]["count"] > 0 else "No", "hyperoptable": "Yes" if s["hyperoptable"]["count"] > 0 else "No",
"buy-Params": len(s["hyperoptable"].get("buy", [])), "buy-Params": str(len(s["hyperoptable"].get("buy", []))),
"sell-Params": len(s["hyperoptable"].get("sell", [])), "sell-Params": str(len(s["hyperoptable"].get("sell", []))),
} }
) )
print(tabulate(objs_to_print, headers="keys", tablefmt="psql", stralign="right")) table = Table(title="Available:")
for header in objs_to_print[0].keys():
table.add_column(header.capitalize(), justify="right")
for row in objs_to_print:
table.add_row(*[row[header] for header in objs_to_print[0].keys()])
console = Console(
color_system="auto" if print_colorized else None,
width=200 if "pytest" in sys.modules else None,
)
console.print(table)
def start_list_strategies(args: Dict[str, Any]) -> None: def start_list_strategies(args: Dict[str, Any]) -> None: