import ccxt
import time
from datetime import datetime, timedelta

# Function to validate the trading pair
def validate_pair(exchange, pair):
    markets = exchange.load_markets()
    return pair in markets

def check_balance(exchange, currency='USDT'):
    balance = exchange.fetch_balance()
    available_balance = balance['free'].get(currency, 0)
    return available_balance

# Example usage:
#available_usdt = check_balance(exchange)
#print(f"Available USDT balance: {available_usdt}")


# Function to place the order
def place_order(exchange, pair, usdt_amount):
    # Check the available balance first
    available_usdt = check_balance(exchange)
    if available_usdt < usdt_amount:
        print(f"Adjusting USDT amount to available balance: {available_usdt}")
        usdt_amount = available_usdt

    try:
        order = exchange.create_order(pair, 'market', 'buy', usdt_amount)
        print("Order placed successfully:", order)
        return True
    except Exception as e:
        print("Order failed:", str(e))
        return False

# Function to get the server time
def get_server_time(exchange):
    server_time = exchange.fetch_time()
    return datetime.utcfromtimestamp(server_time / 1000.0)

# Main script
def main():
    # Replace with your LBANK API credentials
    api_key = 'xxxxxx'
    secret = 'xxxxxx'


    # Initialize the exchange
    exchange = ccxt.lbank({
        'apiKey': api_key,
        'secret': secret,
        'enableRateLimit': True,
        'options': {
            'createMarketBuyOrderRequiresPrice': False,
        },
    })

    # Get and print the server time
    server_time = get_server_time(exchange)
    print(f"Server time: {server_time.strftime('%Y-%m-%d %H:%M:%S.%f')} UTC")

    # Step 1: Input and validate trading pair
    while True:
        pair = input("Enter the trading pair (e.g., BTC/USDT): ").upper()
        if validate_pair(exchange, pair):
            print(f"Trading pair {pair} is valid.")
            break
        else:
            print(f"Trading pair {pair} is invalid. Please try again.")

    # Step 2: Input USDT amount
    usdt_amount = float(input("Enter the total USDT amount to buy: "))

    # Step 3: Input time for order (including milliseconds)
    target_time_str = input("Enter the target time (HH:MM:SS.mmm): ")
    try:
        target_time = datetime.strptime(target_time_str, "%H:%M:%S.%f")
    except ValueError:
        print("Invalid time format! Please use the format HH:MM:SS.mmm (e.g., 16:49:50.100000)")
        return

    # Adjust target time to include the same date as the server time
    target_time = server_time.replace(hour=target_time.hour, minute=target_time.minute, second=target_time.second, microsecond=target_time.microsecond)

    # Countdown until target time with synchronized server time
    while True:
        server_time = get_server_time(exchange)
        if server_time >= target_time:
            print(f"\nTarget time reached: {target_time}. Starting order placement...")
            break
        else:
            remaining_time = (target_time - server_time).total_seconds()
            print(f"Server time: {server_time.strftime('%Y-%m-%d %H:%M:%S.%f')} UTC | Time remaining: {remaining_time:.3f} seconds", end='\r')
            time.sleep(0.001)  # Sleep for 1 millisecond to reduce CPU usage

    # Step 4: Attempt to place the order 5 times if necessary
    for attempt in range(10):
        print(f"\nAttempt {attempt   1} to place the order...")
        if place_order(exchange, pair, usdt_amount):
            break
        else:
            print(f"Retrying in 0.3 second...")
            time.sleep(0.3)

if __name__ == "__main__":
    main()