from func_position_calls import query_existing_order
from func_position_calls import get_open_positions
from func_position_calls import get_active_positions
from func_calcultions import get_trade_details
#from config_ws_connect import ws_public          # <- original code websocket method
from config_ws_connect import ws_public_get_orderbook   # <- new websocket method
# from func_trade_management import capital_long
from exceptions import FailedRequestError, InvalidRequestError
from requests.exceptions import ConnectionError, Timeout, RequestException
from json.decoder import JSONDecodeError

# Check order items
def check_order(ticker, order_id, remaining_capital, direction="Long"):
    exception = True
    try:
        # Get current orderbook
        #orderbook = ws_public.fetch(f"orderBookL2_25.{ticker}")    # <- original code websocket method
        orderbook = ws_public_get_orderbook(ticker)         # <- new websocket method
        
        # print("\n")
        # print('-------------------------------------------------------------------')
        # print("Check Order")
        # print(orderbook)
        # print('-------------------------------------------------------------------')
        # print("\n")

        # Get latest price
        mid_price, _, _ = get_trade_details(orderbook)

        # Get trade details
        order_price, order_quantity, order_status = query_existing_order(ticker, order_id, direction)

        # Get open positions
        position_price, position_quantity = get_open_positions(ticker, direction)

        # Get active positions
        # active_order_price, active_order_quantity = get_active_positions(ticker)
        exception = False
        # Determine action - trade complete - stop placing orders
        if position_price * position_quantity >= remaining_capital and position_quantity > 0:  # originally pos_value > remaining_capital
            print(f"position_value {position_price * position_quantity}", f"remaining_capital {remaining_capital}")
            return ("Trade Complete", exception)

        # Determine action - position filled - buy more
        if order_status == "Filled":
            return ("Position Filled", exception)

        # Determine action - order active - do nothing
        active_items = ["Created", "New"]
        if order_status in active_items:
            return ("Order Active", exception)

        # Determine action - partial filled order - do nothing
        if order_status == "PartiallyFilled":
            return ("Partial Fill", exception)

        # Determine action - order failed - try place order again
        cancel_items = ["Cancelled", "Rejected", "PendingCancel"]
        if order_status in cancel_items:
            return ("Try Again", exception)
    except (ConnectionError, Timeout, RequestException, JSONDecodeError, FailedRequestError, InvalidRequestError) as e:
        print(e)
    return ("",exception)