Files
gogo2/NN/exchanges/mexc_interface.py

410 lines
18 KiB
Python

import logging
import time
from typing import Dict, Any, List, Optional
import requests
import hmac
import hashlib
from urllib.parse import urlencode, quote_plus
from .exchange_interface import ExchangeInterface
logger = logging.getLogger(__name__)
# https://github.com/mexcdevelop/mexc-api-postman/blob/main/MEXC%20V3.postman_collection.json
# MEXC V3.postman_collection.json
class MEXCInterface(ExchangeInterface):
"""MEXC Exchange API Interface"""
def __init__(self, api_key: str = "", api_secret: str = "", test_mode: bool = True, trading_mode: str = 'simulation'):
"""Initialize MEXC exchange interface.
Args:
api_key: MEXC API key
api_secret: MEXC API secret
test_mode: If True, use test/sandbox environment (Note: MEXC doesn't have a true sandbox)
trading_mode: 'simulation', 'testnet', or 'live'. Determines API endpoints used.
"""
super().__init__(api_key, api_secret, test_mode)
self.trading_mode = trading_mode # Store the trading mode
# MEXC API Base URLs
self.base_url = "https://api.mexc.com" # Live API URL
if self.trading_mode == 'testnet':
# Note: MEXC does not have a separate testnet for spot trading.
# We use the live API for 'testnet' mode and rely on 'simulation' for true dry-runs.
logger.warning("MEXC does not have a separate testnet for spot trading. Using live API for 'testnet' mode.")
self.api_version = "api/v3"
self.recv_window = 5000 # 5 seconds window for request validity
# Session for HTTP requests
self.session = requests.Session()
logger.info(f"MEXCInterface initialized in {self.trading_mode} mode. Ensure correct API endpoints are being used.")
def connect(self) -> bool:
"""Test connection to MEXC API by fetching account info."""
if not self.api_key or not self.api_secret:
logger.error("MEXC API key or secret not set. Cannot connect.")
return False
# Test connection by making a small, authenticated request
try:
account_info = self.get_account_info()
if account_info:
logger.info("Successfully connected to MEXC API and retrieved account info.")
return True
else:
logger.error("Failed to connect to MEXC API: Could not retrieve account info.")
return False
except Exception as e:
logger.error(f"Exception during MEXC API connection test: {e}")
return False
def _format_spot_symbol(self, symbol: str) -> str:
"""Formats a symbol to MEXC spot API standard (e.g., 'ETH/USDT' -> 'ETHUSDT')."""
if '/' in symbol:
base, quote = symbol.split('/')
return f"{base.upper()}{quote.upper()}"
return symbol.upper()
def _format_futures_symbol(self, symbol: str) -> str:
"""Formats a symbol to MEXC futures API standard (e.g., 'ETH/USDT' -> 'ETH_USDT')."""
# This method is included for completeness but should not be used for spot trading
return symbol.replace('/', '_').upper()
def _generate_signature(self, timestamp: str, method: str, endpoint: str, params: Dict[str, Any]) -> str:
"""Generate signature for private API calls"""
# Build the string to sign
sign_str = self.api_key + timestamp
if params:
# Append all parameters sorted by key, without URL encoding for signature
query_str = "&".join([f"{k}={v}" for k, v in sorted(params.items()) if k != 'signature'])
if query_str:
sign_str += query_str
logger.debug(f"Signature string: {sign_str}")
# Generate HMAC SHA256 signature
signature = hmac.new(
self.api_secret.encode('utf-8'),
sign_str.encode('utf-8'),
hashlib.sha256
).hexdigest()
return signature
def _send_public_request(self, method: str, endpoint: str, params: Optional[Dict[str, Any]] = None) -> Dict[str, Any]:
"""Send a public API request to MEXC."""
if params is None:
params = {}
url = f"{self.base_url}/{self.api_version}/{endpoint}"
headers = {'Accept': 'application/json'}
try:
response = requests.request(method, url, params=params, headers=headers, timeout=10)
response.raise_for_status() # Raise HTTPError for bad responses (4xx or 5xx)
return response.json()
except requests.exceptions.HTTPError as http_err:
logger.error(f"HTTP error in public request to {endpoint}: {response.status_code} {response.reason}")
logger.error(f"Response content: {response.text}")
return {}
except requests.exceptions.ConnectionError as conn_err:
logger.error(f"Connection error in public request to {endpoint}: {conn_err}")
return {}
except requests.exceptions.Timeout as timeout_err:
logger.error(f"Timeout error in public request to {endpoint}: {timeout_err}")
return {}
except Exception as e:
logger.error(f"Error in public request to {endpoint}: {e}")
return {}
def _send_private_request(self, method: str, endpoint: str, params: Dict[str, Any] = None) -> Optional[Dict[str, Any]]:
"""Send a private request to the exchange with proper signature"""
if params is None:
params = {}
timestamp = str(int(time.time() * 1000))
# Add timestamp and recvWindow to params for signature and request
params['timestamp'] = timestamp
params['recvWindow'] = self.recv_window
signature = self._generate_signature(timestamp, method, endpoint, params)
params['signature'] = signature
headers = {
"X-MEXC-APIKEY": self.api_key,
"Request-Time": timestamp
}
# Ensure endpoint does not start with a slash to avoid double slashes
if endpoint.startswith('/'):
endpoint = endpoint.lstrip('/')
url = f"{self.base_url}/{endpoint}"
try:
if method.upper() == "GET":
response = self.session.get(url, headers=headers, params=params, timeout=10)
elif method.upper() == "POST":
headers["Content-Type"] = "application/x-www-form-urlencoded"
response = self.session.post(url, headers=headers, data=params, timeout=10)
else:
logger.error(f"Unsupported method: {method}")
return None
response.raise_for_status()
data = response.json()
if data.get('success', False):
return data.get('data', data)
else:
logger.error(f"API error: Status Code: {response.status_code}, Response: {response.text}")
return None
except requests.exceptions.HTTPError as http_err:
logger.error(f"HTTP error for {endpoint}: Status Code: {response.status_code}, Response: {response.text}")
logger.error(f"HTTP error details: {http_err}")
return None
except Exception as e:
logger.error(f"Request error for {endpoint}: {e}")
return None
def get_account_info(self) -> Dict[str, Any]:
"""Get account information"""
endpoint = "/api/v3/account"
result = self._send_private_request("GET", endpoint, {})
return result if result is not None else {}
def get_balance(self, asset: str) -> float:
"""Get available balance for a specific asset."""
account_info = self.get_account_info()
if account_info and 'balances' in account_info:
for balance in account_info['balances']:
if balance.get('asset') == asset.upper():
return float(balance.get('free', 0.0))
logger.warning(f"Could not retrieve free balance for {asset}")
return 0.0
def get_ticker(self, symbol: str) -> Optional[Dict[str, Any]]:
"""Get ticker information for a symbol."""
formatted_symbol = self._format_spot_symbol(symbol)
endpoint = "ticker/24hr"
params = {'symbol': formatted_symbol}
response = self._send_public_request('GET', endpoint, params)
if response:
# MEXC ticker returns a dictionary if single symbol, list if all symbols
if isinstance(response, dict):
ticker_data = response
elif isinstance(response, list) and len(response) > 0:
# If the response is a list, try to find the specific symbol
found_ticker = next((item for item in response if item.get('symbol') == formatted_symbol), None)
if found_ticker:
ticker_data = found_ticker
else:
logger.error(f"Ticker data for {formatted_symbol} not found in response list.")
return None
else:
logger.error(f"Unexpected ticker response format: {response}")
return None
# Extract relevant info and format for universal use
last_price = float(ticker_data.get('lastPrice', 0))
bid_price = float(ticker_data.get('bidPrice', 0))
ask_price = float(ticker_data.get('askPrice', 0))
volume = float(ticker_data.get('volume', 0)) # Base asset volume
# Determine price change and percent change
price_change = float(ticker_data.get('priceChange', 0))
price_change_percent = float(ticker_data.get('priceChangePercent', 0))
logger.info(f"MEXC: Got ticker from {endpoint} for {symbol}: ${last_price:.2f}")
return {
'symbol': formatted_symbol,
'last': last_price,
'bid': bid_price,
'ask': ask_price,
'volume': volume,
'high': float(ticker_data.get('highPrice', 0)),
'low': float(ticker_data.get('lowPrice', 0)),
'change': price_change_percent, # This is usually priceChangePercent
'exchange': 'MEXC',
'raw_data': ticker_data
}
logger.error(f"Failed to get ticker for {symbol}")
return None
def place_order(self, symbol: str, side: str, order_type: str, quantity: float, price: Optional[float] = None) -> Dict[str, Any]:
"""Place a new order on MEXC."""
formatted_symbol = self._format_spot_symbol(symbol)
endpoint = "order"
params: Dict[str, Any] = {
'symbol': formatted_symbol,
'side': side.upper(),
'type': order_type.upper(),
'quantity': str(quantity) # Quantity must be a string
}
if price is not None:
params['price'] = str(price) # Price must be a string for limit orders
logger.info(f"MEXC: Placing {side.upper()} {order_type.upper()} order for {quantity} {formatted_symbol} at price {price}")
# For market orders, some parameters might be optional or handled differently.
# Check MEXC API docs for market order specifics (e.g., quoteOrderQty for buy market orders)
if order_type.upper() == 'MARKET' and side.upper() == 'BUY':
# If it's a market buy order, MEXC often expects quoteOrderQty instead of quantity
# Assuming quantity here refers to the base asset, if quoteOrderQty is needed, adjust.
# For now, we will stick to quantity and let MEXC handle the conversion if possible
pass # No specific change needed based on the current params structure
try:
# MEXC API endpoint for placing orders is /api/v3/order (POST)
order_result = self._send_private_request('POST', endpoint, params)
if order_result:
logger.info(f"MEXC: Order placed successfully: {order_result}")
return order_result
else:
logger.error(f"MEXC: Error placing order: {order_result}")
return {}
except Exception as e:
logger.error(f"MEXC: Exception placing order: {e}")
return {}
def cancel_order(self, symbol: str, order_id: str) -> Dict[str, Any]:
"""Cancel an existing order on MEXC."""
formatted_symbol = self._format_spot_symbol(symbol)
endpoint = "order"
params = {
'symbol': formatted_symbol,
'orderId': order_id
}
logger.info(f"MEXC: Cancelling order {order_id} for {formatted_symbol}")
try:
# MEXC API endpoint for cancelling orders is /api/v3/order (DELETE)
cancel_result = self._send_private_request('DELETE', endpoint, params)
if cancel_result:
logger.info(f"MEXC: Order cancelled successfully: {cancel_result}")
return cancel_result
else:
logger.error(f"MEXC: Error cancelling order: {cancel_result}")
return {}
except Exception as e:
logger.error(f"MEXC: Exception cancelling order: {e}")
return {}
def get_order_status(self, symbol: str, order_id: str) -> Dict[str, Any]:
"""Get the status of an order on MEXC."""
formatted_symbol = self._format_spot_symbol(symbol)
endpoint = "order"
params = {
'symbol': formatted_symbol,
'orderId': order_id
}
logger.info(f"MEXC: Getting status for order {order_id} for {formatted_symbol}")
try:
# MEXC API endpoint for order status is /api/v3/order (GET)
status_result = self._send_private_request('GET', endpoint, params)
if status_result:
logger.info(f"MEXC: Order status retrieved: {status_result}")
return status_result
else:
logger.error(f"MEXC: Error getting order status: {status_result}")
return {}
except Exception as e:
logger.error(f"MEXC: Exception getting order status: {e}")
return {}
def get_open_orders(self, symbol: Optional[str] = None) -> List[Dict[str, Any]]:
"""Get all open orders on MEXC for a symbol or all symbols."""
endpoint = "openOrders"
params = {}
if symbol:
params['symbol'] = self._format_spot_symbol(symbol)
logger.info(f"MEXC: Getting open orders for {symbol if symbol else 'all symbols'}")
try:
# MEXC API endpoint for open orders is /api/v3/openOrders (GET)
open_orders = self._send_private_request('GET', endpoint, params)
if open_orders and isinstance(open_orders, list):
logger.info(f"MEXC: Retrieved {len(open_orders)} open orders.")
return open_orders
else:
logger.error(f"MEXC: Error getting open orders: {open_orders}")
return []
except Exception as e:
logger.error(f"MEXC: Exception getting open orders: {e}")
return []
def get_my_trades(self, symbol: str, limit: int = 100) -> List[Dict[str, Any]]:
"""Get trade history for a specific symbol."""
formatted_symbol = self._format_spot_symbol(symbol)
endpoint = "myTrades"
params = {'symbol': formatted_symbol, 'limit': limit}
logger.info(f"MEXC: Getting trade history for {formatted_symbol} (limit: {limit})")
try:
# MEXC API endpoint for trade history is /api/v3/myTrades (GET)
trade_history = self._send_private_request('GET', endpoint, params)
if trade_history and isinstance(trade_history, list):
logger.info(f"MEXC: Retrieved {len(trade_history)} trade records.")
return trade_history
else:
logger.error(f"MEXC: Error getting trade history: {trade_history}")
return []
except Exception as e:
logger.error(f"MEXC: Exception getting trade history: {e}")
return []
def get_server_time(self) -> int:
"""Get current MEXC server time in milliseconds."""
endpoint = "time"
response = self._send_public_request('GET', endpoint)
if response and 'serverTime' in response:
return int(response['serverTime'])
logger.error("Failed to get MEXC server time.")
return int(time.time() * 1000) # Fallback to local time
def get_all_balances(self) -> Dict[str, Dict[str, float]]:
"""Get all asset balances from MEXC account."""
account_info = self.get_account_info()
balances = {}
if account_info and 'balances' in account_info:
for balance in account_info['balances']:
asset = balance.get('asset')
free = float(balance.get('free'))
locked = float(balance.get('locked'))
if asset:
balances[asset.upper()] = {'free': free, 'locked': locked, 'total': free + locked}
return balances
def get_trading_fees(self) -> Dict[str, Any]:
"""Get current trading fee rates from MEXC API"""
endpoint = "account/commission"
response = self._send_private_request('GET', endpoint)
if response and 'data' in response:
fees_data = response['data']
return {
'maker': float(fees_data.get('makerCommission', 0.0)),
'taker': float(fees_data.get('takerCommission', 0.0)),
'default': float(fees_data.get('defaultCommission', 0.0))
}
logger.error("Failed to get trading fees from MEXC API.")
return {}
def get_symbol_trading_fees(self, symbol: str) -> Dict[str, Any]:
"""Get trading fee rates for a specific symbol from MEXC API"""
formatted_symbol = self._format_spot_symbol(symbol)
endpoint = "account/commission"
params = {'symbol': formatted_symbol}
response = self._send_private_request('GET', endpoint, params)
if response and 'data' in response:
fees_data = response['data']
return {
'maker': float(fees_data.get('makerCommission', 0.0)),
'taker': float(fees_data.get('takerCommission', 0.0)),
'default': float(fees_data.get('defaultCommission', 0.0))
}
logger.error(f"Failed to get trading fees for {symbol} from MEXC API.")
return {}