module.py

import json
import requests

def print_blue_text(text):
    blue_text = "\033[34m" + text + "\033[0m"
    print(blue_text)

def print_red_text(text):
    red_text = "\033[31m" + text + "\033[0m"
    print(red_text)

def get_new_access_token(refresh_token, client_id, client_secret):
    token_url = 'https://signin.tradestation.com/oauth/token'
    headers = {'content-type': 'application/x-www-form-urlencoded'}

    payload = {
        'grant_type': 'refresh_token',
        'client_id': client_id,
        'client_secret': client_secret,
        'refresh_token': refresh_token
    }

    response = requests.post(token_url, data=payload, headers=headers)

    if response.status_code == 200:
        data = response.json()
        access_token = data.get('access_token')
        expires_in = data.get('expires_in')
        return access_token, expires_in
    else:
        raise Exception(f"Failed to get a new access token. Response: {response.text}")

def refresh_access_token():
    with open(r"C:\\Users\\INSERT A VALID PATH HERE\\private.json", 'r') as file:
        private_data = json.load(file)

    with open(r"C:\\Users\\INSERT A VALID PATH HERE\\refreshToken.json", 'r') as file:
        refresh = json.load(file)

    client_id = private_data['client_id']
    client_secret = private_data.get('client_secret', '')
    refresh_token = refresh['refresh']

    on = True

    while on == True:
        try:
            access_token, expires_in = get_new_access_token(refresh_token, client_id, client_secret)
            data = {"accessToken": access_token}
            with open(r"C:\\Users\\INSERT A VALID PATH HERE\\access_token.json", "w") as json_file:
                json.dump(data, json_file)
            print(f"New access token aquired. Expires in: {expires_in} seconds")
            time.sleep(expires_in - 60)
        except KeyboardInterrupt:
            on = False
            print_red_text("Shutting down the refresh loop...")
            sys.exit(0)
        except Exception as e:
            print(f"Error: {e}")
            break

def load_access_token():
    path = 'C:\\Users\\admin\\Desktop\\Million Dollar Business\\argonaut\\access_token.json'

    with open(f'{path}') as file:
        data = json.load(file)
        return data['accessToken']

def placeTradeFlat(quantityEnter, direction):
    access_token = load_access_token()
    account_id = 'SIM2504240F'
    symbol = 'MNQU23'
    quantity = quantityEnter
    order_type = 'Market'
    action = direction
    time_in_force = {'Duration': 'GTC'}
    route = 'Intelligent'

    url = 'https://sim-api.tradestation.com/v3/orderexecution/orders'
    headers = {
        'content-type': 'application/json',
        'Authorization': f'Bearer {access_token}',
    }
    data = {
        'AccountID': account_id,
        'Symbol': symbol,
        'Quantity': quantity,
        'OrderType': order_type,
        'TradeAction': action,
        'TimeInForce': time_in_force,
        'Route': route,
    }

    try:
        response = requests.post(url, headers=headers, json=data)
        response_data = response.json()
        print(response_data)
        print(f'Order for {direction} {symbol} placed successfully.')
    except Exception as e:
        print_red_text("\n------ CRITICAL ------")
        print_red_text("Position exit order failed.")
        print_red_text("---- KILL PROGRAM ----")
        raise e

def placeTrade(quantityEnter, quantityEnter2, direction, tsdirec, tspoints):
    access_token = load_access_token()
    account_id = 'SIM2504240F'
    symbol = 'MNQU23'
    quantity = quantityEnter
    quantity2 = quantityEnter2
    order_type = 'Market'
    action = direction
    actionStop = tsdirec
    points = tspoints

    url = 'https://sim-api.tradestation.com/v3/orderexecution/orders'
    headers = {
        'content-type': 'application/json',
        'Authorization': f'Bearer {access_token}',
    }
    data = {
        "AccountID": f"{account_id}",
        "TimeInForce": {
            "Duration": "DAY"
        },
        "Quantity": f"{quantity}",
        "OrderType": f"{order_type}",
        "Symbol": f"{symbol}",
        "TradeAction": f"{action}",
        "Route": "Intelligent",
        "OSOs": [
            {
                "Type": "NORMAL",
                "Orders": [
                    {
                        "AccountID": f"{account_id}",
                        "TimeInForce": {
                            "Duration": "DAY"
                        },
                        "Quantity": f"{quantity2}",
                        "OrderType": "StopMarket",
                        "Symbol": f"{symbol}",
                        "TradeAction": f"{actionStop}",
                        "Route": "Intelligent",
                        "AdvancedOptions": {
                            "TrailingStop": {
                                "Amount": f"{points}"
                            }
                        }
                    }
                ]
            }
        ]
    } 

    try:
        response = requests.post(url, headers=headers, json=data)
        response_data = response.json()
        print(response_data)
        print(f'Order for {direction} {symbol} placed successfully.')
        try:
            first_order = response_data['Orders'][0]
            first_order_id = first_order['OrderID']
            print(f'Order ID for the trailingstop is {first_order_id}')
            path = 'C:\\Users\\admin\\Desktop\\Million Dollar Business\\argonaut\\ts_orderid.txt'
            with open(f'{path}', 'w') as f:
                f.write(first_order_id)
        except Exception as e:
            print_red_text("\n------ WARNING ------")
            print_red_text("Stop ID logging failed.")
            print_red_text("---- END WARNING ----")
    except Exception as e:
        print_red_text("\n------ WARNING ------")
        print_red_text("Order placement loop failed.")
        print_red_text("---- END WARNING ----")

def cancel_order():
    try:
        path = 'C:\\Users\\admin\\Desktop\\Million Dollar Business\\argonaut\\ts_orderid.txt'

        with open(f'{path}', 'r') as f:
            order_id = f.read().strip()
    except FileNotFoundError:
        order_id = None

    access_token = load_access_token()
    url = f"https://sim-api.tradestation.com/v3/orderexecution/orders/{order_id}"
    headers = {"Authorization": f"Bearer {access_token}"}

    response = requests.request("DELETE", url, headers=headers)

    if response.status_code == 200:
        print("\nStop order successfully canceled.")
        path = 'C:\\Users\\admin\\Desktop\\Million Dollar Business\\argonaut\\ts_orderid.txt'
        with open(f'{path}', 'w') as f:
            f.write("")
    else:
        print_red_text(f"Failed to cancel the stop order. Status code: {response.status_code}, Response: {response.text}")

def openTrade():
    url = 'https://sim-api.tradestation.com/v3/brokerage/accounts/SIM2504240F/positions?symbol=MNQU23'
    access_token = load_access_token()
    headers = {
        'content-type': 'application/json',
        'Authorization': f'Bearer {access_token}',
    }

    response = requests.request("GET", url, headers=headers)
    data = response.json()
    try:
        if data["Positions"]:
            return True
        else:
            return False
    except Exception as e:
        print_red_text("\n------ WARNING ------")
        print_red_text("Your access token is expired.")
        print_red_text("---- END WARNING ----")