Automating Algorithmic Trading In Python
Introduction
Python is the lingua franca of quantitative trading. Its ecosystem -- pandas for data manipulation, numpy for numerical computation, scikit-learn for machine learning, and asyncio for concurrent I/O -- makes it uniquely suited for the full trading pipeline from research to production. This article presents a complete automated trading system in Python, covering market data ingestion, strategy logic, order execution, risk management, and production scheduling. Every component is designed for real-world deployment, not just backtesting.System Architecture
A production Python trading system has five layers: ``
Layer 1: Data Ingestion -> Market data feeds, alternative data
Layer 2: Signal Generation -> Strategy logic, alpha models
Layer 3: Risk Management -> Position limits, exposure checks
Layer 4: Execution -> Order routing, smart order routing
Layer 5: Monitoring -> P&L tracking, alerts, logging
`
`python
import logging
import asyncio
from abc import ABC, abstractmethod
from dataclasses import dataclass, field
from datetime import datetime
from typing import Dict, List, Optional
from enum import Enum
Configure structured logging
logging.basicConfig(
level=logging.INFO,
format='%(asctime)s [%(levelname)s] %(name)s: %(message)s'
)
logger = logging.getLogger('TradingSystem')
class OrderSide(Enum):
BUY = 'BUY'
SELL = 'SELL'
class OrderType(Enum):
MARKET = 'MARKET'
LIMIT = 'LIMIT'
@dataclass
class Order:
symbol: str
side: OrderSide
quantity: int
order_type: OrderType
limit_price: Optional[float] = None
timestamp: datetime = field(default_factory=datetime.now)
order_id: str = ''
status: str = 'PENDING'
@dataclass
class Position:
symbol: str
quantity: int
avg_price: float
unrealized_pnl: float = 0.0
realized_pnl: float = 0.0
`
Layer 1: Data Pipeline
`python
import pandas as pd
import numpy as np
from pathlib import Path
import sqlite3
class DataManager:
"""Manages market data ingestion, storage, and retrieval."""
def __init__(self, db_path: str = 'market_data.db'):
self.db_path = db_path
self.conn = sqlite3.connect(db_path)
self._create_tables()
self._cache: Dict[str, pd.DataFrame] = {}
def _create_tables(self):
self.conn.execute('''
CREATE TABLE IF NOT EXISTS daily_bars (
symbol TEXT,
date TEXT,
open REAL,
high REAL,
low REAL,
close REAL,
volume INTEGER,
PRIMARY KEY (symbol, date)
)
''')
self.conn.commit()
def update_from_api(self, symbols: List[str], source: str = 'yfinance'):
"""Fetch latest data and store in database."""
import yfinance as yf
for symbol in symbols:
# Get last date in DB
cursor = self.conn.execute(
'SELECT MAX(date) FROM daily_bars WHERE symbol = ?',
(symbol,)
)
last_date = cursor.fetchone()[0]
if last_date:
start = pd.Timestamp(last_date) + pd.Timedelta(days=1)
else:
start = pd.Timestamp.now() - pd.Timedelta(days=365*5)
df = yf.download(symbol, start=start, progress=False)
if len(df) == 0:
continue
df.columns = [c.lower() for c in df.columns]
records = [
(symbol, idx.strftime('%Y-%m-%d'),
row['open'], row['high'], row['low'], row['close'],
int(row['volume']))
for idx, row in df.iterrows()
]
self.conn.executemany(
'INSERT OR REPLACE INTO daily_bars VALUES (?,?,?,?,?,?,?)',
records
)
self.conn.commit()
logger.info(f"Updated {symbol}: {len(records)} new bars")
# Invalidate cache
self._cache.clear()
def get_prices(self, symbol: str, lookback_days: int = 500) -> pd.DataFrame:
"""Retrieve price data with caching."""
cache_key = f"{symbol}_{lookback_days}"
if cache_key not in self._cache:
query = '''
SELECT date, open, high, low, close, volume
FROM daily_bars
WHERE symbol = ?
ORDER BY date DESC
LIMIT ?
'''
df = pd.read_sql(query, self.conn, params=(symbol, lookback_days))
df['date'] = pd.to_datetime(df['date'])
df = df.set_index('date').sort_index()
self._cache[cache_key] = df
return self._cache[cache_key]
`
Layer 2: Strategy Engine
`python
class StrategyBase(ABC):
"""Abstract base class for all strategies."""
def __init__(self, name: str):
self.name = name
self.logger = logging.getLogger(f'Strategy.{name}')
@abstractmethod
def compute_signals(self, data: Dict[str, pd.DataFrame]) -> Dict[str, float]:
"""
Compute target positions for each symbol.
Returns dict mapping symbol -> target weight [-1.0, 1.0]
"""
pass
class MeanReversionStrategy(StrategyBase):
"""
Bollinger Band mean reversion with volume confirmation.
"""
def __init__(self, lookback: int = 20, entry_std: float = 2.0,
exit_std: float = 0.5, volume_multiplier: float = 1.5):
super().__init__('MeanReversion')
self.lookback = lookback
self.entry_std = entry_std
self.exit_std = exit_std
self.vol_mult = volume_multiplier
def compute_signals(self, data: Dict[str, pd.DataFrame]) -> Dict[str, float]:
signals = {}
for symbol, df in data.items():
if len(df) < self.lookback + 5:
signals[symbol] = 0.0
continue
close = df['close']
volume = df['volume']
# Bollinger Bands
sma = close.rolling(self.lookback).mean()
std = close.rolling(self.lookback).std()
z_score = (close - sma) / std
# Volume confirmation
vol_avg = volume.rolling(self.lookback).mean()
vol_ratio = volume / vol_avg
current_z = z_score.iloc[-1]
current_vol = vol_ratio.iloc[-1]
# Signal logic
if current_z < -self.entry_std and current_vol > self.vol_mult:
# Oversold with high volume = buy
weight = min(1.0, abs(current_z) / 3.0)
signals[symbol] = weight
elif current_z > self.entry_std and current_vol > self.vol_mult:
# Overbought with high volume = sell
weight = min(1.0, abs(current_z) / 3.0)
signals[symbol] = -weight
elif abs(current_z) < self.exit_std:
# Back to mean = flatten
signals[symbol] = 0.0
else:
signals[symbol] = None # Hold current position
self.logger.debug(f"{symbol}: z={current_z:.2f}, vol_ratio={current_vol:.2f}")
return signals
class MomentumStrategy(StrategyBase):
"""
Cross-sectional momentum: rank universe by returns,
go long winners, short losers.
"""
def __init__(self, lookback: int = 60, holding_period: int = 20,
long_pct: float = 0.2, short_pct: float = 0.2):
super().__init__('Momentum')
self.lookback = lookback
self.holding_period = holding_period
self.long_pct = long_pct
self.short_pct = short_pct
def compute_signals(self, data: Dict[str, pd.DataFrame]) -> Dict[str, float]:
# Compute momentum scores
scores = {}
for symbol, df in data.items():
if len(df) < self.lookback + 5:
continue
ret = df['close'].pct_change(self.lookback).iloc[-1]
if not np.isnan(ret):
scores[symbol] = ret
if len(scores) < 5:
return {s: 0.0 for s in data}
# Rank
sorted_symbols = sorted(scores, key=scores.get, reverse=True)
n = len(sorted_symbols)
n_long = max(1, int(n * self.long_pct))
n_short = max(1, int(n * self.short_pct))
signals = {}
for i, sym in enumerate(sorted_symbols):
if i < n_long:
signals[sym] = 1.0 / n_long # Equal-weight long
elif i >= n - n_short:
signals[sym] = -1.0 / n_short # Equal-weight short
else:
signals[sym] = 0.0
return signals
`
Layer 3: Risk Management
`python
@dataclass
class RiskLimits:
max_position_pct: float = 0.10 # Max 10% in any single name
max_gross_exposure: float = 2.0 # Max 200% gross
max_net_exposure: float = 0.30 # Max 30% net (long-short balance)
max_daily_loss_pct: float = 0.02 # Stop trading at 2% daily loss
max_drawdown_pct: float = 0.15 # Halt at 15% drawdown
max_sector_exposure: float = 0.30 # Max 30% in any sector
class RiskManager:
def __init__(self, limits: RiskLimits, portfolio_value: float):
self.limits = limits
self.portfolio_value = portfolio_value
self.daily_pnl = 0.0
self.peak_value = portfolio_value
def check_order(self, order: Order, current_positions: Dict[str, Position],
prices: Dict[str, float]) -> tuple:
"""
Validate order against risk limits.
Returns (approved: bool, reason: str)
"""
# Check daily loss limit
if self.daily_pnl / self.portfolio_value < -self.limits.max_daily_loss_pct:
return False, f"Daily loss limit breached: {self.daily_pnl/self.portfolio_value:.2%}"
# Check drawdown limit
current_dd = (self.portfolio_value - self.peak_value) / self.peak_value
if current_dd < -self.limits.max_drawdown_pct:
return False, f"Max drawdown breached: {current_dd:.2%}"
# Check position concentration
order_value = order.quantity * prices.get(order.symbol, 0)
existing = current_positions.get(order.symbol, Position(order.symbol, 0, 0))
new_position_value = (existing.quantity * existing.avg_price + order_value)
position_pct = abs(new_position_value) / self.portfolio_value
if position_pct > self.limits.max_position_pct:
return False, f"Position limit: {order.symbol} would be {position_pct:.1%}"
# Check gross exposure
gross = sum(abs(p.quantity * prices.get(p.symbol, p.avg_price))
for p in current_positions.values())
gross += abs(order_value)
gross_pct = gross / self.portfolio_value
if gross_pct > self.limits.max_gross_exposure:
return False, f"Gross exposure limit: {gross_pct:.1%}"
return True, "Approved"
`
Layer 4: Execution Engine
`python
class ExecutionEngine:
"""
Handles order generation and submission.
Supports multiple broker APIs through adapter pattern.
"""
def __init__(self, broker_adapter, risk_manager: RiskManager):
self.broker = broker_adapter
self.risk = risk_manager
self.pending_orders: List[Order] = []
def reconcile_positions(self, target_weights: Dict[str, float],
current_positions: Dict[str, Position],
prices: Dict[str, float],
portfolio_value: float) -> List[Order]:
"""Convert target weights to orders."""
orders = []
for symbol, target_weight in target_weights.items():
if target_weight is None:
continue # Hold current
target_value = portfolio_value * target_weight
current_qty = current_positions.get(symbol, Position(symbol, 0, 0)).quantity
current_value = current_qty * prices.get(symbol, 0)
delta_value = target_value - current_value
if abs(delta_value) < 100: # Minimum trade threshold
continue
price = prices[symbol]
delta_shares = int(delta_value / price)
if delta_shares > 0:
order = Order(symbol=symbol, side=OrderSide.BUY,
quantity=delta_shares, order_type=OrderType.MARKET)
elif delta_shares < 0:
order = Order(symbol=symbol, side=OrderSide.SELL,
quantity=abs(delta_shares), order_type=OrderType.MARKET)
else:
continue
# Risk check
approved, reason = self.risk.check_order(
order, current_positions, prices
)
if approved:
orders.append(order)
logger.info(f"Order approved: {order.side.value} {order.quantity} {order.symbol}")
else:
logger.warning(f"Order rejected: {reason}")
return orders
async def submit_orders(self, orders: List[Order]):
"""Submit orders to broker asynchronously."""
tasks = [self.broker.submit(order) for order in orders]
results = await asyncio.gather(*tasks, return_exceptions=True)
for order, result in zip(orders, results):
if isinstance(result, Exception):
logger.error(f"Order failed: {order.symbol} - {result}")
order.status = 'FAILED'
else:
order.status = 'FILLED'
logger.info(f"Filled: {order.side.value} {order.quantity} {order.symbol}")
`
Layer 5: Production Scheduler
`python
import schedule
import time
class TradingScheduler:
"""Orchestrates the daily trading workflow."""
def __init__(self, data_mgr: DataManager, strategies: List[StrategyBase],
execution: ExecutionEngine, symbols: List[str]):
self.data_mgr = data_mgr
self.strategies = strategies
self.execution = execution
self.symbols = symbols
def daily_workflow(self):
"""Complete daily trading pipeline."""
logger.info("=== Starting daily workflow ===")
try:
# Step 1: Update data
self.data_mgr.update_from_api(self.symbols)
# Step 2: Load data for all symbols
data = {s: self.data_mgr.get_prices(s) for s in self.symbols}
# Step 3: Compute signals from all strategies
combined_signals = {}
for strategy in self.strategies:
signals = strategy.compute_signals(data)
for sym, weight in signals.items():
if weight is not None:
combined_signals[sym] = combined_signals.get(sym, 0) + weight
# Step 4: Normalize combined signals
max_weight = max(abs(v) for v in combined_signals.values()) if combined_signals else 1
if max_weight > 1:
combined_signals = {k: v/max_weight for k, v in combined_signals.items()}
# Step 5: Generate and execute orders
prices = {s: data[s]['close'].iloc[-1] for s in self.symbols}
orders = self.execution.reconcile_positions(
combined_signals, {}, prices, self.execution.risk.portfolio_value
)
logger.info(f"Generated {len(orders)} orders")
# Step 6: Submit orders
asyncio.run(self.execution.submit_orders(orders))
except Exception as e:
logger.error(f"Workflow failed: {e}", exc_info=True)
logger.info("=== Daily workflow complete ===")
def start(self):
"""Schedule and run the daily workflow."""
schedule.every().monday.at("09:35").do(self.daily_workflow)
schedule.every().tuesday.at("09:35").do(self.daily_workflow)
schedule.every().wednesday.at("09:35").do(self.daily_workflow)
schedule.every().thursday.at("09:35").do(self.daily_workflow)
schedule.every().friday.at("09:35").do(self.daily_workflow)
logger.info("Scheduler started. Waiting for market hours...")
while True:
schedule.run_pending()
time.sleep(30)
``