Files
hummingbot-dashboard/frontend/components/executors_distribution.py

69 lines
5.0 KiB
Python

import streamlit as st
from frontend.components.st_inputs import get_distribution, normalize, distribution_inputs
def get_executors_distribution_inputs(default_spreads=[0.01, 0.02], default_amounts=[0.2, 0.8]):
default_config = st.session_state.get("default_config", {})
buy_spreads = default_config.get("buy_spreads", default_spreads)
sell_spreads = default_config.get("sell_spreads", default_spreads)
buy_amounts_pct = default_config.get("buy_amounts_pct", default_amounts)
sell_amounts_pct = default_config.get("sell_amounts_pct", default_amounts)
buy_order_levels_def = len(buy_spreads)
sell_order_levels_def = len(sell_spreads)
with st.expander("Executors Configuration", expanded=True):
col_buy, col_sell = st.columns(2)
with col_buy:
st.header("Buy Order Settings")
buy_order_levels = st.number_input("Number of Buy Order Levels", min_value=1, value=buy_order_levels_def,
help="Enter the number of buy order levels (e.g., 2).")
with col_sell:
st.header("Sell Order Settings")
sell_order_levels = st.number_input("Number of Sell Order Levels", min_value=1, value=sell_order_levels_def,
help="Enter the number of sell order levels (e.g., 2).")
if buy_order_levels > buy_order_levels_def:
buy_spreads += [0.01 + max(buy_spreads)] * (buy_order_levels - buy_order_levels_def)
buy_amounts_pct += [0.2 + max(buy_amounts_pct)] * (buy_order_levels - buy_order_levels_def)
elif buy_order_levels < buy_order_levels_def:
buy_spreads = buy_spreads[:buy_order_levels]
buy_amounts_pct = buy_amounts_pct[:buy_order_levels]
if sell_order_levels > sell_order_levels_def:
sell_spreads += [0.01 + max(sell_spreads)] * (sell_order_levels - sell_order_levels_def)
sell_amounts_pct += [0.2 + max(sell_amounts_pct)] * (sell_order_levels - sell_order_levels_def)
elif sell_order_levels < sell_order_levels_def:
sell_spreads = sell_spreads[:sell_order_levels]
sell_amounts_pct = sell_amounts_pct[:sell_order_levels]
col_buy_spreads, col_buy_amounts, col_sell_spreads, col_sell_amounts = st.columns(4)
with col_buy_spreads:
buy_spread_dist_type, buy_spread_start, buy_spread_base, buy_spread_scaling, buy_spread_step, buy_spread_ratio, buy_manual_spreads = distribution_inputs(
col_buy_spreads, "Spread", buy_order_levels, buy_spreads)
with col_buy_amounts:
buy_amount_dist_type, buy_amount_start, buy_amount_base, buy_amount_scaling, buy_amount_step, buy_amount_ratio, buy_manual_amounts = distribution_inputs(
col_buy_amounts, "Amount", buy_order_levels, buy_amounts_pct)
with col_sell_spreads:
sell_spread_dist_type, sell_spread_start, sell_spread_base, sell_spread_scaling, sell_spread_step, sell_spread_ratio, sell_manual_spreads = distribution_inputs(
col_sell_spreads, "Spread", sell_order_levels, sell_spreads)
with col_sell_amounts:
sell_amount_dist_type, sell_amount_start, sell_amount_base, sell_amount_scaling, sell_amount_step, sell_amount_ratio, sell_manual_amounts = distribution_inputs(
col_sell_amounts, "Amount", sell_order_levels, sell_amounts_pct)
# Generate distributions
buy_spread_distributions = get_distribution(buy_spread_dist_type, buy_order_levels, buy_spread_start,
buy_spread_base, buy_spread_scaling, buy_spread_step, buy_spread_ratio,
buy_manual_spreads)
sell_spread_distributions = get_distribution(sell_spread_dist_type, sell_order_levels, sell_spread_start,
sell_spread_base, sell_spread_scaling, sell_spread_step,
sell_spread_ratio, sell_manual_spreads)
buy_amount_distributions = get_distribution(buy_amount_dist_type, buy_order_levels, buy_amount_start, buy_amount_base, buy_amount_scaling,
buy_amount_step, buy_amount_ratio, buy_manual_amounts)
sell_amount_distributions = get_distribution(sell_amount_dist_type, sell_order_levels, sell_amount_start, sell_amount_base,
sell_amount_scaling, sell_amount_step, sell_amount_ratio, sell_manual_amounts)
# Normalize and calculate order amounts
all_orders_amount_normalized = normalize(buy_amount_distributions + sell_amount_distributions)
buy_order_amounts_pct = [amount for amount in all_orders_amount_normalized[:buy_order_levels]]
sell_order_amounts_pct = [amount for amount in all_orders_amount_normalized[buy_order_levels:]]
buy_spread_distributions = [spread / 100 for spread in buy_spread_distributions]
sell_spread_distributions = [spread / 100 for spread in sell_spread_distributions]
return buy_spread_distributions, sell_spread_distributions, buy_order_amounts_pct, sell_order_amounts_pct