76 lines
2.5 KiB
Python
76 lines
2.5 KiB
Python
|
"""Flask backend for IB dashboard.
|
||
|
|
|
||
|
|
Provides simple REST endpoints that the vanilla-JS frontend polls:
|
||
|
|
- GET /api/quotes -> latest market quotes
|
||
|
|
- GET /api/positions -> latest positions
|
||
|
|
- POST /api/subscribe { symbols: ["AAPL","MSFT"] } -> start subscribing
|
||
|
|
|
||
|
|
This app expects an IB Gateway or TWS running locally on port 7496.
|
||
|
|
"""
|
||
|
|
from flask import Flask, jsonify, request, send_from_directory
|
||
|
|
import os
|
||
|
|
import threading
|
||
|
|
|
||
|
|
from ib_client import IBClient
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
# Use an absolute path for the static folder so Flask can reliably serve files
|
||
|
|
STATIC_FOLDER = os.path.abspath(os.path.join(os.path.dirname(__file__), "..", "frontend"))
|
||
|
|
app = Flask(__name__, static_folder=STATIC_FOLDER, static_url_path="/static")
|
||
|
|
|
||
|
|
# instantiate the IB client with a process-unique client id to avoid
|
||
|
|
# collisions when Flask's reloader spawns multiple processes.
|
||
|
|
ib = IBClient(client_id=os.getpid())
|
||
|
|
# Start connection only in the main process (Werkzeug sets WERKZEUG_RUN_MAIN)
|
||
|
|
if os.environ.get("WERKZEUG_RUN_MAIN") == "true" or not os.environ.get("WERKZEUG_RUN_MAIN"):
|
||
|
|
# attempt to start the connection; failures are logged by the client
|
||
|
|
try:
|
||
|
|
ib.connect_and_start()
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/")
|
||
|
|
def index():
|
||
|
|
return send_from_directory(app.static_folder, "index.html")
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/api/quotes")
|
||
|
|
def api_quotes():
|
||
|
|
return jsonify(ib.get_quotes())
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/api/positions")
|
||
|
|
def api_positions():
|
||
|
|
# request a fresh positions snapshot from IB and wait briefly for callbacks
|
||
|
|
try:
|
||
|
|
ib.request_positions()
|
||
|
|
except Exception:
|
||
|
|
pass
|
||
|
|
# wait up to 2 seconds for positions to arrive
|
||
|
|
deadline = time.time() + 2.0
|
||
|
|
while time.time() < deadline:
|
||
|
|
positions = ib.get_positions()
|
||
|
|
if positions:
|
||
|
|
return jsonify(positions)
|
||
|
|
time.sleep(0.1)
|
||
|
|
# return whatever we have (possibly empty)
|
||
|
|
return jsonify(ib.get_positions())
|
||
|
|
|
||
|
|
|
||
|
|
@app.route("/api/subscribe", methods=["POST"])
|
||
|
|
def api_subscribe():
|
||
|
|
data = request.get_json(force=True)
|
||
|
|
symbols = data.get("symbols", []) if isinstance(data, dict) else []
|
||
|
|
if not isinstance(symbols, list):
|
||
|
|
return jsonify({"error": "symbols must be a list"}), 400
|
||
|
|
for s in symbols:
|
||
|
|
ib.subscribe_market_data(s)
|
||
|
|
return jsonify({"subscribed": symbols})
|
||
|
|
|
||
|
|
|
||
|
|
if __name__ == "__main__":
|
||
|
|
# simple dev server; disable reloader to avoid duplicate IB client instances
|
||
|
|
app.run(debug=True, port=8000, use_reloader=False)
|